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.

from io import StringIO

from pydoc import pager

from kamaki.cli import command

from kamaki.cli.command_tree import CommandTree

from kamaki.cli.errors import (

    CLIBaseUrlError, CLIInvalidArgument, raiseCLIError)

from kamaki.clients.cyclades import CycladesNetworkClient, ClientError

from kamaki.cli.argument import (

    FlagArgument, ValueArgument, RepeatableArgument, IntArgument,

    StatusArgument)

from kamaki.cli.commands import _command_init, errors, addLogSettings

from kamaki.cli.commands import (

    _optional_output_cmd, _optional_json, _name_filter, _id_filter)

from kamaki.cli.commands.cyclades import _service_wait

network_cmds = CommandTree('network', 'Network API network commands')

port_cmds = CommandTree('port', 'Network API port commands')

subnet_cmds = CommandTree('subnet', 'Network API subnet commands')

ip_cmds = CommandTree('ip', 'Network API floatingip commands')

_commands = [network_cmds, port_cmds, subnet_cmds, ip_cmds]

about_authentication = '\nUser Authentication:\

    \n  to check authentication: [kamaki] ]user authenticate\

    \n  to set authentication token: \

    [kamaki] config set cloud.<CLOUD>.token <TOKEN>'

port_states = ('BUILD', 'ACTIVE', 'DOWN', 'ERROR')

class _port_wait(_service_wait):

    def _wait(self, port_id, current_status, timeout=60):

        super(_port_wait, self)._wait(

            'Port', port_id, self.client.wait_port, current_status,

            timeout=timeout)

class _init_network(_command_init):

    @errors.generic.all

    @addLogSettings

    def _run(self, service='network'):

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

            base_url = self._custom_url(service) or self._custom_url(

                'network')

            if base_url:

                token = self._custom_token(service) or self._custom_token(

                    'network') or self.config.get_cloud('token')

                self.client = CycladesNetworkClient(

                  base_url=base_url, token=token)

                return

        else:

            self.cloud = 'default'

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

            network_endpoints = self.auth_base.get_service_endpoints(

                self._custom_type('network') or 'network',

                self._custom_version('network') or '')

            base_url = network_endpoints['publicURL']

            token = self.auth_base.token

            self.client = CycladesNetworkClient(base_url=base_url, token=token)

        else:

            raise CLIBaseUrlError(service='network')

    def _filter_by_user_id(self, nets):

        return [net for net in nets if net['user_id'] == self['user_id']] if (

            self['user_id']) else nets

    def main(self):

        self._run()

@command(network_cmds)

class network_list(_init_network, _optional_json, _name_filter, _id_filter):

    """List networks

    Use filtering arguments (e.g., --name-like) to manage long server lists

    """

    arguments = dict(

        detail=FlagArgument('show detailed output', ('-l', '--details')),

        more=FlagArgument(

            'output results in pages (-n to set items per page, default 10)',

            '--more'),

        user_id=ValueArgument(

            'show only networks belonging to user with this id', '--user-id')

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self):

        nets = self.client.list_networks(detail=True)

        nets = self._filter_by_user_id(nets)

        nets = self._filter_by_name(nets)

        nets = self._filter_by_id(nets)

        if not self['detail']:

            nets = [dict(

                _0_id=n['id'],

                _1_name=n['name'],

                _2_public='( %s )' % ('public' if (

                    n.get('public', None)) else 'private')) for n in nets]

            kwargs = dict(title=('_0_id', '_1_name', '_2_public'))

        else:

            kwargs = dict()

        if self['more']:

            kwargs['out'] = StringIO()

            kwargs['title'] = ()

        self._print(nets, **kwargs)

        if self['more']:

            pager(kwargs['out'].getvalue())

    def main(self):

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

        self._run()

@command(network_cmds)

class network_info(_init_network, _optional_json):

    """Get details about a network"""

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.network_id

    def _run(self, network_id):

        net = self.client.get_network_details(network_id)

        self._print(net, self.print_dict)

    def main(self, network_id):

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

        self._run(network_id=network_id)

class NetworkTypeArgument(ValueArgument):

    types = ('MAC_FILTERED', 'CUSTOM', 'IP_LESS_ROUTED', 'PHYSICAL_VLAN')

    @property

    def value(self):

        return getattr(self, '_value', self.types[0])

    @value.setter

    def value(self, new_value):

        if new_value and new_value.upper() in self.types:

            self._value = new_value.upper()

        elif new_value:

            raise CLIInvalidArgument(

                'Invalid network type %s' % new_value, details=[

                    'Valid types: %s' % ', '.join(self.types), ])

@command(network_cmds)

class network_create(_init_network, _optional_json):

    """Create a new network (default type: MAC_FILTERED)"""

    arguments = dict(

        name=ValueArgument('Network name', '--name'),

        shared=FlagArgument(

            'Make network shared (special privileges required)', '--shared'),

        project=ValueArgument('Assign the network to project', '--project'),

        network_type=NetworkTypeArgument(

            'Valid network types: %s' % (', '.join(NetworkTypeArgument.types)),

            '--type')

    )

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.network_type

    def _run(self, network_type):

        net = self.client.create_network(

            network_type, name=self['name'],

            shared=self['shared'],

            project=self['project'])

        self._print(net, self.print_dict)

    def main(self):

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

        self._run(network_type=self['network_type'])

@command(network_cmds)

class network_reassign(_init_network, _optional_json):

    """Assign a network to a different project

    """

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.network_id

    def _run(self, network_id, project):

        self.client.reassign_network(network_id, project)

    def main(self, network_id, project):

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

        self._run(network_id=network_id, project=project)

@command(network_cmds)

class network_delete(_init_network, _optional_output_cmd):

    """Delete a network"""

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.network_id

    def _run(self, network_id):

        r = self.client.delete_network(network_id)

        self._optional_output(r)

    def main(self, network_id):

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

        self._run(network_id=network_id)

@command(network_cmds)

class network_modify(_init_network, _optional_json):

    """Modify network attributes"""

    arguments = dict(new_name=ValueArgument('Rename the network', '--name'))

    required = ['new_name', ]

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.network_id

    def _run(self, network_id):

        r = self.client.update_network(network_id, name=self['new_name'])

        self._print(r, self.print_dict)

    def main(self, network_id):

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

        self._run(network_id=network_id)

@command(subnet_cmds)

class subnet_list(_init_network, _optional_json, _name_filter, _id_filter):

    """List subnets

    Use filtering arguments (e.g., --name-like) to manage long server lists

    """

    arguments = dict(

        detail=FlagArgument('show detailed output', ('-l', '--details')),

        more=FlagArgument(

            'output results in pages (-n to set items per page, default 10)',

            '--more')

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self):

        nets = self.client.list_subnets()

        nets = self._filter_by_name(nets)

        nets = self._filter_by_id(nets)

        if not self['detail']:

            nets = [dict(

                _0_id=n['id'],

                _1_name=n['name'],

                _2_net='( of network %s )' % n['network_id']) for n in nets]

            kwargs = dict(title=('_0_id', '_1_name', '_2_net'))

        else:

            kwargs = dict()

        if self['more']:

            kwargs['out'] = StringIO()

            kwargs['title'] = ()

        self._print(nets, **kwargs)

        if self['more']:

            pager(kwargs['out'].getvalue())

    def main(self):

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

        self._run()

@command(subnet_cmds)

class subnet_info(_init_network, _optional_json):

    """Get details about a subnet"""

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, subnet_id):

        net = self.client.get_subnet_details(subnet_id)

        self._print(net, self.print_dict)

    def main(self, subnet_id):

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

        self._run(subnet_id=subnet_id)

class AllocationPoolArgument(RepeatableArgument):

    @property

    def value(self):

        return super(AllocationPoolArgument, self).value or []

    @value.setter

    def value(self, new_pools):

        if not new_pools:

            return

        new_list = []

        for pool in new_pools:

            start, comma, end = pool.partition(',')

            if not (start and comma and end):

                raise CLIInvalidArgument(

                    'Invalid allocation pool argument %s' % pool, details=[

                    'Allocation values must be of the form:',

                    '  <start address>,<end address>'])

            new_list.append(dict(start=start, end=end))

        self._value = new_list

@command(subnet_cmds)

class subnet_create(_init_network, _optional_json):

    """Create a new subnet"""

    arguments = dict(

        name=ValueArgument('Subnet name', '--name'),

        allocation_pools=AllocationPoolArgument(

            'start_address,end_address of allocation pool (can be repeated)'

            ' e.g., --alloc-pool=123.45.67.1,123.45.67.8',

            '--alloc-pool'),

        gateway=ValueArgument('Gateway IP', '--gateway'),

        subnet_id=ValueArgument('The id for the subnet', '--id'),

        ipv6=FlagArgument('If set, IP version is set to 6, else 4', '--ipv6'),

        enable_dhcp=FlagArgument('Enable dhcp (default: off)', '--with-dhcp'),

        network_id=ValueArgument('Set the network ID', '--network-id'),

        cidr=ValueArgument('Set the CIDR', '--cidr')

    )

    required = ('network_id', 'cidr')

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.network_id

    def _run(self, network_id, cidr):

        net = self.client.create_subnet(

            network_id, cidr,

            self['name'], self['allocation_pools'], self['gateway'],

            self['subnet_id'], self['ipv6'], self['enable_dhcp'])

        self._print(net, self.print_dict)

    def main(self):

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

        self._run(network_id=self['network_id'], cidr=self['cidr'])

# @command(subnet_cmds)

# class subnet_delete(_init_network, _optional_output_cmd):

#     """Delete a subnet"""

#     @errors.generic.all

#     @errors.cyclades.connection

#     def _run(self, subnet_id):

#         r = self.client.delete_subnet(subnet_id)

#         self._optional_output(r)

#     def main(self, subnet_id):

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

#         self._run(subnet_id=subnet_id)

@command(subnet_cmds)

class subnet_modify(_init_network, _optional_json):

    """Modify the attributes of a subnet"""

    arguments = dict(

        new_name=ValueArgument('New name of the subnet', '--name')

    )

    required = ['new_name']

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, subnet_id):

        r = self.client.update_subnet(subnet_id, name=self['new_name'])

        self._print(r, self.print_dict)

    def main(self, subnet_id):

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

        self._run(subnet_id=subnet_id)

@command(port_cmds)

class port_list(_init_network, _optional_json, _name_filter, _id_filter):

    """List all ports"""

    arguments = dict(

        detail=FlagArgument('show detailed output', ('-l', '--details')),

        more=FlagArgument(

            'output results in pages (-n to set items per page, default 10)',

            '--more'),

        user_id=ValueArgument(

            'show only networks belonging to user with this id', '--user-id')

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self):

        detail = bool(self['detail'] or self['user_id'])

        ports = self.client.list_ports(detail=detail)

        ports = self._filter_by_user_id(ports)

        ports = self._filter_by_name(ports)

        ports = self._filter_by_id(ports)

        if detail and not self['detail']:

            ports = [dict(

                id=p['id'], name=p['name'], links=p['links']) for p in ports]

        kwargs = dict()

        if self['more']:

            kwargs['out'] = StringIO()

            kwargs['title'] = ()

        self._print(ports, **kwargs)

        if self['more']:

            pager(kwargs['out'].getvalue())

    def main(self):

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

        self._run()

@command(port_cmds)

class port_info(_init_network, _optional_json):

    """Get details about a port"""

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, port_id):

        port = self.client.get_port_details(port_id)

        self._print(port, self.print_dict)

    def main(self, port_id):

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

        self._run(port_id=port_id)

@command(port_cmds)

class port_delete(_init_network, _optional_output_cmd, _port_wait):

    """Delete a port (== disconnect server from network)"""

    arguments = dict(

        wait=FlagArgument('Wait port to be established', ('-w', '--wait'))

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, port_id):

        if self['wait']:

            status = self.client.get_port_details(port_id)['status']

        r = self.client.delete_port(port_id)

        if self['wait']:

            try:

                self._wait(port_id, status)

            except ClientError as ce:

                if ce.status not in (404, ):

                    raise

                self.error('Port %s is deleted' % port_id)

        self._optional_output(r)

    def main(self, port_id):

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

        self._run(port_id=port_id)

@command(port_cmds)

class port_modify(_init_network, _optional_json):

    """Modify the attributes of a port"""

    arguments = dict(new_name=ValueArgument('New name of the port', '--name'))

    required = ['new_name', ]

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, port_id):

        r = self.client.get_port_details(port_id)

        r = self.client.update_port(

            port_id, r['network_id'], name=self['new_name'])

        self._print(r, self.print_dict)

    def main(self, port_id):

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

        self._run(port_id=port_id)

class _port_create(_init_network, _optional_json, _port_wait):

    def connect(self, network_id, device_id):

        fixed_ips = [dict(ip_address=self['ip_address'])] if (

            self['ip_address']) else None

        if fixed_ips and self['subnet_id']:

            fixed_ips[0]['subnet_id'] = self['subnet_id']

        r = self.client.create_port(

            network_id, device_id,

            name=self['name'],

            security_groups=self['security_group_id'],

            fixed_ips=fixed_ips)

        if self['wait']:

            self._wait(r['id'], r['status'])

            r = self.client.get_port_details(r['id'])

        self._print([r])

@command(port_cmds)

class port_create(_port_create):

    """Create a new port (== connect server to network)"""

    arguments = dict(

        name=ValueArgument('A human readable name', '--name'),

        security_group_id=RepeatableArgument(

            'Add a security group id (can be repeated)',

            ('-g', '--security-group')),

        subnet_id=ValueArgument(

            'Subnet id for fixed ips (used with --ip-address)',

            '--subnet-id'),

        ip_address=ValueArgument(

            'IP address for subnet id', '--ip-address'),

        network_id=ValueArgument('Set the network ID', '--network-id'),

        device_id=ValueArgument(

            'The device is either a virtual server or a virtual router',

            '--device-id'),

        wait=FlagArgument('Wait port to be established', ('-w', '--wait')),

    )

    required = ('network_id', 'device_id')

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.network_id

    @errors.cyclades.server_id

    def _run(self, network_id, server_id):

        self.connect(network_id, server_id)

    def main(self):

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

        self._run(network_id=self['network_id'], server_id=self['device_id'])

@command(port_cmds)

class port_wait(_init_network, _port_wait):

    """Wait for port to finish (default: BUILD)"""

    arguments = dict(

        port_status=StatusArgument(

            'Wait while in this status (%s, default: %s)' % (

                ', '.join(port_states), port_states[0]),

            '--status',

            valid_states=port_states),

        timeout=IntArgument(

            'Wait limit in seconds (default: 60)', '--timeout', default=60)

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, port_id, port_status):

        port = self.client.get_port_details(port_id)

        if port['status'].lower() == port_status.lower():

            self._wait(port_id, port_status, timeout=self['timeout'])

        else:

            self.error(

                'Port %s: Cannot wait for status %s, '

                'status is already %s' % (

                    port_id, port_status, port['status']))

    def main(self, port_id):

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

        port_status = self['port_status'] or port_states[0]

        self._run(port_id=port_id, port_status=port_status)

@command(ip_cmds)

class ip_list(_init_network, _optional_json):

    """List reserved floating IPs"""

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self):

        self._print(self.client.list_floatingips())

    def main(self):

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

        self._run()

@command(ip_cmds)

class ip_info(_init_network, _optional_json):

    """Get details on a floating IP"""

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, ip_id):

        self._print(

            self.client.get_floatingip_details(ip_id), self.print_dict)

    def main(self, ip_id):

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

        self._run(ip_id=ip_id)

@command(ip_cmds)

class ip_create(_init_network, _optional_json):

    """Reserve an IP on a network"""

    arguments = dict(

        network_id=ValueArgument(

            'The network to preserve the IP on', '--network-id'),

        ip_address=ValueArgument('Allocate an IP address', '--address'),

        project_id=ValueArgument('Assign the IP to project', '--project-id'),

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self):

        self._print(

            self.client.create_floatingip(

                self['network_id'],

                floating_ip_address=self['ip_address'],

                project_id=self['project_id']),

            self.print_dict)

    def main(self):

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

        self._run()

@command(ip_cmds)

class ip_reassign(_init_network, _optional_output_cmd):

    """Assign a floating IP to a different project

    """

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, ip, project):

        self._optional_output(self.client.reassign_floating_ip(ip, project))

    def main(self, IP, project):

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

        self._run(ip=IP, project=project)

@command(ip_cmds)

class ip_delete(_init_network, _optional_output_cmd):

    """Unreserve an IP (also delete the port, if attached)"""

    def _run(self, ip_id):

        self._optional_output(self.client.delete_floatingip(ip_id))

    def main(self, ip_id):

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

        self._run(ip_id=ip_id)

@command(ip_cmds)

class ip_attach(_port_create):

    """Attach an IP on a virtual server"""

    arguments = dict(

        name=ValueArgument('A human readable name for the port', '--name'),

        security_group_id=RepeatableArgument(

            'Add a security group id (can be repeated)',

            ('-g', '--security-group')),

        subnet_id=ValueArgument('Subnet id', '--subnet-id'),

        wait=FlagArgument('Wait IP to be attached', ('-w', '--wait')),

        server_id=ValueArgument(

            'Server to attach to this IP', '--server-id')

    )

    required = ('server_id', )

    @errors.generic.all

    @errors.cyclades.connection

    @errors.cyclades.server_id

    def _run(self, ip_or_ip_id, server_id):

        netid = None

        for ip in self.client.list_floatingips():

            if ip_or_ip_id in (ip['floating_ip_address'], ip['id']):

                netid = ip['floating_network_id']

                iparg = ValueArgument(parsed_name='--ip')

                iparg.value = ip['floating_ip_address']

                self.arguments['ip_address'] = iparg

                break

        if netid:

            self.error('Creating a port to attach IP %s to server %s' % (

                ip_or_ip_id, server_id))

            self.connect(netid, server_id)

        else:

            raiseCLIError(

                '%s does not match any reserved IPs or IP ids' % ip_or_ip_id,

                details=[

                    'To reserve an IP:', '  [kamaki] ip create',

                    'To see all reserved IPs:', '  [kamaki] ip list'])

    def main(self, ip_or_ip_id):

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

        self._run(ip_or_ip_id=ip_or_ip_id, server_id=self['server_id'])

@command(ip_cmds)

class ip_detach(_init_network, _port_wait, _optional_json):

    """Detach an IP from a virtual server"""

    arguments = dict(

        wait=FlagArgument('Wait network to disconnect', ('-w', '--wait')),

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, ip_or_ip_id):

        for ip in self.client.list_floatingips():

            if ip_or_ip_id in (ip['floating_ip_address'], ip['id']):

                if not ip['port_id']:

                    raiseCLIError('IP %s is not attached' % ip_or_ip_id)

                self.error('Deleting port %s:' % ip['port_id'])

                self.client.delete_port(ip['port_id'])

                if self['wait']:

                    port_status = self.client.get_port_details(ip['port_id'])[

                        'status']