pastebin (fixed)

function urlEncode( str )
    if str then
        str = string.gsub (str, "\n", "\r\n")
        str = string.gsub (str, "([^%w ])",
        function (c)
            return string.format ("%%%02X", string.byte(c))
        end)
        str = string.gsub (str, " ", "+")
    end
    return str
end


local function printUsage()
    print( "Usages:" )
    print( "pastebin put <filename>" )
    print( "pastebin get <code> <filename>" )
end

local tArgs = { ... }
if #tArgs < 2 then
    printUsage()
    return
end

local sCommand = tArgs[1]
if sCommand == "put" then
    -- Upload a file to pastebin.com
    -- Determine file to upload
    local sFile = tArgs[2]
    local sPath = shell.resolve( sFile )
    if not fs.exists( sPath ) or fs.isDir( sPath ) then
        print( "No such file" )
        return
    end

    -- Read in the file
    local sName = fs.getName( sPath )
    local file = fs.open( sPath, "r" )
    local sText = file.readAll()
    file.close()

    -- POST the contents to pastebin
    write( "Connecting to pastebin.com... " )
    local key = "0ec2eb25b6166c0c27a394ae118ad829"
    local response = http.post(
        "http://pastebin.com/api/api_post.php",
        "api_option=paste&"..
        "api_dev_key="..key.."&"..
        "api_paste_format=lua&"..
        "api_paste_name="..urlEncode(sName).."&"..
        "api_paste_code="..urlEncode(sText)
        )

    if response then
        print( "Success." )

        local sResponse = response.readAll()
        response.close()

        local sCode = string.match( sResponse, "[^/]+$" )
        print( "Uploaded as "..sResponse )
        print( "Run \"pastebin get "..sCode.."\" to download anywhere" )

    else
        print( "Failed." )
    end

elseif sCommand == "get" then
    -- Download a file from pastebin.com
    if #tArgs < 3 then
        printUsage()
        return
    end

    -- Determine file to download
    local sCode = tArgs[2]
    local sFile = tArgs[3]
    local sPath = shell.resolve( sFile )
    if fs.exists( sPath ) then
        print( "File already exists" )
        return
    end

    -- GET the contents from pastebin
    write( "Connecting to pastebin.com... " )
    local response = http.get(
        "http://pastebin.com/raw.php?i="..urlEncode( sCode )
        )

    if response then
        print( "Success." )

        local sResponse = response.readAll()
        response.close()

        local file = fs.open( sPath, "w" )
        file.write( sResponse )
        file.close()

        print( "Downloaded as "..sFile )

    else
        print( "Failed." )
    end

else
    printUsage()
    return
end