automatically map ports via scanning

Author: mikew

src/client/main.go

@@ -122,6 +122,8 @@ var CliClientOpenCommand = cli.Command{
 
 			SshPath: sshPath,
 			Debug:   isDebug,
+
+			TunneledPorts: make(map[string]bool),
 		}
 
 		if nvrhContext.SshPath == "internal" {
@@ -466,6 +468,12 @@ func BuildClientNvimCmd(nvrhContext *context.NvrhContext) *exec.Cmd {
 
 func prepareRemoteNvim(nvrhContext *context.NvrhContext, nv *nvim.Nvim) error {
 	nv.RegisterHandler("tunnel-port", func(v *nvim.Nvim, args []string) {
+		if _, ok := nvrhContext.TunneledPorts[args[0]]; ok {
+			return
+		}
+
+		nvrhContext.TunneledPorts[args[0]] = true
+
 		go nvrhContext.SshClient.TunnelSocket(&ssh_tunnel_info.SshTunnelInfo{
 			Mode:         "port",
 			LocalSocket:  args[0],
@@ -533,6 +541,72 @@ os.execute('chmod +x ' .. browser_script_path)
 return true
 	`, nil, nvrhContext.BrowserScriptPath, nvrhContext.RemoteSocketOrPort(), nv.ChannelID())
 
+	batch.ExecLua(`
+local nvrh_port_scanner = {
+  active_watchers = {},
+
+  mapped_ports = {},
+
+  patterns = {
+    -- "port 3000"
+    "port%s+(%d+)",
+    -- "localhost:3000"
+    "localhost:(%d+)",
+    -- "0.0.0.0:3000"
+    "%d+%.%d+%.%d+%.%d+:(%d+)",
+    -- ":3000" at start of line
+    "^:(%d+)",
+    -- ":3000" but avoid eslint errors (error in foo.tsx:3)
+    "%s+:(%d+)",
+  },
+}
+
+function nvrh_port_scanner.attach_port_watcher(bufnr)
+  if vim.bo[bufnr].buftype ~= "terminal" then
+    return
+  end
+
+  -- Already watching?
+  if nvrh_port_scanner.active_watchers[bufnr] then
+    return
+  end
+
+  local function on_lines(_, _, _, lastline, new_lastline, _)
+    local lines = vim.api.nvim_buf_get_lines(bufnr, lastline, new_lastline, false)
+
+    for _, line in ipairs(lines) do
+      for _, pattern in ipairs(nvrh_port_scanner.patterns) do
+        local port = string.match(line, pattern)
+        if port then
+          if nvrh_port_scanner.mapped_ports[port] then
+            -- Already notified about this port
+          else
+						nvrh_port_scanner.mapped_ports[port] = true
+            vim.rpcnotify(tonumber(os.getenv("NVRH_CHANNEL_ID")), "tunnel-port", { port })
+          end
+          break
+        end
+      end
+    end
+  end
+
+  local detach = vim.api.nvim_buf_attach(bufnr, false, {
+    on_lines = on_lines,
+  })
+
+  nvrh_port_scanner.active_watchers[bufnr] = detach
+end
+
+-- Attach watcher on TermOpen
+vim.api.nvim_create_autocmd("TermOpen", {
+  callback = function(args)
+    nvrh_port_scanner.attach_port_watcher(args.buf)
+  end,
+})
+
+return true
+	`, nil, nil)
+
 	if err := batch.Execute(); err != nil {
 		return err
 	}

src/context/main.go

@@ -30,6 +30,8 @@ type NvrhContext struct {
 	Debug   bool
 
 	SshClient nvrh_base_ssh.BaseNvrhSshClient
+
+	TunneledPorts map[string]bool
 }
 
 func (nc *NvrhContext) LocalSocketOrPort() string {