abt-cli 0.0.2 → 0.0.7

data/lib/abt/providers/harvest.rb

@@ -1,51 +1,18 @@
 # frozen_string_literal: true
 
-Dir.glob("#{File.expand_path(__dir__)}/harvest/*.rb").sort.each do |file|
-  require file
-end
+Dir.glob("#{File.expand_path(__dir__)}/harvest/*.rb").sort.each { |file| require file }
+Dir.glob("#{File.expand_path(__dir__)}/harvest/commands/*.rb").sort.each { |file| require file }
 
 module Abt
   module Providers
-    class Harvest
-      class << self
-        def user_id
-          Abt::GitConfig.prompt_global(
-            'abt.harvest.userId',
-            'Please enter your harvest User ID',
-            'In harvest open "My profile". The ID is the number part of the URL you are taken to'
-          )
-        end
-
-        def access_token
-          Abt::GitConfig.prompt_global(
-            'abt.harvest.accessToken',
-            'Please enter your personal harvest access token',
-            'Create your personal access token here: https://id.getharvest.com/developers'
-          )
-        end
-
-        def account_id
-          Abt::GitConfig.prompt_global(
-            'abt.harvest.accountId',
-            'Please enter the harvest account id',
-            'This information is shown next to your generated access token'
-          )
-        end
-
-        def clear
-          Abt::GitConfig.unset_local('abt.harvest.projectId')
-          Abt::GitConfig.unset_local('abt.harvest.taskId')
-        end
-
-        def clear_global
-          Abt::GitConfig.unset_global('abt.harvest.userId')
-          Abt::GitConfig.unset_global('abt.harvest.accountId')
-          Abt::GitConfig.unset_global('abt.harvest.accessToken')
-        end
+    module Harvest
+      def self.command_names
+        Commands.constants.sort.map { |constant_name| Helpers.const_to_command(constant_name) }
+      end
 
-        def client
-          @client ||= Abt::HarvestClient.new(access_token: access_token, account_id: account_id)
-        end
+      def self.command_class(name)
+        const_name = Helpers.command_to_const(name)
+        Commands.const_get(const_name) if Commands.const_defined?(const_name)
       end
     end
   end

data/lib/abt/providers/harvest/api.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      class Api
+        API_ENDPOINT = 'https://api.harvestapp.com/v2'
+        VERBS = %i[get post patch].freeze
+
+        attr_reader :access_token, :account_id
+
+        def initialize(access_token:, account_id:)
+          @access_token = access_token
+          @account_id = account_id
+        end
+
+        VERBS.each do |verb|
+          define_method(verb) do |*args|
+            request(verb, *args)
+          end
+        end
+
+        def get_paged(path, query = {})
+          result_key = path.split('?').first.split('/').last
+
+          page = 1
+          records = []
+
+          loop do
+            result = get(path, query.merge(page: page))
+            records += result[result_key]
+            break if result['total_pages'] == page
+
+            page += 1
+          end
+
+          records
+        end
+
+        def request(*args)
+          response = connection.public_send(*args)
+
+          if response.success?
+            Oj.load(response.body)
+          else
+            error_class = Abt::HttpError.error_class_for_status(response.status)
+            encoded_response_body = response.body.force_encoding('utf-8')
+            raise error_class, "Code: #{response.status}, body: #{encoded_response_body}"
+          end
+        end
+
+        def connection
+          @connection ||= Faraday.new(API_ENDPOINT) do |connection|
+            connection.headers['Authorization'] = "Bearer #{access_token}"
+            connection.headers['Harvest-Account-Id'] = account_id
+            connection.headers['Content-Type'] = 'application/json'
+          end
+        end
+      end
+    end
+  end
+end

data/lib/abt/providers/harvest/base_command.rb

@@ -2,12 +2,13 @@
 
 module Abt
   module Providers
-    class Harvest
+    module Harvest
       class BaseCommand
-        attr_reader :arg_str, :project_id, :task_id, :cli
+        attr_reader :arg_str, :project_id, :task_id, :cli, :config
 
         def initialize(arg_str:, cli:)
           @arg_str = arg_str
+          @config = Configuration.new(cli: cli)
 
           if arg_str.nil?
             use_current_args
@@ -19,6 +20,10 @@ module Abt
 
         private
 
+        def same_args_as_config?
+          project_id == config.project_id && task_id == config.task_id
+        end
+
         def print_project(project)
           cli.print_provider_command(
             'harvest',
@@ -36,10 +41,8 @@ module Abt
         end
 
         def use_current_args
-          @project_id = Abt::GitConfig.local('abt.harvest.projectId').to_s
-          @project_id = nil if project_id.empty?
-          @task_id = Abt::GitConfig.local('abt.harvest.taskId').to_s
-          @task_id = nil if task_id.empty?
+          @project_id = config.project_id
+          @task_id = config.task_id
         end
 
         def use_arg_str(arg_str)
@@ -53,16 +56,9 @@ module Abt
           @task_id = nil if @task_id.empty?
         end
 
-        def remember_project_id(project_id)
-          Abt::GitConfig.local('abt.harvest.projectId', project_id)
-        end
-
-        def remember_task_id(task_id)
-          if task_id.nil?
-            Abt::GitConfig.unset_local('abt.harvest.taskId')
-          else
-            Abt::GitConfig.local('abt.harvest.taskId', task_id)
-          end
+        def api
+          @api ||= Abt::Providers::Harvest::Api.new(access_token: config.access_token,
+                                                    account_id: config.account_id)
         end
       end
     end

data/lib/abt/providers/harvest/commands/clear.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class Clear < BaseCommand
+          def self.command
+            'clear harvest'
+          end
+
+          def self.description
+            'Clear project/task for current git repository'
+          end
+
+          def call
+            cli.warn 'Clearing Harvest project configuration'
+            config.clear_local
+          end
+        end
+      end
+    end
+  end
+end

data/lib/abt/providers/harvest/commands/clear_global.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class ClearGlobal < BaseCommand
+          def self.command
+            'clear-global harvest'
+          end
+
+          def self.description
+            'Clear all global configuration'
+          end
+
+          def call
+            cli.warn 'Clearing Harvest project configuration'
+            config.clear_global
+          end
+        end
+      end
+    end
+  end
+end

data/lib/abt/providers/harvest/commands/current.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class Current < BaseCommand
+          def self.command
+            'current harvest[:<project-id>[/<task-id>]]'
+          end
+
+          def self.description
+            'Get or set project and or task for current git repository'
+          end
+
+          def call
+            if same_args_as_config? || !config.local_available?
+              show_current_configuration
+            else
+              cli.warn 'Updating configuration'
+              update_configuration
+            end
+          end
+
+          private
+
+          def show_current_configuration
+            if project_id.nil?
+              cli.warn 'No project selected'
+            elsif task_id.nil?
+              print_project(project)
+            else
+              print_task(project, task)
+            end
+          end
+
+          def update_configuration
+            ensure_project_is_valid!
+            config.project_id = project_id
+
+            if task_id.nil?
+              print_project(project)
+              config.task_id = nil
+            else
+              ensure_task_is_valid!
+              config.task_id = task_id
+
+              print_task(project, task)
+            end
+          end
+
+          def ensure_project_is_valid!
+            cli.abort "Invalid project: #{project_id}" if project.nil?
+          end
+
+          def ensure_task_is_valid!
+            cli.abort "Invalid task: #{task_id}" if task.nil?
+          end
+
+          def project
+            @project ||= project_assignment['project'].merge('client' => project_assignment['client'])
+          end
+
+          def task
+            @task ||= project_assignment['task_assignments'].map { |ta| ta['task'] }.find do |task|
+              task['id'].to_s == task_id
+            end
+          end
+
+          def project_assignment
+            @project_assignment ||= begin
+              project_assignments.find { |pa| pa['project']['id'].to_s == project_id }
+            end
+          end
+
+          def project_assignments
+            @project_assignments ||= api.get_paged('users/me/project_assignments')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/abt/providers/harvest/commands/init.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class Init < BaseCommand
+          def self.command
+            'init harvest'
+          end
+
+          def self.description
+            'Pick Harvest project for current git repository'
+          end
+
+          def call
+            cli.abort 'Must be run inside a git repository' unless config.local_available?
+
+            projects # Load projects up front to make it obvious that searches are instant
+            project = find_search_result
+
+            config.project_id = project['id']
+            config.task_id = nil
+
+            print_project(project)
+          end
+
+          private
+
+          def find_search_result
+            cli.warn 'Select a project'
+
+            loop do
+              matches = matches_for_string cli.prompt('Enter search')
+              if matches.empty?
+                warn 'No matches'
+                next
+              end
+
+              cli.warn 'Showing the 10 first matches' if matches.size > 10
+              choice = cli.prompt_choice 'Select a project', matches[0...10], true
+              break choice['project'] unless choice.nil?
+            end
+          end
+
+          def matches_for_string(string)
+            search_string = sanitize_string(string)
+
+            searchable_projects.select do |project|
+              sanitize_string(project['name']).include?(search_string)
+            end
+          end
+
+          def sanitize_string(string)
+            string.downcase.gsub(/[^\w]/, '')
+          end
+
+          def searchable_projects
+            @searchable_projects ||= projects.map do |project|
+              {
+                'name' => "#{project['client']['name']} > #{project['name']}",
+                'project' => project
+              }
+            end
+          end
+
+          def projects
+            @projects ||= begin
+              cli.warn 'Fetching projects...'
+              project_assignments.map do |project_assignment|
+                project_assignment['project'].merge('client' => project_assignment['client'])
+              end
+            end
+          end
+
+          def project_assignments
+            @project_assignments ||= api.get_paged('users/me/project_assignments')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/abt/providers/harvest/commands/pick.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class Pick < BaseCommand
+          def self.command
+            'pick harvest[:<project-id>]'
+          end
+
+          def self.description
+            'Pick task for current git repository'
+          end
+
+          def call
+            cli.abort 'Must be run inside a git repository' unless config.local_available?
+
+            cli.warn project['name']
+            task = cli.prompt_choice 'Select a task', tasks
+
+            config.project_id = project_id # We might have gotten the project ID as an argument
+            config.task_id = task['id']
+
+            print_task(project, task)
+          end
+
+          private
+
+          def project
+            project_assignment['project']
+          end
+
+          def tasks
+            @tasks ||= project_assignment['task_assignments'].map { |ta| ta['task'] }
+          end
+
+          def project_assignment
+            @project_assignment ||= begin
+              project_assignments.find { |pa| pa['project']['id'].to_s == project_id }
+            end
+          end
+
+          def project_assignments
+            @project_assignments ||= api.get_paged('users/me/project_assignments')
+          end
+        end
+      end
+    end
+  end
+end