| # Development tool - deploy/undeploy 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 plugin containing the deploy subcommands""" |
| |
| import os |
| import subprocess |
| import logging |
| from devtool import exec_fakeroot, setup_tinfoil, DevtoolError |
| |
| logger = logging.getLogger('devtool') |
| |
| def deploy(args, config, basepath, workspace): |
| """Entry point for the devtool 'deploy' subcommand""" |
| import re |
| import oe.recipeutils |
| |
| if not args.recipename in workspace: |
| raise DevtoolError("no recipe named %s in your workspace" % |
| args.recipename) |
| try: |
| host, destdir = args.target.split(':') |
| except ValueError: |
| destdir = '/' |
| else: |
| args.target = host |
| |
| deploy_dir = os.path.join(basepath, 'target_deploy', args.target) |
| deploy_file = os.path.join(deploy_dir, args.recipename + '.list') |
| |
| tinfoil = setup_tinfoil() |
| try: |
| rd = oe.recipeutils.parse_recipe_simple(tinfoil.cooker, args.recipename, tinfoil.config_data) |
| except Exception as e: |
| raise DevtoolError('Exception parsing recipe %s: %s' % |
| (args.recipename, e)) |
| recipe_outdir = rd.getVar('D', True) |
| if not os.path.exists(recipe_outdir) or not os.listdir(recipe_outdir): |
| raise DevtoolError('No files to deploy - have you built the %s ' |
| 'recipe? If so, the install step has not installed ' |
| 'any files.' % args.recipename) |
| |
| if args.dry_run: |
| print('Files to be deployed for %s on target %s:' % (args.recipename, args.target)) |
| for root, _, files in os.walk(recipe_outdir): |
| for fn in files: |
| print(' %s' % os.path.join(destdir, os.path.relpath(root, recipe_outdir), fn)) |
| return 0 |
| |
| if os.path.exists(deploy_file): |
| if undeploy(args, config, basepath, workspace): |
| # Error already shown |
| return 1 |
| |
| extraoptions = '' |
| if args.no_host_check: |
| extraoptions += '-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no' |
| if args.show_status: |
| tarextractopts = 'xv' |
| else: |
| tarextractopts = 'x' |
| extraoptions += ' -q' |
| # We cannot use scp here, because it doesn't preserve symlinks |
| ret = exec_fakeroot(rd, 'tar cf - . | ssh %s %s \'tar %s -C %s -f -\'' % (extraoptions, args.target, tarextractopts, destdir), cwd=recipe_outdir, shell=True) |
| if ret != 0: |
| raise DevtoolError('Deploy failed - rerun with -s to get a complete ' |
| 'error message') |
| |
| logger.info('Successfully deployed %s' % recipe_outdir) |
| |
| if not os.path.exists(deploy_dir): |
| os.makedirs(deploy_dir) |
| |
| files_list = [] |
| for root, _, files in os.walk(recipe_outdir): |
| for filename in files: |
| filename = os.path.relpath(os.path.join(root, filename), recipe_outdir) |
| files_list.append(os.path.join(destdir, filename)) |
| |
| with open(deploy_file, 'w') as fobj: |
| fobj.write('\n'.join(files_list)) |
| |
| return 0 |
| |
| def undeploy(args, config, basepath, workspace): |
| """Entry point for the devtool 'undeploy' subcommand""" |
| deploy_file = os.path.join(basepath, 'target_deploy', args.target, args.recipename + '.list') |
| if not os.path.exists(deploy_file): |
| raise DevtoolError('%s has not been deployed' % args.recipename) |
| |
| if args.dry_run: |
| print('Previously deployed files to be un-deployed for %s on target %s:' % (args.recipename, args.target)) |
| with open(deploy_file, 'r') as f: |
| for line in f: |
| print(' %s' % line.rstrip()) |
| return 0 |
| |
| extraoptions = '' |
| if args.no_host_check: |
| extraoptions += '-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no' |
| if not args.show_status: |
| extraoptions += ' -q' |
| |
| ret = subprocess.call("scp %s %s %s:/tmp" % (extraoptions, deploy_file, args.target), shell=True) |
| if ret != 0: |
| raise DevtoolError('Failed to copy file list to %s - rerun with -s to ' |
| 'get a complete error message' % args.target) |
| |
| ret = subprocess.call("ssh %s %s 'xargs -n1 rm -f </tmp/%s'" % (extraoptions, args.target, os.path.basename(deploy_file)), shell=True) |
| if ret == 0: |
| logger.info('Successfully undeployed %s' % args.recipename) |
| os.remove(deploy_file) |
| else: |
| raise DevtoolError('Undeploy failed - rerun with -s to get a complete ' |
| 'error message') |
| |
| return ret |
| |
| |
| def register_commands(subparsers, context): |
| """Register devtool subcommands from the deploy plugin""" |
| parser_deploy = subparsers.add_parser('deploy-target', help='Deploy recipe output files to live target machine') |
| parser_deploy.add_argument('recipename', help='Recipe to deploy') |
| parser_deploy.add_argument('target', help='Live target machine running an ssh server: user@hostname[:destdir]') |
| parser_deploy.add_argument('-c', '--no-host-check', help='Disable ssh host key checking', action='store_true') |
| parser_deploy.add_argument('-s', '--show-status', help='Show progress/status output', action='store_true') |
| parser_deploy.add_argument('-n', '--dry-run', help='List files to be deployed only', action='store_true') |
| parser_deploy.set_defaults(func=deploy) |
| |
| parser_undeploy = subparsers.add_parser('undeploy-target', help='Undeploy recipe output files in live target machine') |
| parser_undeploy.add_argument('recipename', help='Recipe to undeploy') |
| parser_undeploy.add_argument('target', help='Live target machine running an ssh server: user@hostname') |
| parser_undeploy.add_argument('-c', '--no-host-check', help='Disable ssh host key checking', action='store_true') |
| parser_undeploy.add_argument('-s', '--show-status', help='Show progress/status output', action='store_true') |
| parser_undeploy.add_argument('-n', '--dry-run', help='List files to be undeployed only', action='store_true') |
| parser_undeploy.set_defaults(func=undeploy) |