fogbugz 1.0.3 → 1.0.4

data/bin/fogbugz-statuses

@@ -1,22 +1,5 @@
 #!/usr/bin/env ruby
-
-require 'rubygems'
-require 'net/https'
-require 'uri'
-require 'rexml/document'
-require 'optparse'
-
-api_url = ENV['FOGBUGZ_API_URL']
-unless api_url
-  puts "Environment variable FOGBUGZ_API_URL must be set."
-  exit 1
-end
-
-api_token = ENV['FOGBUGZ_API_TOKEN']
-unless api_token
-  puts "Environment variable FOGBUGZ_API_TOKEN must be set."
-  exit 1
-end
+require 'fogbugz/common'
 
 options = {}
 optparse = OptionParser.new do |opts|
@@ -28,33 +11,17 @@ optparse = OptionParser.new do |opts|
   end
 
   options[:resolved] = false
-  opts.on('-r', '--resolved', 'Only show resolved statuses.') do
+  opts.on('--resolved', 'Only show resolved statuses.') do
     options[:resolved] = true
   end
 end
 optparse.parse!
-
-uri = URI format("#{api_url}?cmd=listStatuses&token=%s%s",
-                 URI.escape(api_token),
-                 options[:resolved] ? '&fResolved=1' : '')
-http = Net::HTTP.new(uri.host, uri.port)
-if uri.scheme == 'https'
-  http.use_ssl = true
-  http.verify_mode = OpenSSL::SSL::VERIFY_NONE
-end
-response = http.start { |h| h.request Net::HTTP::Get.new(uri.request_uri) }
-if response.code != '200'
-  puts "HTTP request to #{api_url} failed with code #{response.code}."
-  exit 1
-end
-
-result = REXML::Document.new(response.body)
-error = result.elements['/response/error']
-if error
-  puts "Failed with error: #{error.text}."
+unless ARGV.length == 0
+  puts optparse.help
   exit 1
 end
 
+result = do_api 'listStatuses', options[:resolved] ? { :fResolved => '1' } : {}
 result.elements.to_a('/response/statuses/status').map { |s|
   s.elements['sStatus'].text
 }.uniq!.sort!.each { |s| puts s }

data/bin/fogbugz-stop

@@ -1,49 +1,5 @@
 #!/usr/bin/env ruby
+require 'fogbugz/common'
 
-require 'rubygems'
-require 'net/https'
-require 'uri'
-require 'rexml/document'
-require 'optparse'
-
-api_url = ENV['FOGBUGZ_API_URL']
-unless api_url
-  puts "Environment variable FOGBUGZ_API_URL must be set."
-  exit 1
-end
-
-api_token = ENV['FOGBUGZ_API_TOKEN']
-unless api_token
-  puts "Environment variable FOGBUGZ_API_TOKEN must be set."
-  exit 1
-end
-
-options = {}
-optparse = OptionParser.new do |opts|
-  opts.banner = "usage: #{File::basename(__FILE__)} [options]"
-
-  opts.on_tail('-h', '--help') do
-    puts optparse.help
-    exit 1
-  end
-end
-optparse.parse!
-
-uri = URI format("#{api_url}?cmd=stopWork&token=%s", URI.escape(api_token))
-http = Net::HTTP.new(uri.host, uri.port)
-if uri.scheme == 'https'
-  http.use_ssl = true
-  http.verify_mode = OpenSSL::SSL::VERIFY_NONE
-end
-response = http.start { |h| h.request Net::HTTP::Get.new(uri.request_uri) }
-if response.code != '200'
-  puts "HTTP request to #{api_url} failed with code #{response.code}."
-  exit 1
-end
-
-result = REXML::Document.new(response.body)
-error = result.elements['/response/error']
-if error
-  puts "Failed with error: #{error.text}."
-  exit 1
-end
+parse_opts "usage: #{File::basename(__FILE__)} [options]", 0
+do_api 'stopWork'

data/lib/fogbugz/common.rb

@@ -0,0 +1,370 @@
+require 'net/https'
+require 'uri'
+require 'rexml/document'
+require 'optparse'
+require 'highline'
+require 'tempfile'
+require 'yaml'
+require 'English'
+require 'time'
+require 'chronic'
+
+HighLine.use_color = STDOUT.isatty
+begin
+  require 'win32console' if RUBY_PLATFORM =~ /mingw/
+rescue LoadError
+  HighLine.use_color = false
+end
+
+def r; HighLine.use_color? ? HighLine::RED     : ''; end
+def b; HighLine.use_color? ? HighLine::BLUE    : ''; end
+def g; HighLine.use_color? ? HighLine::GREEN   : ''; end
+def y; HighLine.use_color? ? HighLine::YELLOW  : ''; end
+def m; HighLine.use_color? ? HighLine::MAGENTA : ''; end
+def c; HighLine.use_color? ? HighLine::CLEAR   : ''; end
+
+unless ENV['FOGBUGZ_API_URL']
+  puts "Environment variable FOGBUGZ_API_URL must be set."
+  exit 1
+end
+
+unless ENV['FOGBUGZ_API_TOKEN']
+  puts "Environment variable FOGBUGZ_API_TOKEN must be set."
+  exit 1
+end
+
+def parse_opts(usage, num_args)
+  options = {}
+  optparse = OptionParser.new do |opts|
+    opts.banner = usage
+
+    opts.on_tail('-h', '--help') do
+      puts optparse.help
+      exit 1
+    end
+  end
+  optparse.parse!
+
+  unless ARGV.length == num_args
+    puts optparse.help
+    exit 1
+  end
+  options
+end
+
+def parse_stdin_opts(usage, num_args)
+  options = {}
+  optparse = OptionParser.new do |opts|
+    opts.banner = usage
+
+    opts.on_tail('-h', '--help') do
+      puts optparse.help
+      exit 1
+    end
+
+    options[:file] = nil
+    opts.on('--file=<file>',
+            'Take the case content from the given file. Use - to read from STDIN.') do |file|
+      options[:file] = file
+    end
+
+    options[:template] = nil
+    opts.on('--template=<template>',
+            'Use the file content or - for STDIN as the initial case content.') do |template|
+      options[:template] = template
+    end
+  end
+  optparse.parse!
+
+  unless ARGV.length == num_args or
+      (ARGV.length == num_args + 1 and ARGV.last == '-')
+    puts optparse.help
+    exit 1
+  end
+
+  if ARGV.length == num_args + 1
+    ARGV.pop
+    options[:file] = '-'
+  end
+  options
+end
+
+def get_editor_content(options, template)
+  editor = ENV['EDITOR'] || 'vim'
+  invoke_editor = true
+  old_argv = ARGV.clone
+  if options[:file]
+    if options[:file] == '-'
+      ARGV.replace []
+    else
+      ARGV.replace [options[:file]]
+    end
+    template = ARGF.read
+    invoke_editor = false
+  elsif options[:template]
+    if options[:template] == '-'
+      ARGV.replace []
+    else
+      ARGV.replace [options[:template]]
+    end
+    template = ARGF.read
+  end
+
+  content = template
+  if invoke_editor
+    tempfile = Tempfile.new ['case', '.md']
+    tempfile.write template
+    tempfile.close
+    rc = system "#{editor} #{tempfile.path}"
+    unless rc
+      puts "Editor exited with non-zero status. Aborting."
+      exit 1
+    end
+    tempfile.open
+    content = tempfile.read
+    tempfile.close
+    tempfile.delete
+  end
+
+  data = {}
+  if content =~ /(.*?\n?)^(---\s*$\n?)/m
+    # Combined YAML front matter with text content.
+    begin
+      data = YAML.load($1) || {}
+      data['body'] = $POSTMATCH if $POSTMATCH and not $POSTMATCH.empty?
+    rescue => e
+      puts "Exception reading YAML front matter. #{e.inspect}"
+      exit 1
+    end
+  else
+    begin
+      # YAML only content.
+      data = YAML.load(content)
+      if data.instance_of? String
+        # Text only content.
+        data = { 'body' => content }
+      end
+    rescue => e
+      data = {}
+    end
+  end
+
+  if not data or data.empty?
+    puts "No new content for case. Aborting."
+    exit 1
+  end
+  ARGV.replace old_argv
+  data
+end
+
+def maybe_show(key, xml)
+  if xml and xml.text and not xml.text.empty?
+    format "#{b}%-11.11s#{c}#{xml.text}\n", "#{key}:"
+  else
+    ''
+  end
+end
+
+def maybe_show_array(key, xml)
+  if xml and not xml.empty?
+    format "#{b}%-11.11s#{c}#{xml.join(', ')}\n", "#{key}:"
+  else
+    ''
+  end
+end
+
+def maybe_show_literal(color, xml)
+  if xml and xml.text and not xml.text.empty?
+    "#{color}#{xml.text}#{c}\n"
+  else
+    ''
+  end
+end
+
+def maybe_show_time(key, xml)
+  if xml and xml.text and not xml.text.empty?
+    format "#{b}%-11.11s#{c}#{time_string(xml.text)}\n", "#{key}:"
+  else
+    ''
+  end
+end
+
+def maybe_show_event(color, xml, time_xml)
+  if xml and xml.text and not xml.text.empty?
+    if time_xml and time_xml.text and not time_xml.text.empty?
+      format("#{color}%s at %s.#{c}\n", xml.text, time_string(time_xml.text))
+    else
+      "#{color}#{xml.text}#{c}\n"
+    end
+  else
+    ''
+  end
+end
+
+def maybe_append(key, xml)
+  if xml and xml.text and not xml.text.empty?
+    "# #{key}: #{xml.text}\n"
+  else
+    "# #{key}: <#{key}>\n"
+  end
+end
+
+def maybe_array(key, xml)
+  if xml and not xml.empty?
+    "# #{key}: [#{xml.join(', ')}]\n"
+  else
+    "# #{key}: [<#{key}>]\n"
+  end
+end
+
+def time_short_string(text)
+  if text
+    Time.parse(text).localtime.strftime('%D')
+  else
+    ''
+  end
+end
+
+def time_string(text)
+  if text
+    time = Time.parse(text).localtime
+    format("%s on %s", time.strftime('%-l:%M %p'),
+           time.strftime('%A, %B %e %Y'))
+  else
+    ''
+  end
+end
+
+def parse_time(text)
+  return nil unless text
+  begin
+    time = Time.parse(text)
+    # Ruby 1.8.7 returns the current time on bad parses.
+    raise ArgumentError if Time.now.to_s == time.to_s
+  rescue ArgumentError
+    begin
+      time = Chronic.parse(text).utc.iso8601
+    rescue
+      puts "Unable to parse date #{text}."
+      exit 1
+    end
+  end
+  unless time
+    puts "Unable to parse date #{text}."
+    exit 1
+  end
+  time
+end
+
+def maybe_time(key, xml)
+  if xml and xml.text and not xml.text.empty?
+    "# #{key}: #{time_string(xml.text)}\n"
+  else
+    "# #{key}: <#{key}>\n"
+  end
+end
+
+def maybe_event(xml, time_xml)
+  if xml and xml.text and not xml.text.empty?
+    if time_xml and time_xml.text and not time_xml.text.empty?
+      format("# %s at %s.\n", xml.text, time_string(time_xml.text))
+    else
+      commented = xml.text.gsub(/\n/, "\n# ")
+      "# #{commented}"
+    end
+  else
+    ''
+  end
+end
+
+def maybe_literal(xml)
+  if xml and xml.text and not xml.text.empty?
+    commented = xml.text.gsub(/\n/, "\n# ")
+    "# #{commented}\n"
+  else
+    ''
+  end
+end
+
+def do_api(cmd, params={})
+  api_url = ENV['FOGBUGZ_API_URL']
+  api_token = ENV['FOGBUGZ_API_TOKEN']
+
+  query = ''
+  params.each_pair do |k,v|
+    query << format("&%s=%s", URI.escape(k.to_s), URI.escape(v.to_s)) if v
+  end
+
+  uri = URI "#{api_url}?cmd=#{cmd}&token=#{api_token}#{query}"
+  http = Net::HTTP.new(uri.host, uri.port)
+  if uri.scheme == 'https'
+    http.use_ssl = true
+    http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+  end
+  response = http.start do |h|
+    h.request Net::HTTP::Get.new(uri.request_uri)
+  end
+  if response.code != '200'
+    puts "HTTP request to #{api_url} failed with code #{response.code}."
+    exit 1
+  end
+
+  result = REXML::Document.new(response.body)
+  error = result.elements['/response/error']
+  if error
+    puts "Failed with error: #{error.text}."
+    exit 1
+  end
+  result
+end
+
+def which(cmd)
+  exts = ENV['PATHEXT'] ? ENV['PATHEXT'].split(';') : ['']
+  ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
+    exts.each { |ext|
+      exe = "#{path}/#{cmd}#{ext}"
+      return exe if File.executable? exe
+    }
+  end
+  return nil
+end
+
+def dispatch_subcommand(file, commands = {})
+  basename = File.basename(file)
+  absolute = File.expand_path(file)
+
+  usage = <<HERE
+usage: #{basename} <subcommands> [options]
+
+The subcommands are:
+HERE
+  commands.keys.sort { |a,b| a.to_s <=> b.to_s }.each do |k|
+    usage << format("   %-16.16s%s\n", k, commands[k])
+  end
+  usage << <<HERE
+
+See '#{basename} help <commands>' for more information on a specific command.
+HERE
+
+  if ARGV[0] == 'help'
+    if ARGV[1] and commands[ARGV[1].to_sym]
+      subcommand = ARGV[1]
+      ARGV.replace ['--help']
+      load "#{absolute}-#{subcommand}"
+    elsif ARGV[1] and which("#{basename}-#{ARGV[1]}")
+      command = which("#{basename}-#{ARGV[1]}")
+      system "#{command} --help"
+    else
+      puts usage
+      exit 1
+    end
+  elsif ARGV[0] and commands[ARGV[0].to_sym]
+    load "#{absolute}-#{ARGV.shift}"
+  elsif ARGV[0] and which("#{basename}-#{ARGV[0]}")
+    command = which("#{basename}-#{ARGV.shift}")
+    Process.wait(Process.fork { Process.exec(command, *ARGV) })
+  else
+    puts usage
+    exit 1
+  end
+end