[extregs] initial commit.HEADmain

added a bunch of useful extregs

13 files changed, 282 insertions, 0 deletions

diff --git a/lua/extregs/buffer.lua b/lua/extregs/buffer.lua
new file mode 100644
index 0000000..de7c1af
--- /dev/null
+++ b/lua/extregs/buffer.lua
@@ -0,0 +1,13 @@
+local M = {}
+function M.get_provider(_)
+  return {
+    read = function()
+      return { lines = vim.api.nvim_buf_get_lines(0, 0, -1, false), type = 'V' }
+    end,
+    write = function(_, reg)
+      vim.api.nvim_buf_set_lines(0, 0, -1, false, reg.lines)
+      return true
+    end,
+  }
+end
+return M
diff --git a/lua/extregs/file.lua b/lua/extregs/file.lua
new file mode 100644
index 0000000..0dfe88c
--- /dev/null
+++ b/lua/extregs/file.lua
@@ -0,0 +1,26 @@
+local util = require('extregs.util')
+local storage = require('extregs.storage')
+local M = {}
+
+function M.get_provider(cfg)
+  local function path() return util.path(cfg.path) end
+  return {
+    read = function()
+      local fd, err = io.open(path(), 'rb')
+      if not fd then error('cannot read ' .. path() .. ': ' .. tostring(err)) end
+      local data = fd:read('*a')
+      fd:close()
+      return storage.decode(data)
+    end,
+    write = function(_, reg)
+      local dest = path()
+      vim.fn.mkdir(vim.fn.fnamemodify(dest, ':h'), 'p')
+      local fd, err = io.open(dest, 'wb')
+      if not fd then error('cannot write ' .. dest .. ': ' .. tostring(err)) end
+      assert(fd:write(storage.encode(reg)))
+      assert(fd:close())
+      return true
+    end,
+  }
+end
+return M
diff --git a/lua/extregs/file_basename.lua b/lua/extregs/file_basename.lua
new file mode 100644
index 0000000..de59fea
--- /dev/null
+++ b/lua/extregs/file_basename.lua
@@ -0,0 +1,8 @@
+local util = require('extregs.util')
+local M = {}
+function M.get_provider(_)
+  return util.readonly(function()
+    return vim.fn.fnamemodify(vim.api.nvim_buf_get_name(0), ':t')
+  end)
+end
+return M
diff --git a/lua/extregs/file_dirname.lua b/lua/extregs/file_dirname.lua
new file mode 100644
index 0000000..8ba5e77
--- /dev/null
+++ b/lua/extregs/file_dirname.lua
@@ -0,0 +1,8 @@
+local util = require('extregs.util')
+local M = {}
+function M.get_provider(_)
+  return util.readonly(function()
+    return vim.fn.fnamemodify(vim.api.nvim_buf_get_name(0), ':h')
+  end)
+end
+return M
diff --git a/lua/extregs/file_stem.lua b/lua/extregs/file_stem.lua
new file mode 100644
index 0000000..49ebc24
--- /dev/null
+++ b/lua/extregs/file_stem.lua
@@ -0,0 +1,8 @@
+local util = require('extregs.util')
+local M = {}
+function M.get_provider(_)
+  return util.readonly(function()
+    return vim.fn.fnamemodify(vim.fn.fnamemodify(vim.api.nvim_buf_get_name(0), ':t'), ':r')
+  end)
+end
+return M
diff --git a/lua/extregs/init.lua b/lua/extregs/init.lua
new file mode 100644
index 0000000..b9dcbe1
--- /dev/null
+++ b/lua/extregs/init.lua
@@ -0,0 +1,72 @@
+local M = {}
+
+M.defaults = {
+  file = { path = (vim.env.XDG_RUNTIME_DIR or vim.fn.stdpath('state')) .. '/nvim.regfile' },
+  ssh = { host = nil, path = '~/.local/state/nvim/regfile' },
+  osc52 = { register = '+' },
+  tmux = { buffer = nil },
+  timestamp = { format = '%Y-%m-%dT%H:%M:%S%z' },
+}
+
+-- name, module, configuration section; registration semantics are (write, read).
+local providers = {
+  { 'osc52', 'osc52', 'osc52' },
+  { 'tmux', 'tmux', 'tmux' },
+  { 'buffer', 'buffer' },
+  { 'file.basename', 'file_basename' },
+  { 'file.stem', 'file_stem' },
+  { 'file.dirname', 'file_dirname' },
+  { 'file.backed', 'file', 'file' },
+  { 'ssh.backed', 'ssh', 'ssh' },
+  { 'timestamp', 'timestamp', 'timestamp' },
+  { 'uuid', 'uuid' },
+}
+
+local function notify(msg, level)
+  vim.schedule(function()
+    vim.notify('extregs.nvim: ' .. msg, level or vim.log.levels.ERROR)
+  end)
+end
+
+local function guard(name, fn)
+  return function(...)
+    local ok, result = pcall(fn, ...)
+    if not ok then
+      notify(name .. ': ' .. tostring(result))
+      return false
+    end
+    return result
+  end
+end
+
+local function register(name, provider)
+  assert(type(provider.read) == 'function', name .. ': missing read()')
+  assert(type(provider.write) == 'function', name .. ': missing write()')
+  vim._extregs.register(name,
+    guard(name, function(reg)
+      if provider.readonly then
+        notify(name .. ' is read-only', vim.log.levels.WARN)
+        return false
+      end
+      return provider:write(reg)
+    end),
+    guard(name, function() return provider:read() end))
+end
+
+function M.setup(opts)
+  if vim.fn.has('extregs') ~= 1 or not (vim._extregs and vim._extregs.register) then
+    notify("requires a Neovim build with has('extregs') and vim._extregs.register")
+    return false
+  end
+  if M._initialized then return true end
+  local cfg = vim.tbl_deep_extend('force', vim.deepcopy(M.defaults), opts or {})
+  for _, spec in ipairs(providers) do
+    local name, module, section = unpack(spec)
+    local provider = require('extregs.' .. module).get_provider(cfg[section] or {})
+    register(name, provider)
+  end
+  M._initialized = true
+  return true
+end
+
+return M
diff --git a/lua/extregs/osc52.lua b/lua/extregs/osc52.lua
new file mode 100644
index 0000000..92fa12d
--- /dev/null
+++ b/lua/extregs/osc52.lua
@@ -0,0 +1,15 @@
+local M = {}
+function M.get_provider(cfg)
+  local clipboard = require('vim.ui.clipboard.osc52')
+  return {
+    read = function()
+      return false
+    end,
+    write = function(_, reg)
+      clipboard.copy(cfg.register)(reg.lines)
+      return true
+    end,
+  }
+end
+
+return M
diff --git a/lua/extregs/ssh.lua b/lua/extregs/ssh.lua
new file mode 100644
index 0000000..8102434
--- /dev/null
+++ b/lua/extregs/ssh.lua
@@ -0,0 +1,31 @@
+local util = require('extregs.util')
+local storage = require('extregs.storage')
+local M = {}
+
+local function remote_path(path)
+  -- Expand ~/ on the remote system; quote everything else.
+  if path:sub(1, 2) == '~/' then
+    return '"$HOME"/' .. util.shell_quote(path:sub(3))
+  end
+  return util.shell_quote(path)
+end
+
+function M.get_provider(cfg)
+  assert(type(cfg.path) == 'string' and cfg.path ~= '', 'ssh.path required')
+  local path = remote_path(cfg.path)
+  local function exec(script, input)
+    assert(type(cfg.host) == 'string' and cfg.host:match('^[%w_.%-]+@[%w_.%-]+$'),
+      'ssh.host must be user@host')
+    -- SSH joins remote argv with spaces: quote the *whole* sh -c argument.
+    return util.system({ 'ssh', '-o', 'BatchMode=yes', cfg.host,
+      'sh -c ' .. util.shell_quote(script) }, input, 'ssh')
+  end
+  return {
+    read = function() return storage.decode(exec('cat ' .. path)) end,
+    write = function(_, reg)
+      exec('mkdir -p "$(dirname ' .. path .. ')" && cat > ' .. path, storage.encode(reg))
+      return true
+    end,
+  }
+end
+return M
diff --git a/lua/extregs/storage.lua b/lua/extregs/storage.lua
new file mode 100644
index 0000000..c532485
--- /dev/null
+++ b/lua/extregs/storage.lua
@@ -0,0 +1,24 @@
+local M = {}
+M.cookie = 'NVIM_EXTREGS_JSON_V1'
+
+local function validate(reg)
+  if type(reg) ~= 'table' or type(reg.lines) ~= 'table' or type(reg.type) ~= 'string' then
+    error('invalid yankreg: expected { lines = {...}, type = "v"|"V"|blockwise }')
+  end
+  return reg
+end
+
+function M.encode(reg)
+  return M.cookie .. '\n' .. vim.json.encode(validate(reg)) .. '\n'
+end
+
+function M.decode(data)
+  if data:sub(1, #M.cookie + 1) == M.cookie .. '\n' then
+    return validate(vim.json.decode(data:sub(#M.cookie + 2)))
+  end
+  local lines = vim.split(data, '\n', { plain = true })
+  if #lines > 1 and lines[#lines] == '' then table.remove(lines) end
+  return { lines = lines, type = #lines > 1 and 'V' or 'v' }
+end
+
+return M
diff --git a/lua/extregs/timestamp.lua b/lua/extregs/timestamp.lua
new file mode 100644
index 0000000..6405285
--- /dev/null
+++ b/lua/extregs/timestamp.lua
@@ -0,0 +1,6 @@
+local util = require('extregs.util')
+local M = {}
+function M.get_provider(cfg)
+  return util.readonly(function() return os.date(cfg.format) end)
+end
+return M
diff --git a/lua/extregs/tmux.lua b/lua/extregs/tmux.lua
new file mode 100644
index 0000000..ea0299b
--- /dev/null
+++ b/lua/extregs/tmux.lua
@@ -0,0 +1,24 @@
+local util = require('extregs.util')
+local storage = require('extregs.storage')
+local M = {}
+function M.get_provider(cfg)
+  local function run(cmd, input)
+    local argv = { 'tmux', cmd }
+    if cfg.buffer ~= nil then
+      assert(type(cfg.buffer) == 'string' and cfg.buffer ~= '', 'tmux.buffer must be a nonempty string')
+      vim.list_extend(argv, { '-b', cfg.buffer })
+    end
+    if cmd == 'load-buffer' then argv[#argv + 1] = '-' end
+    return util.system(argv, input, 'tmux')
+  end
+  return {
+    read = function() return storage.decode(run('show-buffer')) end,
+    write = function(_, reg)
+      local data = table.concat(reg.lines, '\n')
+      if reg.type == 'V' then data = data .. '\n' end
+      run('load-buffer', data)
+      return true
+    end,
+  }
+end
+return M
diff --git a/lua/extregs/util.lua b/lua/extregs/util.lua
new file mode 100644
index 0000000..6dd1f00
--- /dev/null
+++ b/lua/extregs/util.lua
@@ -0,0 +1,31 @@
+local M = {}
+
+function M.textreg(s)
+  return { lines = { tostring(s) }, type = 'v' }
+end
+
+function M.readonly(getter)
+  return {
+    read = function() return M.textreg(getter()) end,
+    write = function() return false end,
+    readonly = true,
+  }
+end
+
+function M.path(path)
+  return vim.fn.expand(path)
+end
+
+function M.shell_quote(s)
+  return "'" .. s:gsub("'", "'\\''") .. "'"
+end
+
+function M.system(argv, input, label)
+  local result = vim.system(argv, { text = false, stdin = input }):wait()
+  if result.code ~= 0 then
+    error((label or argv[1]) .. ': ' .. (result.stderr or ('exit status ' .. result.code)))
+  end
+  return result.stdout or ''
+end
+
+return M
diff --git a/lua/extregs/uuid.lua b/lua/extregs/uuid.lua
new file mode 100644
index 0000000..e23a37f
--- /dev/null
+++ b/lua/extregs/uuid.lua
@@ -0,0 +1,16 @@
+local util = require('extregs.util')
+local M = {}
+function M.get_provider(_)
+  return util.readonly(function()
+    local bytes = vim.uv.random(16)
+    assert(bytes and #bytes == 16, 'unable to obtain random bytes')
+    local b = { bytes:byte(1, 16) }
+    b[7] = (b[7] % 16) + 64
+    b[9] = (b[9] % 64) + 128
+    local h = {}
+    for i = 1, 16 do h[i] = string.format('%%02x', b[i]) end
+    return table.concat(h, '', 1, 4) .. '-' .. table.concat(h, '', 5, 6) .. '-' ..
+      table.concat(h, '', 7, 8) .. '-' .. table.concat(h, '', 9, 10) .. '-' .. table.concat(h, '', 11, 16)
+  end)
+end
+return M