trackinator 0.0.7 → 0.0.8

data/bin/trackinate

@@ -1,6 +1,8 @@
 #!/usr/bin/ruby
 
 require 'trackinator/importer'
+require 'trackinator/you_track'
+require 'trackinator/google'
 require 'trollop'
 require 'etc'
 
@@ -34,6 +36,13 @@ Trollop::die :google_username, "is required" if opts[:google_username].nil?
 Trollop::die :google_password, "is required" if opts[:google_password].nil?
 Trollop::die :host, "is required" if opts[:host].nil?
 
+you_track = Trackinator::YouTrack.new opts
+google = Trackinator::Google.new opts
+
+Trollop::die "Check your YouTrack credentials and try again" unless you_track.is_logged_in?
+Trollop::die "Check your Google credentials and try again" unless google.is_logged_in?
+Trollop::die "You must supply a Google doc filename to import" if ARGV.empty? && opts[:create_rc]
+
 if opts[:create_rc]
   unless File.exists?("#{Etc.getpwuid.dir}/.trackinatorrc")
     file = File.new("#{Etc.getpwuid.dir}/.trackinatorrc", "w")
@@ -49,5 +58,13 @@ if opts[:create_rc]
   end
 end
 
-importer = Trackinator::Importer.new opts
-importer.import ARGV[0]
+if ARGV.length == 1
+  importer = Trackinator::Importer.new you_track, google
+  issues = importer.import ARGV[0], "YTTP"
+
+  if issues.length > 0
+    issues.each do |issue|
+      p issue
+    end
+  end
+end

data/lib/trackinator/google.rb

@@ -1,10 +1,16 @@
 module Trackinator
   class Google
-    @client
-
     def initialize opts
       @client = GData::Client::Spreadsheets.new
-      @client.clientlogin(opts[:google_username], opts[:google_password])
+      begin
+        @token = @client.clientlogin(opts[:google_username], opts[:google_password])
+      rescue
+        @token = nil
+      end
+    end
+
+    def is_logged_in?
+      !@token.nil?
     end
 
     def get_tickets file_name
@@ -20,6 +26,8 @@ module Trackinator
       tickets
     end
 
+    private
+
     def get_spreadsheet_key file_name
       doc_feed = @client.get("http://spreadsheets.google.com/feeds/spreadsheets/private/full").to_xml
 

data/lib/trackinator/importer.rb

@@ -3,25 +3,69 @@ require 'CSV'
 
 require 'gdata'
 
-require 'trackinator/google'
-require 'trackinator/you_track'
+require 'trackinator'
 
 module Trackinator
   class Importer
-    @you_track
 
-    def initialize opts
-      @you_track = YouTrack.new opts
-      @google = Google.new opts
+    def initialize you_track, google
+      @you_track = you_track
+      @google = google
     end
 
-    def import file_name
+    def import file_name, project = nil
       ticket_data = @google.get_tickets file_name
 
-      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)
+      issues = validate_tickets(project, ticket_data)
+
+      if issues.length == 0
+        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
+
+      issues
+    end
+
+    private
+
+    def validate_tickets project, tickets
+      issues = []
+
+      issues.concat(@you_track.project_exists?(project))
+      issues.concat(@you_track.you_track_fields_defined?(project, tickets[0].keys))
+
+      tickets.each do |ticket|
+        issues.concat(validate_fields(ticket))
+        issues.concat(validate_formats(ticket))
       end
+
+      issues
+    end
+
+    def validate_fields ticket
+      issues = []
+
+      REQUIRED.each do |req_field|
+        unless ticket.keys.include?(req_field) || ticket["type"].downcase.eql?("story")
+          issues << "Validation Error: Ticket with ID: #{ticket["id"]} is missing required field '#{req_field}'"
+        end
+      end
+
+      issues
+    end
+
+    def validate_formats ticket
+      issues = []
+
+      ticket.keys.each do |key|
+        if Trackinator.const_defined?("#{key.upcase}") && Trackinator.const_get("#{key.upcase}").match(ticket[key].downcase).nil?
+          issues << "Validation Error: Ticket with ID: #{ticket["id"]} has field '#{key}' with invalid format"
+        end
+      end
+
+      issues
     end
   end
 end

data/lib/trackinator/version.rb

@@ -1,3 +1,3 @@
 module Trackinator
-  VERSION = "0.0.7"
+  VERSION = "0.0.8"
 end

data/lib/trackinator/you_track.rb

@@ -1,14 +1,5 @@
 module Trackinator
   class YouTrack
-    @stack
-
-    @cookie
-    @connection
-
-    @host
-    @port
-    @path_prefix
-
     def initialize opts
       @stack = []
 
@@ -19,10 +10,16 @@ module Trackinator
       login opts[:youtrack_username], opts[:youtrack_password]
     end
 
+    def is_logged_in?
+      !@cookie.nil?
+    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]
+      if response.header.msg.eql? "OK"
+        @cookie = response.response['Set-Cookie'].split('; ')[0]
+      end
     end
 
     def create_ticket data
@@ -43,6 +40,43 @@ module Trackinator
       success
     end
 
+    def project_exists?(project)
+      issues = []
+
+      response = @connection.get("#{@path_prefix}rest/admin/project/#{project}", { 'Cookie' => @cookie, 'Content-Type' => 'text/plain; charset=utf-8' })
+      if response.header.msg.eql?("Not Found")
+        issues << "Project Error: Project doesn't exist!"
+      end
+
+      issues
+    end
+
+    def you_track_fields_defined?(project, fields)
+      issues = []
+
+      response = @connection.get("#{@path_prefix}rest/admin/project/#{project}/customfield", { 'Cookie' => @cookie, 'Content-Type' => 'text/plain; charset=utf-8' })
+      response_xml = REXML::Document.new(response.body)
+
+      you_track_fields = []
+
+      response_xml.elements.each('projectCustomFieldRefs/projectCustomField') { |element| you_track_fields << element.attributes["name"].downcase }
+
+      (fields - REQUIRED).each do |document_field|
+        unless you_track_fields.include?(document_field)
+          issues << "Validation Error: Custom field '#{document_field}' not found in YouTrack"
+        end
+      end
+
+      issues
+    end
+
+    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
+
+    private
+
     def update_dependencies issue
       issue_id = issue[0]
       issue_level = issue[1]
@@ -66,13 +100,6 @@ module Trackinator
       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

data/lib/trackinator.rb

@@ -1,5 +1,22 @@
-require "trackinator/version"
+require 'trackinator/version'
+require 'trackinator/google'
+require 'trackinator/you_track'
+
 
 module Trackinator
-  # Your code goes here...
+
+  ID = /^(?:\d+\.?)+$/
+
+  TYPE = /^story$|^feature$|^task$|^bug$/
+
+  PRIORITY = /^low$|^normal$|^high$|^show-stopper$/
+
+  REQUIRED = %w{
+      project
+      id
+      summary
+      description
+      outcome
+  }
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: trackinator
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-01 00:00:00.000000000 Z
+date: 2012-08-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gdata_19