🎨 Provide function main()

Author: Freed-Wu

README.md

@@ -130,6 +130,7 @@ luarocks install prompt-style
 
 ### Other luas
 
+- [lupa](https://github.com/scoder/lupa): provide an extra `python` module in lua.
 - [wezterm](https://github.com/wez/wezterm):
   - `wezterm --config-file /the/path/weztermp`
   - doesn't support shebang

bin/lupa

@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+"""
+https://github.com/scoder/lupa/issues/278
+https://gitlab.archlinux.org/archlinux/packaging/packages/python-lupa/-/issues/2
+"""
+from lupa.lua54 import LuaRuntime
+
+if __name__ == "__main__":
+    lua = LuaRuntime()
+    lua.require('prompt.utils')
+    # use $PYTHONSTARTUP
+    lua.eval('require "prompt.utils".main(arg, {})')

bin/neomuttp

@@ -1,14 +1,14 @@
 #!/usr/bin/env -S sh -c 'neomutt -elua-source\\ $0'
 local utils = require "prompt.utils"
 
-utils.init()
 -- use ~/.config/neomutt/neomuttrc
--- utils.source_configs()
-local parser = utils.get_parser()
-local args = parser:parse(arg)
-if args.v then
-    mutt.command.version()
-    os.exit(0)
-end
-utils.process_args(args, parser)
+utils.main(arg, {}, function(args)
+    if args.v then
+        -- luacheck: ignore 111 113
+        ---@diagnostic disable: undefined-global
+        mutt.command.version()
+        os.exit(0)
+    end
+    return args
+end)
 -- ex: filetype=lua

bin/nvimp

@@ -14,14 +14,12 @@ if require "prompt".name == "lua" then
     end
 end
 
-utils.init()
 -- use ~/.config/nvim/init.vim
--- utils.source_configs()
-local parser = utils.get_parser()
-local args = parser:parse(arg)
-if args.v then
-    vim.cmd.version()
-    os.exit(0)
-end
-utils.process_args(args, parser)
+utils.main(arg, {}, function(args)
+    if args.v then
+        vim.cmd.version()
+        os.exit(0)
+    end
+    return args
+end)
 -- ex: filetype=lua

bin/pandocp

@@ -1,9 +1,5 @@
 #!/usr/bin/env -S pandoc -fmarkdown /dev/null --lua-filter
 local utils = require "prompt.utils"
 
-utils.init()
-utils.source_configs()
-local parser = utils.get_parser()
-local args = parser:parse(arg)
-utils.process_args(args, parser)
+utils.main(arg)
 -- ex: filetype=lua

lua/prompt/utils.lua

@@ -131,7 +131,7 @@ function M.process_args(args, parser)
 
                 if not ok or not m then
                     parser:error('unknown luaJIT command or ' ..
-                        'jit.* modules not installed')
+                    'jit.* modules not installed')
                 end
 
                 m.start(unpack(_args))
@@ -224,7 +224,7 @@ end
 ---@param configs string[] | nil
 function M.source_configs(configs)
     configs = configs or { os.getenv('HOME') .. '/.luaprc.lua',
-        os.getenv('HOME') .. '/.config/luaprc.lua' }
+            os.getenv('HOME') .. '/.config/luaprc.lua' }
     for _, name in ipairs(configs) do
         local f = io.open(name, "r")
         if f ~= nil then
@@ -240,5 +240,20 @@ function M.source_configs(configs)
     end
 end
 
+---**entry for pandocp**
+---@param argv string[] command line arguments
+---@param configs string[] | nil `nil` means default `~/.config/luaprc.lua`
+---@param callback function | nil handle `args`
+function M.main(argv, configs, callback)
+    M.init()
+    M.source_configs(configs)
+    local parser = M.get_parser()
+    local args = parser:parse(argv)
+    if callback then
+        args = callback(args)
+    end
+    M.process_args(args, parser)
+end
+
 ---@export
 return M

template.rockspec

@@ -39,14 +39,6 @@ end
 
 build = {
   type = 'builtin',
-  install = {
-      -- cannot use _VERSION
-      bin = {
-          "bin/nvimp",
-          "bin/pandocp",
-          "bin/neomuttp"
-      },
-  },
 }
 
 deploy = {