Encrypted security card manager

-- Function for Security Card Login (insert or enter ID)
function securityCardLogin(username)
    while true do
        term.clear()
        term.setCursorPos(1, 1)
        print("Security Card Manager")
        print("1. Add Card")
        print("2. Remove Cards")
        print("3. Show All Security Cards")
        print("4. Disable Security Card Login")
        print("5. Back to Main Menu")

        local choice = read()

        if choice == "1" then
            addCardMenu(username)
        elseif choice == "2" then
            removeCardsMenu(username)
        elseif choice == "3" then
            showAllSecurityCards(username)
        elseif choice == "4" then
            disableSecurityCardLogin(username)
        elseif choice == "5" then
            return  -- Back to main menu
        else
            print("Invalid choice. Try again.")
            sleep(2)
        end
    end
end

-- Function for Add Card submenu
function addCardMenu(username)
    while true do
        term.clear()
        term.setCursorPos(1, 1)
        print("Add Card")
        print("1. Insert Security Card")
        print("2. Enter Security Card ID")
        print("3. Back to Security Card Manager")

        local choice = read()

        if choice == "1" then
            grabAndSaveDiskID(username)
        elseif choice == "2" then
            saveDiskID(username)
        elseif choice == "3" then
            return  -- Back to Security Card Manager
        else
            print("Invalid choice. Try again.")
            sleep(2)
        end
    end
end

-- Function for Remove Cards submenu
function removeCardsMenu(username)
    while true do
        term.clear()
        term.setCursorPos(1, 1)
        print("Remove Cards")
        print("1. Insert Security Card to delete")
        print("2. Enter Security Card ID to delete")
        print("3. Back to Security Card Manager")

        local choice = read()

        if choice == "1" then
            grabAndDeleteDiskID(username)
        elseif choice == "2" then
            deleteSpecificCard(username)
        elseif choice == "3" then
            return  -- Back to Security Card Manager
        else
            print("Invalid choice. Try again.")
            sleep(2)
        end
    end
end

-- Function to delete a specific security card by ID
function deleteSpecificCard(username)
    term.clear()
    term.setCursorPos(1, 1)
    print("Enter Card ID to delete:")
    local cardID = read()

    local result = delete(username, cardID)
    print(result.message)

    sleep(2)
end

-- Function to grab disk ID and delete it for a user
function grabAndDeleteDiskID(username)
    term.clear()
    term.setCursorPos(1, 1)
    print("Insert Security Card to delete ID...")

    while true do
        local peripherals = peripheral.getNames()

        for _, name in ipairs(peripherals) do
            if peripheral.getType(name) == "drive" then
                local diskID = disk.getID(name)
                if diskID then
                    local result = delete(username, diskID)
                    print(result.message)

                    return  -- Exit function after deleting ID
                end
            end
        end

        sleep(1)  -- Check every second
    end
end

-- Function to save disk ID for a user
function saveDiskID(username)
    term.clear()
    term.setCursorPos(1, 1)
    print("Enter Card ID:")
    local cardID = read()

    local result = create(username, cardID)
    print(result.status)

    sleep(2)
end

-- Function to grab disk ID and save it for a user
function grabAndSaveDiskID(username)
    term.clear()
    term.setCursorPos(1, 1)
    print("Insert Security Card to grab ID...")

    while true do
        local peripherals = peripheral.getNames()

        for _, name in ipairs(peripherals) do
            if peripheral.getType(name) == "drive" then
                local diskID = disk.getID(name)
                if diskID then
                    local result = create(username, diskID)
                    print(result.status)

                    return  -- Exit function after saving
                end
            end
        end

        sleep(1)  -- Check every second
    end
end

-- Function to disable Security Card Login (delete all IDs)
function disableSecurityCardLogin(username)
    term.clear()
    term.setCursorPos(1, 1)
    print("Are you sure you want to disable Security Card Login? (y/n)")
    local confirm = read()
    if confirm:lower() == "y" then
        local result = delete(username)
        print(result.message)
    else
        print("Operation canceled.")
    end

    sleep(2)
end

-- Function to show all security cards for a user
function showAllSecurityCards(username)
    term.clear()
    term.setCursorPos(1, 1)
    print("Security Cards for User: " .. username)

    local idDir = "/disk/users/" .. username .. "/ID"
    if fs.exists(idDir) then
        local files = fs.list(idDir)
        for _, file in ipairs(files) do
            if fs.isDir(idDir .. "/" .. file) == false then
                print(file)
            end
        end
    else
        print("No security cards found.")
    end

    print("Press any key to continue...")
    read()
end

-- Function for authentication
function authenticate(adminUsername, adminPassword)
    -- Replace with actual admin credential verification
    if adminUsername == "admin" and adminPassword == "password" then
        return true
    else
        return false
    end
end

-- Function to generate a key for encryption and decryption
function generateKey()
    local chars = {}
    for i = 32, 126 do -- ASCII range for printable characters
        table.insert(chars, string.char(i))
    end

    local key = {}
    while #chars > 0 do
        local index = math.random(#chars)
        table.insert(key, table.remove(chars, index))
    end

    return table.concat(key)
end

-- Function to create a map from a key
function createMap(key)
    local map = {}
    for i = 32, 126 do
        map[string.char(i)] = key:sub(i - 31, i - 31)
    end
    return map
end

-- Function to invert a map
function invertMap(map)
    local invMap = {}
    for k, v in pairs(map) do
        invMap[v] = k
    end
    return invMap
end

-- Function to encrypt text using a key
function encrypt(text, key)
    local map = createMap(key)
    local encrypted = {}
    for i = 1, #text do
        local char = text:sub(i, i)
        table.insert(encrypted, map[char] or char)
    end
    return table.concat(encrypted)
end

-- Function to decrypt text using a key
function decrypt(text, key)
    local map = invertMap(createMap(key))
    local decrypted = {}
    for i = 1, #text do
        local char = text:sub(i, i)
        table.insert(decrypted, map[char] or char)
    end
    return table.concat(decrypted)
end

-- Function to save the key to a file
function saveKeyToFile(key, dir)
    local randomNumber = math.random(1, 1000)
    local fileName = dir .. "/key" .. randomNumber .. ".txt"
    local file = fs.open(fileName, "w")
    file.write(key)
    file.close()
    return fileName
end

-- Function to load the key from a file
function loadKeyFromFile(fileName)
    local file = fs.open(fileName, "r")
    local key = file.readAll()
    file.close()
    return key
end

-- Function to find the key file in a directory
function findKeyFile(dir)
    local files = fs.list(dir)
    for _, file in ipairs(files) do
        if file:match("^key%d+%.txt$") then
            return dir .. "/" .. file
        end
    end
    return nil
end

-- API Functions
function create(username, cardID)
    local userDir = "/disk/users/" .. username

    if not fs.exists(userDir) then
        fs.makeDir(userDir)
    end

    local key = generateKey()
    local encryptedCardID = encrypt(cardID, key)
    local keyFileName = saveKeyToFile(key, userDir)

    local cardFile = fs.open(userDir .. "/card_" .. cardID .. ".txt", "w")
    cardFile.write(encryptedCardID)
    cardFile.close()

    return { status = "success", encryptedCardID = encryptedCardID, keyFileName = keyFileName }
end

function delete(username, cardID)
    local userDir = "/disk/users/" .. username
    local keyFileName = findKeyFile(userDir)

    if keyFileName then
        local key = loadKeyFromFile(keyFileName)
        local encryptedCardID = encrypt(cardID, key)
        local filePath = userDir .. "/card_" .. encryptedCardID .. ".txt"

        if fs.exists(filePath) then
            fs.delete(filePath)
            return { status = "success", message = "Security Card deleted successfully." }
        else
            return { status = "error", message = "Security Card not found." }
        end
    else
        return { status = "error", message = "Key file not found. Cannot delete card." }
    end
end

-- Main Program Execution

local function main()
    term.clear()
    term.setCursorPos(1, 1)
    print("Enter admin username:")
    local adminUsername = read()
    print("Enter admin password:")
    local adminPassword = read("*")

    local authenticated = authenticate(adminUsername, adminPassword)

    if authenticated then
        print("Authentication successful.")
        print("Enter username to manage security cards:")
        local username = read()

        securityCardLogin(username)
    else
        print("Authentication failed. Exiting...")
        sleep(2)
        return
    end
end

main()