nlua_news.c

Go to the documentation of this file.

/*
 * See Licensing and Copyright notice in naev.h
 */
#include <lauxlib.h>
 
#include "nlua_news.h"
 
#include "array.h"
#include "land.h"
#include "news.h"
#include "nlua_time.h"
#include "nluadef.h"
#include "ntime.h"
 
extern news_t *news_list; 
extern int     land_loaded;
 
static news_t    *luaL_validnews( lua_State *L, int ind );
static int        lua_isnews( lua_State *L, int ind );
static LuaNews_t *lua_pushnews( lua_State *L, LuaNews_t news );
 
static int newsL_add( lua_State *L );
static int newsL_rm( lua_State *L );
static int newsL_get( lua_State *L );
static int newsL_eq( lua_State *L );
static int newsL_title( lua_State *L );
static int newsL_desc( lua_State *L );
static int newsL_faction( lua_State *L );
static int newsL_date( lua_State *L );
static int newsL_bind( lua_State *L );
 
static const luaL_Reg news_methods[] = {
   { "add", newsL_add },   { "rm", newsL_rm },
   { "get", newsL_get },   { "title", newsL_title },
   { "desc", newsL_desc }, { "faction", newsL_faction },
   { "date", newsL_date }, { "bind", newsL_bind },
   { "__eq", newsL_eq },   { 0, 0 } }; 

int nlua_loadNews( nlua_env env )
{
   nlua_register( env, NEWS_METATABLE, news_methods, 1 );
   return 0; /* No error */
}

LuaNews_t *lua_tonews( lua_State *L, int ind )
{
   return (LuaNews_t *)lua_touserdata( L, ind );
}

LuaNews_t *luaL_checknews( lua_State *L, int ind )
{
   if ( lua_isnews( L, ind ) )
      return lua_tonews( L, ind );
   luaL_typerror( L, ind, NEWS_METATABLE );
   return NULL;
}

static LuaNews_t *lua_pushnews( lua_State *L, LuaNews_t news )
{
   LuaNews_t *la = (LuaNews_t *)lua_newuserdata( L, sizeof( LuaNews_t ) );
   *la           = news;
   luaL_getmetatable( L, NEWS_METATABLE );
   lua_setmetatable( L, -2 );
   return la;
}

int lua_isnews( lua_State *L, int ind )
{
   int ret;
 
   if ( lua_getmetatable( L, ind ) == 0 )
      return 0;
   lua_getfield( L, LUA_REGISTRYINDEX, NEWS_METATABLE );
 
   ret = 0;
   if ( lua_rawequal( L, -1, -2 ) ) /* does it have the correct mt? */
      ret = 1;
 
   lua_pop( L, 2 ); /* remove both metatables */
   return ret;
}

static news_t *luaL_validnews( lua_State *L, int ind )
{
   const LuaNews_t *ln = luaL_checknews( L, ind );
   news_t          *n  = news_get( *ln );
   if ( n == NULL )
      NLUA_ERROR( L, _( "Article is invalid." ) );
   return n;
}

int newsL_add( lua_State *L )
{
   const char *title, *body, *faction;
   ntime_t     date, date_to_rm;
   int         priority;
 
   title   = NULL;
   body    = NULL;
   faction = NULL;
 
   date       = ntime_get();
   date_to_rm = NEWS_FOREVER;
 
   /* If a table is passed in. ugly hack */
   if ( lua_istable( L, 1 ) ) {
      lua_pushnil( L );
 
      /* traverse table */
      while ( lua_next( L, -2 ) ) {
         /* traverse sub table */
         if ( lua_istable( L, -1 ) ) {
            faction = title = body = NULL;
            priority               = 5;
            date                   = ntime_get();
            date_to_rm             = NEWS_FOREVER;
 
            lua_getfield( L, -1, "faction" );
            if ( !lua_isnil( L, -1 ) )
               faction = luaL_checkstring( L, -1 );
            lua_pop( L, 1 );
 
            lua_getfield( L, -1, "head" );
            if ( !lua_isnil( L, -1 ) )
               title = luaL_checkstring( L, -1 );
            lua_pop( L, 1 );
 
            lua_getfield( L, -1, "body" );
            if ( !lua_isnil( L, -1 ) )
               body = luaL_checkstring( L, -1 );
            lua_pop( L, 1 );
 
            lua_getfield( L, -1, "date" );
            if ( !lua_isnil( L, -1 ) ) {
               if ( lua_isnumber( L, -1 ) )
                  date = lua_tonumber( L, -1 );
               else
                  date = luaL_validtime( L, -1 );
            }
            lua_pop( L, 1 );
 
            lua_getfield( L, -1, "date_to_rm" );
            if ( !lua_isnil( L, -1 ) ) {
               if ( lua_isnumber( L, -1 ) )
                  date_to_rm = lua_tonumber( L, -1 );
               else
                  date_to_rm = luaL_validtime( L, -1 );
            }
            lua_pop( L, 1 );
 
            lua_getfield( L, -1, "priority" );
            if ( !lua_isnil( L, -1 ) )
               priority = luaL_checkinteger( L, -1 );
            lua_pop( L, 1 );
 
            if ( title && body && faction )
               news_add( title, body, faction, NULL, date, date_to_rm,
                         priority );
            else
               NLUA_WARN( L, _( "Bad arguments" ) );
         }
         lua_pop( L, 1 );
      }
      lua_pop( L, 1 );
 
      /* If we're landed, we should regenerate the news buffer. */
      if ( landed ) {
         generate_news( land_spob->presence.faction );
         if ( land_loaded )
            bar_regen();
      }
 
      return 0;
   }
 
   if ( !( lua_isstring( L, 1 ) && lua_isstring( L, 2 ) &&
           lua_isstring( L, 3 ) ) ) {
      NLUA_WARN(
         L, _( "\nBad arguments, use "
               "addArticle(\"Faction\",\"Title\",\"Content\",[date,[date_to_rm]"
               "])" ) );
      return 0;
   }
 
   faction  = luaL_checkstring( L, 1 );
   title    = luaL_checkstring( L, 2 );
   body     = luaL_checkstring( L, 3 );
   priority = luaL_optinteger( L, 6, 5 );
 
   /* Get date and date to remove, or leave at defaults. */
   if ( !lua_isnoneornil( L, 4 ) ) {
      if ( lua_istime( L, 4 ) )
         date_to_rm = luaL_validtime( L, 4 );
      else
         date_to_rm = luaL_checklong( L, 4 );
   }
   if ( !lua_isnoneornil( L, 5 ) ) {
      if ( lua_istime( L, 5 ) )
         date = luaL_validtime( L, 5 );
      else
         date = luaL_checklong( L, 5 );
   }
 
   if ( title && body && faction ) {
      LuaNews_t n_article =
         news_add( title, body, faction, NULL, date, date_to_rm, priority );
      lua_pushnews( L, n_article );
   } else
      return NLUA_ERROR( L, _( "Bad arguments" ) );
 
   /* If we're landed, we should regenerate the news buffer. */
   if ( landed ) {
      generate_news( land_spob->presence.faction );
      if ( land_loaded )
         bar_regen();
   }
 
   return 1;
}

int newsL_rm( lua_State *L )
{
   if ( lua_istable( L, 1 ) ) {
      lua_pushnil( L );
      while ( lua_next( L, -2 ) ) {
         const LuaNews_t *Larticle = luaL_checknews( L, -1 );
         news_rm( *Larticle );
         lua_pop( L, 1 );
      }
   } else {
      const LuaNews_t *Larticle = luaL_checknews( L, 1 );
      news_rm( *Larticle );
   }
 
   /* If we're landed, we should regenerate the news buffer. */
   if ( landed ) {
      generate_news( land_spob->presence.faction );
      if ( land_loaded )
         bar_regen();
   }
 
   return 1;
}

int newsL_get( lua_State *L )
{
   ntime_t     date           = -1;
   const char *characteristic = NULL;
   int         print_all      = 0;
   int         narticle       = 1;
 
   if ( lua_isnoneornil( L, 1 ) ) /* Case no argument */
      print_all = 1;
   else if ( lua_isnumber( L, 1 ) )
      date = (ntime_t)lua_tonumber( L, 1 );
   else if ( lua_isstring( L, 1 ) )
      characteristic = lua_tostring( L, 1 );
   else
      NLUA_INVALID_PARAMETER( L, 1 ); /* Bad Parameter */
 
   /* Now put all the matching articles in a table. */
   lua_newtable( L );
   for ( int i = 0; i < array_size( news_list ); i++ ) {
      const news_t *n = &news_list[i];
 
      if ( ( n->title == NULL ) || ( n->desc == NULL ) ||
           ( n->faction == NULL ) )
         continue;
 
      if ( print_all || date == n->date ||
           ( characteristic &&
             ( ( strcmp( n->title, characteristic ) == 0 ) ||
               ( strcmp( n->desc, characteristic ) == 0 ) ||
               ( strcmp( n->faction, characteristic ) == 0 ) ||
               ( n->tag != NULL &&
                 strcmp( n->tag, characteristic ) == 0 ) ) ) ) {
         lua_pushnews( L, n->id ); /* value */
         lua_rawseti( L, -2, narticle++ );
      }
   }
 
   return 1;
}

int newsL_eq( lua_State *L )
{
   const LuaNews_t *a1, *a2;
   a1 = luaL_checknews( L, 1 );
   a2 = luaL_checknews( L, 2 );
   lua_pushboolean( L, *a1 == *a2 );
   return 1;
}

int newsL_title( lua_State *L )
{
   news_t *article_ptr = luaL_validnews( L, 1 );
   lua_pushstring( L, article_ptr->title );
   return 1;
}

int newsL_desc( lua_State *L )
{
   news_t *article_ptr = luaL_validnews( L, 1 );
   lua_pushstring( L, article_ptr->desc );
   return 1;
}

int newsL_faction( lua_State *L )
{
   news_t *article_ptr = luaL_validnews( L, 1 );
   lua_pushstring( L, article_ptr->faction );
   return 1;
}

int newsL_date( lua_State *L )
{
   news_t *article_ptr = luaL_validnews( L, 1 );
   lua_pushinteger( L, (lua_Integer)article_ptr->date );
   return 1;
}

int newsL_bind( lua_State *L )
{
   news_t    *article_ptr;
   LuaNews_t *a = NULL;
 
   if ( lua_istable( L, 1 ) ) {
      const char *tag = luaL_checkstring( L, 2 );
      lua_pop( L, 1 );
      lua_pushnil( L );
 
      /* traverse table */
      while ( lua_next( L, -2 ) ) {
         if ( !( a = luaL_checknews( L, -1 ) ) ) {
            NLUA_WARN(
               L, _( "Bad argument to news.date(), must be article or a table "
                     "of articles" ) );
            return 0;
         }
         article_ptr = news_get( *a );
         if ( article_ptr == NULL ) {
            NLUA_WARN( L, _( "Article not valid" ) );
            return 0;
         }
         article_ptr->tag = strdup( tag );
         lua_pop( L, 1 );
      }
   } else {
      const char *tag;
      if ( !( a = luaL_checknews( L, 1 ) ) ) {
         NLUA_WARN(
            L, _( "Bad argument to news.date(), must be article or a table of "
                  "articles" ) );
         return 0;
      }
      article_ptr = news_get( *a );
      if ( article_ptr == NULL ) {
         NLUA_WARN( L, _( "Article not valid" ) );
         return 0;
      }
 
      tag              = luaL_checkstring( L, 2 );
      article_ptr->tag = strdup( tag );
   }
   return 1;
}