Nims backend adds nimble's pkgs2 dirs to the import paths.

Author: RowDaBoat

inimpkg/nimsBackend.nim

@@ -7,10 +7,24 @@ const errorPrefix = "Script Error: "
 type NimsBackend* = object
   addins: VMAddins
 
-proc nimsBackend*(): NimsBackend =
-  #exportTo(module, help)
-  let addins = implNimScriptModule(module)
-  NimsBackend(addins: addins)
+proc getNimblePackagesDir(): string =
+  var nimbleDir = getEnv("NIMBLE_DIR")
+
+  if nimbleDir.len == 0:
+    nimbleDir = getHomeDir() / ".nimble"
+
+  result = nimbleDir / "pkgs2"
+
+proc getImportPaths(): seq[string] =
+  let nimblePkgsDir = getNimblePackagesDir()
+  result = @[ getCurrentDir() ]
+
+  if not nimblePkgsDir.dirExists:
+    return result
+
+  for entry in walkDir(nimblePkgsDir, relative = false):
+    if entry.kind == pcDir:
+      result.add(entry.path)
 
 proc formatResult(exitCode: int, output: string): (string, int) =
   let hasFailed = exitCode != 0
@@ -21,19 +35,23 @@ proc formatResult(exitCode: int, output: string): (string, int) =
   else:
     (output, exitCode)
 
+proc nimsBackend*(): NimsBackend =
+  #exportTo(module, help)
+  let addins = implNimScriptModule(module)
+  NimsBackend(addins: addins)
+
 proc runCode*(self: NimsBackend, source: string): (string, int) =
   let tempFile = getTempDir() / "nimrepl_capture.txt"
   let oldStdout = stdout
   let outFile = open(tempFile, fmWrite)
-  stdout = outFile
+  let scriptPath = NimScriptPath(source)
+  let searchPaths = getImportPaths()
   var exitCode = 0
+  stdout = outFile
 
   try:
-    let interpreterOpt = loadScript(
-      NimScriptPath(source),
-      self.addins,
-      searchPaths = @[ getCurrentDir() ]
-    )
+    let interpreterOpt = loadScript(scriptPath, self.addins, searchPaths = searchPaths)
+
     if interpreterOpt.isNone:
       exitCode = 1
   except: