From edc25012747368e79f7ca3d5036c8b93752eea6d Mon Sep 17 00:00:00 2001
From: Amir Sarabadani <Ladsgroup@gmail.com>
Date: Sat, 15 Aug 2020 16:25:33 +0200
Subject: [PATCH] Add Gerritbot

To make automatic patches for new wikis

Bug: T253439
---
 .gitignore |   2 +
 gerrit.py  | 108 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 110 insertions(+)
 create mode 100644 gerrit.py

diff --git a/.gitignore b/.gitignore
index 3b37a32..556c033 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,5 +1,7 @@
 # My credentials
 creds.json
+private_key
+gerrit-creds.json
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
diff --git a/gerrit.py b/gerrit.py
new file mode 100644
index 0000000..0fc0700
--- /dev/null
+++ b/gerrit.py
@@ -0,0 +1,108 @@
+import requests
+import json
+import subprocess
+import tempfile
+import urllib
+import os
+from contextlib import contextmanager
+
+
+with open('gerrit-creds.json', 'r') as f:
+    creds = json.loads(f.read())
+
+def load_ssh_key():
+    mixin = ShellMixin()
+    mixin.check_call(['ssh-add', '/home/amsa/Phabricator-maintenance-bot/private_key'])
+
+@contextmanager
+def cd(dirname):
+    cwd = os.getcwd()
+    os.chdir(dirname)
+    try:
+        yield dirname
+    finally:
+        os.chdir(cwd)
+
+def gerrit_url(repo: str, user=None, ssh=False) -> str:
+    if user is not None:
+        prefix = user + '@' 
+    else:
+        prefix = ''
+    if ssh:
+        return f'ssh://{prefix}gerrit.wikimedia.org:29418/{repo}'
+    else:
+        return f'https://{prefix}gerrit.wikimedia.org/r/{repo}.git'
+
+
+class ShellMixin:
+    def check_call(self, args: list, stdin='', env=None,
+                   ignore_returncode=False) -> str:
+        debug = self.log if hasattr(self, 'log') else print
+        debug('$ ' + ' '.join(args))
+        res = subprocess.run(
+            args,
+            input=stdin.encode(),
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            env=env
+        )
+        debug(res.stdout.decode())
+        if not ignore_returncode:
+            res.check_returncode()
+        return res.stdout.decode()
+    def clone(self, repo):
+        url = gerrit_url(repo, user=creds['name'])
+        self.check_call(['git', 'clone', url, 'repo', '--depth=1'])
+        os.chdir('repo')
+        self.check_call(['git', 'config', 'user.name', creds['name']])
+        self.check_call(['git', 'config', 'user.email', creds['email']])
+        self.check_call(['git', 'submodule', 'update', '--init'])
+        self.check_call(['scp', '-p', '-P', '29418', creds['name'] + '@gerrit.wikimedia.org:hooks/commit-msg', '.git/hooks/'])
+
+    def build_push_command(self, options: dict) -> list:
+        per = '%topic=new-wikis-patches'
+        for hashtag in options['hashtags']:
+            per += ',t=' + hashtag
+        if options.get('vote'):
+            per += ',l=' + options['vote']
+            # If we're not automerging, vote V+1 to trigger jenkins (T254070)
+        if options.get('message'):
+            per += ',m=' + urllib.parse.quote_plus(options['message'])
+        return ['git', 'push',
+                gerrit_url(options['repo'], creds['name'], ssh=True),
+                'HEAD:refs/for/master' + per]
+
+class GerritBot(ShellMixin):
+    def __init__(self, name, commit_message):
+        self.name = name
+        self.commit_message = commit_message
+
+    def run(self):
+        with tempfile.TemporaryDirectory() as tmpdirname:
+            with cd(tmpdirname):
+                self.clone(self.name)
+                self.changes()
+                self.commit()
+
+
+    def changes(self):
+        files = [
+            'i18n/wikimediainterwikisearchresults/en.json',
+            'i18n/wikimediainterwikisearchresults/qqq.json'
+        ]
+        for file_ in files:
+            with open(file_, 'r') as f:
+                result = json.load(f)
+            with open(file_, 'w') as f:
+                f.write(json.dumps(result, ensure_ascii=False, indent='\t', sort_keys=True))
+
+    def commit(self):
+        self.check_call(['git', 'add', '.'])
+        with open('.git/COMMIT_EDITMSG', 'w') as f:
+            f.write(self.commit_message)
+        self.check_call(['git', 'commit', '-F', '.git/COMMIT_EDITMSG'])
+        load_ssh_key()
+        self.check_call(self.build_push_command({'hashtags': ['automated-wiki-creation'], 'repo': self.name}))
+
+gerritbot = GerritBot('mediawiki/extensions/WikimediaMessages', "Order entries by alphabetical order\n\nThis would make creating automated patches easier")
+gerritbot.run()
\ No newline at end of file