nrepl.lua

-- luacheck: globals unpack vim
local impromptu = require("impromptu")
local connections = require('acid.connections')
local nrepl = require('acid.nrepl')
local acid_utils = require('acid.utils')
local acid = require("acid")
local eval = require("acid.ops").eval

local find_value = function(tbl, val)
  for ix, v in ipairs(tbl) do
    if v == val then
      return ix
    end
  end
  return -1
end

local jazz_nrepl = {}

local select_portno = function(handler)
  return impromptu.new.form{
    title = "🎵 Select port number:",
    questions = {portno = {description = "Port number"}},
    handler = function(session, result)
      return handler(session, result)
    end
  }
end

local existing = function(obj)
  return select_portno(function(_, handler)
    local ix = connections.add{"127.0.0.1", tonumber(handler.portno)}
    connections.select(obj.pwd, ix)
    return true
  end)
end

local run_alias = function(obj)
  local options = {
    start = {
      description = "Start",
      hl = "Function"
    }
  }

  local spawn_options = {
    pwd = obj.pwd,
    alias = {}
  }

  local new_session = obj.session:stack(impromptu.new.ask{
    title = "🎵 From deps.edn",
    options = options,
    handler = function(session, selected)
      if selected.tp == "alias" then
        table.insert(spawn_options.alias, selected.description)
        session.lines[selected.description].hl = "String"
      else
        nrepl.start(spawn_options)
        return true
      end
    end
  })
  local admin = acid.admin_session()
  acid.run(eval{
      code = '(map println (keys (:aliases (clojure.edn/read-string (slurp "' ..
      obj.pwd .. "/" ..obj.file ..
    '")))))'}:with_handler(function(data)
      if data.status or data.value then
        return
      end

      for _, v in ipairs(acid_utils.split_lines(data.out)) do
        options[v] = {
          description = v,
          tp = "alias",
          hl = "Comment"
        }
      end

      new_session:render()
    end),
    admin)

  return false
end

local toolsdeps = function(obj)
  local files = vim.api.nvim_call_function("expand", {"**/*.edn", true, true})

  if #files == 1 and files[1] == "deps.edn" then
    local admin = acid.admin_session()
    if admin == nil then
      nrepl.start{pwd = obj.pwd}
      return true
    else
        return run_alias{file = files[1], session = obj.session, pwd = obj.pwd}
    end
  end

  local options = {}

  for _, v in ipairs(files) do
    options[v] = {
      description = v,
      hl = "Function"
    }
  end

  obj.session:stack(impromptu.new.ask{
      title = "🎵 Select deps.edn file",
      options = options,
      handler = function(session, selected)
        return run_alias{file = selected.description, session = session, pwd = obj.pwd}
      end
    })

  return false
end

local connect_nrepl = function(obj)
  return impromptu.new.form{
    title = "🎵 Connect nrepl to:",
    questions = {
      portno = {description = "Port number"},
      host = {description = "Host address"}
    },
    handler = function(session, result)
      obj.port = tonumber(result.portno)
      obj.host = result.host
      obj.connect = true

      session:pop()

      session.lines.connect.description = "Connect to remote nrepl? (true)"
      session.lines.connect.hl = "String"

      session.lines.host.description = "Connect to address(" .. result.host .. ")"
      session.lines.host.hl = "String"

      session.lines.port.description = "Port number (" .. result.portno .. ")"
      session.lines.port.hl = "String"
      return false
    end
  }
end

local custom_nrepl = function(obj)
  local opts = {}
  local nrepl_config = {middlewares = nrepl.default_middlewares, pwd = obj.pwd}

  for k, _ in pairs(nrepl.middlewares) do
    opts[k] = {
      description = k,
      hl = "Comment"
    }
  end

  for _, k in ipairs(nrepl.default_middlewares) do
    opts[k].hl = "String"
  end

  opts.port = {
    description = "Port number (auto)",
    hl = "Comment"
  }

  --opts.bind = {
  --description = "Bind to address (127.0.0.1)",
  --hl = "Comment"
  --}

  --opts.host = {
  --description = "Connect to address (127.0.0.1)",
  --hl = "Comment"
  --}

  --opts.connect = {
  --description = "Connect to remote nrepl? (false)",
  --hl = "Comment"
  --}

  --opts.pwd = {
  --description = "Directory (" .. obj.pwd .. ")",
  --hl = "Comment"
  --}

  opts.start = {description = "Start with custom configuration"}
  opts.abort = {
    description = "Abort",
    key = "q"
  }


  return impromptu.new.ask{
    title = "🎵 Configure the custom nrepl:",
    quitable = false,
    options = opts,
    handler = function(session, selected)
      if selected.index == "start" then
        nrepl.start(nrepl_config)
        return true
      elseif selected.index == "port" then
        session:stack(select_portno(function(ss, result)
          ss:pop()
          ss.lines.port.description = "Port number (" .. result.portno .. ")"
          ss.lines.port.hl = "String"
          return false
        end))
      elseif selected.index == "connect" then
        session:stack(connect_nrepl(nrepl_config))
      elseif selected.index == "abort" then
        session:pop()
      else
        local ix = find_value(nrepl_config.middlewares, selected.index)
        if ix == -1 then
          table.insert(nrepl_config.middlewares, selected.index)
          selected.hl = "String"
        else
          table.remove(nrepl_config.middlewares, ix)
          selected.hl = "Comment"
        end
      end
    end
  }
end


jazz_nrepl.nrepl_menu = function(pwd)
  pwd = pwd or vim.api.nvim_call_function("getcwd", {})
  if not acid_utils.ends_with(pwd, "/") then
    pwd = pwd .. "/"
  end

  local current = require('acid.connections').current[pwd]
  local opts = {}

  local check

  if current ~= nil then
    current = connections.store[current]
    check = function (v)
      return v[2] == current[2] and v[1] == current[1]
    end
  else
    check = function(_) return false end
  end

  for ix, v in pairs(connections.store) do
    if #v == 2 then
      local opt = {}
      local str = "nrepl://" .. v[1] .. ":" .. v[2]

      if check(v) then
        opts.close = {
          description = "Close connection to " .. str ,
          index = ix,
          hl = "Function"
        }

        opts.refresh = {
          description = "Refresh connections",
          index = ix,
          hl = "Function"
        }

        str = str .. " (current)"
        opt.hl = "String"
      end

      opt.description = str
      opt.index = ix
      opts["conn" .. ix] = opt
    end
  end

  opts.new = {
    description = "Spawn new nrepl",
    hl = "Function"
  }

  opts.custom = {
    description = "Spawn custom nrepl",
    hl = "Function"
  }

  opts.existing = {
    description = "Connect to existing nrepl",
    hl = "Function"
  }

  opts.toolsdeps = {
    description = "Spawn from deps.edn",
    hl = "Function"
  }

  impromptu.ask{
    title = "🎵 Select nrepl to connect to:",
    options = opts,
    handler = function(session, selected)
      if selected.index == "new" then
        nrepl.start{pwd = pwd}
      elseif selected.index == "close" then
        nrepl.stop{pwd = pwd}
      elseif selected.index == "refresh" then
        nrepl.stop{pwd = pwd}
        nrepl.start{pwd = pwd}
      elseif selected.index == "toolsdeps" then
        return toolsdeps{pwd = pwd, session = session}
      elseif selected.index == "existing" then
        session:stack(existing{pwd = pwd})
        return false
      elseif selected.index == "custom" then
        session:stack(custom_nrepl{pwd = pwd})
        return false
      else
        connections.select(pwd, selected.index)
      end
      return true
    end
  }
end

return jazz_nrepl