Add local route handler.

Author: GirlBossRush

src/node/cli.ts

@@ -53,6 +53,7 @@ export interface Args extends VsArgs {
   "new-window"?: boolean
 
   link?: OptionalString
+  "local-directory"?: OptionalString
 }
 
 interface Option<T> {
@@ -179,6 +180,10 @@ const options: Options<Required<Args>> = {
     `,
     beta: true,
   },
+  "local-directory": {
+    type: OptionalString,
+    description: "Serve files from a given directory",
+  },
 }
 
 export const optionDescriptions = (): string[] => {

src/node/routes/index.ts

@@ -12,7 +12,7 @@ import { plural } from "../../common/util"
 import { AuthType, DefaultedArgs } from "../cli"
 import { rootPath } from "../constants"
 import { Heart } from "../heart"
-import { commonTemplateVars } from "../http"
+import { commonTemplateVars, ensureAuthenticated } from "../http"
 import { loadPlugins } from "../plugin"
 import * as domainProxy from "../proxy"
 import { getMediaMime, paths } from "../util"
@@ -118,6 +118,16 @@ export const register = async (
 
   app.use("/", _static.router)
 
+  if (args["local-directory"]?.value) {
+    const directoryPath = path.resolve(args["local-directory"].value)
+    logger.info(`Serving files from local directory “${directoryPath}” at “/local”`)
+
+    const localDirectoryRouter = express.Router()
+    _static.createServeDirectoryHandler(localDirectoryRouter, "/local", directoryPath)
+
+    app.use(ensureAuthenticated, localDirectoryRouter)
+  }
+
   app.use("/update", update.router)
 
   await loadPlugins(app, args)

src/node/routes/static.ts

@@ -1,20 +1,34 @@
-import { Router, static as createStaticRouteHandler } from "express"
+import { Router, static as serveStatic } from "express"
 import { resolve } from "path"
+import { ServeStaticOptions } from "serve-static"
 import { commit, rootPath } from "../constants"
 
 export const router = Router()
 
+export const createDefaultStaticRouteHandlerOptions = (): ServeStaticOptions => ({
+  cacheControl: commit !== "development",
+})
+
+export const createServeDirectoryHandler = (
+  router: Router,
+  publicRoute: string,
+  directoryPath: string,
+  options: ServeStaticOptions = {},
+): void => {
+  router.use(
+    publicRoute,
+    serveStatic(directoryPath, {
+      ...createDefaultStaticRouteHandlerOptions(),
+      ...options,
+    }),
+  )
+}
+
 const staticPaths: Record<string, string> = {
   "/static": "dist",
   "/static/media": "src/browser/public/media",
 }
 
 for (const [publicRoute, sourcePath] of Object.entries(staticPaths)) {
-  router.use(
-    publicRoute,
-    createStaticRouteHandler(resolve(rootPath, sourcePath), {
-      index: false,
-      cacheControl: commit !== "development",
-    }),
-  )
+  createServeDirectoryHandler(router, publicRoute, resolve(rootPath, sourcePath), { index: false })
 }

src/node/routes/vscode.ts

@@ -1,6 +1,6 @@
 import { field, logger } from "@coder/logger"
 import * as crypto from "crypto"
-import { RequestHandler, Router, static as createStaticRouteHandler } from "express"
+import { RequestHandler, Router } from "express"
 import { promises as fs } from "fs"
 import * as path from "path"
 import { resolve } from "path"
@@ -15,6 +15,7 @@ import { authenticated, commonTemplateVars, ensureAuthenticated, redirect } from
 import { getMediaMime, pathToFsPath } from "../util"
 import { VscodeProvider } from "../vscode"
 import { Router as WsRouter } from "../wsRouter"
+import { createServeDirectoryHandler } from "./static"
 
 export const router = Router()
 
@@ -103,13 +104,8 @@ wsRouter.ws("/", ensureAuthenticated, async (req) => {
   )
   await vscode.sendWebsocket(req.ws, req.query)
 })
-router.use(
-  "/lib",
-  createStaticRouteHandler(resolve(rootPath, "lib"), {
-    index: false,
-    cacheControl: commit !== "development",
-  }),
-)
+
+createServeDirectoryHandler(router, "/lib", resolve(rootPath, "lib"))
 
 interface TarHandlerQueryParams extends ParsedQs {
   filePath?: string | string[]