Synnefo » ./kamaki

# Copyright 2011-2013 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.command

from json import load, loads

from os.path import abspath

from kamaki.cli import command

from kamaki.clients.astakos import SynnefoAstakosClient

from kamaki.cli.commands import (

    _command_init, errors, _optional_json, addLogSettings)

from kamaki.cli.command_tree import CommandTree

from kamaki.cli.errors import CLIBaseUrlError, CLISyntaxError, CLIError

from kamaki.cli.argument import (

    FlagArgument, ValueArgument, IntArgument, CommaSeparatedListArgument)

from kamaki.cli.utils import format_size

user_commands = CommandTree('user', 'Astakos/Identity API commands')

admin_commands = CommandTree('admin', 'Astakos/Account API commands')

project_commands = CommandTree('project', 'Astakos project API commands')

_commands = [user_commands, admin_commands, project_commands]

def with_temp_token(foo):

    """ Set token to self.client.token, run foo, recover old token """

    def wrap(self, *args, **kwargs):

        try:

            token = kwargs.pop('token')

        except KeyError:

            raise CLISyntaxError('A token is needed for %s' % foo)

        token_bu = self.client.token

        try:

            self.client.token = token or token_bu

            return foo(self, *args, **kwargs)

        finally:

            self.client.token = token_bu

    return wrap

class _init_synnefo_astakosclient(_command_init):

    @errors.generic.all

    @errors.user.load

    @errors.user.astakosclient

    @addLogSettings

    def _run(self):

        if getattr(self, 'cloud', None):

            base_url = self._custom_url('astakos')

            if base_url:

                token = self._custom_token(

                    'astakos') or self.config.get_cloud(

                    self.cloud, 'token')

                token = token.split()[0] if ' ' in token else token

                self.client = SynnefoAstakosClient(

                    auth_url=base_url, token=token)

                return

        else:

            self.cloud = 'default'

        if getattr(self, 'auth_base', None):

            self.client = self.auth_base.get_client()

            return

        raise CLIBaseUrlError(service='astakos')

    def main(self):

        self._run()

@command(user_commands)

class user_authenticate(_init_synnefo_astakosclient, _optional_json):

    """Authenticate a user and get all authentication information"""

    @errors.generic.all

    @errors.user.authenticate

    @errors.user.astakosclient

    @with_temp_token

    def _run(self):

        self._print(self.client.authenticate(), self.print_dict)

    def main(self, token=None):

        super(self.__class__, self)._run()

        self._run(token=token)

@command(user_commands)

class user_uuid2name(_init_synnefo_astakosclient, _optional_json):

    """Get user name(s) from uuid(s)"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, uuids):

        r = self.client.get_usernames(uuids)

        self._print(r, self.print_dict)

        unresolved = set(uuids).difference(r)

        if unresolved:

            self.error('Unresolved uuids: %s' % ', '.join(unresolved))

    def main(self, uuid, *more_uuids):

        super(self.__class__, self)._run()

        self._run(uuids=((uuid, ) + more_uuids))

@command(user_commands)

class user_name2uuid(_init_synnefo_astakosclient, _optional_json):

    """Get user uuid(s) from name(s)"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, usernames):

        r = self.client.get_uuids(usernames)

        self._print(r, self.print_dict)

        unresolved = set(usernames).difference(r)

        if unresolved:

            self.error('Unresolved usernames: %s' % ', '.join(unresolved))

    def main(self, username, *more_usernames):

        super(self.__class__, self)._run()

        self._run(usernames=((username, ) + more_usernames))

@command(user_commands)

class user_quotas(_init_synnefo_astakosclient, _optional_json):

    """Get user quotas"""

    _to_format = set(['cyclades.disk', 'pithos.diskspace', 'cyclades.ram'])

    arguments = dict(

        bytes=FlagArgument('Show data size in bytes', '--bytes')

    )

    def _print_quotas(self, quotas, *args, **kwargs):

        if not self['bytes']:

            for category in quotas.values():

                for service in self._to_format.intersection(category):

                    for attr, v in category[service].items():

                        category[service][attr] = format_size(v)

        self.print_dict(quotas, *args, **kwargs)

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print(self.client.get_quotas(), self._print_quotas)

    def main(self):

        super(self.__class__, self)._run()

        self._run()

#  command user session

@command(user_commands)

class user_session(_init_synnefo_astakosclient):

    """User session commands (cached identity calls for kamaki sessions)"""

@command(user_commands)

class user_session_info(_init_synnefo_astakosclient, _optional_json):

    """Get info for (current) session user"""

    arguments = dict(

        uuid=ValueArgument('Query user with uuid', '--uuid'),

        name=ValueArgument('Query user with username/email', '--username')

    )

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        if self['uuid'] and self['name']:

            raise CLISyntaxError(

                'Arguments uuid and username are mutually exclusive',

                details=['Use either uuid OR username OR none, not both'])

        uuid = self['uuid'] or (self._username2uuid(self['name']) if (

            self['name']) else None)

        try:

            token = self.auth_base.get_token(uuid) if uuid else None

        except KeyError:

            msg = ('id %s' % self['uuid']) if (

                self['uuid']) else 'username %s' % self['name']

            raise CLIError(

                'No user with %s in the cached session list' % msg, details=[

                    'To see all cached session users',

                    '  /user session list',

                    'To authenticate and add a new user in the session list',

                    '  /user session authenticate <new token>'])

        self._print(self.auth_base.user_info(token), self.print_dict)

@command(user_commands)

class user_session_authenticate(_init_synnefo_astakosclient, _optional_json):

    """Authenticate a user by token and update cache"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, token=None):

        ask = token and token not in self.auth_base._uuids

        self._print(self.auth_base.authenticate(token), self.print_dict)

        if ask and self.ask_user(

                'Token is temporarily stored in memory. If it is stored in'

                ' kamaki configuration file, it will be available in later'

                ' sessions. Do you want to permanently store this token?'):

            tokens = self.auth_base._uuids.keys()

            tokens.remove(self.auth_base.token)

            self['config'].set_cloud(

                self.cloud, 'token', ' '.join([self.auth_base.token] + tokens))

            self['config'].write()

    def main(self, new_token=None):

        super(self.__class__, self)._run()

        self._run(token=new_token)

@command(user_commands)

class user_session_list(_init_synnefo_astakosclient, _optional_json):

    """List cached users"""

    arguments = dict(

        detail=FlagArgument('Detailed listing', ('-l', '--detail'))

    )

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print([u if self['detail'] else (dict(

            id=u['id'], name=u['name'])) for u in self.auth_base.list_users()])

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(user_commands)

class user_session_select(_init_synnefo_astakosclient):

    """Pick a user from the cached list to be the current session user"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, uuid):

        try:

            first_token = self.auth_base.get_token(uuid)

        except KeyError:

            raise CLIError(

                'No user with uuid %s in the cached session list' % uuid,

                details=[

                    'To see all cached session users',

                    '  /user session list',

                    'To authenticate and add a new user in the session list',

                    '  /user session authenticate <new token>'])

        if self.auth_base.token != first_token:

            self.auth_base.token = first_token

            msg = 'User with id %s is now the current session user.\n' % uuid

            msg += 'Do you want future sessions to also start with this user?'

            if self.ask_user(msg):

                tokens = self.auth_base._uuids.keys()

                tokens.remove(self.auth_base.token)

                tokens.insert(0, self.auth_base.token)

                self['config'].set_cloud(

                    self.cloud, 'token',  ' '.join(tokens))

                self['config'].write()

                self.error('User is selected for next sessions')

            else:

                self.error('User is not permanently selected')

        else:

            self.error('User was already the selected session user')

    def main(self, user_uuid):

        super(self.__class__, self)._run()

        self._run(uuid=user_uuid)

@command(user_commands)

class user_session_remove(_init_synnefo_astakosclient):

    """Delete a user (token) from the cached list of session users"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, uuid):

        if uuid == self.auth_base.user_term('id'):

            raise CLIError('Cannot remove current session user', details=[

                'To see all cached session users',

                '  /user session list',

                'To see current session user',

                '  /user session info',

                'To select a different session user',

                '  /user session select <user uuid>'])

        try:

            self.auth_base.remove_user(uuid)

        except KeyError:

            raise CLIError('No user with uuid %s in session list' % uuid,

                details=[

                    'To see all cached session users',

                    '  /user session list',

                    'To authenticate and add a new user in the session list',

                    '  /user session authenticate <new token>'])

        if self.ask_user(

                'User is removed from current session, but will be restored in'

                ' the next session. Remove the user from future sessions?'):

            self['config'].set_cloud(

                self.cloud, 'token', ' '.join(self.auth_base._uuids.keys()))

            self['config'].write()

    def main(self, user_uuid):

        super(self.__class__, self)._run()

        self._run(uuid=user_uuid)

#  command admin

@command(admin_commands)

class admin_service(_init_synnefo_astakosclient):

    """Manage commissions (special privileges required)"""

@command(admin_commands)

class admin_service_list(_init_synnefo_astakosclient, _optional_json):

    """List available services"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print(self.client.get_services())

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(admin_commands)

class admin_service_uuid2username(_init_synnefo_astakosclient, _optional_json):

    """Get service username(s) from uuid(s)"""

    @errors.generic.all

    @errors.user.astakosclient

    @with_temp_token

    def _run(self, uuids):

        if 1 == len(uuids):

            self._print(self.client.service_get_username(uuids[0]))

        else:

            self._print(

                self.client.service_get_usernames(uuids),

                self.print_dict)

    def main(self, service_token, uuid, *more_uuids):

        super(self.__class__, self)._run()

        self._run([uuid] + list(more_uuids), token=service_token)

@command(admin_commands)

class admin_service_username2uuid(_init_synnefo_astakosclient, _optional_json):

    """Get service uuid(s) from username(s)"""

    @errors.generic.all

    @errors.user.astakosclient

    @with_temp_token

    def _run(self, usernames):

        if 1 == len(usernames):

            self._print(self.client.service_get_uuid(usernames[0]))

        else:

            self._print(

                self.client.service_get_uuids(usernames),

                self.print_dict)

    def main(self, service_token, usernames, *more_usernames):

        super(self.__class__, self)._run()

        self._run([usernames] + list(more_usernames), token=service_token)

@command(admin_commands)

class admin_service_quotas(_init_synnefo_astakosclient, _optional_json):

    """Get service quotas"""

    arguments = dict(

        uuid=ValueArgument('A user uuid to get quotas for', '--uuid')

    )

    @errors.generic.all

    @errors.user.astakosclient

    @with_temp_token

    def _run(self):

        self._print(self.client.service_get_quotas(self['uuid']))

    def main(self, service_token):

        super(self.__class__, self)._run()

        self._run(token=service_token)

@command(admin_commands)

class admin_commission(_init_synnefo_astakosclient):

    """Manage commissions (special privileges required)"""

@command(admin_commands)

class admin_commission_pending(_init_synnefo_astakosclient, _optional_json):

    """List pending commissions (special privileges required)"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print(self.client.get_pending_commissions())

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(admin_commands)

class admin_commission_info(_init_synnefo_astakosclient, _optional_json):

    """Get commission info (special privileges required)"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, commission_id):

        commission_id = int(commission_id)

        self._print(

            self.client.get_commission_info(commission_id), self.print_dict)

    def main(self, commission_id):

        super(self.__class__, self)._run()

        self._run(commission_id)

@command(admin_commands)

class admin_commission_accept(_init_synnefo_astakosclient):

    """Accept a pending commission  (special privileges required)"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, commission_id):

        commission_id = int(commission_id)

        self.client.accept_commission(commission_id)

    def main(self, commission_id):

        super(self.__class__, self)._run()

        self._run(commission_id)

@command(admin_commands)

class admin_commission_reject(_init_synnefo_astakosclient):

    """Reject a pending commission (special privileges required)"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, commission_id):

        commission_id = int(commission_id)

        self.client.reject_commission(commission_id)

    def main(self, commission_id):

        super(self.__class__, self)._run()

        self._run(commission_id)

@command(admin_commands)

class admin_commission_resolve(_init_synnefo_astakosclient, _optional_json):

    """Resolve multiple commissions (special privileges required)"""

    arguments = dict(

        accept=CommaSeparatedListArgument(

            'commission ids to accept (e.g., --accept=11,12,13,...',

            '--accept'),

        reject=CommaSeparatedListArgument(

            'commission ids to reject (e.g., --reject=11,12,13,...',

            '--reject')

    )

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self.writeln('accepted ', self['accept'])

        self.writeln('rejected ', self['reject'])

        self._print(

            self.client.resolve_commissions(self['accept'], self['reject']),

            self.print_dict)

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(admin_commands)

class admin_commission_issue(_init_synnefo_astakosclient, _optional_json):

    """Issue commissions as a json string (special privileges required)

    Parameters:

    holder      -- user's id (string)

    source      -- commission's source (ex system) (string)

    provisions  -- resources with their quantity (json-dict from string to int)

    name        -- description of the commission (string)

    """

    arguments = dict(

        force=FlagArgument('Force commission', '--force'),

        accept=FlagArgument('Do not wait for verification', '--accept')

    )

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, holder, source, provisions, name=''):

        provisions = loads(provisions)

        self._print(self.client.issue_one_commission(

            holder, source, provisions, name,

            self['force'], self['accept']))

    def main(self, user_uuid, source, provisions_file, name=''):

        super(self.__class__, self)._run()

        self._run(user_uuid, source, provisions_file, name)

@command(admin_commands)

class admin_resources(_init_synnefo_astakosclient, _optional_json):

    """List user resources"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print(self.client.get_resources(), self.print_dict)

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(admin_commands)

class admin_feedback(_init_synnefo_astakosclient):

    """Send feedback to server"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, msg, more_info=None):

        self.client.send_feedback(msg, more_info or '')

    def main(self, message, more_info=None):

        super(self.__class__, self)._run()

        self._run(message, more_info)

@command(admin_commands)

class admin_endpoints(_init_synnefo_astakosclient, _optional_json):

    """Get endpoints service endpoints"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print(self.client.get_endpoints(), self.print_dict)

    def main(self):

        super(self.__class__, self)._run()

        self._run()

#  command project

_project_specs = """

    {

        "name": name,

        "owner": uuid,

        "homepage": homepage,         # optional

        "description": description,   # optional

        "comments": comments,         # optional

        "start_date": date,           # optional

        "end_date": date,

        "join_policy": "auto" | "moderated" | "closed",  # default: "moderated"

        "leave_policy": "auto" | "moderated" | "closed", # default: "auto"

        "resources": {

            "cyclades.vm": {

                "project_capacity": int or null,

                 "member_capacity": int

            }

        }

  }

  """

def apply_notification(foo):

    def wrap(self, *args, **kwargs):

        r = foo(self, *args, **kwargs)

        self.writeln('Application is submitted successfully')

        return r

    return wrap

@command(project_commands)

class project_list(_init_synnefo_astakosclient, _optional_json):

    """List all projects"""

    arguments = dict(

        name=ValueArgument('Filter by name', ('--with-name', )),

        state=ValueArgument('Filter by state', ('--with-state', )),

        owner=ValueArgument('Filter by owner', ('--with-owner', ))

    )

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print(self.client.get_projects(

            self['name'], self['state'], self['owner']))

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(project_commands)

class project_info(_init_synnefo_astakosclient, _optional_json):

    """Get details for a project"""

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, project_id):

        self._print(

            self.client.get_project(project_id), self.print_dict)

    def main(self, project_id):

        super(self.__class__, self)._run()

        self._run(project_id)

@command(project_commands)

class project_create(_init_synnefo_astakosclient, _optional_json):

    """Apply for a new project (input a json-dict)

    Project details must be provided as a json-formated dict from the standard

    input, or through a file

    """

    __doc__ += _project_specs

    arguments = dict(

        specs_path=ValueArgument(

            'Specification file path (content must be in json)', '--spec-file')

    )

    @errors.generic.all

    @errors.user.astakosclient

    @apply_notification

    def _run(self):

        input_stream = open(abspath(self['specs_path'])) if (

            self['specs_path']) else self._in

        specs = load(input_stream)

        self._print(

            self.client.create_project(specs), self.print_dict)

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(project_commands)

class project_modify(_init_synnefo_astakosclient, _optional_json):

    """Modify a project (input a json-dict)

    Project details must be provided as a json-formated dict from the standard

    input, or through a file

    """

    __doc__ += _project_specs

    arguments = dict(

        specs_path=ValueArgument(

            'Specification file path (content must be in json)', '--spec-file')

    )

    @errors.generic.all

    @errors.user.astakosclient

    @apply_notification

    def _run(self, project_id):

        input_stream = open(abspath(self['specs_path'])) if (

            self['specs_path']) else self._in

        specs = load(input_stream)

        self._print(

            self.client.modify_project(project_id, specs),

            self.print_dict)

    def main(self, project_id):

        super(self.__class__, self)._run()

        self._run(project_id)

class _project_action(_init_synnefo_astakosclient):

    action = ''

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self, project_id, quote_a_reason):

        self.client.project_action(project_id, self.action, quote_a_reason)

    def main(self, project_id, quote_a_reason=''):

        super(_project_action, self)._run()

        self._run(project_id, quote_a_reason)

@command(project_commands)

class project_suspend(_project_action):

    """Suspend a project (special privileges needed)"""

    action = 'suspend'

@command(project_commands)

class project_unsuspend(_project_action):

    """Resume a suspended project (special privileges needed)"""

    action = 'unsuspend'

@command(project_commands)

class project_terminate(_project_action):

    """Terminate a project (special privileges needed)"""

    action = 'terminate'

@command(project_commands)

class project_reinstate(_project_action):

    """Reinstate a terminated project (special privileges needed)"""

    action = 'reinstate'

@command(project_commands)

class project_application(_init_synnefo_astakosclient):

    """Application management commands"""

@command(project_commands)

class project_application_list(_init_synnefo_astakosclient, _optional_json):

    """List all applications (old and new)"""

    arguments = dict(

        project=IntArgument('Filter by project id', '--with-project-id')

    )

    @errors.generic.all

    @errors.user.astakosclient

    def _run(self):

        self._print(self.client.get_applications(self['project']))

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(project_commands)

class project_application_info(_init_synnefo_astakosclient, _optional_json):