server/src/bind_hashtable.c

/* vi: set ts=2:
+-------------------+
|                   |  Enno Rehling <enno@eressea.de>
| Eressea PBEM host |  Christian Schlittchen <corwin@amber.kn-bremen.de>
| (c) 1998 - 2008   |  Katja Zedel <katze@felidae.kn-bremen.de>
|                   |  Henning Peters <faroul@beyond.kn-bremen.de>
+-------------------+

This program may not be used, modified or distributed
without prior permission by the authors of Eressea.
*/

#include <platform.h>
#include "bind_hashtable.h"

#include <kernel/building.h>
#include <kernel/unit.h>
#include <kernel/ship.h>
#include <kernel/region.h>

#include <attributes/object.h>

#include <util/variant.h>
#include <util/attrib.h>

#include <tolua.h>
#include <lua.h>

#include <string.h>
#include <assert.h>

static int tolua_hashtable_get(lua_State * L)
{
  hashtable self = (hashtable) tolua_tousertype(L, 1, 0);
  const char *name = tolua_tostring(L, 2, 0);
  attrib *a = a_find(*self, &at_object);

  for (; a && a->type == &at_object; a = a->next) {
    const char *obj_name = object_name(a);
    if (obj_name && name && strcmp(obj_name, name) == 0) {
      variant val;
      object_type type;

      object_get(a, &type, &val);
      switch (type) {
        case TNONE:
          lua_pushnil(L);
          break;
        case TINTEGER:
          lua_pushnumber(L, (lua_Number) val.i);
          break;
        case TREAL:
          lua_pushnumber(L, (lua_Number) val.f);
          break;
        case TREGION:
          tolua_pushusertype(L, val.v, TOLUA_CAST "region");
          break;
        case TBUILDING:
          tolua_pushusertype(L, val.v, TOLUA_CAST "building");
          break;
        case TUNIT:
          tolua_pushusertype(L, val.v, TOLUA_CAST "unit");
          break;
        case TSHIP:
          tolua_pushusertype(L, val.v, TOLUA_CAST "ship");
          break;
        case TSTRING:
          tolua_pushstring(L, (const char *)val.v);
          break;
        default:
          assert(!"not implemented");
      }
      return 1;
    }
  }
  lua_pushnil(L);
  return 1;
}

static int tolua_hashtable_set_number(lua_State * L)
{
  hashtable self = (hashtable) tolua_tousertype(L, 1, 0);
  const char *name = tolua_tostring(L, 2, 0);
  lua_Number value = tolua_tonumber(L, 3, 0);
  attrib *a = a_find(*self, &at_object);
  variant val;

  val.f = (float)value;

  for (; a && a->type == &at_object; a = a->next) {
    if (strcmp(object_name(a), name) == 0) {
      object_set(a, TREAL, val);
      return 0;
    }
  }

  a = a_add(self, object_create(name, TREAL, val));
  return 0;
}

static int tolua_hashtable_set_string(lua_State * L)
{
  hashtable self = (hashtable) tolua_tousertype(L, 1, 0);
  const char *name = tolua_tostring(L, 2, 0);
  const char *value = tolua_tostring(L, 3, 0);
  attrib *a = a_find(*self, &at_object);
  variant val;

  val.v = _strdup(value);

  for (; a && a->type == &at_object; a = a->next) {
    if (strcmp(object_name(a), name) == 0) {
      object_set(a, TSTRING, val);
      return 0;
    }
  }

  a = a_add(self, object_create(name, TSTRING, val));
  return 0;
}

static int tolua_hashtable_set_usertype(lua_State * L, int type)
{
  hashtable self = (hashtable) tolua_tousertype(L, 1, 0);
  const char *name = tolua_tostring(L, 2, 0);
  unit *value = tolua_tousertype(L, 3, 0);
  attrib *a = a_find(*self, &at_object);
  variant val;

  val.v = value;

  for (; a && a->type == &at_object; a = a->next) {
    if (strcmp(object_name(a), name) == 0) {
      object_set(a, type, val);
      return 0;
    }
  }

  a = a_add(self, object_create(name, type, val));
  return 0;
}

static int tolua_hashtable_set(lua_State * L)
{
  tolua_Error tolua_err;
  if (tolua_isnumber(L, 3, 0, &tolua_err)) {
    return tolua_hashtable_set_number(L);
  } else if (tolua_isusertype(L, 3, TOLUA_CAST "unit", 0, &tolua_err)) {
    return tolua_hashtable_set_usertype(L, TUNIT);
  } else if (tolua_isusertype(L, 3, TOLUA_CAST "faction", 0, &tolua_err)) {
    return tolua_hashtable_set_usertype(L, TFACTION);
  } else if (tolua_isusertype(L, 3, TOLUA_CAST "ship", 0, &tolua_err)) {
    return tolua_hashtable_set_usertype(L, TSHIP);
  } else if (tolua_isusertype(L, 3, TOLUA_CAST "building", 0, &tolua_err)) {
    return tolua_hashtable_set_usertype(L, TBUILDING);
  } else if (tolua_isusertype(L, 3, TOLUA_CAST "region", 0, &tolua_err)) {
    return tolua_hashtable_set_usertype(L, TREGION);
  }
  return tolua_hashtable_set_string(L);
}

void tolua_hashtable_open(lua_State * L)
{
  /* register user types */
  tolua_usertype(L, TOLUA_CAST "hashtable");

  tolua_module(L, NULL, 0);
  tolua_beginmodule(L, NULL);
  {
    tolua_cclass(L, TOLUA_CAST "hashtable", TOLUA_CAST "hashtable",
      TOLUA_CAST "", NULL);
    tolua_beginmodule(L, TOLUA_CAST "hashtable");
    {
      tolua_function(L, TOLUA_CAST "get", tolua_hashtable_get);
      tolua_function(L, TOLUA_CAST "set", tolua_hashtable_set);
    }
    tolua_endmodule(L);
  }
  tolua_endmodule(L);
}