aboutsummaryrefslogtreecommitdiff
path: root/plugins/exec.lua
blob: b8f61c56b9f6042f9bc8c588af0e705fc826103e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
-- mod-version:3
local core = require "core"
local command = require "core.command"


local function exec(cmd, keep_newline)
  local fp = io.popen(cmd, "r")
  local res = fp:read("*a")
  fp:close()
  return keep_newline and res or res:gsub("%\n$", "")
end


local function shell_quote(str)
  return "'" .. str:gsub("'", "'\\''") .. "'"
end


local printfb_sub = {
  ["\\"] = "\\\\",
  ["\0"] = "\\0000",
  ["'"] = "'\\''",
}
local function printfb_quote(str)
  return "'" .. str:gsub(".", printfb_sub) .. "'"
end


command.add("core.docview", {
  ["exec:insert"] = function()
    core.command_view:enter("Insert Result Of Command", {
      submit = function(cmd)
        core.active_view.doc:text_input(exec(cmd))
      end
    })
  end,

  ["exec:replace"] = function()
    core.command_view:enter("Replace With Result Of Command", {
      submit = function(cmd)
        core.active_view.doc:replace(function(str)
          return exec(
            "printf %b " .. printfb_quote(str:gsub("%\n$", "") .. "\n") .. " | eval '' " .. shell_quote(cmd),
            str:find("%\n$")
          )
        end)
      end
    })
  end,
})