console.c

Go to the documentation of this file.

/*
 * See Licensing and Copyright notice in naev.h
 */
#define lua_c
#include <ctype.h>
#include <lauxlib.h>
#include <lua.h>
#include <lualib.h>
#include <stdlib.h>
 
#include "naev.h"
#include "console.h"
 
#include "array.h"
#include "conf.h"
#include "font.h"
#include "log.h"
#include "menu.h"
#include "naev.h"
#include "ndata.h"
#include "nfile.h"
#include "nlua.h"
#include "nlua_audio.h"
#include "nlua_bkg.h"
#include "nlua_camera.h"
#include "nlua_cli.h"
#include "nlua_colour.h"
#include "nlua_linopt.h"
#include "nlua_music.h"
#include "nlua_tex.h"
#include "nlua_tk.h"
#include "nluadef.h"
#include "nstring.h"
#include "toolkit.h"
 
#define BUTTON_WIDTH    50 
#define BUTTON_HEIGHT   20 
/*
 * Global stuff.
 */
static nlua_env cli_env     = LUA_NOREF; 
static lua_State *cli_thread= NULL; 
static int cli_thread_ref   = LUA_NOREF; 
static glFont *cli_font     = NULL; 
/*
 * Buffers.
 */
#define CLI_MAX_INPUT      STRMAX_SHORT 
#define CLI_WIDTH          (SCREEN_W - 100) 
#define CLI_HEIGHT         (SCREEN_H - 100) 
static char **cli_buffer; 
static char *cli_prompt; 
static int cli_history     = 0; 
static int cli_scroll_pos  = -1; 
static int cli_firstOpen   = 1; 
static int cli_firstline   = 1; 
static int cli_height      = 0;
static int cli_max_lines   = 0;
#define CLI_MAX_LINES (cli_max_lines)
 
/*
 * CLI stuff.
 */
static int cli_script( lua_State *L );
static const luaL_Reg cli_methods[] = {
   { "script", cli_script },
   { "warn", cli_warn },
   {NULL, NULL}
}; 
/*
 * Prototypes.
 */
static int cli_keyhandler( unsigned int wid, SDL_Keycode key, SDL_Keymod mod, int isrepeat );
static void cli_render( double bx, double by, double w, double h, void *data );
static void cli_printCoreString( const char *s, int escape );
static int cli_printCore( lua_State *L, int cli_only, int escape );
static void cli_addMessage( const char *msg );
static void cli_addMessageMax( const char *msg, const int l );
void cli_tabComplete( unsigned int wid );
static int cli_initLua (void);
 
static char* cli_escapeString( int *len_out, const char *s, int len )
{
   char *buf = malloc( 2*len+1 ); /* worst case */
   int b = 0;
   for (int i=0; i<len; i++) {
      if (s[i]==FONT_COLOUR_CODE)
         buf[b++] = FONT_COLOUR_CODE;
      buf[b++] = s[i];
   }
   buf[b] = '\0';
   *len_out = b;
   return buf;
}
 
static void cli_printCoreString( const char *s, int escape )
{
   int len;
   glPrintLineIterator iter;
 
   gl_printLineIteratorInit( &iter, cli_font, s, CLI_WIDTH-40 );
   while (gl_printLineIteratorNext( &iter )) {
      if (escape) {
         char *buf = cli_escapeString( &len, &s[iter.l_begin], iter.l_end - iter.l_begin );
         cli_addMessageMax( buf, len );
         free(buf);
      }
      else
         cli_addMessageMax( &s[iter.l_begin], iter.l_end - iter.l_begin );
   }
}
 
static int cli_printCore( lua_State *L, int cli_only, int escape )
{
   int n = lua_gettop(L); /* Number of arguments. */
 
   lua_getglobal(L, "tostring");
   for (int i=1; i<=n; i++) {
      const char *s;
      lua_pushvalue(L, -1);  /* function to be called */
      lua_pushvalue(L, i);   /* value to print */
      if (lua_pcall(L, 1, 1, 0) != 0) {
         WARN(_("Error calling 'tostring':\n%s"), lua_tostring(L,-1));
         lua_pop(L,1);
         continue;
      }
      s = lua_tostring(L, -1);  /* get result */
      if (s == NULL)
         return NLUA_ERROR(L, LUA_QL("tostring") " must return a string to "
               LUA_QL("print"));
      if (!cli_only)
         LOG( "%s", s );
 
      /* Add to console. */
      cli_printCoreString( s, escape );
 
      lua_pop(L, 1);  /* pop result */
   }
 
   return 0;
}
 
int cli_warn( lua_State *L )
{
   const char *msg = luaL_checkstring(L,1);
   WARN("%s", msg );
   /* Add to console. */
   cli_printCoreString( msg, 1 );
   return 0;
}
 
int cli_print( lua_State *L )
{
   return cli_printCore( L, 0, 1 );
}
 
int cli_printRaw( lua_State *L )
{
   return cli_printCore( L, 1, 0 );
}
 
static int cli_script( lua_State *L )
{
   const char *fname;
   char *buf;
   size_t blen;
   int n;
 
   /* Handle parameters. */
   fname = luaL_checkstring(L, 1);
   n = lua_gettop(L);
 
   /* Reset loaded buffer. */
   if (cli_env != LUA_NOREF) {
      nlua_getenv( L, cli_env, "_LOADED" );
      if (lua_istable(L,-1)) {
         lua_pushnil(L);                     /* t, nil */
         while (lua_next(L, -2) != 0) {      /* t, key, val */
            lua_pop(L,1);                    /* t, key */
            lua_pushvalue(L,-1);             /* t, key, key */
            lua_pushnil(L);                  /* t, key, key, nil */
            lua_rawset(L,-4);                /* t, key */
         }                                   /* t */
      }
      lua_pop(L,1);                          /* */
   }
 
   /* Do the file from PHYSFS. */
   buf = ndata_read( fname, &blen );
   if (luaL_loadbuffer( L, buf, blen, fname ) != 0)
      lua_error(L);
   free( buf );
 
   /* Return the stuff. */
   nlua_pushenv(L, cli_env);
   lua_setfenv(L, -2);
   if (lua_pcall(L, 0, LUA_MULTRET, 0) != 0) {
      WARN(_("Error running 'script':\n%s"), lua_tostring(L,-1));
      lua_pop(L,1);
      return 0;
   }
   return lua_gettop(L) - n;
}
 
static void cli_addMessage( const char *msg )
{
   char *buf;
   /* Not initialized. */
   if (cli_env == LUA_NOREF)
      return;
   buf = strdup((msg != NULL) ? msg : "");
   array_grow(&cli_buffer) = buf;
   cli_history = array_size(cli_buffer) - 1;
}
 
static void cli_addMessageMax( const char *msg, const int l )
{
   char *buf;
   /* Not initialized. */
   if (cli_env == LUA_NOREF)
      return;
   buf = strndup((msg != NULL) ? msg : "", l);
   array_grow(&cli_buffer) = buf;
   cli_history = array_size(cli_buffer) - 1;
}
 
static void cli_render( double bx, double by, double w, double h, void *data )
{
   (void) data;
   int start;
   const glColour col = COL_ALPHA( cBlack, 0.5 );
 
   gl_renderRect( bx, by, w, h, &col );
 
   if (cli_scroll_pos == -1)
      start = MAX(0, array_size(cli_buffer)-CLI_MAX_LINES);
   else
      start = cli_scroll_pos;
 
   for (int i=start; i<array_size(cli_buffer); i++)
      gl_printMaxRaw( cli_font, w, bx,
            by + h - (i-start)*(cli_font->h+5),
            &cFontWhite, -1., cli_buffer[i] );
}
 
static int cli_keyhandler( unsigned int wid, SDL_Keycode key, SDL_Keymod mod, int isrepeat )
{
   (void) mod;
   (void) isrepeat;
 
   switch (key) {
 
      /* Go up in history. */
      case SDLK_UP:
         for (int i=cli_history; i>=0; i--) {
            if (strncmp(cli_buffer[i], "#C>", 3) == 0) {
               /* Strip escape codes from beginning and end */
               char *str = strndup(cli_buffer[i]+5, strlen(cli_buffer[i])-7);
               if (i == cli_history &&
                  strcmp(window_getInput(wid, "inpInput"), str) == 0) {
                  free(str);
                  continue;
               }
               window_setInput( wid, "inpInput", str );
               free(str);
               cli_history = i;
               return 1;
            }
         }
         return 1;
 
      /* Go down in history. */
      case SDLK_DOWN:
         /* Clears buffer. */
         if (cli_history >= array_size(cli_buffer)-1) {
            window_setInput( wid, "inpInput", NULL );
            return 1;
         }
 
         /* Find next buffer. */
         for (int i=cli_history+1; i<array_size(cli_buffer); i++) {
            if (strncmp(cli_buffer[i], "#C>", 3) == 0) {
               char *str = strndup(cli_buffer[i]+5, strlen(cli_buffer[i])-7);
               window_setInput( wid, "inpInput", str );
               free(str);
               return 1;
            }
         }
         cli_history = array_size(cli_buffer)-1;
         window_setInput( wid, "inpInput", NULL );
         return 1;
 
      /* Scroll up */
      case SDLK_PAGEUP:
         if (cli_scroll_pos == -1)
            cli_scroll_pos = MAX(0, array_size(cli_buffer) - CLI_MAX_LINES);
         cli_scroll_pos = MAX(0, cli_scroll_pos - CLI_MAX_LINES);
         return 1;
 
      /* Scroll down */
      case SDLK_PAGEDOWN:
         if (cli_scroll_pos != -1) {
            cli_scroll_pos = cli_scroll_pos + CLI_MAX_LINES;
            if (cli_scroll_pos > (array_size(cli_buffer) - CLI_MAX_LINES))
               cli_scroll_pos = -1;
         }
         return 1;
 
      /* Tab completion */
      case SDLK_TAB:
         cli_tabComplete(wid);
         return 1;
 
      default:
         break;
   }
 
   return 0;
}
 
void cli_tabComplete( unsigned int wid )
{
   const char *match, *old;
   char *str, *cur, *new;
 
   old = window_getInput( wid, "inpInput" );
   str = strdup(old);
 
   nlua_pushenv(naevL, cli_env);
   cur = str;
   for (int i=0; str[i] != '\0'; i++) {
      if (str[i] == '.' || str[i] == ':') {
         str[i] = '\0';
         lua_getfield(naevL, -1, cur);
 
         /* If not indexable, replace with blank table */
         if (!lua_istable(naevL, -1)) {
            if (luaL_getmetafield(naevL, -1, "__index")) {
               if (lua_istable(naevL, -1)) {
                  /* Handles the metatables used by Naev's userdatas */
                  lua_remove(naevL, -2);
               } else {
                  lua_pop(naevL, 2);
                  lua_newtable(naevL);
               }
            } else {
               lua_pop(naevL, 1);
               lua_newtable(naevL);
            }
         }
 
         lua_remove(naevL, -2);
         cur = str + i + 1;
      /* Start over on other non-identifier character */
      } else if (!isalnum(str[i]) && str[i] != '_') {
         lua_pop(naevL, 1);
         nlua_pushenv(naevL, cli_env);
         cur = str + i + 1;
      }
   }
 
   if (strlen(cur) > 0) {
      lua_pushnil(naevL);
      while (lua_next(naevL, -2) != 0) {
         if (lua_isstring(naevL, -2)) {
            match = lua_tostring(naevL, -2);
            if (strncmp(cur, match, strlen(cur)) == 0) {
               new = malloc(strlen(old) + strlen(match) - strlen(cur) + 1);
               strcpy(new, old);
               strcat(new, match + strlen(cur));
               window_setInput( wid, "inpInput", new);
               free(new);
               lua_pop(naevL, 2);
               break;
            }
         }
         lua_pop(naevL, 1);
      }
   }
   free(str);
   lua_pop(naevL, 1);
}
 
static int cli_initLua (void)
{
   int status;
   size_t blen;
   /* Already loaded. */
   if (cli_env != LUA_NOREF)
      return 0;
 
   /* Create the state. */
   cli_env = nlua_newEnv();
   nlua_loadStandard( cli_env );
   nlua_loadTex( cli_env );
   nlua_loadCol( cli_env );
   nlua_loadBackground( cli_env );
   nlua_loadCLI( cli_env );
   nlua_loadCamera( cli_env );
   nlua_loadMusic( cli_env );
   nlua_loadTk( cli_env );
   nlua_loadLinOpt( cli_env );
 
   nlua_pushenv( naevL, cli_env );
   luaL_register( naevL, NULL, cli_methods );
   lua_settop( naevL, 0 );
 
   if (conf.lua_repl) {
      char *buf = ndata_read( "rep.lua", &blen );
      status = nlua_dobufenv( cli_env, buf, blen, "@rep.lua" );
      free( buf );
      if (status) {
         lua_settop( naevL, 0 );
         return status;
      }
      if (lua_gettop( naevL ) != 1) {
         WARN( _("rep.lua failed to return a single coroutine.") );
         lua_settop( naevL, 0 );
         return -1;
      }
      cli_thread = lua_tothread( naevL, -1 );
      cli_thread_ref = luaL_ref( naevL, LUA_REGISTRYINDEX );
      if (cli_thread == NULL) {
         WARN( _("rep.lua failed to return a single coroutine.") );
         lua_settop( naevL, 0 );
         return -1;
      }
      status = lua_resume( cli_thread, 0 );
      cli_prompt = strdup( lua_tostring( cli_thread, -1 ) );
      if (status == 0) {
         WARN( _("REPL thread exited.") );
         return 1;
      }
      else if (status != LUA_YIELD) {
         WARN( "%s", cli_prompt );
         return -1;
      }
   }
   return 0;
}
 
int cli_init (void)
{
   /* Set the font. */
   cli_font    = malloc( sizeof(glFont) );
   gl_fontInit( cli_font, _(FONT_MONOSPACE_PATH), conf.font_size_console, FONT_PATH_PREFIX, 0 );
 
   /* Allocate the buffer. */
   cli_buffer = array_create(char*);
 
   cli_initLua();
 
   return 0;
}
 
void cli_exit (void)
{
   /* Destroy the state. */
   luaL_unref( naevL, LUA_REGISTRYINDEX, cli_thread_ref );
   nlua_freeEnv( cli_env );
   cli_thread = NULL;
   cli_env = cli_thread_ref = LUA_NOREF;
 
   gl_freeFont( cli_font );
   free( cli_font );
   cli_font = NULL;
 
   /* Free the buffer. */
   for (int i=0; i<array_size(cli_buffer); i++)
      free(cli_buffer[i]);
   array_free(cli_buffer);
   cli_buffer = NULL;
   free(cli_prompt);
   cli_prompt = NULL;
}
 
static void cli_input( unsigned int wid, const char *unused )
{
   (void) unused;
   int status, len;
   const char *str, *prompt;
   char *escaped;
   char buf[CLI_MAX_INPUT+7];
 
   /* Get the input. */
   str = window_getInput( wid, "inpInput" );
 
   /* Ignore useless stuff. */
   if (str == NULL || (conf.lua_repl && cli_thread == NULL))
      return;
 
   /* Put the message in the console. */
   escaped = cli_escapeString( &len, str, strlen(str) );
   if (conf.lua_repl)
      prompt = cli_prompt; /* Remembered from last time lua-repl requested input */
   else
      prompt = cli_firstline ? "> " : ">>";
   snprintf( buf, CLI_MAX_INPUT+7, "#C%s %s#0", prompt, escaped );
   free(escaped);
   cli_printCoreString( buf, 0 );
 
   if (conf.lua_repl) {
      /* Resume the REPL. */
      lua_pushstring( cli_thread, str );
      status = lua_resume( cli_thread, 1 );
      if (status == 0) {
         cli_printCoreString( _("REPL thread exited."), 0 );
         luaL_unref( naevL, LUA_REGISTRYINDEX, cli_thread_ref );
         cli_thread = NULL;
         cli_thread_ref = LUA_NOREF;
      }
      else {
         free( cli_prompt );
         cli_prompt = strdup( lua_tostring( cli_thread, -1 ) );
         lua_settop( cli_thread, 0 );
         if (status != LUA_YIELD)
            cli_printCoreString( cli_prompt, 0 );
      }
   }
   else {
      /* Set up for concat. */
      if (!cli_firstline)               /* o */
         lua_pushliteral(naevL, "\n");  /* o \n */
 
      /* Load the string. */
      lua_pushstring( naevL, str );     /* s */
 
      /* Concat. */
      if (!cli_firstline)               /* o \n s */
         lua_concat(naevL, 3);          /* s */
 
      status = luaL_loadbuffer( naevL, lua_tostring(naevL,-1), lua_strlen(naevL,-1), "=cli" );
 
      /* String isn't proper Lua yet. */
      if (status == LUA_ERRSYNTAX) {
         size_t lmsg;
         const char *msg = lua_tolstring(naevL, -1, &lmsg);
         const char *tp = msg + lmsg - (sizeof(LUA_QL("<eof>")) - 1);
         if (strstr(msg, LUA_QL("<eof>")) == tp) {
            /* Pop the loaded buffer. */
            lua_pop(naevL, 1);
            cli_firstline = 0;
         }
         else {
            /* Real error, spew message and break. */
            const char *s = lua_tostring(naevL, -1);
            WARN( "%s", s );
            cli_printCoreString( s, 1 );
            lua_settop(naevL, 0);
            cli_firstline = 1;
         }
      }
      /* Print results - all went well. */
      else if (status == 0) {
         lua_remove(naevL,1);
 
         nlua_pushenv(naevL, cli_env);
         lua_setfenv(naevL, -2);
 
         if (nlua_pcall(cli_env, 0, LUA_MULTRET)) {
            cli_printCoreString( lua_tostring(naevL, -1), 1 );
            lua_pop(naevL, 1);
         }
 
         if (lua_gettop(naevL) > 0) {
            nlua_getenv(naevL, cli_env, "print");
            lua_insert(naevL, 1);
            if (lua_pcall(naevL, lua_gettop(naevL)-1, 0, 0) != 0)
               cli_addMessage( _("Error printing results.") );
         }
 
         /* Clear stack. */
         lua_settop(naevL, 0);
         cli_firstline = 1;
      }
   }
 
   /* Clear the box now. */
   window_setInput( wid, "inpInput", NULL );
 
   /* Scroll to bottom */
   cli_scroll_pos = -1;
}
 
void cli_open (void)
{
   unsigned int wid;
   int line_height;
 
   /* Lazy loading. */
   if (cli_env == LUA_NOREF)
      if (cli_init())
         return;
 
   if (menu_isOpen(MENU_MAIN) || cli_isOpen())
      return;
 
   /* Put a friendly message at first. */
   if (cli_firstOpen) {
      char *buf;
      cli_addMessage( "" );
      cli_addMessage( _("#gWelcome to the Lua console!") );
      SDL_asprintf( &buf, "#g "APPNAME" v%s", naev_version(0) );
      cli_printCoreString( buf, 0 );
      free( buf );
      cli_addMessage( "" );
      cli_firstOpen = 0;
   }
 
   /* Create the window. */
   wid = window_create( "wdwLuaConsole", _("Lua Console"), -1, -1, CLI_WIDTH, CLI_HEIGHT );
 
   /* Window settings. */
   window_setAccept( wid, cli_input );
   window_setCancel( wid, window_close );
   window_handleKeys( wid, cli_keyhandler );
 
   /* Input box. */
   window_addInput( wid, 20, 20,
         CLI_WIDTH-60-BUTTON_WIDTH, BUTTON_HEIGHT,
         "inpInput", CLI_MAX_INPUT, 1, cli_font );
 
   /* Buttons. */
   window_addButton( wid, -20, 20, BUTTON_WIDTH, BUTTON_HEIGHT,
         "btnClose", _("Close"), window_close );
 
   /* Custom console widget. */
   line_height = cli_font->h*1.5;
   cli_max_lines = ((CLI_HEIGHT-60-BUTTON_HEIGHT+(line_height-cli_font->h)) / line_height);
   cli_height = line_height * cli_max_lines - (line_height - cli_font->h);
   window_addCust( wid, 20, -40,
         CLI_WIDTH-40, cli_height,
         "cstConsole", 0, cli_render, NULL, NULL, NULL, NULL );
 
   /* Reinitilaized. */
   cli_firstline = 1;
}
 
int cli_isOpen (void)
{
   return window_exists( "wdwLuaConsole" );
}