epc 1.0.2

data/lib/epc/command/login_command.rb

@@ -0,0 +1,32 @@
+module EPC::Command
+  class LoginCommand < BaseCommand
+
+    dont_require_login
+
+    def execute(email = nil)
+      if blank?(email)
+        say("You must supply an email address")
+        return 1
+      end
+      
+      password = options[:password]
+      password = ask("Password: ") { |q| q.echo = '*' } unless password
+      
+      begin
+        status, response, headers = client.post("#{EPC::Config::TOKENS_PATH}/retrieve", { :email => email, :password => password }, true)
+        if status.successful? && response
+          EPC::Config.store_caller_id(response[:caller_id])
+          EPC::Config.store_auth_token(response[:authentication_token])
+          EPC::Config.store_username(email)
+          say("Successfully logged into [#{target_url}]")
+        elsif response
+          say("Login failed [#{response[:message]}]. Please make sure you entered the correct email and password.")
+        end
+        return status
+      rescue Exception => ex
+        say("Login failed [#{ex}].")
+        return 1
+      end    
+    end
+  end
+end

data/lib/epc/command/logout_command.rb

@@ -0,0 +1,20 @@
+module EPC::Command
+  class LogoutCommand < BaseCommand
+    def execute
+      
+      begin
+        if EPC::Config.remove_auth_token
+          EPC::Config.remove_username
+          say("Successfully logged out of [#{target_url}]")
+          return 0
+        else
+          say("Logout failed [#{response[:message]}]. Please try again later.")
+          return 1
+        end
+      rescue Exception => ex
+        say("Logout failed [#{ex}].")
+        return 1
+      end
+    end
+  end
+end

data/lib/epc/command/objectrole/create_objectrole_command.rb

@@ -0,0 +1,19 @@
+module EPC::Command
+  class CreateObjectroleCommand < BaseCommand
+    def execute(*params)
+      obj_type, obj_id = params[0].split(":") rescue [nil, nil]
+      name = params[1]
+
+      raise FatalError, "Parameters incorrectly specified" if obj_type.blank? || obj_id.blank? || name.blank?
+
+      status, response, headers = client.post(EPC::Config::ROLES_PATH, {:name => name, :owner_type => obj_type, :owner_id => obj_id})
+
+      if status.successful?
+        say("Role [#{name}] created")
+      else
+        say("Request failed: [#{response[:message]}]")
+      end
+
+    end
+  end
+end

data/lib/epc/command/project/create_project_command.rb

@@ -0,0 +1,78 @@
+module EPC::Command
+  class CreateProjectCommand < BaseCommand
+    def execute(*params)
+
+      target_type, target_id = params.first.split(":") rescue [nil, nil]
+      project_name = params[1]
+      uri_name = params[2]
+
+      raise FatalError, "You have to specify a project name to run this command" if project_name.blank?
+      raise InputError, "Target type must be: [Solution]" if target_type.present? && target_type.downcase != "solution"
+       
+      if numeric?(@options[:type])
+        type_id = @options[:type].to_i
+      elsif @options[:type].present?
+        type_id = get_resource_id(EPC::Config::PROJECT_TYPES_PATH, :name, @options[:type], :case_sensitivity => false)
+      end
+
+      if type_id.blank? || type_id == 0
+        choose do |menu|
+          menu.prompt = "Please choose the project type: " 
+
+          status, response, headers = client.get(EPC::Config::PROJECT_TYPES_PATH)
+          if status.failure?
+            say("Request failed: [#{response[:message]}")
+            return status
+          end
+
+          if response.empty?
+            say("No project types defined yet")
+            return status
+          end
+
+          response.each do |type|
+            menu.choice(type[:name]) { type_id = type[:id]}
+          end
+
+        end
+      end
+
+      path = File.expand_path(".")
+
+      solution_id, solution_name = infer_solution_context(target_id, path, :get_solution_name => true)
+
+      raise InputError, "Request failed: [Solution not found]" if solution_name.nil?
+
+      uri_name = project_name if uri_name.nil?
+      
+      proceed = ask_yn("You are creating the '#{project_name}' project as part of the '#{solution_name}' solution. Correct? [Yn]: ")
+      if proceed.upcase == 'N'
+        return 1
+      end
+      
+      begin
+        status, response, headers = client.post("#{EPC::Config::PROJECTS_PATH}",
+          { :project_name => project_name, :solution_name => solution_name, :uri_name => uri_name, :project_type_id => type_id})
+        if status.successful?
+          if options[:nodir]
+            say("Successfully created the project with [#{target_url}]")
+            return status
+          end
+          result = mkdir(project_name)
+          if result == :ok
+            EPC::Config.write_project_metadata(response[:id], client)
+            say("Successfully created the project with [#{target_url}] and created your local directory.")
+          else
+            say("Successfully created the project with [#{target_url}], but FAILED to create your local directory.")
+          end
+        else
+          say("Project creation failed [#{response[:message]}].")
+        end
+      rescue Exception => ex
+        say("Project creation failed [#{ex}].")
+        return 1
+      end
+      return status
+    end
+  end
+end

data/lib/epc/command/pull_command.rb

@@ -0,0 +1,209 @@
+require 'rubygems'
+require 'fileutils'
+require 'zip/zipfilesystem'
+
+module EPC::Command
+  class PullCommand < BaseCommand
+    SLEEP_TIME  = 1
+    
+    # Numerators are in secs
+    TICKER_TICKS  = 25/SLEEP_TIME
+    GIVEUP_TICKS  = 120/SLEEP_TIME
+    
+    def execute(solution_name = nil)
+      first_time_pull = false
+      
+      path = File.expand_path(".")
+      create_dir = false
+
+      if numeric?(solution_name)
+        solution_name = get_resource_name(EPC::Config::SOLUTIONS_PATH, :id, solution_name)
+      end
+
+      if !EPC::Config.is_solution_dir?(path) && !EPC::Config.is_project_dir?(path)
+        if solution_name.nil?
+          raise FatalError, "Solution could not be inferred"
+        else
+          create_dir = true
+          first_time_pull = true
+        end
+      elsif EPC::Config.is_project_dir?(path)
+        path = File.join(path, "..")
+      end
+
+      return unless check_pull_directory(project_solution_path(path))
+      
+      solution_name = EPC::Config.get_solution_value(project_solution_path(path), "name") if solution_name.nil?
+
+      proceed = ask_yn("You are pulling the '#{solution_name}' solution. Correct? [Yn]: ")
+      if proceed.upcase == 'N'
+        return 1
+      end
+
+      
+      begin
+        status, id = create_pull(solution_name)
+        if status.successful?
+          say("Pulling - ")
+          shown, url = poll_for_pull_url(id)
+          if shown
+            tmpdir = Dir.tmpdir
+            pulled, zip_path = pull_zip(url, tmpdir)
+            if pulled
+              if create_dir
+                return 1 unless prep_solution_directory(path, solution_name)
+                path = File.join(path, solution_name)
+              end
+              unzip_solution(zip_path, path)
+              confirm_pull(id)
+              refresh_command = EPC::Command::RefreshSolutionCommand.new(client, @options)
+              refresh_command.options[:prompt_refresh] = false if first_time_pull
+              refresh_command.execute(path)
+              return 0
+            end
+          end
+        end
+      rescue Exception => ex
+        say("Pull failed [#{ex.to_s}].")
+      end
+      return 1
+    end
+    
+    private
+
+    def create_pull(solution_name)
+      status, response, headers = client.post("#{EPC::Config::PULLS_PATH}", { :solution_name => solution_name })
+      if status.successful?
+        say("Successfully created a pull with [#{target_url}]")
+        return status, response[:id]
+      else
+        raise Exception, response[:message]
+        return status
+      end
+    end
+    
+    def confirm_pull(id)
+      status, response, headers = client.put("#{EPC::Config::PULLS_PATH}/#{id}/confirm", { :pulled => 'yes' })
+      if status.successful?
+        say("Successfully confirmed the pull with [#{target_url}]")
+        true
+      else
+        say("Pull confirmation failed [#{response[:message]}].")
+        false
+      end
+    end
+    
+    def sign_pull(id)
+      status, response, headers = client.put("#{EPC::Config::PULLS_PATH}/#{id}/sign")
+      if status.successful?
+        return status, nil, response[:signed_url]
+      else
+        return status, response[:error_code], nil
+      end
+    end
+
+    def prep_solution_directory(path, solution_name)
+      solution_id = get_resource_id(EPC::Config::SOLUTIONS_PATH, :name, solution_name)
+      if solution_id.nil?
+        say("Solution not found")
+        return false
+      end
+
+      begin
+        Dir.mkdir(File.join(path, solution_name))
+      rescue Exception => ex
+        say(ex.to_s)
+        return false
+      end
+
+      begin
+        File.open(File.join(path, solution_name, ".epc_solution"), 'w') do |f|
+          f.write({:id => solution_id, :name => solution_name}.to_json)
+        end
+      rescue Exception => ex
+        say(ex.to_s)
+        return false
+      end
+      return true
+    end
+    
+    def poll_for_pull_url(id)
+      count = 0
+      failed = false
+      
+      while count <= GIVEUP_TICKS
+        unless count > TICKER_TICKS
+          STDOUT.print('.')
+          STDOUT.flush
+        end 
+        sleep(SLEEP_TIME)
+        
+        status, code, url = sign_pull(id)
+        if status.successful? || (status == 404 && code == 801)
+          count += 1
+          next if url.nil? || url.empty?
+          break
+        else
+          say("Poller failed to get the pull URL. Aborting.")
+          failed = true
+          break
+        end
+      end
+
+
+      if url
+        [true, url]
+      elsif failed
+        [false, nil]
+      else
+        if count > GIVEUP_TICKS
+          say("\nServer is taking too long to update the pull URL. Please contact the AgileMethods support team.")
+        end
+        [false, nil]
+      end
+    end
+    
+    def check_pull_directory(path)
+      unless File.exists?(path)
+        say("Pull path does not exist")
+        return false
+      end
+      
+      unless File.directory?(path)
+        say("Pull path is not a directory")
+        return false
+      end
+      return true
+    end
+    
+    def pull_zip(url, path)
+      status, response, headers = http_client.get(url)
+      if status.successful?
+        say("Successfully pulled the zip file from the build repository")
+
+        zip_path = File.join(path, "pulled#{rand(9999)}.zip")
+        open(zip_path, "wb") { |io| io.write(response) }
+        [true, zip_path]
+      else
+        say("Failed to pull the zip file from the build repository [status = #{status}]. Aborting pull command.")
+        [false, nil]
+      end
+    end
+    
+    def unzip_solution(zip_path, solution_path)
+      Zip::ZipFile.foreach(zip_path) do |entry|
+        file_path = File.join(solution_path, entry.to_s)
+        FileUtils.mkdir_p File.dirname(file_path)
+        if entry.file?
+          FileUtils.rm(file_path) if File.exists?(file_path)
+          entry.extract(file_path)
+        end
+      end
+
+      say("Solution skeleton has been extracted to #{solution_path}")
+
+      ensure File.delete zip_path if File.exists? zip_path
+    end
+    
+  end
+end

data/lib/epc/command/push_command.rb

@@ -0,0 +1,194 @@
+require 'rubygems'
+require 'digest/sha1'
+require 'fileutils'
+require 'find'
+require 'pathname'
+require 'zip/zipfilesystem'
+
+module EPC::Command
+  class PushCommand < BaseCommand
+    ZIP_DIR_EXCLUDES = %w(target)
+    ZIP_FILE_EXCLUDES = %w(pom.xml)
+    ZIP_EXT_EXCLUDES = ['.class', '.jar']
+
+    def execute(project_name = nil)
+      path = "."  
+      path = File.expand_path(path)
+
+
+      if @options[:no_poll].present? && @options[:timeout].present?
+        raise InputError, "You cannot specify both --nopoll and --timeout options at the same time"
+      end
+
+      @timeout = GIVEUP_TICKS
+      parse_timeout_value
+
+      solution_id, solution_name = infer_solution_context(nil, path)
+
+      raise FatalError, "Solution could not be inferred" if solution_id.nil?
+
+      if EPC::Config.is_solution_dir?(path)
+        if project_name.nil?
+          proceed = ask_yn("You are pushing the [#{solution_name}] solution. Correct: [Yn]? ") 
+          paths = Dir.glob(path +"/*/").select{|dir| EPC::Config.is_project_dir?(dir)}
+        else
+          project_id, p_name = infer_project_context(project_name, path, solution_id, {:get_project_name => true})
+          raise FatalError, "Project not found. Please run this command with the correct project name." if !project_name.nil? && (p_name.nil? || p_name.empty?)
+          proceed = ask_yn("You are pushing the [#{p_name}] project belonging to the [#{solution_name}] solution. Correct: [Yn]? ") 
+          paths = [path + "/#{project_name}"]
+        end
+      elsif EPC::Config.is_project_dir?(path)
+        project_id, project_name = infer_project_context(p_name, path, solution_id, false) 
+        raise FatalError if project_id.nil?
+        proceed = ask_yn("You are pushing the [#{project_name}] project belonging to the [#{solution_name}] solution. Correct: [Yn]? ") 
+        paths = [path]
+      else
+        say("This command must be run from a solution/project directory")
+        return 1
+      end
+
+      if (proceed.upcase == "N" rescue true )
+        return 1
+      end
+
+      raise FatalError, "Nothing to push" if paths.empty?
+
+      @versions = []
+      paths.each do |path|
+        project_id, project_name = infer_project_context(nil, path, solution_id, {:get_project_name => true})
+        say("\nPushing #{project_name}...")
+        begin
+          zip_path = zip_source(path)
+          return 1 unless zip_path
+          sha1 = sha1(zip_path)
+          status, url, signature, id = create_push(project_name, solution_name, sha1, @options[:note])
+          if status.successful?
+            pushed = push_zip(zip_path, url, signature)
+            resp = confirm_push(id)
+            if resp[:build_id].present? && !@options[:skip_build]
+              poll_for_build_status(resp[:build_id]) unless @options[:no_poll].present?
+            end
+            @versions << {project_name => resp[:project_version]} if pushed
+          else
+            return 1
+          end
+        rescue Exception => ex
+          say("Push failed [#{ex}].")
+          return 1
+        end
+      end
+      return 0
+    end
+
+    def execute_internal
+      code = execute
+      return code, @versions
+    end
+
+    private
+
+    def create_push(project_name, solution_name, sha1, note = nil)
+      status, response, headers = client.post("#{EPC::Config::PUSHES_PATH}",
+                                              { :project_name => project_name, :solution_name => solution_name, :sha1 => sha1, :note => note})
+      if status.successful?
+        say("Successfully created a push with [#{target_url}]")
+        return status, response[:url], response[:signature], response[:id]
+      else
+        raise BasicError, "Push creation failed [#{response[:message]}]. Aborting push command."
+      end
+    end
+
+    def confirm_push(id)
+      status, response, headers = client.put("#{EPC::Config::PUSHES_PATH}/#{id}/confirm", { :pushed => 'yes' })
+      if status.successful?
+        say("Successfully confirmed the push with [#{target_url}]")
+        response
+      else
+        say("Push confirmation failed [#{response[:message]}].")
+        nil
+      end
+    end
+
+    def zip_source(path)
+      if check_push_directory(path)
+        tmpdir = Dir.tmpdir
+        zip_path = File.join(tmpdir, "#{rand(9999)}pushed.zip")
+        FileUtils.rm_f(zip_path) if File.exists?(zip_path)
+        Zip::ZipFile.open(zip_path, Zip::ZipFile::CREATE) { |zf|
+          full_paths, relative_paths = recurse_project_directory(path)
+          full_paths.each_index do |i|
+            zf.add(relative_paths[i], full_paths[i])
+          end
+        }
+      end
+
+      zip_path
+    end
+
+    def recurse_project_directory(projdir)
+      full_paths = Array.new
+      relative_paths = Array.new
+      toplevel = Pathname(projdir)
+
+
+      Find.find(projdir) { |path|
+        if File.directory?(path)
+          if ZIP_DIR_EXCLUDES.include?(File.basename(path.downcase))
+            Find.prune  # skip this dir
+          else
+            full_paths << path
+
+            current = Pathname.new(path)
+            relative_paths << current.relative_path_from(toplevel)
+          end
+        else
+          current = Pathname.new(path)
+          relative_path = current.relative_path_from(toplevel)
+
+          unless ZIP_FILE_EXCLUDES.include?(File.basename(relative_path.to_s.downcase)) ||
+            (ZIP_EXT_EXCLUDES.include?(File.extname(relative_path.to_s.downcase)) && relative_path.to_s !~ /^lib/)
+            full_paths << path
+            relative_paths << relative_path.to_s
+          end
+        end    
+      }
+
+      return full_paths, relative_paths
+    end
+
+    def push_zip(zip_path, url, signature)
+      # create API returns a URL that includes protocol://host/solution_name/project_name/version
+
+      status, response, headers = http_client.post(url,
+                                                   { :file => File.new(zip_path, "rb") }, nil, { 'X-AGILEMETHODS-SIGNATURE' => signature }, true)
+      if status.successful?
+        say("Successfully pushed the zip file to the build area")
+        true
+      else
+        raise BasicError, "Failed to push the zip file to the build area [status = #{status}]. Aborting push command."
+      end
+
+    ensure File.delete zip_path if File.exists? zip_path
+    end
+
+    def check_push_directory(path)
+      project_type = EPC::Config.get_project_value(path, "project_type")["name"] rescue ""
+      unless File.exists?(path)
+        say("#{path} does not exist")
+        return false
+      end
+
+      unless File.directory?(path)
+        say("#{path} is not a directory")
+        return false
+      end
+
+      if project_type == "Java Web Archive" || project_type == "Java Library"
+        src_path = File.join(path, "src")
+        raise BasicError, "#{path} does not include a src directory" if !File.exists?(src_path) && !File.directory?(src_path)
+      end
+
+      return true
+    end
+  end
+end