61 lines
1.9 KiB
TypeScript
61 lines
1.9 KiB
TypeScript
import { field, logger } from "@coder/logger"
|
|
import * as fs from "fs"
|
|
import * as path from "path"
|
|
import * as util from "util"
|
|
import { Args } from "./cli"
|
|
import { HttpServer } from "./http"
|
|
|
|
/* eslint-disable @typescript-eslint/no-var-requires */
|
|
|
|
export type Activate = (httpServer: HttpServer, args: Args) => void
|
|
|
|
export interface Plugin {
|
|
activate: Activate
|
|
}
|
|
|
|
/**
|
|
* Intercept imports so we can inject code-server when the plugin tries to
|
|
* import it.
|
|
*/
|
|
const originalLoad = require("module")._load
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
require("module")._load = function (request: string, parent: object, isMain: boolean): any {
|
|
return originalLoad.apply(this, [request.replace(/^code-server/, path.resolve(__dirname, "../..")), parent, isMain])
|
|
}
|
|
|
|
const loadPlugin = async (pluginPath: string, httpServer: HttpServer, args: Args): Promise<void> => {
|
|
try {
|
|
const plugin: Plugin = require(pluginPath)
|
|
plugin.activate(httpServer, args)
|
|
logger.debug("Loaded plugin", field("name", path.basename(pluginPath)))
|
|
} catch (error) {
|
|
if (error.code !== "MODULE_NOT_FOUND") {
|
|
logger.warn(error.message)
|
|
} else {
|
|
logger.error(error.message)
|
|
}
|
|
}
|
|
}
|
|
|
|
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> => {
|
|
try {
|
|
await _loadPlugins(httpServer, args)
|
|
} catch (error) {
|
|
if (error.code !== "ENOENT") {
|
|
logger.warn(error.message)
|
|
}
|
|
}
|
|
|
|
if (process.env.PLUGIN_DIR) {
|
|
await loadPlugin(process.env.PLUGIN_DIR, httpServer, args)
|
|
}
|
|
}
|