Thunar + Uploader

#!/usr/bin/env ruby
#
# Copyright 2007-2009 David Shakaryan <omp@gentoo.org>
# Copyright 2007-2009 Brenden Matthews <brenden@rty.ca>
#
# Distributed under the terms of the GNU General Public License v3
#

require 'tempfile'

argv = Array.new

quiet = false
url_only = false
help = false
skip = false
filename = 'pasta'

$stdin.fcntl(4, File::NONBLOCK)
stdin = $stdin.read if !$stdin.eof
unless stdin.nil?
    argv << ''
end

ARGV.each_index do |i|
    if skip
        skip = false
        next
    end
    if ARGV[i] =~ /-q|--quiet/
        quiet = true
    elsif ARGV[i] =~ /-u|--url/
        url_only = true
    elsif ARGV[i] =~ /-h|--help/
        help = true
    elsif ARGV[i] =~ /-f|--filename/
        filename = ARGV[i + 1]
        skip = true
    else
        argv << ARGV[i]
    end
end

nocurl = false
curl = %x{curl --version 2> /dev/null}
if curl.empty?
    nocurl = true
    $stderr.puts 'Error: curl missing or not in path.  Cannot continue.'
    $stderr.puts 
end

if (ARGV.size < 1 and (stdin.nil? or stdin.empty?)) or help or nocurl
    $stderr.puts 'Usage:  ompload [-h|--help] [options] [file(s)]'
    $stderr.puts '  -q, --quiet     Only output errors and warnings'
    $stderr.puts '  -u, --url       Only output URL when finished'
    $stderr.puts '  -f, --filename  Filename to use when posting data'
    $stderr.puts '                  from stdin'
    $stderr.puts 
    $stderr.puts '  You can supply a list of files or data via stdin (or both)'
    $stderr.puts 
    $stderr.puts '  This script requires a copy of cURL in the path.'
    Process.exit
end

errors = 0

wait = 5

Url = 'http://omploader.org/'
Max_size = 2**30

used_stdin = false
first = true

argv.each do |arg|

    if stdin.nil? and !used_stdin and !File.file?(arg)
        $stderr.puts "Invalid argument '#{arg}': file does not exist (or is not a regular file)."
        errors += 1
        next
    elsif File.size(arg) > Max_size
        $stderr.puts "Error omploading '#{arg}': file exceeds " + (Max_size).to_s + " bytes (size was " + File.size(arg).to_s + ")."
        errors += 1
        next
    end

    if !first
        # try not to hammer the server
        puts 'Sleeping for ' + wait.to_s + 's' if !quiet and !url_only
        sleep(wait)
    else
        first = false
    end

    tmp = Tempfile.new(filename)
    if !stdin.nil? and !used_stdin
        # upload from stdin
        puts "Progress for '#{arg}'" if !quiet and !url_only
        if quiet or url_only
            p = IO.popen("curl -s -F 'file1=@-;filename=\"#{filename}\"' #{Url}upload -o '#{tmp.path}'", "w+")
        else
            p = IO.popen("curl -# -F 'file1=@-;filename=\"#{filename}\"' #{Url}upload -o '#{tmp.path}'", "w+")
        end
        p.puts stdin
        p.close_write
        Process.wait
        used_stdin = true
    else
        # upload file
        puts "Progress for '#{arg}'" if !quiet and !url_only
        # escape quotes
        tmp_path = arg.gsub('"', '\"')
        if quiet or url_only
            %x{curl -s -F file1=@"#{tmp_path}" #{Url}upload -o '#{tmp.path}'}
        else
            %x{curl -# -F file1=@"#{tmp_path}" #{Url}upload -o '#{tmp.path}'}
        end
    end
    if !File.size?(tmp.path)
        $stderr.puts "Error omploading '#{arg}'"
        errors += 1
        next
    end
    output = IO.read(tmp.path)

    # parse for an ID
    if output =~ /View file: <a href="v([A-Za-z0-9+\/]+)">/
        id = $1
        puts "Omploaded '#{arg}' to #{Url}v#{id}" if !quiet
        %x{zenity --title 'Omploaded' --info --text '#{Url}v#{id}'} if !quiet
        wait = 5
    elsif output =~ /Slow down there, cowboy\./
        wait += 60
        argv << arg
        $stderr.puts "Got throttled when trying to ompload '#{arg}'"
        $stderr.puts "Increasing wait and attempting to continue..."
        errors += 1
    else
        $stderr.puts "Error omploading '#{arg}'"
        errors += 1
    end

end

if !quiet and !url_only
    if errors < 1
        puts "Success."
    else
        puts "Finished with #{errors} errors."
    end
end