💥 Support \directlua{}

Author: Freed-Wu

README.md

@@ -102,10 +102,10 @@ Refer [luatex](#luatex).
 luatex use `kpse` to search lua modules and binary lua modules.
 
 ```sh
-kpsewhich -progname=luatex -var-value=LUAINPUTS
-kpsewhich -progname=luatex -var-value=CLUAINPUTS
-kpsewhich -progname=luajittex -var-value=LUAINPUTS
-kpsewhich -progname=luajittex -var-value=CLUAINPUTS
+$ kpsewhich -progname=luatex -format=clua prompt
+/run/current-system/sw/lib/lua/5.1/prompt.so
+$ kpsewhich -progname=luatex -format=lua prompt/utils
+/home/wzy/.local/share/lua/5.1/prompt/utils.lua
 ```
 
 You can edit web2c config to modify its default values. Such as, default config
@@ -130,6 +130,31 @@ export LUAINPUTS="$HOME/.local/share/lua/5.3"
 export CLUAINPUTS="$HOME/.local/lib/lua/5.3"
 ```
 
+Check your changes:
+
+```sh
+kpsewhich -progname=luatex -var-value=LUAINPUTS
+kpsewhich -progname=luatex -var-value=CLUAINPUTS
+kpsewhich -progname=luajittex -var-value=LUAINPUTS
+kpsewhich -progname=luajittex -var-value=CLUAINPUTS
+```
+
+If you want to use lua REPL in `\directlua{}`, you can insert:
+
+```tex
+\directlua{require'prompt.tex'}
+```
+
+As [example](examples/main.tex):
+
+```sh
+luajittex --lua=/the/path/of/prompt/tex.lua main.tex
+```
+
+Then you will get:
+
+![directlua](https://github.com/user-attachments/assets/45279ebe-76aa-4ff4-826e-03c00e9e79a8)
+
 #### [neomutt](https://neomutt.org)
 
 Broken in [upstream](https://github.com/neomutt/neomutt/issues/4328)
@@ -173,6 +198,3 @@ luarocks install prompt-style
   - `xmake l /the/path/xmake.lua`
   - doesn't support shebang
   - xmake doesn't provide `require()`
-- luatex's `\directlua{}`:
-  - weird. Even if `kpse.lookup('prompt.so')` return correct path,
-    `require'prompt'` still throw `module 'prompt' not found`.

examples/main.tex

@@ -0,0 +1,2 @@
+\directlua{require'prompt.tex'}
+\bye

lua/prompt/style.lua

@@ -15,7 +15,13 @@ local function wakatime(cmd)
         "--entity-type=app --entity=lua --alternate-language=lua --project=%s &")
     local s, _ = cmd:find("%s")
     if s ~= nil then
-        local project = io.popen("git rev-parse --show-toplevel 2> /dev/null"):read()
+        local f = io.popen("git rev-parse --show-toplevel 2> /dev/null")
+        -- specific command execution disabled
+        if f == nil then
+            return ""
+        end
+        local project = f:read()
+        f:close()
         if project == nil then
             project = lfs.currentdir()
         end
@@ -48,63 +54,63 @@ local function get_distribution()
         end
         f:close()
         if result:match("arch") then
-          return "arch"
+            return "arch"
         elseif result:match("debian") then
-          return "debian"
+            return "debian"
         elseif result:match("raspbian") then
-          return "raspbian"
+            return "raspbian"
         elseif result:match("ubuntu") then
-          return "ubuntu"
+            return "ubuntu"
         elseif result:match("elementary") then
-          return "elementary"
+            return "elementary"
         elseif result:match("fedora") then
-          return "fedora"
+            return "fedora"
         elseif result:match("coreos") then
-          return "coreos"
+            return "coreos"
         elseif result:match("kali") then
-          return "kali"
+            return "kali"
         elseif result:match("gentoo") then
-          return "gentoo"
+            return "gentoo"
         elseif result:match("mageia") then
-          return "mageia"
+            return "mageia"
         elseif result:match("centos") then
-          return "centos"
+            return "centos"
         elseif result:match("opensuse") then
-          return "opensuse"
+            return "opensuse"
         elseif result:match("tumbleweed") then
-          return "opensuse"
+            return "opensuse"
         elseif result:match("sabayon") then
-          return "sabayon"
+            return "sabayon"
         elseif result:match("slackware") then
-          return "slackware"
+            return "slackware"
         elseif result:match("linuxmint") then
-          return "mint"
+            return "mint"
         elseif result:match("alpine") then
-          return "alpine"
+            return "alpine"
         elseif result:match("aosc") then
-          return "aosc"
+            return "aosc"
         elseif result:match("nixos") then
-          return "nixos"
+            return "nixos"
         elseif result:match("devuan") then
-          return "devuan"
+            return "devuan"
         elseif result:match("manjaro") then
-          return "manjaro"
+            return "manjaro"
         elseif result:match("void") then
-          return "void"
+            return "void"
         elseif result:match("artix") then
-          return "artix"
+            return "artix"
         elseif result:match("rhel") then
-          return "rhel"
+            return "rhel"
         elseif result:match("amzn") then
-          return "amzn"
+            return "amzn"
         elseif result:match("endeavouros") then
-          return "endeavouros"
+            return "endeavouros"
         elseif result:match("rocky") then
-          return "rocky"
+            return "rocky"
         elseif result:match("guix") then
-          return "guix"
+            return "guix"
         elseif result:match("dock") then
-          return "docker"
+            return "docker"
         end
     end
     return "linux"

lua/prompt/tex.lua

@@ -0,0 +1,31 @@
+-- luacheck: ignore 111 112 113
+---@diagnostic disable: undefined-global
+if not texconfig.prompt_init then
+  -- FIXME: no format given, quitting
+  -- kpse.set_program_name(status.list().luatex_engine)
+  local configs = { os.getenv('HOME') .. '/.luaprc.lua',
+    os.getenv('HOME') .. '/.config/luaprc.lua' }
+  for _, name in ipairs(configs) do
+    local f = io.open(name, "r")
+    if f ~= nil then
+      io.close(f)
+      local chunk, message = loadfile(name)
+      if chunk then
+        chunk()
+      else
+        print(message)
+      end
+      break
+    end
+  end
+
+  require 'prompt'
+  texconfig.prompt_init = true
+  return
+end
+local prompt = require 'prompt'
+local utils = require "prompt.utils"
+
+utils.init()
+print()
+prompt.enter()

lua/prompt/utils.lua

@@ -1,6 +1,7 @@
 ---prompt utils.
 local argparse = require "argparse"
 local prompt = require "prompt"
+local style = require "prompt.style"
 local M = {}
 
 ---get a parser with same command line options as lua
@@ -212,7 +213,7 @@ function M.init()
         prompt.name = arg[0]:gsub(".*/", ""):gsub("p$", "")
     end
 
-    prompt.prompts = prompt.prompts or { '>  ', '>> ' }
+    prompt.prompts = { style.generate_ps1(), "    " }
     if prompt.colorize == nil then
         prompt.colorize = true
     end