implements extension/runTask (#261)

Author: jmgomez

nimlangserver.nim

@@ -65,7 +65,7 @@ proc registerRoutes(srv: RpcSocketServer, ls: LanguageServer) =
   srv.register("extension/capabilities", wrapRpc(partial(extensionCapabilities, ls)))
   srv.register("extension/suggest", wrapRpc(partial(extensionSuggest, ls)))
   srv.register("extension/tasks", wrapRpc(partial(tasks, ls)))
-
+  srv.register("extension/runTask", wrapRpc(partial(runTask, ls)))
   #Notifications
   srv.register("$/cancelRequest", wrapRpc(partial(cancelRequest, ls)))
   srv.register("initialized", wrapRpc(partial(initialized, ls)))

protocol/types.nim

@@ -1034,3 +1034,10 @@ type
   NimbleTask* = object
     name*: string 
     description*: string
+
+  RunTaskParams* = object
+    command*: seq[string] #command and args
+  
+  RunTaskResult* = object
+    command*: seq[string] #command and args
+    output*: seq[string] #output lines

routes.nim

@@ -17,7 +17,7 @@ import
   chronicles,
   asyncprocmonitor,
   json_serialization,
-  std/[strscans, times, json, parseutils],
+  std/[strscans, times, json, parseutils, strutils],
   ls,
   stew/[byteutils]
 
@@ -753,21 +753,23 @@ proc exit*(
   result = newJNull()
   await p.onExit()
 
-proc tasks*(
-    ls: LanguageServer, conf: JsonNode
-): Future[seq[NimbleTask]] {.async, gcsafe.} =
-  let rootPath: string = ls.initializeParams.getRootPath
-
-  debug "Received tasks ", rootPath = rootPath
-  delEnv "NIMBLE_DIR"
-  let process = await startProcess(
+proc startNimbleProcess(ls: LanguageServer, args: seq[string]): Future[AsyncProcessRef] {.async.} =
+  await startProcess(
     "nimble",
-    arguments = @["tasks"],
+    arguments = args,
     options = {UsePath},
-    workingDir = rootPath,
+    workingDir = ls.initializeParams.getRootPath,
     stdoutHandle = AsyncProcess.Pipe,
     stderrHandle = AsyncProcess.Pipe,
   )
+
+proc tasks*(
+    ls: LanguageServer, conf: JsonNode
+): Future[seq[NimbleTask]] {.async.} =
+  let rootPath: string = ls.initializeParams.getRootPath
+  debug "Received tasks ", rootPath = rootPath
+  delEnv "NIMBLE_DIR"
+  let process = await ls.startNimbleProcess(@["tasks"])
   let res =
     await process.waitForExit(InfiniteDuration) #TODO handle error (i.e. no nimble file)
   let output = await process.stdoutStream.readLine()
@@ -778,6 +780,23 @@ proc tasks*(
       if name.isWord:
         result.add NimbleTask(name: name.strip(), description: desc.strip())
 
+proc runTask*(
+    ls: LanguageServer, params: RunTaskParams
+): Future[RunTaskResult] {.async.} =
+  let process = await ls.startNimbleProcess(params.command) 
+  let res = await process.waitForExit(InfiniteDuration)
+  result.command = params.command
+  let prefix = "\""
+  while not process.stdoutStream.atEof():
+    var lines = process.stdoutStream.readLine().await.splitLines
+    for line in lines.mitems:
+      if line.startsWith(prefix):
+        line = line.unescape(prefix)
+      if line != "":  
+        result.output.add line  
+        
+  debug "Ran nimble cmd/task", command = $params.command, output = $result.output
+
 #Notifications
 proc initialized*(ls: LanguageServer, _: JsonNode): Future[void] {.async.} =
   debug "Client initialized."

tests/projects/tasks/tasks.nimble

@@ -14,7 +14,7 @@ bin           = @["tasks"]
 requires "nim >= 2.1.99"
 
 task helloWorld, "hello world":
-  echo "helo world"
+  echo "hello world"
 
 task anotherTask, "Another task":
   echo "another task"