159 lines
6.6 KiB
Python
Executable file
159 lines
6.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Copyright 2019 Google LLC
|
|
#
|
|
# Licensed under the the Apache License v2.0 with LLVM Exceptions (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# https://llvm.org/LICENSE.txt
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import argparse
|
|
import json
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
from typing import List, Optional, Tuple
|
|
|
|
from phabricator import Phabricator
|
|
from git import Repo
|
|
|
|
class ApplyPatch:
|
|
|
|
def __init__(self, diff_id:str, comment_file_path: str, token: str, url: str, store_json_diff: str):
|
|
# TODO: turn os.environ parameter into command line arguments
|
|
# this would be much clearer and easier for testing
|
|
self.comment_file_path = comment_file_path
|
|
self.conduit_token = token # type: Optional[str]
|
|
self.host = url # type: Optional[str]
|
|
self._load_arcrc()
|
|
self.diff_id = diff_id # type: str
|
|
self.diff_json_path = store_json_diff # type: str
|
|
if not self.host.endswith('/api/'):
|
|
self.host += '/api/'
|
|
self.phab = Phabricator(token=self.conduit_token, host=self.host)
|
|
self.git_hash = None # type: Optional[str]
|
|
self.msg = [] # type: List[str]
|
|
self.repo = Repo(os.getcwd()) # type: Repo
|
|
|
|
def _load_arcrc(self):
|
|
"""Load arc configuration from file if not set."""
|
|
if self.conduit_token is not None or self.host is not None:
|
|
return
|
|
print('Loading configuration from ~/.arcrc file')
|
|
with open(os.path.expanduser('~/.arcrc'), 'r') as arcrc_file:
|
|
arcrc = json.load(arcrc_file)
|
|
# use the first host configured in the file
|
|
self.host = next(iter(arcrc['hosts']))
|
|
self.conduit_token = arcrc['hosts'][self.host]['token']
|
|
|
|
def run(self):
|
|
"""try to apply the patch from phabricator
|
|
|
|
Write to `self.comment_file` for showing error messages on Phabricator.
|
|
"""
|
|
self.phab.update_interfaces()
|
|
|
|
try:
|
|
print('Checking out master...')
|
|
self.repo.git.checkout('master')
|
|
revision_id, dependencies = self._get_dependencies()
|
|
print('Analyzing {}'.format(diff_to_str(revision_id)))
|
|
if len(dependencies) > 0:
|
|
print('This diff depends on: {}'.format(diff_list_to_str(dependencies)))
|
|
missing, landed = self._get_missing_landed_dependencies(dependencies)
|
|
print(' These have already landed: {}'.format(diff_list_to_str(landed)))
|
|
print(' These are missing on master: {}'.format(diff_list_to_str(missing)))
|
|
# TODO for revision in missing:
|
|
for revision in landed:
|
|
self._apply_revision(revision)
|
|
self._apply_diff(self.diff_id)
|
|
finally:
|
|
self._write_error_message()
|
|
|
|
def _get_diff(self, diff_id: str):
|
|
return self.phab.differential.getdiff(diff_id=diff_id)
|
|
|
|
def _get_revision(self, revision_id: int):
|
|
return self.phab.differential.query(ids=[revision_id])[0]
|
|
|
|
def _get_revisions(self, phids: str = None, ids: int = None):
|
|
if phids is not None:
|
|
return self.phab.differential.query(phids=phids)
|
|
|
|
def _get_dependencies(self) -> List[int]:
|
|
revision_id = int(self._get_diff(self.diff_id).revisionID)
|
|
revision = self._get_revision(revision_id)
|
|
dependency_ids = revision['auxiliary']['phabricator:depends-on']
|
|
revisions = self._get_revisions(dependency_ids)
|
|
diff_ids = [int(rev['id']) for rev in revisions]
|
|
# It seems Phabricator lists the dependencies in the opposite order,
|
|
# so we reverse the order before returning the list, so that they
|
|
# can be applied in this order
|
|
diff_ids.reverse()
|
|
return revision_id, diff_ids
|
|
|
|
def _apply_diff(self, diff_id: int):
|
|
print('Applying diff {}...'.format(diff_id))
|
|
diff = self.phab.differential.getrawdiff(diffID=diff_id).response
|
|
proc = subprocess.run('git apply', input=diff, shell=True, text=True,
|
|
stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
if proc.returncode != 0:
|
|
raise Exception('Applying patch failed:\n{}'.format(proc.stdout + proc.stderr))
|
|
|
|
def _apply_revision(self, revision_id: int):
|
|
revision = self._get_revisions([revision_id])
|
|
print(revision.response)
|
|
|
|
def _write_error_message(self):
|
|
"""Write the log message to a file."""
|
|
if self.comment_file_path is None:
|
|
return
|
|
|
|
if len(self.msg) == 0:
|
|
return
|
|
print('writing error message to {}'.format(self.comment_file_path))
|
|
with open(self.comment_file_path, 'a') as comment_file:
|
|
text = '\n\n'.join(self.msg)
|
|
comment_file.write(text)
|
|
|
|
def _get_landed_revisions(self, limit: int = 1000):
|
|
diff_regex = re.compile(r'^Differential Revision: https:\/\/reviews\.llvm\.org\/(.*)$', re.MULTILINE)
|
|
for commit in self.repo.iter_commits("master", max_count=limit):
|
|
result = diff_regex.search(commit.message)
|
|
if result is not None:
|
|
yield result.group(1)
|
|
return
|
|
|
|
def _get_missing_landed_dependencies(self, dependencies: List[int]) -> Tuple[List[int], List[int]]:
|
|
landed_deps = []
|
|
missing_deps = []
|
|
for dependency in dependencies:
|
|
if diff_to_str(dependency) in self._get_landed_revisions():
|
|
landed_deps.append(dependency)
|
|
else:
|
|
missing_deps.append(dependency)
|
|
return missing_deps, landed_deps
|
|
|
|
def diff_to_str(diff: int) -> str:
|
|
return 'D{}'.format(diff)
|
|
|
|
def diff_list_to_str(diffs: List[int]) -> str:
|
|
return ', '.join([diff_to_str(d) for d in diffs])
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description='Apply Phabricator patch to working directory.')
|
|
parser.add_argument('diff_id', type=str)
|
|
parser.add_argument('--comment-file', type=str, dest='comment_file_path', default=None)
|
|
parser.add_argument('--token', type=str, default=None)
|
|
parser.add_argument('--url', type=str, default=None)
|
|
parser.add_argument('--store-json-diff', dest='store_json_diff', type=str, default=None)
|
|
args = parser.parse_args()
|
|
patcher = ApplyPatch(args.diff_id, args.comment_file_path, args.token, args.url, args.store_json_diff)
|
|
patcher.run()
|
|
|