lua.c

Go to the documentation of this file.

 
#ifndef LUA_COMPAT_ALL
#define LUA_COMPAT_ALL
#endif
#ifndef LUA_COMPAT_5_1
#define LUA_COMPAT_5_1
#endif
 
#include "config.h"
#include <lauxlib.h>
#include <lua.h>
#include <lualib.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include "mutt/lib.h"
#include "config/lib.h"
#include "core/lib.h"
#include "parse/lib.h"
#include "muttlib.h"
#include "version.h"

static int lua_handle_panic(lua_State *l)
{
  mutt_debug(LL_DEBUG1, "lua runtime panic: %s\n", lua_tostring(l, -1));
  mutt_error("Lua runtime panic: %s", lua_tostring(l, -1));
  lua_pop(l, 1);
  return -1;
}

static int lua_handle_error(lua_State *l)
{
  mutt_debug(LL_DEBUG1, "lua runtime error: %s\n", lua_tostring(l, -1));
  mutt_error("Lua runtime error: %s", lua_tostring(l, -1));
  lua_pop(l, 1);
  return -1;
}

static int lua_cb_global_call(lua_State *l)
{
  mutt_debug(LL_DEBUG2, "enter\n");
  struct Buffer *buf = buf_pool_get();
  struct ParseContext *pc = parse_context_new();
  struct ParseError *pe = parse_error_new();
  const struct Command *cmd = NULL;
  int rc = 0;
 
  if (lua_gettop(l) == 0)
  {
    buf_pool_release(&buf);
    parse_context_free(&pc);
    parse_error_free(&pe);
    luaL_error(l, "Error command argument required");
    return -1;
  }
 
  cmd = commands_get(&NeoMutt->commands, lua_tostring(l, 1));
  if (!cmd)
  {
    buf_pool_release(&buf);
    parse_context_free(&pc);
    parse_error_free(&pe);
    luaL_error(l, "Error command %s not found", lua_tostring(l, 1));
    return -1;
  }
 
  for (int i = 2; i <= lua_gettop(l); i++)
  {
    buf_addstr(buf, lua_tostring(l, i));
    buf_addch(buf, ' ');
  }
  buf_seek(buf, 0);
 
  if (cmd->parse(cmd, buf, pc, pe))
  {
    luaL_error(l, "NeoMutt error: %s", buf_string(pe->message));
    rc = -1;
  }
  else
  {
    if (!lua_pushstring(l, buf_string(pe->message)))
      lua_handle_error(l);
    else
      rc++;
  }
 
  buf_pool_release(&buf);
  parse_context_free(&pc);
  parse_error_free(&pe);
  return rc;
}

static int lua_cb_global_set(lua_State *l)
{
  const char *param = lua_tostring(l, -2);
  mutt_debug(LL_DEBUG2, "%s\n", param);
 
  struct Buffer *err = buf_pool_get();
  struct HashElem *he = cs_subset_lookup(NeoMutt->sub, param);
  if (!he)
  {
    // In case it is a my_var, we have to create it
    if (mutt_str_startswith(param, "my_"))
    {
      struct ConfigDef my_cdef = { 0 };
      my_cdef.name = param;
      my_cdef.type = DT_MYVAR;
      he = cs_create_variable(NeoMutt->sub->cs, &my_cdef, err);
      if (!he)
        return -1;
    }
    else
    {
      luaL_error(l, "NeoMutt parameter not found %s", param);
      return -1;
    }
  }
 
  struct ConfigDef *cdef = he->data;
 
  int rc = 0;
 
  switch (CONFIG_TYPE(cdef->type))
  {
    case DT_ADDRESS:
    case DT_ENUM:
    case DT_EXPANDO:
    case DT_MBTABLE:
    case DT_MYVAR:
    case DT_PATH:
    case DT_REGEX:
    case DT_SLIST:
    case DT_SORT:
    case DT_STRING:
    {
      const char *value = lua_tostring(l, -1);
      size_t val_size = lua_rawlen(l, -1);
      struct Buffer *value_buf = buf_pool_get();
      buf_strcpy_n(value_buf, value, val_size);
      if (CONFIG_TYPE(he->type) == DT_PATH)
        expand_path(value_buf, false);
 
      int rv = cs_subset_he_string_set(NeoMutt->sub, he, buf_string(value_buf), err);
      buf_pool_release(&value_buf);
      if (CSR_RESULT(rv) != CSR_SUCCESS)
        rc = -1;
      break;
    }
    case DT_LONG:
    case DT_NUMBER:
    case DT_QUAD:
    {
      const intptr_t value = lua_tointeger(l, -1);
      int rv = cs_subset_he_native_set(NeoMutt->sub, he, value, err);
      if (CSR_RESULT(rv) != CSR_SUCCESS)
        rc = -1;
      break;
    }
    case DT_BOOL:
    {
      const intptr_t value = lua_toboolean(l, -1);
      int rv = cs_subset_he_native_set(NeoMutt->sub, he, value, err);
      if (CSR_RESULT(rv) != CSR_SUCCESS)
        rc = -1;
      break;
    }
    default:
      luaL_error(l, "Unsupported NeoMutt parameter type %d for %s",
                 CONFIG_TYPE(cdef->type), param);
      rc = -1;
      break;
  }
 
  buf_pool_release(&err);
  return rc;
}

static int lua_cb_global_get(lua_State *l)
{
  const char *param = lua_tostring(l, -1);
  mutt_debug(LL_DEBUG2, "%s\n", param);
 
  struct HashElem *he = cs_subset_lookup(NeoMutt->sub, param);
  if (!he)
  {
    mutt_debug(LL_DEBUG2, "error\n");
    luaL_error(l, "NeoMutt parameter not found %s", param);
    return -1;
  }
 
  struct ConfigDef *cdef = he->data;
 
  switch (CONFIG_TYPE(cdef->type))
  {
    case DT_ADDRESS:
    case DT_ENUM:
    case DT_EXPANDO:
    case DT_MBTABLE:
    case DT_MYVAR:
    case DT_PATH:
    case DT_REGEX:
    case DT_SLIST:
    case DT_SORT:
    case DT_STRING:
    {
      struct Buffer *value = buf_pool_get();
      int rc = cs_subset_he_string_get(NeoMutt->sub, he, value);
      if (CSR_RESULT(rc) != CSR_SUCCESS)
      {
        buf_pool_release(&value);
        return -1;
      }
 
      struct Buffer *escaped = buf_pool_get();
      escape_string(escaped, buf_string(value));
      lua_pushstring(l, buf_string(escaped));
      buf_pool_release(&value);
      buf_pool_release(&escaped);
      return 1;
    }
    case DT_QUAD:
      lua_pushinteger(l, (unsigned char) cdef->var);
      return 1;
    case DT_LONG:
      lua_pushinteger(l, (signed long) cdef->var);
      return 1;
    case DT_NUMBER:
      lua_pushinteger(l, (signed short) cdef->var);
      return 1;
    case DT_BOOL:
      lua_pushboolean(l, (bool) cdef->var);
      return 1;
    default:
      luaL_error(l, "NeoMutt parameter type %d unknown for %s", cdef->type, param);
      return -1;
  }
}

static int lua_cb_global_enter(lua_State *l)
{
  mutt_debug(LL_DEBUG2, "enter\n");
  struct Buffer *line = buf_pool_get();
  struct ParseContext *pc = parse_context_new();
  struct ParseError *pe = parse_error_new();
 
  buf_strcpy(line, lua_tostring(l, -1));
  int rc = 0;
 
  if (parse_rc_line(line, pc, pe))
  {
    luaL_error(l, "NeoMutt error: %s", buf_string(pe->message));
    rc = -1;
  }
  else
  {
    if (!lua_pushstring(l, buf_string(pe->message)))
      lua_handle_error(l);
    else
      rc++;
  }
 
  buf_pool_release(&line);
  parse_context_free(&pc);
  parse_error_free(&pe);
 
  return rc;
}

static int lua_cb_global_message(lua_State *l)
{
  mutt_debug(LL_DEBUG2, "enter\n");
  const char *msg = lua_tostring(l, -1);
  if (msg)
    mutt_message("%s", msg);
  return 0;
}

static int lua_cb_global_error(lua_State *l)
{
  mutt_debug(LL_DEBUG2, "enter\n");
  const char *msg = lua_tostring(l, -1);
  if (msg)
    mutt_error("%s", msg);
  return 0;
}

static void lua_expose_command(lua_State *l, const struct Command *cmd)
{
  char buf[1024] = { 0 };
  snprintf(buf, sizeof(buf), "mutt.command.%s = function (...); mutt.call('%s', ...); end",
           cmd->name, cmd->name);
  (void) luaL_dostring(l, buf);
}

static const luaL_Reg LuaMuttCommands[] = {
  // clang-format off
  { "set",     lua_cb_global_set },
  { "get",     lua_cb_global_get },
  { "call",    lua_cb_global_call },
  { "enter",   lua_cb_global_enter },
  { "print",   lua_cb_global_message },
  { "message", lua_cb_global_message },
  { "error",   lua_cb_global_error },
  { NULL, NULL },
  // clang-format on
};

static int lua_expose_commands(lua_State *l)
{
  mutt_debug(LL_DEBUG2, "enter\n");
  luaL_newlib(l, LuaMuttCommands);
  int lib_idx = lua_gettop(l);
 
  // clang-format off
  lua_pushstring(l, "VERSION");     lua_pushstring(l, mutt_make_version()); lua_settable(l, lib_idx);
  lua_pushstring(l, "QUAD_YES");    lua_pushinteger(l, MUTT_YES);           lua_settable(l, lib_idx);
  lua_pushstring(l, "QUAD_NO");     lua_pushinteger(l, MUTT_NO);            lua_settable(l, lib_idx);
  lua_pushstring(l, "QUAD_ASKYES"); lua_pushinteger(l, MUTT_ASKYES);        lua_settable(l, lib_idx);
  lua_pushstring(l, "QUAD_ASKNO");  lua_pushinteger(l, MUTT_ASKNO);         lua_settable(l, lib_idx);
  // clang-format on
 
  return 1;
}

static void lua_expose_mutt(lua_State *l)
{
  luaL_requiref(l, "mutt", lua_expose_commands, 1);
  (void) luaL_dostring(l, "mutt.command = {}");
 
  const struct Command **cp = NULL;
  ARRAY_FOREACH(cp, &NeoMutt->commands)
  {
    const struct Command *cmd = *cp;
 
    lua_expose_command(l, cmd);
  }
}

bool lua_init_state(lua_State **l)
{
  if (!l)
    return false;
  if (*l)
    return true;
 
  mutt_debug(LL_DEBUG2, "enter\n");
  *l = luaL_newstate();
 
  if (!*l)
  {
    mutt_error(_("Error: Couldn't load the lua interpreter"));
    return false;
  }
 
  lua_atpanic(*l, lua_handle_panic);
 
  /* load various Lua libraries */
  luaL_openlibs(*l);
  lua_expose_mutt(*l);
 
  return true;
}