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,
                             const struct ParseContext *pc, struct ParseError *pe)
{
  struct Buffer *err = pe->message;
 
  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,
                                    const struct ParseContext *pc, struct ParseError *pe)
{
  struct Buffer *err = pe->message;
 
  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;
  }
 
  expand_path(token, false);
 
  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;
}

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

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