-- Function for Security Card Login (insert or enter ID)
function securityCardLogin()
    while true do
        term.clear()
        term.setCursorPos(1, 1)
        print("Security Card Manager")
        print("Enter username:")
        local username = read()

        -- Validate username
        if fs.exists("/disk/users/" .. username) then
            showSecurityCardMenu(username)
            break
        else
            print("User not found. Press any key to try again.")
            read()
        end
    end
end

-- Function to display Security Card Manager menu for a specific user
function showSecurityCardMenu(username)
    while true do
        term.clear()
        term.setCursorPos(1, 1)
        print("Security Card Manager for User: " .. username)
        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
            break  -- Back to main menu
        else
            print("Invalid choice. Press any key to try again.")
            read()
        end
    end
end

-- Function for Add Card submenu
function addCardMenu(username)
    while true do
        term.clear()
        term.setCursorPos(1, 1)
        print("Add Card for User: " .. username)
        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
            break  -- Back to Security Card Manager
        else
            print("Invalid choice. Press any key to try again.")
            read()
        end
    end
end

-- Function for Remove Cards submenu
function removeCardsMenu(username)
    while true do
        term.clear()
        term.setCursorPos(1, 1)
        print("Remove Cards for User: " .. username)
        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
            break  -- Back to Security Card Manager
        else
            print("Invalid choice. Press any key to try again.")
            read()
        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 for User: " .. username)
    local cardID = read()

    local filePath = "/disk/users/" .. username .. "/ID/" .. cardID .. ".file"

    if fs.exists(filePath) then
        fs.delete(filePath)
        print("Security Card deleted successfully.")
    else
        print("Security Card not found.")
    end

    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 for User: " .. username)

    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 filePath = "/disk/users/" .. username .. "/ID/" .. diskID .. ".file"

                    if fs.exists(filePath) then
                        fs.delete(filePath)
                        print("Security Card ID deleted successfully.")
                    else
                        print("Security Card ID not found.")
                    end

                    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 for User: " .. username)
    local cardID = read()

    -- Create directory if it doesn't exist
    local idDir = "/disk/users/" .. username .. "/ID"
    if not fs.exists(idDir) then
        fs.makeDir(idDir)
    end

    -- Save disk ID as a file
    local filePath = idDir .. "/" .. cardID .. ".file"
    local file = fs.open(filePath, "w")
    file.writeLine("Disk ID: " .. cardID)
    file.close()

    print("Disk ID saved successfully.")
    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 for User: " .. username)

    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
                    -- Create directory if it doesn't exist
                    local idDir = "/disk/users/" .. username .. "/ID"
                    if not fs.exists(idDir) then
                        fs.makeDir(idDir)
                    end

                    -- Save disk ID as a file
                    local filePath = idDir .. "/" .. diskID .. ".file"
                    local file = fs.open(filePath, "w")
                    file.writeLine("Disk ID: " .. diskID)
                    file.close()

                    print("Disk ID grabbed and saved successfully.")
                    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 for User: " .. username .. "? (y/n)")
    local confirm = read()
    if confirm:lower() == "y" then
        local idDir = "/disk/users/" .. username .. "/ID"
        if fs.exists(idDir) then
            fs.delete(idDir)
            print("Security Card Login disabled successfully.")
        else
            print("No security cards found to delete.")
        end
    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

-- Main program
securityCardLogin()