worktree 0.1.0

data/lib/worktree/config.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require 'yaml'
+
+module Worktree
+  module Config
+    def config_file
+      xdg_config_home = ENV.fetch('XDG_CONFIG_HOME') { "#{ENV['HOME']}/.config" }
+      _config_file = "#{xdg_config_home}/worktree/worktree.yml"
+      unless File.exist?(_config_file)
+        raise Worktree::Error, "config file #{_config_file} not found!"
+      end
+
+      _config_file
+    end
+
+    def config
+      YAML.load_file(Worktree::Config.config_file)
+    end
+
+    module_function :config_file, :config
+  end
+end

data/lib/worktree/db_manager.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require 'yaml'
+
+module Worktree
+  class DbManager
+    attr_reader :spec
+
+    def initialize(config_file, environment = 'development')
+      @spec = YAML.load_file(config_file)
+      @environment = environment
+    end
+
+    def environment_spec
+      @spec.fetch(@environment, {})
+    end
+
+    def multi?
+      environment_spec.key? 'primary'
+    end
+
+    def db_port
+      if multi?
+        environment_spec.dig('primary', 'port')
+      else
+        environment_spec['port']
+      end
+    end
+
+    def template
+      if multi?
+        environment_spec.dig('primary', 'database')
+      else
+        environment_spec['database']
+      end
+    end
+
+    def createdb!(db_name)
+      cmd = if db_port
+              "createdb -h localhost -p #{db_port} -T #{template} #{db_name}"
+            else
+              "createdb -h localhost -T #{template} #{db_name}"
+            end
+      Worktree.run_command cmd
+    end
+
+    def dropdb!
+      cmd = if db_port
+              "dropdb -h localhost -p #{db_port} #{template}"
+            else
+              "dropdb -h localhost #{template}"
+            end
+      Worktree.run_command cmd
+    end
+  end
+end

data/lib/worktree/error.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module Worktree
+  class Error < RuntimeError
+  end
+end

data/lib/worktree/feature/clone_dbs.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require 'yaml'
+
+module Worktree
+  module Feature
+    class CloneDbs
+
+      def initialize(project_dir:, branch:)
+        @project_dir = project_dir
+        @branch = branch
+      end
+
+      def run!
+        @db_manager = DbManager.new("#{@project_dir}/master/config/database.yml")
+        @db_manager.createdb!(db_name)
+
+        write!
+      rescue StandardError => e
+        # bypass error
+        Worktree.logger.error { e.message }
+      end
+
+      private
+
+      def write!
+        new_spec = @db_manager.spec.dup
+        if @db_manager.multi?
+          new_spec['development']['primary']['database'] = db_name
+        else
+          new_spec['development']['database'] = db_name
+        end
+        # write changed database config back
+        File.write("#{@project_dir}/#{@branch}/config/database.yml", new_spec.to_yaml)
+      end
+
+      def db_name
+        # db name cannot be > 63 bytes
+        db_suffix = '_development'
+        max_db_prefix_length = 62 - db_suffix.length
+        db_prefix = @branch[0..max_db_prefix_length]
+        "#{db_prefix}#{db_suffix}"
+      end
+    end
+  end
+end

data/lib/worktree/feature/copy_files.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'fileutils'
+
+module Worktree
+  module Feature
+    class CopyFiles
+
+      def initialize(project_dir:, branch:)
+        @project_dir = project_dir
+        @branch = branch
+      end
+
+      def run!
+        project_key = Worktree::Project.project_key_for(@branch)
+        paths = Worktree::Config.config.dig('projects', project_key, 'copy_files') || []
+        paths.each { |path| copy_file(path) }
+      end
+
+      private
+
+      def copy_file(file)
+        master_path = "#{@project_dir}/master/#{file}"
+        if File.exist?(master_path)
+          FileUtils.cp_r master_path, "#{@project_dir}/#{@branch}/#{file}"
+        else
+          print "The path #{master_path} was not found!"
+        end
+      end
+    end
+  end
+end

data/lib/worktree/feature/jira.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+require 'jira-ruby'
+
+module Worktree
+  module Feature
+    class Jira
+      def initialize(project_dir:, branch:)
+        @project_dir = project_dir
+        @branch = branch
+      end
+
+      def run!
+        if jira_issue?
+          prompt = "Jira issue #{jira_issue_id} status: #{jira_issue.status.name}. Would you like to change it?"
+          jira_process! unless TTY::Prompt.new.no?(prompt)
+        end
+
+        super
+      end
+
+      private
+
+      def jira_client
+        @jira_client ||= JIRA::Client.new(jira_client_options)
+      end
+
+      def jira_client_options
+        {
+          username: ENV['JIRA_USERNAME'],
+          password: ENV['JIRA_PASSWORD'],
+          site: ENV['JIRA_SITE'],
+          context_path: '',
+          auth_type: :basic
+        }
+      end
+
+      def jira_issue?
+        return false unless jira_issue_id
+
+        jira_issue_id =~ Worktree::JIRA_ISSUE_ID_REGEX
+      end
+
+      def jira_process!
+        transition = choose_transition
+        apply_transition!(transition) if transition != -1
+      rescue StandardError => e
+        Worktree.logger.error { e.message }
+      end
+
+      def jira_issue_id
+        (@branch.match(/^\w+\-\d+/) || [])[0]
+      end
+
+      def jira_issue
+        @jira_issue ||= jira_client.Issue.find(jira_issue_id)
+      end
+
+      def apply_transition!(transition)
+        jira_issue.transitions.build.save!(transition: { id: transition.id })
+      end
+
+      def choose_transition
+        TTY::Prompt.new.select('Choose a transition?', cycle: true) do |menu|
+          menu.enum '.'
+          menu.choice 'Skip it', -1
+          jira_issue.transitions.all.each do |s|
+            menu.choice s.name, s
+          end
+        end
+      end
+    end
+  end
+end

data/lib/worktree/feature/tmux.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+module Worktree
+  module Feature
+    class Tmux
+
+      class VimEditor
+        attr_reader :window_name
+
+        def initialize(cwd:)
+          @cwd = cwd
+          @window_name = 'vim'
+        end
+
+        # open Gemfile if present
+        def cmd
+          if File.exist?("#{@cwd}/Gemfile")
+            'vim Gemfile'
+          else
+            'vim'
+          end
+        end
+      end
+
+      def initialize(project_dir:, branch:)
+        @project_dir = project_dir
+        @branch = branch
+        @working_directory = "#{@project_dir}/#{@branch}".chomp('/')
+      end
+
+      def run!(session_name)
+        if session_exist?(session_name)
+          Worktree.logger.info { "TMUX session #{session_name} already exist" }
+          # TODO: ask for attach to it
+          return
+        end
+
+        Worktree.run_command "tmux new-session -t #{session_name} -d", chdir: @working_directory
+        Worktree.run_command "tmux new-window -d -t #{session_name} -n #{editor.window_name}", chdir: @working_directory
+        Worktree.run_command "tmux send-keys -t #{session_name}:2 \"#{editor.cmd}\" C-m"
+        Worktree.run_command "tmux select-window -t #{session_name}:2" # select vim window
+        if inside_tmux?
+          Kernel.system "tmux switch -t #{session_name}"
+        else
+          Kernel.system "tmux attach-session -t #{session_name}"
+        end
+      end
+
+      private
+
+      def editor
+        return @editor if defined?(@editor)
+
+        @editor = VimEditor.new(cwd: @working_directory)
+      end
+
+      def inside_tmux?
+        Worktree.run_command('echo $TMUX').out.strip.present?
+      end
+
+      def session_exist?(name)
+        cmd = "tmux list-sessions -F '#S' | awk '/'#{name}/' {print $1}'"
+        Worktree.run_command(cmd).out.strip.present?
+      end
+    end
+  end
+end

data/lib/worktree/project.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Worktree
+  class Project
+    def self.project_key_for(branch)
+      project_keys = Worktree::Config.config['projects'].keys
+      return nil if project_keys.empty?
+
+      re = Regexp.new("^(#{project_keys.join('|')})\-")
+      (branch.match(re) || [])[1]
+    end
+
+    def self.resolve(branch)
+      new(project_key_for(branch))
+    end
+
+    def initialize(key)
+      @key = key
+    end
+
+    def root
+      if @key
+        Worktree::Config.config.dig('projects', @key, 'root').chomp('/')
+      else
+        Dir.pwd
+      end
+    end
+  end
+end

data/lib/worktree/tab_completion.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+require 'worktree'
+
+module Worktree
+  module TabCompletion # :nodoc:
+  end
+end

data/lib/worktree/tab_completion/branch_completion.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require 'jira-ruby'
+
+module Worktree
+  module TabCompletion
+    class BranchCompletion
+      def initialize(compl)
+        @compl = compl
+      end
+
+      def list
+        issue_id = find_jira_issue_by(@compl)
+        if issue_id
+          jira_issue = jira_client.Issue.find(issue_id)
+          ["#{issue_id}-#{clean_jira_summary(jira_issue)}"]
+        else
+          []
+        end
+      end
+
+      private
+
+      def find_jira_issue_by(comp_line)
+        (comp_line.match(Worktree::JIRA_ISSUE_ID_REGEX) || [])[0]
+      end
+
+      def jira_client
+        @jira_client ||= JIRA::Client.new(
+          username: ENV['JIRA_USERNAME'],
+          password: ENV['JIRA_PASSWORD'],
+          site: ENV['JIRA_SITE'],
+          context_path: '',
+          auth_type: :basic
+        )
+      end
+
+      def clean_jira_summary(jira_issue)
+        raw_summary = jira_issue.summary
+        raw_summary = raw_summary.strip
+
+        # translate raw summary to branch name
+        summary = raw_summary.split(' ').map(&:underscore).join('-')
+        summary.gsub!('&&', 'and')
+        summary.gsub!(/\(|\)/, '') # remove brackets
+        summary.gsub!(/"|'|”|“|«|»/, '') # remove quotes
+        summary.gsub!(/\.$/, '') # remove end period
+        summary.gsub!(%r{/}, '-') # change back slash to minus
+        summary.gsub!(/:|;/, '')
+        summary
+      end
+    end
+  end
+end

data/lib/worktree/tab_completion/cli.rb

@@ -0,0 +1,140 @@
+# frozen_string_literal: true
+
+require 'worktree/tab_completion'
+
+module Worktree
+  module TabCompletion
+    class CLI # :nodoc:
+      COMMAND_NEW = 'new'
+      COMMAND_CHERRY_PICK = 'cherry_pick'
+      COMMAND_REMOVE = 'remove'
+      COMMAND_CONFIGURE = 'configure'
+      COMMAND_OPEN = 'open'
+
+      def find_matches_for(command_line)
+        command_parameters = command_line.split(' ', -1)[1..-1]
+
+        # 'worktree '
+        # 'worktree n'
+        if command_parameters.empty? || command_parameters.size == 1
+          return [
+            COMMAND_NEW,
+            COMMAND_CHERRY_PICK,
+            COMMAND_REMOVE,
+            COMMAND_CONFIGURE,
+            COMMAND_OPEN
+          ]
+        end
+
+        # 'worktree new '
+        # 'worktree new DAPI'
+        if command_parameters.size == 2
+          command, compl = command_parameters
+
+          return BranchCompletion.new(compl).list if command == COMMAND_NEW
+          return WorktreeCompletion.new(compl).list if command == COMMAND_OPEN
+          return WorktreeCompletion.new(compl).list if command == COMMAND_REMOVE
+          # TODO: cherry_pick/remove command
+        end
+
+        # 'worktree new BRANCHNAME '
+        # 'worktree new BRANCHNAME --project-dir='
+        # 'worktree new BRANCHNAME --project-dir=../'
+        # 'worktree new BRANCHNAME --project-dir=/tmp/path'
+        # 'worktree new BRANCHNAME --project-dir=~/path'
+        # 'worktree new BRANCHNAME --from='
+        # 'worktree new BRANCHNAME --from=upstream/'
+        if command_parameters.size == 3
+          command, branch, compl = command_parameters
+
+          if command == COMMAND_NEW
+            if compl.starts_with?('--project-dir=')
+              return ProjectDirCompletion.new(compl[14..-1]).list.
+                map { |dir| "--project-dir=#{dir}" }
+            elsif command_parameters[2].starts_with?('--from=')
+              project_dir = Project.resolve(branch).root
+              return RemoteBranchCompletion.new(compl[7..-1], project_dir: project_dir).list.
+                map { |b| "--from=#{b}" }
+            else
+              return ['--from=', '--project-dir=']
+            end
+          end
+
+          if command == COMMAND_CHERRY_PICK
+            if compl.starts_with?('--project-dir=')
+              return ProjectDirCompletion.new(compl[14..-1]).list.
+                map { |dir| "--project-dir=#{dir}" }
+            elsif command_parameters[2].starts_with?('--to=')
+              return RemoteBranchCompletion.new(compl[5..-1]).list.
+                map { |dir| "--to=#{dir}" }
+            else
+              return ['--to=', '--project-dir=']
+            end
+          end
+
+          if command == COMMAND_REMOVE
+            if compl.starts_with?('--project-dir=')
+              return ProjectDirCompletion.new(compl[14..-1]).list.
+                map { |dir| "--project-dir=#{dir}" }
+            else
+              return ['--project-dir=']
+            end
+          end
+
+          if command == COMMAND_OPEN
+            if compl.starts_with?('--project-dir=')
+              return ProjectDirCompletion.new(compl[14..-1]).list.
+                map { |dir| "--project-dir=#{dir}" }
+            else
+              return ['--project-dir=']
+            end
+          end
+        end
+
+        if command_parameters.size == 4
+          command, _branch_name, project_dir_or_from, compl = command_parameters
+
+          if command == COMMAND_NEW
+            if project_dir_or_from.starts_with?('--project-dir=')
+              if compl.starts_with?('--from=')
+                project_dir = project_dir_or_from[14..-1]
+                return RemoteBranchCompletion.new(compl[7..-1], project_dir: project_dir).list.
+                  map { |dir| "--from=#{dir}" }
+              else
+                return ['--from=']
+              end
+            elsif project_dir_or_from.starts_with?('--from=')
+              if compl.starts_with?('--project-dir=')
+                return ProjectDirCompletion.new(compl[14..-1]).list.
+                  map { |dir| "--project-dir=#{dir}" }
+              else
+                return ['--project-dir=']
+              end
+            end
+          end
+
+          if command == COMMAND_CHERRY_PICK
+            if project_dir_or_from.starts_with?('--project-dir=')
+              if compl.starts_with?('--to=')
+                project_dir = project_dir_or_from[14..-1]
+                return RemoteBranchCompletion.new(compl[5..-1], project_dir: project_dir).list.
+                  map { |dir| "--to=#{dir}" }
+              else
+                return ['--to=']
+              end
+            elsif project_dir_or_from.starts_with?('--to=')
+              if compl.starts_with?('--project-dir=')
+                return ProjectDirCompletion.new(compl[14..-1]).list.
+                  map { |dir| "--project-dir=#{dir}" }
+              else
+                return ['--project-dir=']
+              end
+            end
+          end
+        end
+
+        []
+      end
+    end
+  end
+end