~drizzle-trunk/drizzle/development

/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-

 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:

 *

 *  Copyright (C) 2010 Sun Microsystems

 *

 *  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 St, Fifth Floor, Boston, MA  02110-1301  USA

 */

#include "config.h"

#include <plugin/function_engine/function.h>

#include <plugin/function_engine/cursor.h>

#include <string>

using namespace std;

using namespace drizzled;

static SchemaIdentifier INFORMATION_SCHEMA_IDENTIFIER("INFORMATION_SCHEMA");

static SchemaIdentifier DATA_DICTIONARY_IDENTIFIER("DATA_DICTIONARY");

Function::Function(const std::string &name_arg) :

  drizzled::plugin::StorageEngine(name_arg,

                                  HTON_ALTER_NOT_SUPPORTED |

                                  HTON_HAS_SCHEMA_DICTIONARY |

                                  HTON_SKIP_STORE_LOCK |

                                  HTON_TEMPORARY_NOT_SUPPORTED)

{

}

Cursor *Function::create(TableShare &table, memory::Root *mem_root)

{

  return new (mem_root) FunctionCursor(*this, table);

}

int Function::doGetTableDefinition(Session &,

                                   const TableIdentifier &identifier,

                                   message::Table &table_proto)

{

  string tab_name(identifier.getPath());

  transform(tab_name.begin(), tab_name.end(),

            tab_name.begin(), ::tolower);

  drizzled::plugin::TableFunction *function= getFunction(tab_name);

  if (not function)

  {

    return ENOENT;

  }

  function->define(table_proto);

  return EEXIST;

}

void Function::doGetTableNames(drizzled::CachedDirectory&, 

                               drizzled::SchemaIdentifier &schema_identifier,

                               set<string> &set_of_names)

{

  drizzled::plugin::TableFunction::getNames(schema_identifier.getSchemaName(), set_of_names);

}

void Function::doGetSchemaIdentifiers(SchemaIdentifierList& schemas)

{

  schemas.push_back(INFORMATION_SCHEMA_IDENTIFIER);

  schemas.push_back(DATA_DICTIONARY_IDENTIFIER);

}

bool Function::doGetSchemaDefinition(SchemaIdentifier &schema_identifier, message::Schema &schema_message)

{

  if (schema_identifier == INFORMATION_SCHEMA_IDENTIFIER)

  {

    schema_message.set_name("information_schema");

    schema_message.set_collation("utf8_general_ci");

  }

  else if (schema_identifier == DATA_DICTIONARY_IDENTIFIER)

  {

    schema_message.set_name("data_dictionary");

    schema_message.set_collation("utf8_general_ci");

  }

  else

  {

    return false;

  }

  return true;

}

bool Function::doCanCreateTable(const drizzled::TableIdentifier &table_identifier)

{

  if (static_cast<const SchemaIdentifier&>(table_identifier) == INFORMATION_SCHEMA_IDENTIFIER)

  {

    return false;

  }

  else if (static_cast<const SchemaIdentifier&>(table_identifier) == DATA_DICTIONARY_IDENTIFIER)

  {

    return false;

  }

  return true;

}

bool Function::doDoesTableExist(Session&, const TableIdentifier &identifier)

{

  drizzled::plugin::TableFunction *function= getFunction(identifier.getPath());

  if (function)

    return true;

  return false;

}

void Function::doGetTableIdentifiers(drizzled::CachedDirectory&,

                                     drizzled::SchemaIdentifier &schema_identifier,

                                     drizzled::TableIdentifiers &set_of_identifiers)

{

  set<string> set_of_names;

  drizzled::plugin::TableFunction::getNames(schema_identifier.getSchemaName(), set_of_names);

  for (set<string>::iterator iter= set_of_names.begin(); iter != set_of_names.end(); iter++)

  {

    set_of_identifiers.push_back(TableIdentifier(schema_identifier, *iter));

  }

}

static int init(drizzled::module::Context &context)

{

  context.add(new Function("FunctionEngine"));

  return 0;

}

DRIZZLE_DECLARE_PLUGIN

{

  DRIZZLE_VERSION_ID,

  "FunctionEngine",

  "1.0",

  "Brian Aker",

  "Function Engine provides the infrastructure for Table Functions,etc.",

  PLUGIN_LICENSE_GPL,

  init,     /* Plugin Init */

  NULL,               /* system variables */

  NULL                /* config options   */

}

DRIZZLE_DECLARE_PLUGIN_END;