You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.7 KiB
111 lines
3.7 KiB
#!/usr/bin/env python
|
|
#
|
|
# Copyright 2019 Google Inc.
|
|
#
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
|
|
from __future__ import print_function
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
|
|
from infra import git
|
|
from infra import go
|
|
|
|
_TOOLS_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
_REPO_ROOT = os.path.realpath(os.path.join(_TOOLS_DIR, os.pardir))
|
|
_INFRA_BOTS = os.path.join(_REPO_ROOT, 'infra', 'bots')
|
|
sys.path.insert(0, _INFRA_BOTS)
|
|
import git_utils
|
|
|
|
|
|
REFS_HEADS_PREFIX = 'refs/heads/'
|
|
CHROME_REF_PREFIX = REFS_HEADS_PREFIX + 'chrome/m'
|
|
SK_MILESTONE_H = os.path.join('include', 'core', 'SkMilestone.h')
|
|
SK_MILESTONE_TMPL = r'#define SK_MILESTONE %s'
|
|
SK_MILESTONE_RE = SK_MILESTONE_TMPL % r'(\d+)'
|
|
SKIA_REPO = 'https://skia.googlesource.com/skia.git'
|
|
SUPPORTED_CHROME_BRANCHES = 3
|
|
UPDATE_MILESTONE_COMMIT_MSG = '''Update Skia milestone to %d'''
|
|
|
|
|
|
def get_current_milestone():
|
|
'''Read SkMilestone.h and parse out the current milestone.'''
|
|
sk_milestone = os.path.join(_REPO_ROOT, SK_MILESTONE_H)
|
|
with open(sk_milestone, 'r') as f:
|
|
contents = f.read()
|
|
for line in contents.splitlines():
|
|
m = re.match(SK_MILESTONE_RE, line)
|
|
if m:
|
|
return int(m.groups()[0])
|
|
print >> sys.stderr, (
|
|
'Failed to parse %s; has the format changed?' % SK_MILESTONE_H)
|
|
sys.exit(1)
|
|
|
|
|
|
def create_new_branch(new_branch, branch_at):
|
|
'''Create a temporary checkout of the repo, create the new branch and push.'''
|
|
b = new_branch[len(REFS_HEADS_PREFIX):]
|
|
with git_utils.NewGitCheckout(SKIA_REPO, local=_REPO_ROOT):
|
|
git.git('checkout', '-b', b)
|
|
git.git('reset', '--hard', branch_at)
|
|
git.git('push', '--set-upstream', 'origin', b)
|
|
|
|
|
|
def update_milestone(m):
|
|
'''Update SkMilestone.h to match the given milestone number.'''
|
|
with git_utils.NewGitCheckout(SKIA_REPO, local=_REPO_ROOT):
|
|
with git_utils.GitBranch(
|
|
'update_milestone', UPDATE_MILESTONE_COMMIT_MSG % m):
|
|
with open(SK_MILESTONE_H, 'r+') as f:
|
|
contents = re.sub(
|
|
SK_MILESTONE_RE, SK_MILESTONE_TMPL % str(m), f.read(), flags=re.M)
|
|
f.seek(0)
|
|
f.write(contents)
|
|
f.truncate()
|
|
git.git('diff')
|
|
|
|
|
|
def update_infra_config(old_branch, new_branch):
|
|
'''Create a CL to add infra support for the new branch and remove the old.'''
|
|
owner = git.git('config', 'user.email').rstrip()
|
|
if not owner:
|
|
print >> sys.stderr, ('No configured git user; please run '
|
|
'"git config user.email <your email>".')
|
|
sys.exit(1)
|
|
go.mod_download()
|
|
go.install(go.INFRA_GO+'/go/supported_branches/cmd/new-branch')
|
|
subprocess.check_call(['new-branch',
|
|
'--branch', new_branch[len(REFS_HEADS_PREFIX):],
|
|
'--delete', old_branch[len(REFS_HEADS_PREFIX):],
|
|
'--owner', owner,
|
|
'--exclude-trybots=chromium.*',
|
|
'--exclude-trybots=.*Android_Framework.*',
|
|
'--exclude-trybots=.*G3_Framework.*',
|
|
'--exclude-trybots=.*CanvasKit.*',
|
|
'--exclude-trybots=.*PathKit.*',
|
|
'--submit'])
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) != 2 or '--help' in sys.argv or '-h' in sys.argv:
|
|
print >> sys.stderr, 'Usage: %s <commit hash for branch>' % sys.argv[0]
|
|
sys.exit(1)
|
|
go.check()
|
|
branch_at = sys.argv[1]
|
|
m = get_current_milestone()
|
|
new_branch = '%s%d' % (CHROME_REF_PREFIX, m)
|
|
old_branch = '%s%d' % (CHROME_REF_PREFIX, m-SUPPORTED_CHROME_BRANCHES)
|
|
print('Creating branch %s and removing support (eg. CQ) for %s' % (
|
|
new_branch, old_branch))
|
|
create_new_branch(new_branch, branch_at)
|
|
update_milestone(m+1)
|
|
update_infra_config(old_branch, new_branch)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|