From 29dff9d21baf740c35ec42cbf5da2f9df04a0a71 Mon Sep 17 00:00:00 2001
From: Mitchell Hanberg <mitch@mitchellhanberg.com>
Date: Mon, 26 Jun 2023 23:21:57 -0400
Subject: [PATCH] fix: cancel current progress messages when changing/saving
 file

Fixes #40
---
 lib/next_ls.ex | 31 +++++++++++++++++++++----------
 1 file changed, 21 insertions(+), 10 deletions(-)

diff --git a/lib/next_ls.ex b/lib/next_ls.ex
index b81c5ed4..5dc9072b 100644
--- a/lib/next_ls.ex
+++ b/lib/next_ls.ex
@@ -252,6 +252,11 @@ defmodule NextLS do
         },
         %{assigns: %{ready: true}} = lsp
       ) do
+    for task <- Task.Supervisor.children(lsp.assigns.task_supervisor),
+        task != lsp.assigns.runtime_task.pid do
+      Process.exit(task, :kill)
+    end
+
     token = token()
 
     progress_start(lsp, token, "Compiling...")
@@ -339,15 +344,7 @@ defmodule NextLS do
     Process.demonitor(ref, [:flush])
     {{token, msg}, refs} = Map.pop(refs, ref)
 
-    GenLSP.notify(lsp, %GenLSP.Notifications.DollarProgress{
-      params: %GenLSP.Structures.ProgressParams{
-        token: token,
-        value: %WorkDoneProgressEnd{
-          kind: "end",
-          message: msg
-        }
-      }
-    })
+    progress_end(lsp, token, msg)
 
     lsp =
       case resp do
@@ -371,7 +368,9 @@ defmodule NextLS do
 
   def handle_info({:DOWN, ref, :process, _pid, _reason}, %{assigns: %{refresh_refs: refs}} = lsp)
       when is_map_key(refs, ref) do
-    {_token, refs} = Map.pop(refs, ref)
+    {{token, _}, refs} = Map.pop(refs, ref)
+
+    progress_end(lsp, token)
 
     {:noreply, assign(lsp, refresh_refs: refs)}
   end
@@ -425,6 +424,18 @@ defmodule NextLS do
     })
   end
 
+  defp progress_end(lsp, token, msg \\ nil) do
+    GenLSP.notify(lsp, %GenLSP.Notifications.DollarProgress{
+      params: %GenLSP.Structures.ProgressParams{
+        token: token,
+        value: %WorkDoneProgressEnd{
+          kind: "end",
+          message: msg
+        }
+      }
+    })
+  end
+
   defp token() do
     8
     |> :crypto.strong_rand_bytes()