Synnefo » snf-ganeti

#

#

# Copyright (C) 2006, 2007, 2008 Google Inc.

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# 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.

"""Remote API connection map.

"""

import cgi

import re

from ganeti import constants

from ganeti import http

from ganeti.rapi import baserlib

from ganeti.rapi import rlib2

_NAME_PATTERN = r"[\w\._-]+"

# the connection map is created at the end of this file

CONNECTOR = {}

class Mapper:

  """Map resource to method.

  """

  def __init__(self, connector=CONNECTOR):

    """Resource mapper constructor.

    @param connector: a dictionary, mapping method name with URL path regexp

    """

    self._connector = connector

  def getController(self, uri):

    """Find method for a given URI.

    @param uri: string with URI

    @return: None if no method is found or a tuple containing

        the following fields:

            - method: name of method mapped to URI

            - items: a list of variable intems in the path

            - args: a dictionary with additional parameters from URL

    """

    if '?' in uri:

      (path, query) = uri.split('?', 1)

      args = cgi.parse_qs(query)

    else:

      path = uri

      query = None

      args = {}

    result = None

    for key, handler in self._connector.iteritems():

      # Regex objects

      if hasattr(key, "match"):

        m = key.match(path)

        if m:

          result = (handler, list(m.groups()), args)

          break

      # String objects

      elif key == path:

        result = (handler, [], args)

        break

    if result:

      return result

    else:

      raise http.HttpNotFound()

class R_root(baserlib.R_Generic):

  """/ resource.

  """

  def GET(self):

    """Show the list of mapped resources.

    @return: a dictionary with 'name' and 'uri' keys for each of them.

    """

    root_pattern = re.compile('^R_([a-zA-Z0-9]+)$')

    rootlist = []

    for handler in CONNECTOR.values():

      m = root_pattern.match(handler.__name__)

      if m:

        name = m.group(1)

        if name != 'root':

          rootlist.append(name)

    return baserlib.BuildUriList(rootlist, "/%s")

def _getResources(id):

  """Return a list of resources underneath given id.

  This is to generalize querying of version resources lists.

  @return: a list of resources names.

  """

  r_pattern = re.compile('^R_%s_([a-zA-Z0-9]+)$' % id)

  rlist = []

  for handler in CONNECTOR.values():

    m = r_pattern.match(handler.__name__)

    if m:

      name = m.group(1)

      rlist.append(name)

  return rlist

class R_2(baserlib.R_Generic):

  """ /2 resource, the root of the version 2 API.

  """

  def GET(self):

    """Show the list of mapped resources.

    @return: a dictionary with 'name' and 'uri' keys for each of them.

    """

    return baserlib.BuildUriList(_getResources("2"), "/2/%s")

def GetHandlers(node_name_pattern, instance_name_pattern, job_id_pattern):

  """Returns all supported resources and their handlers.

  """

  return {

    "/": R_root,

    "/version": rlib2.R_version,

    "/2": R_2,

    "/2/nodes": rlib2.R_2_nodes,

    re.compile(r'^/2/nodes/(%s)$' % node_name_pattern):

      rlib2.R_2_nodes_name,

    re.compile(r'^/2/nodes/(%s)/tags$' % node_name_pattern):

      rlib2.R_2_nodes_name_tags,

    re.compile(r'^/2/nodes/(%s)/role$' % node_name_pattern):

      rlib2.R_2_nodes_name_role,

    re.compile(r'^/2/nodes/(%s)/evacuate$' % node_name_pattern):

      rlib2.R_2_nodes_name_evacuate,

    re.compile(r'^/2/nodes/(%s)/migrate$' % node_name_pattern):

      rlib2.R_2_nodes_name_migrate,

    re.compile(r'^/2/nodes/(%s)/storage$' % node_name_pattern):

      rlib2.R_2_nodes_name_storage,

    re.compile(r'^/2/nodes/(%s)/storage/modify$' % node_name_pattern):

      rlib2.R_2_nodes_name_storage_modify,

    re.compile(r'^/2/nodes/(%s)/storage/repair$' % node_name_pattern):

      rlib2.R_2_nodes_name_storage_repair,

    "/2/instances": rlib2.R_2_instances,

    re.compile(r'^/2/instances/(%s)$' % instance_name_pattern):

      rlib2.R_2_instances_name,

    re.compile(r'^/2/instances/(%s)/info$' % instance_name_pattern):

      rlib2.R_2_instances_name_info,

    re.compile(r'^/2/instances/(%s)/tags$' % instance_name_pattern):

      rlib2.R_2_instances_name_tags,

    re.compile(r'^/2/instances/(%s)/reboot$' % instance_name_pattern):

      rlib2.R_2_instances_name_reboot,

    re.compile(r'^/2/instances/(%s)/reinstall$' % instance_name_pattern):

      rlib2.R_2_instances_name_reinstall,

    re.compile(r'^/2/instances/(%s)/replace-disks$' % instance_name_pattern):

      rlib2.R_2_instances_name_replace_disks,

    re.compile(r'^/2/instances/(%s)/shutdown$' % instance_name_pattern):

      rlib2.R_2_instances_name_shutdown,

    re.compile(r'^/2/instances/(%s)/startup$' % instance_name_pattern):

      rlib2.R_2_instances_name_startup,

    "/2/jobs": rlib2.R_2_jobs,

    re.compile(r'/2/jobs/(%s)$' % job_id_pattern):

      rlib2.R_2_jobs_id,

    "/2/tags": rlib2.R_2_tags,

    "/2/info": rlib2.R_2_info,

    "/2/os": rlib2.R_2_os,

    }

CONNECTOR.update(GetHandlers(_NAME_PATTERN, _NAME_PATTERN,

                             constants.JOB_ID_TEMPLATE))