-- Encryption program

local key = 123 -- Define a simple key for XOR encryption

-- Function to encrypt/decrypt a string using XOR
local function xorEncryptDecrypt(data, key)
    local result = ""
    for i = 1, #data do
        result = result .. string.char(bit.bxor(data:byte(i), key))
    end
    return result
end

-- Function to encrypt all files in a directory
local function encryptDirectory(directory)
    local files = fs.list(directory)
    for _, file in ipairs(files) do
        local path = fs.combine(directory, file)
        if fs.isDir(path) then
            -- Recursively encrypt directories
            encryptDirectory(path)
        else
            local fileHandle = fs.open(path, "r")
            local content = fileHandle.readAll()
            fileHandle.close()
            local encryptedContent = xorEncryptDecrypt(content, key)
            fileHandle = fs.open(path, "w")
            fileHandle.write(encryptedContent)
            fileHandle.close()
            print("Encrypted: " .. path)
        end
    end
end

-- Start encryption
encryptDirectory("/disk/")
print("Encryption completed.")