gitloggl 0.1.0

data/lib/gitloggl/commands/sync.rb

@@ -0,0 +1,133 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Commands
+    class Sync < Gitloggl::Command
+      attr_accessor :date_from, :date_to
+      attr_accessor :toggl_bar, :gitlab_bar
+
+      def execute
+        self.date_from, self.date_to = select_dates
+
+        setup_toggl_hooks
+        setup_gitlab_hooks
+
+        toggl.run
+
+        print_tables
+      end
+
+      private
+
+      def print_tables
+        if updated.any?
+          puts pastel.green('Updated tracks')
+          puts(updated_table)
+        end
+
+        if rejected.any?
+          puts pastel.yellow('Rejected tracks')
+          puts(rejected_table)
+        end
+      end
+
+      def setup_toggl_hooks
+        toggl.on_first_page do |response|
+          self.toggl_bar = bars.register("toggl [:bar] :percent", total: response.total_pages)
+          self.gitlab_bar = bars.register("gitlab [:bar] :percent", total: response.total_count)
+        end
+
+        toggl.on_each_page do
+          toggl_bar.advance
+        end
+
+        toggl.on_each_page do |response|
+          issues = response.body.fetch('data').map do |row|
+            Gitlab::Issue.new(
+              spent_ms: Integer(row.fetch('dur')),
+              description: row.fetch('description')
+            )
+          end
+
+          gitlab.batch(issues)
+        end
+      end
+
+      def setup_gitlab_hooks
+        gitlab.on_completed do
+          gitlab_bar.advance
+        end
+
+        gitlab.on_skipped do
+          gitlab_bar.advance
+        end
+
+        gitlab.on_rejected do |issue, reason|
+          rejected.push([issue, reason])
+        end
+
+        gitlab.on_updated do |issue, diff|
+          updated.push([issue, diff])
+        end
+      end
+
+      def rejected
+        @rejected ||= []
+      end
+
+      def updated
+        @updated ||= []
+      end
+
+      def rejected_table
+        table = TTY::Table.new header: %w[Track Reason]
+
+        rejected.uniq { |(i)| i.description }.each do |(issue, reason)|
+          table << [issue.description, reason]
+        end
+
+        table.render :unicode, padding: 1
+      end
+
+      def updated_table
+        table = TTY::Table.new header: %w[Track Diff]
+
+        updated.each do |(issue, diff)|
+          table << [issue.description, diff]
+        end
+
+        table.render :unicode, padding: 1
+      end
+
+      def bars
+        @bars ||= TTY::ProgressBar::Multi.new("sync in progress #{date_from} - #{date_to} [:bar] :percent")
+      end
+
+      def toggl
+        @toggl ||= Toggle::Cli.new do |c|
+          c.verbose = verbose?
+          c.workspace_id = config.fetch(Const::TOGGL_WORKSPACE_ID)
+          c.token = config.fetch(Const::TOGGL_TOKEN)
+          c.date_from = date_from
+          c.date_to = date_to
+        end
+      end
+
+      def gitlab
+        @gitlab ||= Gitlab::Cli.new do |c|
+          c.verbose = verbose?
+          c.url = config.fetch(Const::GITLAB_URL)
+          c.token = config.fetch(Const::GITLAB_TOKEN)
+        end
+      end
+
+      def select_dates
+        prompt.select('Select period for sync. Tracks will be added to gitlab if diff will be detected') do |menu|
+          menu.choice 'Past 3 weeks', [3.weeks.ago.to_date, Date.current]
+          menu.choice 'Past 2 weeks', [2.weeks.ago.to_date, Date.current]
+          menu.choice 'Past 7 days', [7.days.ago.to_date, Date.current]
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/commands/toggl_cfg.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Commands
+    class TogglCfg < Gitloggl::Command
+      def execute(*)
+        render_config_config
+
+        prompt.select('') do |menu|
+          menu.enum ')'
+          menu_back(menu)
+          menu.choice 'Change', -> { change }
+        end
+      end
+
+      private
+
+      def change
+        workspace_id = ask_workspace_id
+        token = ask_token
+
+        return unless prompt.yes? <<~ECHO
+          Change settings? WorkspaceID: #{pastel.green(workspace_id)}, token: #{pastel.green(token)}
+        ECHO
+
+        config.set(Const::TOGGL_WORKSPACE_ID, value: workspace_id)
+        config.set(Const::TOGGL_TOKEN, value: token)
+
+        config.write(force: true)
+      end
+
+      def ask_workspace_id
+        prompt.ask('Enter workspaceID:') do |q|
+          q.required true
+          q.validate /\A.+\Z/
+        end
+      end
+
+      def ask_token
+        prompt.ask('Enter token:') do |q|
+          q.required true
+          q.validate /\A.+\Z/
+        end
+      end
+
+      def render_config_config
+        table = TTY::Table.new header: %w[WorkspaceID Token]
+        table << [config.fetch(Const::TOGGL_WORKSPACE_ID), config.fetch(Const::TOGGL_TOKEN)]
+        puts table.render :unicode, padding: 1
+      end
+    end
+  end
+end

data/lib/gitloggl/connection.rb

@@ -0,0 +1,39 @@
+module Gitloggl
+  class Connection
+    USER_AGENT = 'Gitloggl bot'
+
+    attr_reader :url, :headers, :verbose
+
+    delegate :get, :post, :in_parallel, to: :transport
+
+    def initialize(url, headers: {}, verbose: false)
+      @url = url
+      @headers = headers
+      @verbose = verbose
+
+      yield(self) if block_given?
+    end
+
+    def transport
+      @transport ||= Faraday.new(url: url, headers: default_headers.merge(headers)) do |conn|
+        conn.use Faraday::Request::UrlEncoded
+        conn.use Faraday::Response::RaiseError
+        conn.response :json
+        conn.use Faraday::Response::Logger, self, bodies: true if verbose
+        conn.adapter :typhoeus
+      end
+    end
+
+    def default_headers
+      @default_headers ||= {
+        'User-Agent' => USER_AGENT
+      }
+    end
+
+    %i[debug info warn error fatal].each do |name|
+      define_method name do |label, &block|
+        puts("[#{label}] #{block.call}")
+      end
+    end
+  end
+end

data/lib/gitloggl/const.rb

@@ -0,0 +1,8 @@
+module Gitloggl
+  module Const
+    TOGGL_WORKSPACE_ID = :toggl_workspace_id
+    TOGGL_TOKEN = :toggl_token
+    GITLAB_URL = :gitlab_url
+    GITLAB_TOKEN = :gitlab_token
+  end
+end

data/lib/gitloggl/gitlab/cli.rb

@@ -0,0 +1,86 @@
+module Gitloggl
+  module Gitlab
+    class Cli
+      include Hooks
+      include Hooks::InstanceHooks
+
+      define_hook :on_rejected,  scope: ->(*) { nil }
+      define_hook :on_updated,  scope: ->(*) { nil }
+      define_hook :on_completed,  scope: ->(*) { nil }
+      define_hook :on_skipped,  scope: ->(*) { nil }
+
+      attr_accessor :token, :url, :verbose
+
+      def initialize(params = {})
+        params.each { |k, v| public_send("#{k}=", v) }
+        yield(self) if block_given?
+      end
+
+      def batch(issues)
+        stack = Middleware::Builder.new do |b|
+          b.use Stack::Filter, where: ->(issue) do
+            next true if issue.path.present?
+
+            rejected(issue, %{cant recognize gitlab issue-path})
+
+            false
+          end
+
+          b.use Stack::GroupAgg, callback: ->(issues) do
+            issues.each { skipped }
+          end
+
+          b.use Stack::DetectProject
+
+          b.use Stack::Filter, where: -> (issue) do
+            next true if issue.path.project_id.present?
+
+            rejected(issue, %{cant recognize gitlab projectID for issue})
+
+            false
+          end
+
+          b.use Stack::LoadNotes
+
+          b.use Stack::UpdateSpent, callback: ->(issue, diff_sec) do
+            updated(issue, %{added +#{ChronicDuration.output(diff_sec, keep_zero: true)} })
+          end
+        end
+
+        env = OpenStruct.new(cli: self, issues: issues)
+
+        stack.call(env)
+      end
+
+      def rejected(issue, reason)
+        run_hook(:on_completed, issue)
+        run_hook(:on_rejected, issue, reason)
+      end
+
+      def updated(issue, diff)
+        run_hook(:on_completed, issue)
+        run_hook(:on_updated, issue, diff)
+      end
+
+      def skipped
+        run_hook(:on_skipped)
+      end
+
+      def connection
+        @connection ||= Connection.new(url, headers: { 'PRIVATE-TOKEN' => token }, verbose: verbose).tap do |c|
+          c.transport.basic_auth(token, 'api_token')
+        end
+      end
+
+      # @return [Array<JSON>]
+      def projects
+        @projects ||= connection.get("/api/v4/projects").body
+      end
+
+      # @return [Integer]
+      def current_user_id
+        @current_user_id ||= connection.get("/api/v4/user").body.fetch('id')
+      end
+    end
+  end
+end

data/lib/gitloggl/gitlab/issue.rb

@@ -0,0 +1,25 @@
+module Gitloggl
+  module Gitlab
+    class Issue < Struct.new(:description, :project_id, :spent_ms, keyword_init: true)
+      ISSUE_PATH_RE = /([[[:alnum:]]_-]*)#(\d+)/ # "poslogic-partner#471"
+
+      attr_accessor :notes
+
+      def spent_sec
+        spent_ms.fdiv(1000).round
+      end
+
+      def path
+        return nil unless description =~ ISSUE_PATH_RE
+        return nil unless $1.present? && $2.present?
+
+        @path ||= OpenStruct.new(project_path: $1, issue_id: $2)
+      end
+
+      def +(issue)
+        self.spent_ms = spent_ms + issue.spent_ms
+        self
+      end
+    end
+  end
+end

data/lib/gitloggl/gitlab/stack/abstract.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Gitlab
+    module Stack
+      class Abstract
+        attr_reader :app, :env, :options
+
+        def self.opt(*names)
+          names.each do |name|
+            define_method "#{name}!" do
+              options.fetch(name)
+            end
+          end
+        end
+
+        def initialize(app, options = {})
+          @app = app
+          @options = options
+        end
+
+        def call(env)
+          @env = env
+
+          before_call
+
+          around_call do
+            @app.call(env)
+          end
+
+          after_call
+        end
+
+        protected
+
+        def before_call
+          #:stub:
+        end
+
+        def around_call
+          yield
+        end
+
+        def after_call
+          #:stub:
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/gitlab/stack/detect_project.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Gitlab
+    module Stack
+      class DetectProject < Abstract
+        def before_call
+          env.issues.each do |issue|
+            env.cli.projects.find do |row|
+              next unless row.fetch('path_with_namespace') =~ %r[#{issue.path.project_path}]
+
+              issue.path.project_id = row.fetch('id')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/gitlab/stack/filter.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Gitlab
+    module Stack
+      class Filter < Abstract
+        opt :where
+
+        def before_call
+          env.issues.select!(&where!)
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/gitlab/stack/group_agg.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Gitlab
+    module Stack
+      class GroupAgg < Abstract
+        opt :callback
+
+        def before_call
+          env.issues = env.issues.group_by(&:path).each_with_object([]) do |(_, group), object|
+            object.push(group.inject { |a, b| a + b })
+
+            next unless group.many?
+
+            callback!.call(group[1..-1])
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/gitlab/stack/load_notes.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Gitlab
+    module Stack
+      class LoadNotes < Abstract
+        def before_call
+          result = {}
+
+          env.cli.connection.in_parallel do
+            issue_paths.each do |path|
+              response = env.cli.connection.get("/api/v4/projects/#{path.project_id}/issues/#{path.issue_id}/notes")
+              result[path] = response
+            end
+          end
+
+          env.issues.each do |issue|
+            issue.notes = result[issue.path]&.body || []
+          end
+        end
+
+        private
+
+
+        def issue_paths
+          env.issues.map(&:path).uniq.compact
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/gitlab/stack/update_spent.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Gitloggl
+  module Gitlab
+    module Stack
+      class UpdateSpent < Abstract
+        GITLAB_SPENT_NOTE_RE = /added (.+) of time spent/
+
+        opt :callback
+
+        def before_call
+          queue = env.issues.each_with_object([]) do |issue, object|
+            diff_sec = issue.spent_sec  - spent_sec(issue.notes)
+            diff_sec = diff_sec > 0 ? diff_sec : 0
+
+            callback!.call(issue, diff_sec)
+
+            next if diff_sec.zero?
+
+            object << [
+              "/api/v4/projects/#{issue.path.project_id}/issues/#{issue.path.issue_id}/add_spent_time",
+              duration: "#{diff_sec}sec"
+            ]
+          end
+
+          env.cli.connection.in_parallel do
+            queue.each { |argv| env.cli.connection.post(*argv) }
+          end
+        end
+
+        private
+
+        # @return [Integer]
+        def spent_sec(notes)
+          notes.inject(0) do |memo, row|
+            next memo unless row.fetch('author').fetch('id') == env.cli.current_user_id
+            next memo unless row.fetch('body') =~ GITLAB_SPENT_NOTE_RE
+
+            memo + ChronicDuration.parse($1).to_i
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/templates/.gitkeep

@@ -0,0 +1 @@
+#

data/lib/gitloggl/templates/toggle_conf/.gitkeep

@@ -0,0 +1 @@
+#

data/lib/gitloggl/toggle/cli.rb

@@ -0,0 +1,60 @@
+module Gitloggl
+  module Toggle
+    class Cli
+      include Hooks
+      include Hooks::InstanceHooks
+
+      define_hook :on_first_page,  scope: ->(*) { nil }
+      define_hook :on_each_page,  scope: ->(*) { nil }
+
+      attr_accessor :token, :workspace_id, :date_from, :date_to, :verbose
+
+      def initialize(params = {})
+        params.each { |k, v| public_send("#{k}=", v) }
+        yield(self) if block_given?
+      end
+
+      def run(page: 1)
+        body = request(page)
+
+        total_count = body.fetch('total_count')
+        per_page = body.fetch('per_page')
+        total_pages = total_count.fdiv(per_page).ceil
+
+        result = OpenStruct.new(
+          total_count: total_count,
+          per_page: per_page,
+          total_pages: total_pages,
+          page: page,
+          body: body
+        )
+
+        run_hook(:on_first_page, result) if page == 1
+        run_hook(:on_each_page, result)
+
+        run(page: page.next) if body['data'].length >= per_page
+      end
+
+      private
+
+      # @return [JSON]
+      def request(page)
+        response = connection.get('/reports/api/v2/details', {
+          workspace_id: workspace_id,
+          since: date_from,
+          until: date_to,
+          user_agent: Connection::USER_AGENT,
+          page: page
+        })
+
+        response.body
+      end
+
+      def connection
+        @connection ||= Connection.new('https://toggl.com/reports/api/v2/details', verbose: verbose).tap do |c|
+          c.transport.basic_auth(token, 'api_token')
+        end
+      end
+    end
+  end
+end

data/lib/gitloggl/version.rb

@@ -0,0 +1,3 @@
+module Gitloggl
+  VERSION = '0.1.0'
+end

data/lib/gitloggl.rb

@@ -0,0 +1,58 @@
+require 'tty-table'
+require 'tty-progressbar'
+require 'ostruct'
+require 'faraday'
+require 'faraday_middleware'
+require 'typhoeus'
+require 'json'
+require 'chronic_duration'
+require 'middleware'
+require 'hooks'
+require 'active_support/core_ext/object/blank'
+require 'active_support/core_ext/enumerable'
+require 'active_support/dependencies/autoload'
+require 'active_support/core_ext/module/delegation'
+require 'active_support/duration'
+require 'active_support/core_ext/date'
+require 'active_support/core_ext/numeric/time'
+require 'gitloggl/const'
+require 'gitloggl/version'
+require 'gitloggl/cli'
+require 'gitloggl/command'
+require 'gitloggl/connection'
+require 'gitloggl/commands/main'
+
+module Gitloggl
+  module Commands
+    extend ActiveSupport::Autoload
+
+    autoload :Main
+    autoload :TogglCfg
+    autoload :GitlabCfg
+    autoload :Sync
+  end
+
+  module Toggle
+    extend ActiveSupport::Autoload
+
+    autoload :Cli
+  end
+
+  module Gitlab
+    extend ActiveSupport::Autoload
+
+    autoload :Issue
+    autoload :Cli
+
+    module Stack
+      extend ActiveSupport::Autoload
+
+      autoload :Abstract
+      autoload :Filter
+      autoload :GroupAgg
+      autoload :DetectProject
+      autoload :LoadNotes
+      autoload :UpdateSpent
+    end
+  end
+end