Add variables to better customize plugin directories

Author: code-asher

src/node/plugin.ts

@@ -9,6 +9,9 @@ import { HttpServer } from "./http"
 
 export type Activate = (httpServer: HttpServer, args: Args) => void
 
+/**
+ * Plugins must implement this interface.
+ */
 export interface Plugin {
   activate: Activate
 }
@@ -23,6 +26,9 @@ require("module")._load = function (request: string, parent: object, isMain: boo
   return originalLoad.apply(this, [request.replace(/^code-server/, path.resolve(__dirname, "../..")), parent, isMain])
 }
 
+/**
+ * Load a plugin and run its activation function.
+ */
 const loadPlugin = async (pluginPath: string, httpServer: HttpServer, args: Args): Promise<void> => {
   try {
     const plugin: Plugin = require(pluginPath)
@@ -37,24 +43,34 @@ const loadPlugin = async (pluginPath: string, httpServer: HttpServer, args: Args
   }
 }
 
-const _loadPlugins = async (httpServer: HttpServer, args: Args): Promise<void> => {
-  const pluginPath = path.resolve(__dirname, "../../plugins")
-  const files = await util.promisify(fs.readdir)(pluginPath, {
-    withFileTypes: true,
-  })
-  await Promise.all(files.map((file) => loadPlugin(path.join(pluginPath, file.name), httpServer, args)))
-}
-
-export const loadPlugins = async (httpServer: HttpServer, args: Args): Promise<void> => {
+/**
+ * Load all plugins in the specified directory.
+ */
+const _loadPlugins = async (pluginDir: string, httpServer: HttpServer, args: Args): Promise<void> => {
   try {
-    await _loadPlugins(httpServer, args)
+    const files = await util.promisify(fs.readdir)(pluginDir, {
+      withFileTypes: true,
+    })
+    await Promise.all(files.map((file) => loadPlugin(path.join(pluginDir, file.name), httpServer, args)))
   } catch (error) {
     if (error.code !== "ENOENT") {
       logger.warn(error.message)
     }
   }
+}
+
+/**
+ * Load all plugins from the plugin directory. Defaults to `plugins` inside the
+ * code-server directory but can be set with `PLUGIN_DIR`.
+
+ * Also load any individual plugins found in `PLUGIN_DIRS` (colon-separated).
+ * This allows you to test and develop plugins without having to move them into
+ * one directory.
+ */
+export const loadPlugins = async (httpServer: HttpServer, args: Args): Promise<void> => {
+  await _loadPlugins(path.resolve(process.env.PLUGIN_DIR || path.join(__dirname, "../../plugins")), httpServer, args)
 
-  if (process.env.PLUGIN_DIR) {
-    await loadPlugin(process.env.PLUGIN_DIR, httpServer, args)
+  if (process.env.PLUGIN_DIRS) {
+    await Promise.all(process.env.PLUGIN_DIRS.split(":").map((dir) => loadPlugin(dir, httpServer, args)))
   }
 }