stdio.lua

Branches > main > aeda3bc > files / nvim / pack / plugins / opt / conjure / lua / conjure / client / hy / stdio.lua

-- [nfnl] fnl/conjure/client/hy/stdio.fnl
local _local_1_ = require("conjure.nfnl.module")
local autoload = _local_1_.autoload
local define = _local_1_.define
local core = autoload("conjure.nfnl.core")
local extract = autoload("conjure.extract")
local str = autoload("conjure.nfnl.string")
local stdio = autoload("conjure.remote.stdio")
local config = autoload("conjure.config")
local text = autoload("conjure.text")
local mapping = autoload("conjure.mapping")
local client = autoload("conjure.client")
local log = autoload("conjure.log")
local ts = autoload("conjure.tree-sitter")
local M = define("conjure.client.hy.stdio")
config.merge({client = {hy = {stdio = {eval = {raw_out = false}, command = "hy -iu -c=\"Ready!\"", prompt_pattern = "=> "}}}})
if config["get-in"]({"mapping", "enable_defaults"}) then
  config.merge({client = {hy = {stdio = {mapping = {start = "cs", stop = "cS", interrupt = "ei"}}}}})
else
end
local cfg = config["get-in-fn"]({"client", "hy", "stdio"})
local state
local function _3_()
  return {repl = nil}
end
state = client["new-state"](_3_)
M["buf-suffix"] = ".hy"
M["comment-prefix"] = "; "
M["form-node?"] = ts["node-surrounded-by-form-pair-chars?"]
local function with_repl_or_warn(f, opts)
  local repl = state("repl")
  if repl then
    return f(repl)
  else
    return log.append({(M["comment-prefix"] .. "No REPL running"), (M["comment-prefix"] .. "Start REPL with " .. config["get-in"]({"mapping", "prefix"}) .. cfg({"mapping", "start"}))})
  end
end
local function display_result(msg)
  local prefix
  if (true == cfg({"eval", "raw_out"})) then
    prefix = ""
  else
    local _5_
    if msg.err then
      _5_ = "(err)"
    else
      _5_ = "(out)"
    end
    prefix = (M["comment-prefix"] .. _5_ .. " ")
  end
  local function _8_(_241)
    return (prefix .. _241)
  end
  local function _9_(_241)
    return ("" ~= _241)
  end
  return log.append(core.map(_8_, core.filter(_9_, str.split((msg.err or msg.out), "\n"))))
end
local function prep_code(s)
  return (s .. "\n")
end
M["eval-str"] = function(opts)
  local last_value = nil
  local function _10_(repl)
    local function _11_(msg)
      log.dbg("msg", msg)
      local msgs
      local function _12_(_241)
        return not ("" == _241)
      end
      msgs = core.filter(_12_, str.split((msg.err or msg.out), "\n"))
      last_value = (core.last(msgs) or last_value)
      display_result(msg)
      if msg["done?"] then
        log.append({""})
        if opts["on-result"] then
          return opts["on-result"](last_value)
        else
          return nil
        end
      else
        return nil
      end
    end
    return repl.send(prep_code(opts.code), _11_)
  end
  return with_repl_or_warn(_10_)
end
M["eval-file"] = function(opts)
  return log.append({(M["comment-prefix"] .. "Not implemented")})
end
M["doc-str"] = function(opts)
  local obj
  if ("." == string.sub(opts.code, 1, 1)) then
    obj = extract.prompt("Specify object or module: ")
  else
    obj = nil
  end
  local obj0 = ((obj or "") .. opts.code)
  local code = ("(if (in (hy.mangle '" .. obj0 .. ") _hy_macros)\n                    (help (get _hy_macros (hy.mangle '" .. obj0 .. ")))\n                    (help " .. obj0 .. "))")
  local function _16_(repl)
    local function _17_(msg)
      local content = (msg.err or msg.out)
      local _18_
      if msg.err then
        _18_ = "(err) "
      else
        _18_ = "(doc) "
      end
      log.append(text["prefixed-lines"](content, (M["comment-prefix"] .. _18_)))
      if opts["on-result"] then
        return opts["on-result"](content)
      else
        return nil
      end
    end
    return repl.send(prep_code(code), _17_)
  end
  return with_repl_or_warn(_16_)
end
local function display_repl_status(status)
  local repl = state("repl")
  if repl then
    return log.append({(M["comment-prefix"] .. core["pr-str"](core["get-in"](repl, {"opts", "cmd"})) .. " (" .. status .. ")")}, {["break?"] = true})
  else
    return nil
  end
end
M.stop = function()
  local repl = state("repl")
  if repl then
    repl.destroy()
    display_repl_status("stopped")
    return core.assoc(state(), "repl", nil)
  else
    return nil
  end
end
M.start = function()
  if state("repl") then
    return log.append({(M["comment-prefix"] .. "Can't start, REPL is already running."), (M["comment-prefix"] .. "Stop the REPL with " .. config["get-in"]({"mapping", "prefix"}) .. cfg({"mapping", "stop"}))}, {["break?"] = true})
  else
    local function _23_()
      display_repl_status("started")
      local function _24_(repl)
        return repl.send(prep_code("(import sys) (setv sys.ps2 \"\") (del sys)"))
      end
      return with_repl_or_warn(_24_)
    end
    local function _25_(err)
      return display_repl_status(err)
    end
    local function _26_(code, signal)
      if (("number" == type(code)) and (code > 0)) then
        log.append({(M["comment-prefix"] .. "process exited with code " .. code)})
      else
      end
      if (("number" == type(signal)) and (signal > 0)) then
        log.append({(M["comment-prefix"] .. "process exited with signal " .. signal)})
      else
      end
      return M.stop()
    end
    local function _29_(msg)
      return display_result(msg)
    end
    return core.assoc(state(), "repl", stdio.start({["prompt-pattern"] = cfg({"prompt_pattern"}), cmd = cfg({"command"}), ["on-success"] = _23_, ["on-error"] = _25_, ["on-exit"] = _26_, ["on-stray-output"] = _29_}))
  end
end
M["on-load"] = function()
  return M.start()
end
M["on-exit"] = function()
  return M.stop()
end
M.interrupt = function()
  log.dbg("sending interrupt message", "")
  local function _31_(repl)
    log.append({(M["comment-prefix"] .. " Sending interrupt signal.")}, {["break?"] = true})
    return repl["send-signal"]("sigint")
  end
  return with_repl_or_warn(_31_)
end
M["on-filetype"] = function()
  local function _32_()
    return M.start()
  end
  mapping.buf("HyStart", cfg({"mapping", "start"}), _32_, {desc = "Start the REPL"})
  local function _33_()
    return M.stop()
  end
  mapping.buf("HyStop", cfg({"mapping", "stop"}), _33_, {desc = "Stop the REPL"})
  return mapping.buf("HyInterrupt", cfg({"mapping", "interrupt"}), M.interrupt, {desc = "Interrupt the current evaluation"})
end
return M