lib/toolshed/commands/create_ticket_comment.rb in toolshed-1.0.2 vs lib/toolshed/commands/create_ticket_comment.rb in toolshed-1.0.3

@@ -1,21 +1,37 @@
 require 'toolshed/commands/base'
+require 'toolshed/git'
 
 module Toolshed
   module Commands
     class CreateTicketComment < Base
 
       def initialize(options={})
         super(options)
       end
 
+      def self.cli_options
+        {
+          banner: 'Usage: create_ticket_comment [options]',
+          options: {
+            use_defaults: {
+              short_on: '-d'
+            }
+          }
+        }
+      end
+
+      def branch
+        @branch ||= Toolshed::Git::Branch.new
+      end
+
       def execute(args, options = {})
         ticket_tracker_class =  Object.const_get("Toolshed::TicketTracking::#{Toolshed::Client.ticket_tracking_tool.camel_case}")
 
         options = use_ticket_tracker_project_id(options)
         options = use_ticket_tracker_project_name(options)
 
-        default_ticket_id = Toolshed::TicketTracking::story_id_from_branch_name(Toolshed::Git::Base.branch_name)
+        default_ticket_id = Toolshed::TicketTracking::story_id_from_branch_name(branch.name)
         ticket_id = read_user_input("Ticket ID (Default: #{default_ticket_id}):", options.merge!({ default: default_ticket_id }))
         options.merge!({ ticket_id: ticket_id })
 
         ticket_tracker = ticket_tracker_class.create_instance(options)