# # Copyright (c) 2012, 2013, 2014 Hewlett-Packard Development Company, L.P. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # 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. # from git_upstream.commands import GitUpstreamCommand from git_upstream.lib.importupstream import ImportUpstream from git_upstream.lib.importupstream import ImportUpstreamError from git_upstream.lib.strategies import ImportStrategiesFactory from git_upstream.lib.strategies import LocateChangesWalk from git_upstream.log import LogDedentMixin class ImportCommand(LogDedentMixin, GitUpstreamCommand): """Import code from specified upstream branch. Creates an import branch from the specified upstream branch, and optionally merges additional branches given as arguments. Current branch, unless overridden by the --into option, is used as the target branch from which a list of changes to apply onto the new import is constructed based on the specified strategy. Once complete it will merge and replace the contents of the target branch with those from the import branch, unless --no-merge is specified. """ name = "import" usage = """ %(prog)s [-i] [options] [--onto ] [--import-branch ] [] [ ...] %(prog)s [--finish] [options] [--onto ] [--import-branch ] """ def __init__(self, *args, **kwargs): super(ImportCommand, self).__init__(*args, **kwargs) self.parser.add_argument( '-i', '--interactive', action='store_true', default=False, help='Let the user edit the list of commits before applying.') self.parser.add_argument( '-d', '--dry-run', dest='dry_run', action='store_true', default=False, help='Only print out the list of commits that would be applied.') self.parser.add_argument( '-f', '--force', dest='force', required=False, action='store_true', default=False, help='Force overwrite of existing import branch if it exists.') # finish options self.parser.add_argument( '--finish', dest='finish', required=False, action='store_true', default=False, help='Merge the specified import branch into the target') # result behaviour options self.parser.add_argument( '--merge', dest='merge', required=False, action='store_true', default=True, help='Merge the resulting import branch into the target branch ' 'once complete') self.parser.add_argument( '--no-merge', dest='merge', required=False, action='store_false', help='Disable merge of the resulting import branch') # search/include options self.parser.add_argument( '--search-refs', action='append_replace', metavar='', default=['upstream/*'], dest='search_refs', help='Refs to search for previous import commit. May be ' 'specified multiple times.') self.parser.add_argument( '-s', '--strategy', metavar='', choices=ImportStrategiesFactory.list_strategies(), default=LocateChangesWalk.get_strategy_name(), help='Use the given strategy to re-apply locally carried ' 'changes to the import branch. (default: %(default)s)') self.parser.add_argument( '--into', dest='branch', metavar='', default='HEAD', help='Branch to take changes from, and replace with imported ' 'branch.') self.parser.add_argument( '--import-branch', metavar='', default='import/{describe}', help='Name of import branch to use') # data args self.parser.add_argument( 'upstream_branch', metavar='', nargs='?', default='upstream/master', help='Upstream branch to import. Must be specified if you wish to ' 'provide additional branches.') self.parser.add_argument( 'branches', metavar='', nargs='*', help='Branches to additionally merge into the import branch using ' 'default git merging behaviour') def validate(self): """Perform more complex validation of args that cannot be mixed""" # check if --finish set with --no-merge if self.args.finish and self.args.merge is False: self.parser.error( "--finish cannot be used with '--no-merge'") def _finish(self, import_upstream): self.log.notice("Merging import to requested branch '%s'", self.args.branch) if import_upstream.finish(): self.log.notice( """ Successfully finished import: target branch: '%s' upstream branch: '%s' import branch: '%s'""", self.args.branch, self.args.upstream_branch, import_upstream.import_branch) if self.args.branches: for branch in self.args.branches: self.log.notice(" extra branch: '%s'", branch, dedent=False) return True else: return False def execute(self): import_upstream = ImportUpstream( branch=self.args.branch, upstream=self.args.upstream_branch, import_branch=self.args.import_branch, extra_branches=self.args.branches) self.log.notice("Searching for previous import") strategy = ImportStrategiesFactory.create_strategy( self.args.strategy, branch=self.args.branch, upstream=self.args.upstream_branch, search_refs=self.args.search_refs) if len(strategy) == 0: raise ImportUpstreamError("Cannot find previous import") # if last commit in the strategy was a merge, then the additional # branches that were merged in previously can be extracted based on # the commits merged. prev_import_merge = strategy[-1] if len(prev_import_merge.parents) > 1: idxs = [idx for idx, commit in enumerate(prev_import_merge.parents) if commit.hexsha != strategy.searcher.commit.hexsha] if idxs: additional_commits = [prev_import_merge.parents[i] for i in idxs] if additional_commits and len(self.args.branches) == 0: self.log.warning(""" **************** WARNING **************** Previous import merged additional branches but none have been specified on the command line for this import.\n""") if self.args.dry_run: commit_list = [c.hexsha[:6] + " - " + c.summary[:60] + (c.summary[60:] and "...") for c in list(strategy.filtered_iter())] self.log.notice(""" Requested a dry-run: printing the list of commit that should be rebased %s """, "\n ".join(commit_list)) return True # finish and return if thats all if self.args.finish: return self._finish(import_upstream) # otherwise perform fresh import self.log.notice("Starting import of upstream") import_upstream.create_import(force=self.args.force) self.log.notice("Successfully created import branch") if not import_upstream.apply(strategy, self.args.interactive): self.log.notice("Import cancelled") return False if not self.args.merge: self.log.notice( """ Import complete, not merging to target branch '%s' as requested. """, self.args.branch) return True return self._finish(import_upstream) # vim:sw=4:sts=4:ts=4:et: