post にエラートラップ追加

gisty.rb

require 'pathname'
require 'net/http'
require 'open-uri'
require 'rubygems'
require 'nokogiri'

class Gisty
  VERSION   = '0.0.11'
  GIST_URL  = 'http://gist.github.com/'
  GISTY_URL = 'http://github.com/swdyh/gisty/tree/master'

  class UnsetAuthInfoException < Exception
  end

  class InvalidFileException < Exception
  end

  def self.extract_ids str
    doc = Nokogiri::HTML str
    doc.css('.file .info a').map { |i| i['href'].sub('/', '') }
  end

  def self.extract url
    doc = Nokogiri::HTML open(url)
    {
      :id => url.split('/').last,
      :author => doc.css('#owner a').inner_text,
      :files => doc.css('.meta .info').map { |i| i.inner_text.strip },
      :clone => doc.css('a[rel="#git-clone"]').inner_text,
    }
  end

  def initialize path, login = nil, token = nil
    @auth = (login && token) ? { :login => login, :token => token } : auth
    raise UnsetAuthInfoException if @auth[:login].nil? || @auth[:token].nil?
    @auth_query = "login=#{@auth[:login]}&token=#{@auth[:token]}"
    @dir  = Pathname.pwd.realpath.join path
    FileUtils.mkdir_p @dir unless @dir.exist?
  end

  def next_link str
    doc = Nokogiri::HTML str
    a = doc.at('.pagination a[hotkey="l"]')
    a ? a['href'] : nil
  end

  def map_pages
    result = []
    base_url = GIST_URL.sub(/\/$/, '')
    path = "/mine?page=1"
    loop do
      url = base_url + path + "&#{@auth_query}"
      page = open(url).read
      result << yield(url, page)
      path = next_link page
      break unless path
    end
    result
  end

  def remote_ids
    map_pages { |url, page| Gisty.extract_ids page }.flatten.uniq.sort
  end

  def clone id
    FileUtils.cd @dir do
      c = "git clone git@gist.github.com:#{id}.git"
      Kernel.system c
    end
  end

  def list
    dirs = Pathname.glob(@dir.to_s + '/*').map do |i|
      [i.basename.to_s,
       Pathname.glob(i.to_s + '/*').map { |i| i.basename.to_s }]
    end
    re_pub = /^\d+$/
    pub = dirs.select { |i| re_pub.match(i.first) }.sort_by { |i| i.first.to_i }.reverse
    pri = dirs.select { |i| !re_pub.match(i.first) }.sort_by { |i| i.first }
    { :public => pub, :private => pri }
  end

  def local_ids
    dirs = Pathname.glob(@dir.to_s + '/*')
    dirs.map { |i| i.basename.to_s }
  end

  def delete id
    FileUtils.rm_rf @dir.join(id) if @dir.join(id).exist?
  end

  def sync delete = false
    remote = remote_ids
    local  = local_ids

    if delete
      (local - remote).each do |id|
        print "delete #{id}? [y/n]"
        confirm = $stdin.gets.strip
        if confirm == 'y' || confirm == 'yes'
          puts "delete #{id}"
          delete id
        else
          puts "skip #{id}"
        end
      end
      ids = remote
    else
      ids = (remote + local).uniq
    end

    FileUtils.cd @dir do
      ids.each do |id|
        if File.exist? id
#           FileUtils.cd id do
#             c = "git pull"
#             Kernel.system c
#           end
        else
          c = "git clone git@gist.github.com:#{id}.git"
          Kernel.system c
        end
      end
    end
  end

  def pull_all
    ids = local_ids
    FileUtils.cd @dir do
      ids.each do |id|
        if File.exist? id
          FileUtils.cd id do
            c = "git pull"
            Kernel.system c
          end
        end
      end
    end
  end

  def auth
    user  = `git config --global github.user`.strip
    token = `git config --global github.token`.strip
    user.empty? ? {} : { :login => user, :token => token }
  end

  def post params
    url = URI.parse('http://gist.github.com/gists')
    res = Net::HTTP.post_form(url, params)
    if res.is_a?(Net::HTTPClientError) or res.is_a?(Net::HTTPServerError) then
      raise RuntimeError, res.inspect
    end
    res['Location']
  end

  def build_params paths
    list = (Array === paths ? paths : [paths]).map { |i| Pathname.new i }
    raise InvalidFileException if list.any?{ |i| !i.file? }

    params = {}
    list.each_with_index do |i, index|
      params["file_ext[gistfile#{index + 1}]"] = i.extname
      params["file_name[gistfile#{index + 1}]"] = i.basename.to_s
      params["file_contents[gistfile#{index + 1}]"] = IO.read(i)
    end
    params
  end

  def create paths, opt = { :private => false }
    params = build_params paths
    if opt[:private]
      params['private'] = 'on'
    end
    post params.merge(auth)
  end

  # `figlet -f contributed/bdffonts/clb8x8.flf gisty`.gsub('#', 'm')
  AA = <<-EOS
            mm                mm             
                              mm             
  mmmmmm  mmmm      mmmmm   mmmmmm   mm  mm  
 mm   mm    mm     mm         mm     mm  mm  
 mm   mm    mm      mmmm      mm     mm  mm  
  mmmmmm    mm         mm     mm      mmmmm  
      mm  mmmmmm   mmmmm       mmm       mm  
  mmmmm                               mmmm   
  EOS
end