abt-cli 0.0.3 → 0.0.4

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,26 @@
+# 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 initialize(**); end
+
+          def call
+            cli.warn 'Clearing Harvest project configuration'
+            config.clear
+          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,81 @@
+# 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 ||= api.get("projects/#{project_id}")
+          end
+
+          def task
+            project_task_assignments
+              .map { |assignment| assignment['task'] }
+              .find { |task| task['id'].to_s == task_id }
+          end
+
+          def project_task_assignments
+            @project_task_assignments ||= begin
+              api.get_paged("projects/#{project_id}/task_assignments", is_active: true)
+            rescue Abt::HttpError::HttpError # rubocop:disable Layout/RescueEnsureAlignment
+              []
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,66 @@
+# 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
+            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 unless choice.nil?
+            end
+          end
+
+          def matches_for_string(string)
+            search_string = sanitize_string(string)
+
+            projects.select do |project|
+              sanitize_string(project['name']).include?(search_string)
+            end
+          end
+
+          def sanitize_string(string)
+            string.downcase.gsub(/[^\w]/, '')
+          end
+
+          def projects
+            @projects ||= begin
+              cli.warn 'Fetching projects...'
+              api.get_paged('projects')
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,49 @@
+# 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 ||= api.get("projects/#{project_id}")
+          end
+
+          def tasks
+            project_task_assignments.map { |assignment| assignment['task'] }
+          end
+
+          def project_task_assignments
+            @project_task_assignments ||= begin
+              api.get_paged("projects/#{project_id}/task_assignments", is_active: true)
+            rescue Abt::HttpError::HttpError # rubocop:disable Layout/RescueEnsureAlignment
+              []
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class Projects < BaseCommand
+          def self.command
+            'projects harvest'
+          end
+
+          def self.description
+            'List all available projects - useful for piping into grep etc.'
+          end
+
+          def call
+            projects.map do |project|
+              print_project(project)
+            end
+          end
+
+          private
+
+          def projects
+            @projects ||= begin
+              cli.warn 'Fetching projects...'
+              api.get_paged('projects', is_active: true)
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class Share < BaseCommand
+          def self.command
+            'share harvest[:<project-id>[/<task-id>]]'
+          end
+
+          def self.description
+            'Print project/task config string'
+          end
+
+          def call
+            if project_id.nil?
+              cli.warn 'No project selected'
+            elsif task_id.nil?
+              cli.print_provider_command('harvest', project_id)
+            else
+              cli.print_provider_command('harvest', "#{project_id}/#{task_id}")
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+module Abt
+  module Providers
+    module Harvest
+      module Commands
+        class Start < BaseCommand
+          def self.command
+            'start harvest[:<project-id>/<task-id>]'
+          end
+
+          def self.description
+            'Start tracker for current or specified task. Add a relevant provider to link the time entry: E.g. `abt start harvest asana`' # rubocop:disable Layout/LineLength
+          end
+
+          def call
+            abort 'No current/provided task' if task_id.nil?
+
+            maybe_override_current_task
+
+            print_task(project, task)
+
+            cli.abort('No task selected') if task_id.nil?
+
+            create_time_entry
+
+            cli.warn 'Tracker successfully started'
+          rescue Abt::HttpError::HttpError => e
+            cli.warn e
+            cli.abort 'Unable to start tracker'
+          end
+
+          private
+
+          def maybe_override_current_task
+            return if arg_str.nil?
+            return if same_args_as_config?
+            return unless config.local_available?
+
+            should_override = cli.prompt_boolean 'Set selected task as current?'
+            Current.new(arg_str: arg_str, cli: cli).call if should_override
+          end
+
+          def create_time_entry
+            body = Oj.dump({
+              project_id: project_id,
+              task_id: task_id,
+              user_id: config.user_id,
+              spent_date: Date.today.iso8601
+            }.merge(external_link_data), mode: :json)
+            api.post('time_entries', body)
+          end
+
+          def project
+            @project ||= api.get("projects/#{project_id}")
+          end
+
+          def task
+            @task ||= api.get("tasks/#{task_id}")
+          end
+
+          def external_link_data
+            @external_link_data ||= begin
+              arg_strs = cli.args.join(' ')
+              lines = `#{$PROGRAM_NAME} harvest-time-entry-data #{arg_strs}`.split("\n")
+
+              return {} if lines.empty?
+
+              # TODO: Make user choose which reference to use by printing the urls
+              if lines.length > 1
+                cli.abort('Multiple providers had harvest reference data, only one is supported at a time') # rubocop:disable Layout/LineLength
+              end
+
+              Oj.load(lines.first)
+            end
+          end
+        end
+      end
+    end
+  end
+end