Edit File: clselectctlruby.py
# -*- coding: utf-8 -*- # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT from __future__ import print_function from __future__ import absolute_import from __future__ import division import getopt import glob import os import shutil import sys import traceback from future.utils import iteritems from . import clpassenger from . import clselectctl from . import utils from .clselectprint import clprint from .clselectexcept import ClSelectExcept, BaseClSelectException from .clselectruby import environments, extensions, interpreters from clcommon.cpapi import userdomains, docroot, CP_NAME from clcommon.cpapi.cpapiexceptions import NoDomain, NotSupported INTERPRETER = 'ruby' def usage(): print(' -v | --version VERSION : Specify alternative version') print(' -u | --user USERNAME : Username') print(' --domain DOMAIN : Domain (or subdomain); users main domain as default') print(' -l | --list : List alternatives for interpreter') print(' -G | --list-extensions : List global set of packages') print(' -K | --list-extensions-version : List version set of packages') print(' -s | --user-summary : List user summary of webapps') print(' -y | --create-webapp : Create user webapp') print(' -n | --destroy-webapp : Destroy user webapp') print(' -f | --relocate-webapp : Change webapp directory (files must be moved manually)') print(' -F | --transit-webapp : Change webapp domain/alias') print(' -Z | --restart-webapp : Restart webapp') print(' -c | --user-current : Show currently selected alternative for user webapp') print(' -b | --set-user-current : Set alternative as user webapp default') print(' -e | --enable-user-extensions GEMS : Install comma-separated list of gems for user webapp') print(' -d | --disable-user-extensions GEMS : Uninstall comma-separated list of gems for user webapp') print(" : Use '-' (minus) for all gems") print(' -r | --replace-user-extensions GEMS : Update comma-separated list of gems for user webapp') print(" : Use '-' (minus) for all gems") print(' -g | --list-user-extensions : List installed gems for user webapp') print(' -p | --print-summary : If specified along with setting an alternative prints user summary') print(' -j | --json : Print data as JSON') def _create_environment(user, directory, version, env_name=None): prefix = _get_prefix(user, directory) if not env_name: env_name = version environment = environments.Environment(env_name, user, prefix) if not environment.exists(): try: interpreter = interpreters.interpreters(key='version')[version] except KeyError: raise ClSelectExcept.NoSuchAlternativeVersion(version) environment.create(interpreter) return environment def _get_environment(user, directory, app_summary=None): prefix = _get_prefix(user, directory) if app_summary is None: user_summary = clpassenger.summary(user) app_summary = user_summary.get(directory) if not app_summary: raise ClSelectExcept.NoSuchApplication('No such application (or application not configured) "%s"' % directory) binary = app_summary['binary'] env_name = os.path.basename(os.path.dirname(os.path.dirname(binary))) environment = environments.Environment(env_name, user, prefix) return environment def _get_prefix(user, directory): _, rel_dir = utils.get_abs_rel(user, directory) return os.path.join(environments.DEFAULT_PREFIX, clselectctl.get_prefix(rel_dir)) def create(user, directory, alias, version=None, env_name=None, doc_root=None): if version is None: raise ClSelectExcept.WrongData('Not passed version as argument') user = clselectctl.get_user(user) clselectctl.check_directory(directory) alias = clselectctl.get_alias(alias) environment = _create_environment(user, directory, version, env_name) binary = environment.interpreter().binary clpassenger.configure(user, directory, alias, INTERPRETER, binary, doc_root=doc_root) clpassenger.restart(user, directory) def current(user, directory, version=None, env_name=None): user = clselectctl.get_user(user) old_environment = _get_environment(user, directory) if not version: return {old_environment.name: old_environment.interpreter().as_dict()} old_extensions = set(old_environment.extensions()) new_environment = _create_environment(user, directory, version, env_name) new_extensions = set(new_environment.extensions()) for extension in new_extensions - old_extensions: try: new_environment.extension_uninstall(extension) except ClSelectExcept.ExternalProgramFailed: pass for extension in old_extensions - new_extensions: try: new_environment.extension_install(extension) except ClSelectExcept.ExternalProgramFailed: pass app_summary = clpassenger.summary(user)[directory] alias = app_summary['alias'] doc_root = app_summary['docroot'] htaccess_path = app_summary['htaccess'] binary = new_environment.interpreter().binary clpassenger._unconfigure(htaccess=htaccess_path) clpassenger.configure(user, directory, alias, INTERPRETER, binary, doc_root=doc_root) clpassenger.restart(user, directory) def destroy(user, directory): user = clselectctl.get_user(user) prefix = _get_environment(user, directory).prefix abs_dir, _ = utils.get_abs_rel(user, prefix) try: shutil.rmtree(abs_dir) except OSError: pass clpassenger.unconfigure(user, directory) try: clpassenger.restart(user, directory) except ClSelectExcept.MissingApprootDirectory: pass def install(user, directory, extension): user = clselectctl.get_user(user) environment = _get_environment(user, directory) environment.extension_install(extension) def list_extensions(user, directory): user = clselectctl.get_user(user) environment = _get_environment(user, directory) return environment.extensions() def relocate(user, old_directory, new_directory, fmt): if '/' in new_directory: raise ClSelectExcept.WrongData('You cannot move appication to subdir') user = clselectctl.get_user(user) clselectctl.check_directory(new_directory) old_abs, old_rel = utils.get_abs_rel(user, old_directory) new_abs, new_rel = utils.get_abs_rel(user, new_directory) old_user_summary = clpassenger.summary(user) if new_directory in old_user_summary: raise ClSelectExcept.WebAppError("Specified directory already used by '%s'" % new_abs) if old_directory not in old_user_summary: raise ClSelectExcept.WrongData("No such application (or application not configured) \"%s\"" % old_directory) htaccess_path = old_user_summary[old_directory]['htaccess'] doc_root = old_user_summary[old_directory]['docroot'] alias = old_user_summary[old_directory]['alias'] env_name = _get_environment(user, old_directory).name old_prefix = clselectctl.get_prefix(old_rel) new_prefix = clselectctl.get_prefix(new_rel) _old_envs, _ = utils.get_abs_rel(user, os.path.join( environments.DEFAULT_PREFIX, old_prefix)) old_envs = os.path.join(_old_envs, '') _new_envs, _ = utils.get_abs_rel(user, os.path.join( environments.DEFAULT_PREFIX, new_prefix)) new_envs = os.path.join(_new_envs, '') old_prompt = '(' + old_rel + ':' new_prompt = '(' + new_rel + ':' shutil.move(old_envs, new_envs) for activate in glob.glob(os.path.join(new_envs, '*', 'bin', '*')): old_activate = utils.file_read(activate) if old_prompt in old_activate: new_activate = old_activate.replace(old_prompt, new_prompt) utils.file_write(activate, new_activate, 'w') if not os.path.exists(new_abs): os.rename(old_abs, new_abs) prefix = _get_prefix(user, new_directory) environment = environments.Environment(env_name, user, prefix) binary = environment.interpreter().binary clpassenger._unconfigure(htaccess=htaccess_path) clpassenger.configure(user, new_directory, alias, INTERPRETER, binary, doc_root=doc_root) clpassenger.restart(user, new_directory) def restart(user, directory): user = clselectctl.get_user(user) apps_summary = clpassenger.summary(user) if directory not in apps_summary: raise ClSelectExcept.WrongData("No such application (or application not configured) \"%s\"" % directory) clpassenger.restart(user, directory) def summary(user): user = clselectctl.get_user(user) summ = {} for directory, data in iteritems(clpassenger.summary(user)): if data['interpreter'] != INTERPRETER: continue environment = _get_environment(user, directory, data).as_deepdict() summ[directory] = { 'domain': data['domain'], 'alias': data['alias'], 'environment': environment['name'], 'interpreter': environment['interpreter'], 'extensions': environment['extensions'], } # add only list with additions domains if "domains" in data and len(data["domains"]) > 1: summ[directory]["domains"] = data["domains"] return summ def transit(user, directory, alias, doc_root=None): user = clselectctl.get_user(user) apps_summary = clpassenger.summary(user) if directory not in apps_summary: raise ClSelectExcept.WrongData("No such application (or application not configured) \"%s\"" % directory) old_app_summary = apps_summary[directory] old_alias = old_app_summary['alias'] old_doc_root = old_app_summary['docroot'] new_alias = clselectctl.get_alias(alias) environment = _get_environment(user, directory) binary = environment.interpreter().binary clpassenger.configure(user, directory, new_alias, INTERPRETER, binary, True, 'transit', doc_root=doc_root) clpassenger.move(user, directory, old_alias, new_alias, old_doc_root=old_doc_root, new_doc_root=doc_root) clpassenger.restart(user, directory) def uninstall(user, directory, extension): user = clselectctl.get_user(user) environment = _get_environment(user, directory) environment.extension_uninstall(extension) def update(user, directory, extension): user = clselectctl.get_user(user) environment = _get_environment(user, directory) environment.extension_update(extension) def main(): try: opts, args = getopt.getopt( sys.argv[1:], 'hi:v:u:lGsynfFZcbe:d:r:gpjK:', [ 'help', 'interpreter=', 'version=', 'user=', 'domain=', 'list', 'list-extensions', 'user-summary', 'create-webapp', 'destroy-webapp', 'relocate-webapp', 'transit-webapp', 'restart-webapp', 'user-current', 'set-user-current', 'enable-user-extensions=', 'disable-user-extensions=', 'replace-user-extensions=', 'list-user-extensions', 'print-summary', 'json', 'list-extensions-version=' ]) except getopt.GetoptError as err: sys.stderr.write(str(err)) usage() sys.exit(1) ext_list = '' fmt = 'text' print_summary = False user = None domain = None action = '' version = None if not opts: usage() sys.exit(1) for o, a in opts: if o in ('-i', '--interpreter'): pass elif o in ('-l', '--list'): action = 'list' elif o in ('-y', '--create-webapp'): action = 'create-webapp' elif o in ('-n', '--destroy-webapp'): action = 'destroy-webapp' elif o in ('-f', '--relocate-webapp'): action = 'relocate-webapp' elif o in ('-F', '--transit-webapp'): action = 'transit-webapp' elif o in ('-Z', '--restart-webapp'): action = 'restart-webapp' elif o in ('-c', '--user-current'): action = 'user-current' elif o in ('-b', '--set-user-current'): action = 'set-user-current' elif o in ('-g', '--list-user-extensions'): action = 'list-user-extensions' elif o in ('-e', '--enable-user-extensions'): action = 'enable-user-extensions' ext_list = a elif o in ('-s', '--user-summary'): action = 'user-summary' elif o in ('-j', '--json'): fmt = 'json' elif o in ('-r', '--replace-user-extensions'): action = 'replace-user-extensions' ext_list = a elif o in ('-d', '--disable-user-extensions'): action = 'disable-user-extensions' ext_list = a elif o in ('-v', '--version'): version = a elif o in ('-G', '--list-extensions'): action = 'list-extensions' elif o in ('-u', '--user'): user = a elif o == '--domain': domain = a elif o in ('-p', '--print-summary'): print_summary = True elif o in ('-K', '--list-extensions-version'): action = 'list-extensions-version' ext_list = a else: sys.stderr.write('unhandled option') sys.exit(1) if action == '': sys.stderr.write('ERROR:you must provide option for interpreter ruby') sys.exit(1) if action in ('create-webapp', 'destroy-webapp', 'relocate-webapp', 'transit-webapp', 'restart-webapp', 'enable-user-extensions', 'list-user-extensions', 'replace-user-extensions', 'disable-user-extensions', ): if not args: sys.stderr.write('webapp must be specified') sys.exit(3) if domain: try: doc_root, user_ = docroot(domain) except NoDomain: clprint.print_diag(fmt, {'status': 'ERROR', 'message': 'No such domain: "%s"' % domain}) sys.exit(1) except NotSupported: clprint.print_diag(fmt, {'status': 'ERROR', 'message': 'Ruby selector not supported for %s' % CP_NAME}) sys.exit(1) if not user: # we can obtain user name for domain user = user_ elif user != user_: clprint.print_diag(fmt, {'status': 'ERROR', 'message': 'domain %s is not owned by the user %s' % (domain, user)}) sys.exit(1) elif user and not domain: try: domain_list = userdomains(user) except NotSupported: clprint.print_diag(fmt, {'status': 'ERROR', 'message': 'Ruby selector not supported for %s' % CP_NAME}) sys.exit(1) _, doc_root = domain_list[0] # get document root for main domain if ext_list == '-': if action == 'enable-user-extensions': sys.stderr.write('installlation of all extensions is not possible') sys.exit(4) elif ext_list: _exts = [_f for _f in ext_list.split(',') if _f] try: error = 0 ok = 0 result = {} if action == 'list-extensions': result = extensions.ExtensionInfo().list_extensions_cached() elif action == 'list-extensions-version': result = extensions.ExtensionInfo().list_extensions_version(_exts) ext_list = '' elif action == 'list': result = interpreters.interpreters_dict('version') elif action == 'user-summary': result = summary(user) elif action == 'create-webapp': create(user, args[0], args[1], version, doc_root=doc_root) elif action == 'destroy-webapp': destroy(user, args[0]) elif action == 'relocate-webapp': relocate(user, args[0], args[1], fmt) elif action == 'transit-webapp': new_doc_root = None if domain: # remove app to new domain if domain present new_doc_root = doc_root transit(user, args[0], args[1], doc_root=new_doc_root) elif action == 'restart-webapp': restart(user, args[0]) elif action == 'user-current': result = current(user, args[0]) elif action == 'set-user-current': current(user, args[0], version) elif action == 'list-user-extensions': result = list_extensions(user, args[0]) else: alias = args[0] if ext_list == '-': _exts = list_extensions(user, alias) for extension in _exts: try: if action == 'enable-user-extensions': install(user, alias, extension) elif action == 'disable-user-extensions': uninstall(user, alias, extension) elif action == 'replace-user-extensions': update(user, alias, extension) result.update({extension: {'status': 'OK'}}) ok += 1 except (ValueError, ClSelectExcept.ExternalProgramFailed) as err: result.update( {extension: {'status': 'ERROR', 'message': str(err)}}) error += 1 except BaseClSelectException as err: clprint.print_diag(fmt, {'status': 'ERROR', 'message': str(err)}) sys.exit(1) except Exception as err: msg = traceback.format_exc() clprint.print_diag(fmt, {'status': 'ERROR', 'message': msg}) sys.exit(1) if not result and print_summary: result = summary(user) if error and ok: status = 'PARTIAL' exit_status = 2 elif error: status = 'ERROR' exit_status = 5 else: if ext_list and ok < 2: if print_summary: clprint.print_data(fmt, summary(user)) else: clprint.print_data(fmt, {}) else: clprint.print_data(fmt, result) sys.exit(0) message = '\n'.join( '%s: %s' % (k, v.get('message', v.get('status', ''))) for k, v in iteritems(result)) clprint.print_diag(fmt, {'status': status, 'message': message}) sys.exit(exit_status)