Forked from defunkt's gist.rb

gist.rb

#!/usr/bin/env ruby
require 'open-uri'
require 'net/http'
require 'uri'
require 'pp'
require 'rubygems'
require 'json'

# This is my modified version of defunkt's gist.rb
# http://github.com/defunkt/gist/blob/master/gist.rb
# more scripts that harness the Github Gist API can be found at
# http://groups.google.com/group/github/browse_thread/thread/ffcaad439aa3fb4a/39e99208a32c23b4

# = USAGE
#  gist < file.txt
#  echo secret | gist -p  # or --private
#  gist 1234 > something.txt
#
# = INSTALL
#  curl http://github.com/evaryont/gist/raw/master/gist.rb > gist &&
#  chmod 755 gist &&
#  mv gist /usr/local/bin/gist

module Gist
  extend self
  GIST_URL = 'http://gist.github.com/%s.txt'
  GIST_URL_REGEXP = /https?:\/\/gist.github.com\/\d+$/

  @proxy = ENV['http_proxy'] ? URI(ENV['http_proxy']) : nil

  def read(gist_id)
    return help if gist_id.nil? || gist_id[/^\-h|help$/]
    return open(GIST_URL % gist_id).read unless gist_id.to_i.zero?
    return open(gist_id + '.txt').read if gist_id[GIST_URL_REGEXP]
  end

  def write(content, private_gist = false)
    url = URI.parse('http://gist.github.com/gists')
    if @proxy
      proxy = Net::HTTP::Proxy(@proxy.host, @proxy.port)
      req = proxy.post_form(url, data(nil, nil, content, private_gist))
    else
      req = Net::HTTP.post_form(url, data(nil, nil, content, private_gist))
    end
    copy req['Location']
  end
  
  def list name, is_short = false
    res = Net::HTTP.post_form(URI.parse("http://gist.github.com/api/v1/json/gists/#{name}"), {})		        
    result = JSON.parse(res.body)["gists"]
    
    if is_short
      result.collect { |r| r["repo"] }
    else
      result
    end
  end
  
  def info gist_id
    res = Net::HTTP.post_form(URI.parse("http://gist.github.com/api/v1/json/#{gist_id}"), {})		        
    result = JSON.parse(res.body)["gists"]
  end

  def help
    help = File.read(__FILE__).scan(/# = USAGE(.+?)# = INSTALL/m)[0][0]
    "usage: \n" + help.strip.gsub(/^# ?/, '')
  end

  private
  
  def copy(content)
    case RUBY_PLATFORM
    when /darwin/
      return content unless system("which pbcopy")
      IO.popen('pbcopy', 'r+') { |clip| clip.print content }
      `open #{content}`
    when /linux/
      return content unless system("which xclip  2> /dev/null")
      IO.popen('xclip', 'r+') { |clip| clip.print content }
    when /i386-cygwin/
      return content if `which putclip`.strip == ''
      IO.popen('putclip', 'r+') { |clip| clip.print content }
    end

    content
  end

  def data(name, ext, content, private_gist)
    return {
      'file_ext[gistfile1]'      => ext,
      'file_name[gistfile1]'     => name,
      'file_contents[gistfile1]' => content
    }.merge(private_gist ? { 'action_button' => 'private' } : {}).merge(auth)
  end

  def auth
    user  = `git config --global github.user`.strip
    token = `git config --global github.token`.strip

    user.empty? ? {} : { :login => user, :token => token }
  end
end

if $stdin.tty?
  puts Gist.read(ARGV.first)
else
  puts Gist.write($stdin.read, %w( -p --private ).include?(ARGV.first))
end