trackinator 0.0.2 → 0.0.4

data/bin/trackinate

@@ -9,9 +9,9 @@ opts = Trollop::options do
   opt :youtrack_password, "Your YouTrack password", :type => :string, :short => "p"
   opt :google_username, "Your Google username", :type => :string, :short => "g"
   opt :google_password, "Your Google password", :type => :string, :short => "a"
-  opt :youtrack_host, "YouTrack host", :type => :string, :short => "o"
-  opt :youtrack_port, "YouTrack port", :type => :int, :default => 80, :short => "r"
-  opt :youtrack_path_prefix, "YouTrack path prefix (e.g. '/youtrack/')", :type => :string, :default => "/", :short =>"e"
+  opt :host, "YouTrack host", :type => :string, :short => "o"
+  opt :port, "YouTrack port", :type => :int, :default => 80, :short => "r"
+  opt :path_prefix, "YouTrack path prefix (e.g. '/youtrack/')", :type => :string, :default => "/", :short =>"e"
   opt :create_rc, "Create a .trackinatorrc file in your home dir", :default => false, :short => "c"
 end
 
@@ -32,7 +32,7 @@ Trollop::die :youtrack_username, "is required" if opts[:youtrack_username].nil?
 Trollop::die :youtrack_password, "is required" if opts[:youtrack_password].nil?
 Trollop::die :google_username, "is required" if opts[:google_username].nil?
 Trollop::die :google_password, "is required" if opts[:google_password].nil?
-Trollop::die :youtrack_host, "is required" if opts[:youtrack_host].nil?
+Trollop::die :host, "is required" if opts[:host].nil?
 
 if opts[:create_rc]
   unless File.exists?("#{Etc.getpwuid.dir}/.trackinatorrc")

data/lib/trackinator/google.rb

@@ -0,0 +1,43 @@
+module Trackinator
+  class Google
+    @client
+
+    def initialize opts
+      @client = GData::Client::Spreadsheets.new
+      @client.clientlogin(opts[:google_username], opts[:google_password])
+    end
+
+    def get_tickets file_name
+      spreadsheet_feed = @client.get("http://spreadsheets.google.com/feeds/worksheets/#{get_spreadsheet_key(file_name)}/private/full").to_xml
+      spreadsheet_list_data = @client.get(spreadsheet_feed.elements[1, 'entry'].elements[1, 'content'].attributes['src']).to_xml
+
+      tickets = []
+
+      spreadsheet_list_data.elements.each('entry') do |entry|
+        tickets << get_ticket_data(entry)
+      end
+
+      tickets
+    end
+
+    def get_spreadsheet_key file_name
+      doc_feed = @client.get("http://spreadsheets.google.com/feeds/spreadsheets/private/full").to_xml
+
+      doc_feed.elements.each ('entry') do |entry|
+        if entry.elements['title'].text.eql? file_name
+          return entry.elements[1].text[/spreadsheets\/(.*)/, 1]
+        end
+      end
+    end
+
+    def get_ticket_data entry
+      data = {}
+
+      REXML::XPath.match(entry, 'gsx:*').each do |col|
+        data[col.name] = URI.escape(col.text) unless col.text.nil?
+      end
+
+      data
+    end
+  end
+end

data/lib/trackinator/importer.rb

@@ -3,174 +3,25 @@ require 'CSV'
 
 require 'gdata'
 
+require 'trackinator/google'
+require 'trackinator/you_track'
+
 module Trackinator
   class Importer
-    @youtrack_cookie
-    @youtrack_connection
-
-    @google_headers
-    @stack
-
-    @col_count
-
-    @youtrack_host
-    @youtrack_port
-    @youtrack_path_prefix
+    @you_track
 
     def initialize opts
-      @stack = []
-      @google_connection = Net::HTTP
-      @google_client = GData::Client::Spreadsheets.new
-
-      google_login = opts[:google_username]
-      youtrack_login = opts[:youtrack_username]
-
-      @youtrack_host = opts[:youtrack_host]
-      @youtrack_port = opts[:youtrack_port]
-      @youtrack_path_prefix = opts[:youtrack_path_prefix]
-
-      google_password = opts[:google_password]
-      youtrack_password = opts[:youtrack_password]
-
-      login_youtrack youtrack_login, youtrack_password
-
-      @google_client.clientlogin(google_login, google_password)
+      @you_track = YouTrack.new opts
+      @google = Google.new opts
     end
 
     def import file_name
-      spreadsheet_feed = @google_client.get("http://spreadsheets.google.com/feeds/worksheets/#{get_spreadsheet_key(file_name)}/private/full").to_xml
-      spreadsheet_list_data = @google_client.get(spreadsheet_feed.elements[1, 'entry'].elements[1, 'content'].attributes['src']).to_xml
-
-      spreadsheet_list_data.elements.each('entry') do |entry|
-        ticket_data = get_ticket_data entry
-        issue_id = is_issue_exists? ticket_data
-        false unless !issue_id.nil? ? update_ticket(issue_id, ticket_data) : create_ticket(ticket_data)
-      end
-    end
-
-    private
-
-    # Google API methods
-
-    def get_spreadsheet_key file_name
-      doc_feed = @google_client.get("http://spreadsheets.google.com/feeds/spreadsheets/private/full").to_xml
+      ticket_data = @google.get_tickets file_name
 
-      doc_feed.elements.each ('entry') do |entry|
-        if entry.elements['title'].text.eql? file_name
-          return entry.elements[1].text[/spreadsheets\/(.*)/, 1]
-        end
+      ticket_data.each do |entry|
+        issue_id = @you_track.is_issue_exists? entry
+        false unless !issue_id.nil? ? @you_track.update_ticket(issue_id, entry) : @you_track.create_ticket(entry)
       end
     end
-
-    def get_ticket_data entry
-      data = {}
-
-      REXML::XPath.match(entry, 'gsx:*').each do |col|
-        data[col.name] = URI.escape(col.text) unless col.text.nil?
-      end
-
-      data
-    end
-
-    # YouTrack API methods
-
-    def login_youtrack user_name, password
-      @youtrack_connection = Net::HTTP.new @youtrack_host, @youtrack_port
-      response = @youtrack_connection.post "#{@youtrack_path_prefix}rest/user/login", "login=#{user_name}&password=#{password}"
-      @youtrack_cookie = response.response['Set-Cookie'].split('; ')[0]
-    end
-
-    def create_ticket data
-      issue_id, create_response = create_youtrack_ticket data
-      success = create_response.eql? "Created"
-
-      success ? (success = update_ticket(issue_id, data)) : (return success)
-      success ? update_dependencies([issue_id, data['id']]) : success
-    end
-
-    def update_ticket issue_id, data
-      success = set_platform(issue_id, data['platform'])
-      success ? (success = set_summary_and_description(issue_id, data['summary'], data['description'])) : (return success)
-      success ? (success = set_type(issue_id, data['type'])) : (return success)
-      success ? (success = set_import_identifier(issue_id, "#{data['project']}-#{data['id']}")) : (return success)
-      success ? (success = set_design_reference(issue_id, "#{data['references']}")) : (return success) unless data['references'].nil?
-
-      success
-    end
-
-    def update_dependencies issue
-      issue_id = issue[0]
-      issue_level = issue[1]
-
-      if @stack.empty?
-        @stack.push [issue_id, issue_level]
-      else
-        last_issue = @stack.last
-        last_issue_id = last_issue[0]
-        last_issue_level = last_issue[1]
-
-        if issue_level.length <= last_issue_level.length
-          @stack.pop
-          update_dependencies issue
-        else
-          success = create_dependency last_issue_id, issue_id
-          @stack.push issue
-
-          success
-        end
-      end
-    end
-
-    # YouTrack API calls
-
-    def is_issue_exists? data
-      find_response_xml = REXML::Document.new(@youtrack_connection.get("#{@youtrack_path_prefix}rest/issue/byproject/#{data['project']}?filter=Import+Identifier:+#{data['project']}-#{data['id']}", { 'Cookie' => @youtrack_cookie, 'Content-Type' => 'text/plain; charset=utf-8' }).body)
-      find_response_xml.elements['issues'].length > 0 ? find_response_xml.elements['issues/issue'].attributes['id'] : nil
-    end
-
-    def create_youtrack_ticket data
-      response = @youtrack_connection.put("#{@youtrack_path_prefix}rest/issue?project=#{data['project']}&summary=#{data['summary']}&description=#{data['description']}&priority=#{data['priority']}", nil, { 'Cookie' => @youtrack_cookie, 'Content-Type' => 'text/plain; charset=utf-8' })
-      return response.header["Location"].split("/").last, response.header.msg
-    end
-
-    def create_dependency parent_id, child_id
-      response = @youtrack_connection.post("#{@youtrack_path_prefix}rest/issue/#{parent_id}/execute?command=parent+for+#{child_id}&disableNotifications=true", nil, { 'Cookie' => @youtrack_cookie })
-      response.header.msg.eql? "OK"
-    end
-
-    def set_summary_and_description issue_id, summary, description
-      return true if summary.nil?
-
-      response = @youtrack_connection.post("#{@youtrack_path_prefix}rest/issue/#{issue_id}/?summary=#{summary}&description=#{description}&disableNotifications=true", nil, { 'Cookie' => @youtrack_cookie })
-      response.header.msg.eql? "OK"
-    end
-
-    def set_platform issue_id, platform
-      return true if platform.nil?
-
-      response = @youtrack_connection.post("#{@youtrack_path_prefix}rest/issue/#{issue_id}/execute?command=Platform+#{platform}&disableNotifications=true", nil, { 'Cookie' => @youtrack_cookie })
-      response.header.msg.eql? "OK"
-    end
-
-    def set_type issue_id, type
-      return true if type.nil?
-
-      response = @youtrack_connection.post("#{@youtrack_path_prefix}rest/issue/#{issue_id}/execute?command=Type+#{type}&disableNotifications=true", nil, { 'Cookie' => @youtrack_cookie })
-      response.header.msg.eql? "OK"
-    end
-
-    def set_import_identifier issue_id, import_id
-      return true if import_id.nil?
-
-      response = @youtrack_connection.post("#{@youtrack_path_prefix}rest/issue/#{issue_id}/execute?command=Import+Identifier+#{import_id}&disableNotifications=true", nil, { 'Cookie' => @youtrack_cookie })
-      response.header.msg.eql? "OK"
-    end
-
-    def set_design_reference issue_id, reference
-      return true if reference.nil?
-
-      response = @youtrack_connection.post("#{@youtrack_path_prefix}rest/issue/#{issue_id}/execute?command=Design+Reference+#{reference}&disableNotifications=true", nil, { 'Cookie' => @youtrack_cookie })
-      response.header.msg.eql? "OK"
-    end
   end
 end

data/lib/trackinator/version.rb

@@ -1,3 +1,3 @@
 module Trackinator
-  VERSION = "0.0.2"
+  VERSION = "0.0.4"
 end

data/lib/trackinator/you_track.rb

@@ -0,0 +1,122 @@
+module Trackinator
+  class YouTrack
+    @stack
+
+    @cookie
+    @connection
+
+    @host
+    @port
+    @path_prefix
+
+    def initialize opts
+      @stack = []
+
+      @host = opts[:host]
+      @port = opts[:port]
+      @path_prefix = opts[:path_prefix]
+
+      login opts[:youtrack_username], opts[:youtrack_password]
+    end
+
+    def login username, password
+      @connection = Net::HTTP.new @host, @port
+      response = @connection.post "#{@path_prefix}rest/user/login", "login=#{username}&password=#{password}"
+      @cookie = response.response['Set-Cookie'].split('; ')[0]
+    end
+
+    def create_ticket data
+      issue_id, create_response = create_youtrack_ticket data
+      success = create_response.eql? "Created"
+
+      success ? (success = update_ticket(issue_id, data)) : (return success)
+      success ? update_dependencies([issue_id, data['id']]) : success
+    end
+
+    def update_ticket issue_id, data
+      success = set_platform(issue_id, data['platform'])
+      success ? (success = set_summary_and_description(issue_id, data['summary'], data['description'])) : (return success)
+      success ? (success = set_type(issue_id, data['type'])) : (return success)
+      success ? (success = set_import_identifier(issue_id, "#{data['project']}-#{data['id']}")) : (return success)
+      success ? (success = set_design_reference(issue_id, "#{data['references']}")) : (return success) unless data['references'].nil?
+
+      success
+    end
+
+    def update_dependencies issue
+      issue_id = issue[0]
+      issue_level = issue[1]
+
+      if @stack.empty?
+        @stack.push [issue_id, issue_level]
+      else
+        last_issue = @stack.last
+        last_issue_id = last_issue[0]
+        last_issue_level = last_issue[1]
+
+        if issue_level.length <= last_issue_level.length
+          @stack.pop
+          update_dependencies issue
+        else
+          success = create_dependency last_issue_id, issue_id
+          @stack.push issue
+
+          success
+        end
+      end
+    end
+
+    # YouTrack API calls
+
+    def is_issue_exists? data
+      find_response_xml = REXML::Document.new(@connection.get("#{@path_prefix}rest/issue/byproject/#{data['project']}?filter=Import+Identifier:+#{data['project']}-#{data['id']}", { 'Cookie' => @cookie, 'Content-Type' => 'text/plain; charset=utf-8' }).body)
+      find_response_xml.elements['issues'].length > 0 ? find_response_xml.elements['issues/issue'].attributes['id'] : nil
+    end
+
+    def create_youtrack_ticket data
+      response = @connection.put("#{@path_prefix}rest/issue?project=#{data['project']}&summary=#{data['summary']}&description=#{data['description']}&priority=#{data['priority']}", nil, { 'Cookie' => @cookie, 'Content-Type' => 'text/plain; charset=utf-8' })
+      return response.header["Location"].split("/").last, response.header.msg
+    end
+
+    def create_dependency parent_id, child_id
+      response = @connection.post("#{@path_prefix}rest/issue/#{parent_id}/execute?command=parent+for+#{child_id}&disableNotifications=true", nil, { 'Cookie' => @cookie })
+      response.header.msg.eql? "OK"
+    end
+
+    def set_summary_and_description issue_id, summary, description
+      return true if summary.nil? || description.nil?
+
+      response = @connection.post("#{@path_prefix}rest/issue/#{issue_id}/?summary=#{summary}&description=#{description}&disableNotifications=true", nil, { 'Cookie' => @cookie })
+      response.header.msg.eql? "OK"
+    end
+
+    def set_platform issue_id, platform
+      return true if platform.nil?
+
+      response = @connection.post("#{@path_prefix}rest/issue/#{issue_id}/execute?command=Platform+#{platform}&disableNotifications=true", nil, { 'Cookie' => @cookie })
+      response.header.msg.eql? "OK"
+    end
+
+    def set_type issue_id, type
+      return true if type.nil?
+
+      response = @connection.post("#{@path_prefix}rest/issue/#{issue_id}/execute?command=Type+#{type}&disableNotifications=true", nil, { 'Cookie' => @cookie })
+      response.header.msg.eql? "OK"
+    end
+
+    def set_import_identifier issue_id, import_id
+      return true if import_id.nil?
+
+      response = @connection.post("#{@path_prefix}rest/issue/#{issue_id}/execute?command=Import+Identifier+#{import_id}&disableNotifications=true", nil, { 'Cookie' => @cookie })
+      response.header.msg.eql? "OK"
+    end
+
+    def set_design_reference issue_id, reference
+      return true if reference.nil?
+
+      response = @connection.post("#{@path_prefix}rest/issue/#{issue_id}/execute?command=Design+Reference+#{reference}&disableNotifications=true", nil, { 'Cookie' => @cookie })
+      response.header.msg.eql? "OK"
+    end
+
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: trackinator
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-31 00:00:00.000000000 Z
+date: 2012-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gdata_19
@@ -62,8 +62,10 @@ files:
 - Rakefile
 - bin/trackinate
 - lib/trackinator.rb
+- lib/trackinator/google.rb
 - lib/trackinator/importer.rb
 - lib/trackinator/version.rb
+- lib/trackinator/you_track.rb
 - trackinator.gemspec
 homepage: https://github.com/justincbeck/trackinator
 licenses: []