blob: a605059ba3ffaba29176b2a3216285ffc97f99a4 [file] [log] [blame]
#!/usr/bin/env vpython3
# Copyright 2020 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Migrate local repository onto new default branch."""
import fix_encoding
import gerrit_util
import git_common
import metrics
import scm
import sys
import logging
from six.moves import urllib
def GetGerritProject(remote_url):
"""Returns Gerrit project name based on remote git URL."""
if remote_url is None:
raise RuntimeError('can\'t detect Gerrit project.')
project = urllib.parse.urlparse(remote_url).path.strip('/')
if project.endswith('.git'):
project = project[:-len('.git')]
# * hosts ensure that Git/Gerrit projects don't start with
# 'a/' prefix, because 'a/' prefix is used to force authentication in
# gitiles/git-over-https protocol. E.g.,
# refers to the same repo/project
# as
if project.startswith('a/'):
project = project[len('a/'):]
return project
def GetGerritHost(git_host):
parts = git_host.split('.')
parts[0] = parts[0] + '-review'
return '.'.join(parts)
def main():
remote ='remote')
# Use first remote as source of truth
remote = remote.split("\n")[0]
if not remote:
raise RuntimeError('Could not find any remote')
url = scm.GIT.GetConfig(git_common.repo_root(), 'remote.%s.url' % remote)
host = urllib.parse.urlparse(url).netloc
if not host:
raise RuntimeError('Could not find remote host')
project_head = gerrit_util.GetProjectHead(GetGerritHost(host),
if project_head != 'refs/heads/main':
raise RuntimeError("The repository is not migrated yet.")'fetch', remote)
branches = git_common.get_branches_info(True)
if 'master' in branches:"Migrating master branch...")
if 'main' in branches:'You already have master and main branch, consider removing '
'master manually:\n'
' $ git branch -d master\n')
else:'branch', '-m', 'master', 'main')
branches = git_common.get_branches_info(True)
for name in branches:
branch = branches[name]
if not branch:
if 'master' in branch.upstream:"Migrating %s branch..." % name)
new_upstream = branch.upstream.replace('master', 'main')'branch', '--set-upstream-to', new_upstream, name)
if __name__ == '__main__':
with metrics.collector.print_notice_and_exit():
try:"Starting migration")
main()"Migration completed")
except RuntimeError as e:
logging.error("Error %s" % str(e))