| # Development tool - upgrade command plugin |
| # |
| # Copyright (C) 2014-2015 Intel Corporation |
| # |
| # This program is free software; you can redistribute it and/or modify |
| # it under the terms of the GNU General Public License version 2 as |
| # published by the Free Software Foundation. |
| # |
| # This program is distributed in the hope that it will be useful, |
| # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| # GNU General Public License for more details. |
| # |
| # You should have received a copy of the GNU General Public License along |
| # with this program; if not, write to the Free Software Foundation, Inc., |
| # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| # |
| """Devtool upgrade plugin""" |
| |
| import os |
| import sys |
| import re |
| import shutil |
| import tempfile |
| import logging |
| import argparse |
| import scriptutils |
| import errno |
| import bb |
| import oe.recipeutils |
| from devtool import standard |
| from devtool import exec_build_env_command, setup_tinfoil, DevtoolError, parse_recipe, use_external_build |
| |
| logger = logging.getLogger('devtool') |
| |
| def _run(cmd, cwd=''): |
| logger.debug("Running command %s> %s" % (cwd,cmd)) |
| return bb.process.run('%s' % cmd, cwd=cwd) |
| |
| def _get_srctree(tmpdir): |
| srctree = tmpdir |
| dirs = os.listdir(tmpdir) |
| if len(dirs) == 1: |
| srctree = os.path.join(tmpdir, dirs[0]) |
| return srctree |
| |
| def _copy_source_code(orig, dest): |
| for path in standard._ls_tree(orig): |
| dest_dir = os.path.join(dest, os.path.dirname(path)) |
| bb.utils.mkdirhier(dest_dir) |
| dest_path = os.path.join(dest, path) |
| shutil.move(os.path.join(orig, path), dest_path) |
| |
| def _get_checksums(rf): |
| import re |
| checksums = {} |
| with open(rf) as f: |
| for line in f: |
| for cs in ['md5sum', 'sha256sum']: |
| m = re.match("^SRC_URI\[%s\].*=.*\"(.*)\"" % cs, line) |
| if m: |
| checksums[cs] = m.group(1) |
| return checksums |
| |
| def _remove_patch_dirs(recipefolder): |
| for root, dirs, files in os.walk(recipefolder): |
| for d in dirs: |
| shutil.rmtree(os.path.join(root,d)) |
| |
| def _recipe_contains(rd, var): |
| rf = rd.getVar('FILE', True) |
| varfiles = oe.recipeutils.get_var_files(rf, [var], rd) |
| for var, fn in varfiles.items(): |
| if fn and fn.startswith(os.path.dirname(rf) + os.sep): |
| return True |
| return False |
| |
| def _rename_recipe_dirs(oldpv, newpv, path): |
| for root, dirs, files in os.walk(path): |
| # Rename directories with the version in their name |
| for olddir in dirs: |
| if olddir.find(oldpv) != -1: |
| newdir = olddir.replace(oldpv, newpv) |
| if olddir != newdir: |
| shutil.move(os.path.join(path, olddir), os.path.join(path, newdir)) |
| # Rename any inc files with the version in their name (unusual, but possible) |
| for oldfile in files: |
| if oldfile.endswith('.inc'): |
| if oldfile.find(oldpv) != -1: |
| newfile = oldfile.replace(oldpv, newpv) |
| if oldfile != newfile: |
| os.rename(os.path.join(path, oldfile), os.path.join(path, newfile)) |
| |
| def _rename_recipe_file(oldrecipe, bpn, oldpv, newpv, path): |
| oldrecipe = os.path.basename(oldrecipe) |
| if oldrecipe.endswith('_%s.bb' % oldpv): |
| newrecipe = '%s_%s.bb' % (bpn, newpv) |
| if oldrecipe != newrecipe: |
| shutil.move(os.path.join(path, oldrecipe), os.path.join(path, newrecipe)) |
| else: |
| newrecipe = oldrecipe |
| return os.path.join(path, newrecipe) |
| |
| def _rename_recipe_files(oldrecipe, bpn, oldpv, newpv, path): |
| _rename_recipe_dirs(oldpv, newpv, path) |
| return _rename_recipe_file(oldrecipe, bpn, oldpv, newpv, path) |
| |
| def _write_append(rc, srctree, same_dir, no_same_dir, rev, copied, workspace, d): |
| """Writes an append file""" |
| if not os.path.exists(rc): |
| raise DevtoolError("bbappend not created because %s does not exist" % rc) |
| |
| appendpath = os.path.join(workspace, 'appends') |
| if not os.path.exists(appendpath): |
| bb.utils.mkdirhier(appendpath) |
| |
| brf = os.path.basename(os.path.splitext(rc)[0]) # rc basename |
| |
| srctree = os.path.abspath(srctree) |
| pn = d.getVar('PN',True) |
| af = os.path.join(appendpath, '%s.bbappend' % brf) |
| with open(af, 'w') as f: |
| f.write('FILESEXTRAPATHS_prepend := "${THISDIR}/${PN}:"\n\n') |
| f.write('inherit externalsrc\n') |
| f.write(('# NOTE: We use pn- overrides here to avoid affecting' |
| 'multiple variants in the case where the recipe uses BBCLASSEXTEND\n')) |
| f.write('EXTERNALSRC_pn-%s = "%s"\n' % (pn, srctree)) |
| b_is_s = use_external_build(same_dir, no_same_dir, d) |
| if b_is_s: |
| f.write('EXTERNALSRC_BUILD_pn-%s = "%s"\n' % (pn, srctree)) |
| f.write('\n') |
| if rev: |
| f.write('# initial_rev: %s\n' % rev) |
| if copied: |
| f.write('# original_path: %s\n' % os.path.dirname(d.getVar('FILE', True))) |
| f.write('# original_files: %s\n' % ' '.join(copied)) |
| return af |
| |
| def _cleanup_on_error(rf, srctree): |
| rfp = os.path.split(rf)[0] # recipe folder |
| rfpp = os.path.split(rfp)[0] # recipes folder |
| if os.path.exists(rfp): |
| shutil.rmtree(b) |
| if not len(os.listdir(rfpp)): |
| os.rmdir(rfpp) |
| srctree = os.path.abspath(srctree) |
| if os.path.exists(srctree): |
| shutil.rmtree(srctree) |
| |
| def _upgrade_error(e, rf, srctree): |
| if rf: |
| cleanup_on_error(rf, srctree) |
| logger.error(e) |
| raise DevtoolError(e) |
| |
| def _get_uri(rd): |
| srcuris = rd.getVar('SRC_URI', True).split() |
| if not len(srcuris): |
| raise DevtoolError('SRC_URI not found on recipe') |
| # Get first non-local entry in SRC_URI - usually by convention it's |
| # the first entry, but not always! |
| srcuri = None |
| for entry in srcuris: |
| if not entry.startswith('file://'): |
| srcuri = entry |
| break |
| if not srcuri: |
| raise DevtoolError('Unable to find non-local entry in SRC_URI') |
| srcrev = '${AUTOREV}' |
| if '://' in srcuri: |
| # Fetch a URL |
| rev_re = re.compile(';rev=([^;]+)') |
| res = rev_re.search(srcuri) |
| if res: |
| srcrev = res.group(1) |
| srcuri = rev_re.sub('', srcuri) |
| return srcuri, srcrev |
| |
| def _extract_new_source(newpv, srctree, no_patch, srcrev, branch, keep_temp, tinfoil, rd): |
| """Extract sources of a recipe with a new version""" |
| |
| def __run(cmd): |
| """Simple wrapper which calls _run with srctree as cwd""" |
| return _run(cmd, srctree) |
| |
| crd = rd.createCopy() |
| |
| pv = crd.getVar('PV', True) |
| crd.setVar('PV', newpv) |
| |
| tmpsrctree = None |
| uri, rev = _get_uri(crd) |
| if srcrev: |
| rev = srcrev |
| if uri.startswith('git://'): |
| __run('git fetch') |
| __run('git checkout %s' % rev) |
| __run('git tag -f devtool-base-new') |
| md5 = None |
| sha256 = None |
| else: |
| __run('git checkout devtool-base -b devtool-%s' % newpv) |
| |
| tmpdir = tempfile.mkdtemp(prefix='devtool') |
| try: |
| md5, sha256 = scriptutils.fetch_uri(tinfoil.config_data, uri, tmpdir, rev) |
| except bb.fetch2.FetchError as e: |
| raise DevtoolError(e) |
| |
| tmpsrctree = _get_srctree(tmpdir) |
| srctree = os.path.abspath(srctree) |
| |
| # Delete all sources so we ensure no stray files are left over |
| for item in os.listdir(srctree): |
| if item in ['.git', 'oe-local-files']: |
| continue |
| itempath = os.path.join(srctree, item) |
| if os.path.isdir(itempath): |
| shutil.rmtree(itempath) |
| else: |
| os.remove(itempath) |
| |
| # Copy in new ones |
| _copy_source_code(tmpsrctree, srctree) |
| |
| (stdout,_) = __run('git ls-files --modified --others --exclude-standard') |
| for f in stdout.splitlines(): |
| __run('git add "%s"' % f) |
| |
| useroptions = [] |
| oe.patch.GitApplyTree.gitCommandUserOptions(useroptions, d=rd) |
| __run('git %s commit -q -m "Commit of upstream changes at version %s" --allow-empty' % (' '.join(useroptions), newpv)) |
| __run('git tag -f devtool-base-%s' % newpv) |
| |
| (stdout, _) = __run('git rev-parse HEAD') |
| rev = stdout.rstrip() |
| |
| if no_patch: |
| patches = oe.recipeutils.get_recipe_patches(crd) |
| if len(patches): |
| logger.warn('By user choice, the following patches will NOT be applied') |
| for patch in patches: |
| logger.warn("%s" % os.path.basename(patch)) |
| else: |
| __run('git checkout devtool-patched -b %s' % branch) |
| skiptag = False |
| try: |
| __run('git rebase %s' % rev) |
| except bb.process.ExecutionError as e: |
| skiptag = True |
| if 'conflict' in e.stdout: |
| logger.warn('Command \'%s\' failed:\n%s\n\nYou will need to resolve conflicts in order to complete the upgrade.' % (e.command, e.stdout.rstrip())) |
| else: |
| logger.warn('Command \'%s\' failed:\n%s' % (e.command, e.stdout)) |
| if not skiptag: |
| if uri.startswith('git://'): |
| suffix = 'new' |
| else: |
| suffix = newpv |
| __run('git tag -f devtool-patched-%s' % suffix) |
| |
| if tmpsrctree: |
| if keep_temp: |
| logger.info('Preserving temporary directory %s' % tmpsrctree) |
| else: |
| shutil.rmtree(tmpsrctree) |
| |
| return (rev, md5, sha256) |
| |
| def _create_new_recipe(newpv, md5, sha256, srcrev, srcbranch, workspace, tinfoil, rd): |
| """Creates the new recipe under workspace""" |
| |
| bpn = rd.getVar('BPN', True) |
| path = os.path.join(workspace, 'recipes', bpn) |
| bb.utils.mkdirhier(path) |
| copied, _ = oe.recipeutils.copy_recipe_files(rd, path) |
| |
| oldpv = rd.getVar('PV', True) |
| if not newpv: |
| newpv = oldpv |
| origpath = rd.getVar('FILE', True) |
| fullpath = _rename_recipe_files(origpath, bpn, oldpv, newpv, path) |
| logger.debug('Upgraded %s => %s' % (origpath, fullpath)) |
| |
| newvalues = {} |
| if _recipe_contains(rd, 'PV') and newpv != oldpv: |
| newvalues['PV'] = newpv |
| |
| if srcrev: |
| newvalues['SRCREV'] = srcrev |
| |
| if srcbranch: |
| src_uri = oe.recipeutils.split_var_value(rd.getVar('SRC_URI', False) or '') |
| changed = False |
| replacing = True |
| new_src_uri = [] |
| for entry in src_uri: |
| scheme, network, path, user, passwd, params = bb.fetch2.decodeurl(entry) |
| if replacing and scheme in ['git', 'gitsm']: |
| branch = params.get('branch', 'master') |
| if rd.expand(branch) != srcbranch: |
| # Handle case where branch is set through a variable |
| res = re.match(r'\$\{([^}@]+)\}', branch) |
| if res: |
| newvalues[res.group(1)] = srcbranch |
| # We know we won't change SRC_URI now, so break out |
| break |
| else: |
| params['branch'] = srcbranch |
| entry = bb.fetch2.encodeurl((scheme, network, path, user, passwd, params)) |
| changed = True |
| replacing = False |
| new_src_uri.append(entry) |
| if changed: |
| newvalues['SRC_URI'] = ' '.join(new_src_uri) |
| |
| newvalues['PR'] = None |
| |
| if md5 and sha256: |
| newvalues['SRC_URI[md5sum]'] = md5 |
| newvalues['SRC_URI[sha256sum]'] = sha256 |
| |
| rd = oe.recipeutils.parse_recipe(tinfoil.cooker, fullpath, None) |
| oe.recipeutils.patch_recipe(rd, fullpath, newvalues) |
| |
| return fullpath, copied |
| |
| def upgrade(args, config, basepath, workspace): |
| """Entry point for the devtool 'upgrade' subcommand""" |
| |
| if args.recipename in workspace: |
| raise DevtoolError("recipe %s is already in your workspace" % args.recipename) |
| if not args.version and not args.srcrev: |
| raise DevtoolError("You must provide a version using the --version/-V option, or for recipes that fetch from an SCM such as git, the --srcrev/-S option") |
| if args.srcbranch and not args.srcrev: |
| raise DevtoolError("If you specify --srcbranch/-B then you must use --srcrev/-S to specify the revision" % args.recipename) |
| |
| tinfoil = setup_tinfoil(basepath=basepath, tracking=True) |
| try: |
| rd = parse_recipe(config, tinfoil, args.recipename, True) |
| if not rd: |
| return 1 |
| |
| pn = rd.getVar('PN', True) |
| if pn != args.recipename: |
| logger.info('Mapping %s to %s' % (args.recipename, pn)) |
| if pn in workspace: |
| raise DevtoolError("recipe %s is already in your workspace" % pn) |
| |
| if args.srctree: |
| srctree = os.path.abspath(args.srctree) |
| else: |
| srctree = standard.get_default_srctree(config, pn) |
| |
| standard._check_compatible_recipe(pn, rd) |
| old_srcrev = rd.getVar('SRCREV', True) |
| if old_srcrev == 'INVALID': |
| old_srcrev = None |
| if old_srcrev and not args.srcrev: |
| raise DevtoolError("Recipe specifies a SRCREV value; you must specify a new one when upgrading") |
| if rd.getVar('PV', True) == args.version and old_srcrev == args.srcrev: |
| raise DevtoolError("Current and upgrade versions are the same version") |
| |
| rf = None |
| try: |
| rev1 = standard._extract_source(srctree, False, 'devtool-orig', False, rd) |
| rev2, md5, sha256 = _extract_new_source(args.version, srctree, args.no_patch, |
| args.srcrev, args.branch, args.keep_temp, |
| tinfoil, rd) |
| rf, copied = _create_new_recipe(args.version, md5, sha256, args.srcrev, args.srcbranch, config.workspace_path, tinfoil, rd) |
| except bb.process.CmdError as e: |
| _upgrade_error(e, rf, srctree) |
| except DevtoolError as e: |
| _upgrade_error(e, rf, srctree) |
| standard._add_md5(config, pn, os.path.dirname(rf)) |
| |
| af = _write_append(rf, srctree, args.same_dir, args.no_same_dir, rev2, |
| copied, config.workspace_path, rd) |
| standard._add_md5(config, pn, af) |
| logger.info('Upgraded source extracted to %s' % srctree) |
| logger.info('New recipe is %s' % rf) |
| finally: |
| tinfoil.shutdown() |
| return 0 |
| |
| def register_commands(subparsers, context): |
| """Register devtool subcommands from this plugin""" |
| |
| defsrctree = standard.get_default_srctree(context.config) |
| |
| parser_upgrade = subparsers.add_parser('upgrade', help='Upgrade an existing recipe', |
| description='Upgrades an existing recipe to a new upstream version. Puts the upgraded recipe file into the workspace along with any associated files, and extracts the source tree to a specified location (in case patches need rebasing or adding to as a result of the upgrade).', |
| group='starting') |
| parser_upgrade.add_argument('recipename', help='Name of recipe to upgrade (just name - no version, path or extension)') |
| parser_upgrade.add_argument('srctree', nargs='?', help='Path to where to extract the source tree. If not specified, a subdirectory of %s will be used.' % defsrctree) |
| parser_upgrade.add_argument('--version', '-V', help='Version to upgrade to (PV)') |
| parser_upgrade.add_argument('--srcrev', '-S', help='Source revision to upgrade to (required if fetching from an SCM such as git)') |
| parser_upgrade.add_argument('--srcbranch', '-B', help='Branch in source repository containing the revision to use (if fetching from an SCM such as git)') |
| parser_upgrade.add_argument('--branch', '-b', default="devtool", help='Name for new development branch to checkout (default "%(default)s")') |
| parser_upgrade.add_argument('--no-patch', action="store_true", help='Do not apply patches from the recipe to the new source code') |
| group = parser_upgrade.add_mutually_exclusive_group() |
| group.add_argument('--same-dir', '-s', help='Build in same directory as source', action="store_true") |
| group.add_argument('--no-same-dir', help='Force build in a separate build directory', action="store_true") |
| parser_upgrade.add_argument('--keep-temp', action="store_true", help='Keep temporary directory (for debugging)') |
| parser_upgrade.set_defaults(func=upgrade) |