Implement squash command

Author: kellda

homu/main.py

@@ -141,6 +141,7 @@ class PullReqState:
     num = 0
     priority = 0
     rollup = 0
+    squash = False
     title = ''
     body = ''
     head_ref = ''
@@ -347,7 +348,7 @@ def get_test_on_fork_repo(self):
     def save(self):
         db_query(
             self.db,
-            'INSERT OR REPLACE INTO pull (repo, num, status, merge_sha, title, body, head_sha, head_ref, base_ref, assignee, approved_by, priority, try_, rollup, delegate) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)',  # noqa
+            'INSERT OR REPLACE INTO pull (repo, num, status, merge_sha, title, body, head_sha, head_ref, base_ref, assignee, approved_by, priority, try_, rollup, squash, delegate) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)',  # noqa
             [
                 self.repo_label,
                 self.num,
@@ -363,6 +364,7 @@ def save(self):
                 self.priority,
                 self.try_,
                 self.rollup,
+                self.squash,
                 self.delegate,
             ])
 
@@ -722,6 +724,20 @@ def parse_commands(body, username, user_id, repo_label, repo_cfg, state,
 
             state.save()
 
+        elif command.action == 'squash':
+            if not _try_auth_verified():
+                continue
+            state.squash = True
+
+            state.save()
+
+        elif command.action == 'unsquash':
+            if not _try_auth_verified():
+                continue
+            state.squash = False
+
+            state.save()
+
         elif command.action == 'force' and realtime:
             if not _try_auth_verified():
                 continue
@@ -1004,6 +1020,29 @@ def create_merge(state, repo_cfg, branch, logger, git_cfg,
                     desc = 'Auto-squashing failed'
                     comment = ''
                     ok = False
+            if state.squash:
+                try:
+                    merge_base_sha = subprocess.check_output(
+                        git_cmd(
+                            'merge-base',
+                            base_sha,
+                            state.head_sha)).decode('ascii').strip()
+                    git_editor = os.environ['GIT_EDITOR']
+                    os.environ['GIT_EDITOR'] = "sed -i '2,/^$/s/^pick\b/s/'"
+                    utils.logged_call(git_cmd(
+                        '-c',
+                        'user.name=' + git_cfg['name'],
+                        '-c',
+                        'user.email=' + git_cfg['email'],
+                        'rebase',
+                        '-i',
+                        '--onto',
+                        merge_base_sha, base_sha))
+                    os.environ['GIT_EDITOR'] = git_editor
+                except subprocess.CalledProcessError:
+                    desc = 'Squashing failed'
+                    comment = ''
+                    ok = False
 
             if ok:
                 utils.logged_call(git_cmd('checkout', '-B', branch, base_sha))
@@ -1735,6 +1774,7 @@ def main():
         priority INTEGER,
         try_ INTEGER,
         rollup INTEGER,
+        squash INTEGER,
         delegate TEXT,
         UNIQUE (repo, num)
     )''')
@@ -1779,6 +1819,10 @@ def main():
         db_query(db, 'SELECT treeclosed_src FROM repos LIMIT 0')
     except sqlite3.OperationalError:
         db_query(db, 'ALTER TABLE repos ADD COLUMN treeclosed_src TEXT')
+    try:
+        db_query(db, 'SELECT squash FROM pull LIMIT 0')
+    except sqlite3.OperationalError:
+        db_query(db, 'ALTER TABLE pull ADD COLUMN squash INT')
 
     for repo_label, repo_cfg in cfg['repo'].items():
         repo_cfgs[repo_label] = repo_cfg
@@ -1796,9 +1840,9 @@ def main():
 
         db_query(
             db,
-            'SELECT num, head_sha, status, title, body, head_ref, base_ref, assignee, approved_by, priority, try_, rollup, delegate, merge_sha FROM pull WHERE repo = ?',   # noqa
+            'SELECT num, head_sha, status, title, body, head_ref, base_ref, assignee, approved_by, priority, try_, rollup, squash, delegate, merge_sha FROM pull WHERE repo = ?',   # noqa
             [repo_label])
-        for num, head_sha, status, title, body, head_ref, base_ref, assignee, approved_by, priority, try_, rollup, delegate, merge_sha in db.fetchall():  # noqa
+        for num, head_sha, status, title, body, head_ref, base_ref, assignee, approved_by, priority, try_, rollup, squash, delegate, merge_sha in db.fetchall():  # noqa
             state = PullReqState(num, head_sha, status, db, repo_label, mergeable_que, gh, repo_cfg['owner'], repo_cfg['name'], repo_cfg.get('labels', {}), repos, repo_cfg.get('test-on-fork'))  # noqa
             state.title = title
             state.body = body
@@ -1810,6 +1854,7 @@ def main():
             state.priority = int(priority)
             state.try_ = bool(try_)
             state.rollup = rollup
+            state.squash = bool(squash)
             state.delegate = delegate
             builders = []
             if merge_sha:

homu/parse_issue_comment.py

@@ -70,6 +70,14 @@ def rollup(cls, rollup_value):
         command.rollup_value = rollup_value
         return command
 
+    @classmethod
+    def squash(cls):
+        return cls('squash')
+
+    @classmethod
+    def unsquash(cls):
+        return cls('unsquash')
+
     @classmethod
     def force(cls):
         return cls('force')
@@ -227,6 +235,12 @@ def parse_issue_comment(username, body, sha, botname, hooks=[]):
             rollup_value = WORDS_TO_ROLLUP[word]
             commands.append(IssueCommentCommand.rollup(rollup_value))
 
+        elif word == 'squash':
+            commands.append(IssueCommentCommand.squash())
+
+        elif word == 'squash-':
+            commands.append(IssueCommentCommand.unsquash())
+
         elif word == 'force':
             commands.append(IssueCommentCommand.force())