commands.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 <stdbool.h>
#include <stdio.h>
#include "mutt/lib.h"
#include "core/lib.h"
#include "lib.h"
#include "parse/lib.h"
#include "muttlib.h"
 
extern lua_State *LuaState;
 
bool lua_init_state(lua_State **l);

enum CommandResult parse_lua(const struct Command *cmd, struct Buffer *line, struct Buffer *err)
{
  if (!MoreArgs(line))
  {
    buf_printf(err, _("%s: too few arguments"), cmd->name);
    return MUTT_CMD_WARNING;
  }
 
  struct Buffer *token = buf_pool_get();
  enum CommandResult rc = MUTT_CMD_ERROR;
 
  parse_extract_token(token, line, TOKEN_NO_FLAGS);
 
  lua_init_state(&LuaState);
  mutt_debug(LL_DEBUG2, "%s\n", buf_string(token));
 
  if (luaL_dostring(LuaState, buf_string(token)) != LUA_OK)
  {
    mutt_debug(LL_DEBUG2, "%s -> failure\n", buf_string(token));
    buf_printf(err, "%s: %s", buf_string(token), lua_tostring(LuaState, -1));
    /* pop error message from the stack */
    lua_pop(LuaState, 1);
    goto done;
  }
  mutt_debug(LL_DEBUG2, "%s -> success\n", buf_string(token));
  buf_reset(line); // Clear the rest of the line
 
  rc = MUTT_CMD_SUCCESS;
 
done:
  buf_pool_release(&token);
  return rc;
}

enum CommandResult parse_lua_source(const struct Command *cmd,
                                    struct Buffer *line, struct Buffer *err)
{
  if (!MoreArgs(line))
  {
    buf_printf(err, _("%s: too few arguments"), cmd->name);
    return MUTT_CMD_WARNING;
  }
 
  struct Buffer *token = buf_pool_get();
  enum CommandResult rc = MUTT_CMD_ERROR;
 
  mutt_debug(LL_DEBUG2, "enter\n");
 
  lua_init_state(&LuaState);
 
  if (parse_extract_token(token, line, TOKEN_NO_FLAGS) != 0)
  {
    buf_printf(err, _("source: error at %s"), line->dptr);
    goto done;
  }
  if (MoreArgs(line))
  {
    buf_printf(err, _("%s: too many arguments"), cmd->name);
    rc = MUTT_CMD_WARNING;
    goto done;
  }
 
  buf_expand_path(token);
 
  if (luaL_dofile(LuaState, buf_string(token)) != LUA_OK)
  {
    mutt_error(_("Couldn't source lua source: %s"), lua_tostring(LuaState, -1));
    lua_pop(LuaState, 1);
    goto done;
  }
 
  rc = MUTT_CMD_SUCCESS;
 
done:
  buf_pool_release(&token);
  return rc;
}

static const struct Command LuaCommands[] = {
  // clang-format off
  { "lua", parse_lua, 0,
        N_("Run a Lua expression or call a Lua function"),
        N_("lua '<lua-commands>'"),
        "optionalfeatures.html#lua-commands" },
  { "lua-source", parse_lua_source, 0,
        N_("Execute a Lua script file"),
        N_("lua-source <file>"),
        "optionalfeatures.html#lua-commands" },
 
  { NULL, NULL, 0, NULL, NULL, NULL, CF_NO_FLAGS },
  // clang-format on
};

void lua_init(void)
{
  commands_register(&NeoMutt->commands, LuaCommands);
}

void lua_cleanup(void)
{
  if (LuaState)
  {
    lua_close(LuaState);
    LuaState = NULL;
  }
}