#!/usr/bin/env ruby
# GIST
# Paste to Gist.GitHub.com from the Shell
# 
# Forked by Thomas Aylott / subtleGradient 
# at 4096a998b6c6a0043e1ce158d43639b5db78a61c
# 
# Original "gistie" by Pieter de Bie <frimmirf@gmail.com>
# Based on a "Pastie" task by someone
# 
 
require "tempfile"
 
class Gist
  GIST_URL       = 'http://gist.github.com/gists'
  GIST_LOGIN_URL = 'https://gist.github.com/session'
 
  def login_cookie
    begin
      username = File.read(File.expand_path('~/.gist_username')).chomp
      password = File.read(File.expand_path('~/.gist_pass')).chomp
    rescue => e
      return nil
    end
    
    headers = `curl --insecure #{GIST_LOGIN_URL} -s -i -F "login=#{username}" -F "password=#{password}"`
    if headers =~ /Set-Cookie: (.*); do/
      return $1
    end
    return nil
  end
  
  def initialize(filename, extension, data, login=false)
    cookie = login_cookie if login
    
    text_file = Tempfile.open('w+')
    text_file << data
    text_file.flush
 
    cmd = "curl #{GIST_URL} \\\n"
    cmd += "-b '#{cookie}' \\\n" if cookie
    cmd += <<-EOS
    -s -L -o /dev/null -w "%{url_effective}" \
    -F "file_ext[gistfile1]=#{extension}" \
    -F "file_name[gistfile1]=#{filename}" \
    -F "file_contents[gistfile1]=<#{text_file.path}" \
    -F "x=27" \
    -F "y=27"
    EOS
 
    @url = %x{#{cmd}}
    text_file.close(true)
  end
  
  def to_s
    @url
  end
  
end
 
if __FILE__ == $0
  
  login = !ARGV.delete('-a')
  
  if ARGV.include? "-p"
    data = `pbpaste`
    
  elsif !ARGV.empty?
    @filename = ARGV.shift
    data = File.read(File.expand_path(@filename))
    
  elsif ENV["FILE"]
    @filename = ENV["FILE"]
    data = File.read(File.expand_path(ENV["FILE"]))
    
  else
    data = STDIN.read
    
  end
  
  @filename ||= "Unnamed File"
  @extension = File.extname(@filename)
  @extension = "txt" if @extension == ""
 
  puts Gist.new(@filename, @extension, data, login)
  
end