@eighty4/dank 0.0.5-0 → 0.0.5-2

package/lib/services.ts

@@ -1,114 +1,228 @@
-import { type ChildProcess, spawn } from 'node:child_process'
+import {
+    type ChildProcess,
+    type ChildProcessWithoutNullStreams,
+    execSync,
+    spawn,
+} from 'node:child_process'
+import EventEmitter from 'node:events'
 import { basename, isAbsolute, resolve } from 'node:path'
 import type { DevService, ResolvedDankConfig } from './config.ts'
 
-export type DevServices = {
-    http: HttpServices
-}
+export class ManagedServiceLabel {
+    #command: string
+    #cwd: string
+
+    constructor(spec: DevService) {
+        this.#command = spec.command
+        this.#cwd = !spec.cwd
+            ? './'
+            : spec.cwd.startsWith('/')
+              ? `/.../${basename(spec.cwd)}`
+              : spec.cwd.startsWith('.')
+                ? spec.cwd
+                : `./${spec.cwd}`
+    }
 
-export type HttpServices = {
-    running: Array<HttpService>
+    get command(): string {
+        return this.#command
+    }
+
+    get cwd(): string {
+        return this.#cwd
+    }
 }
 
 export type HttpService = NonNullable<DevService['http']>
 
-// up to date representation of dank.config.ts services
-const running: Array<{ s: DevService; process: ChildProcess | null }> = []
-
-let signal: AbortSignal
-
-// batch of services that must be stopped before starting new services
-let updating: null | {
-    stopping: Array<DevService>
-    starting: Array<DevService>
-} = null
-
-export function startDevServices(
-    services: ResolvedDankConfig['services'],
-    _signal: AbortSignal,
-): DevServices {
-    signal = _signal
-    if (services?.length) {
-        for (const s of services) {
-            running.push({ s, process: startService(s) })
-        }
+export type DevServiceEvents = {
+    error: [label: ManagedServiceLabel, cause: string]
+    exit: [label: ManagedServiceLabel, code: number | string]
+    launch: [label: ManagedServiceLabel]
+    stdout: [label: ManagedServiceLabel, output: Array<string>]
+    stderr: [label: ManagedServiceLabel, output: Array<string>]
+}
+
+class ManagedService extends EventEmitter<DevServiceEvents> {
+    #label: ManagedServiceLabel
+    #process: ChildProcess | null
+    #spec: DevService
+    // #status: ManagedServiceStatus = 'starting'
+
+    constructor(spec: DevService) {
+        super()
+        this.#label = new ManagedServiceLabel(spec)
+        this.#spec = spec
+        this.#process = this.#start()
     }
-    return {
-        http: {
-            get running(): Array<HttpService> {
-                return running.map(({ s }) => s.http).filter(http => !!http)
-            },
-        },
+
+    get spec(): DevService {
+        return this.#spec
     }
-}
 
-export function updateDevServices(services: ResolvedDankConfig['services']) {
-    if (!services?.length) {
-        if (running.length) {
-            if (updating === null) {
-                updating = { stopping: [], starting: [] }
-            }
-            running.forEach(({ s, process }) => {
-                if (process) {
-                    stopService(s, process)
-                } else {
-                    removeFromUpdating(s)
-                }
-            })
-            running.length = 0
-        }
-    } else {
-        if (updating === null) {
-            updating = { stopping: [], starting: [] }
-        }
-        const keep = []
-        const next: Array<DevService> = []
-        for (const s of services) {
-            let found = false
-            for (let i = 0; i < running.length; i++) {
-                const p = running[i].s
-                if (matchingConfig(s, p)) {
-                    found = true
-                    keep.push(i)
-                    break
-                }
-            }
-            if (!found) {
-                next.push(s)
-            }
-        }
-        for (let i = running.length - 1; i >= 0; i--) {
-            if (!keep.includes(i)) {
-                const { s, process } = running[i]
-                if (process) {
-                    stopService(s, process)
-                } else {
-                    removeFromUpdating(s)
-                }
-                running.splice(i, 1)
+    get httpSpec(): HttpService | undefined {
+        return this.#spec.http
+    }
+
+    matches(other: DevService): boolean {
+        return matchingConfig(this.#spec, other)
+    }
+
+    kill() {
+        if (this.#process) killProcess(this.#process)
+    }
+
+    #start(): ChildProcess {
+        const { path, args } = parseCommand(this.#spec.command)
+        const env = this.#spec.env
+            ? { ...process.env, ...this.#spec.env }
+            : undefined
+        const cwd =
+            !this.#spec.cwd || isAbsolute(this.#spec.cwd)
+                ? this.#spec.cwd
+                : resolve(process.cwd(), this.#spec.cwd)
+        const spawned = spawnProcess(path, args, env, cwd)
+        this.emit('launch', this.#label)
+        spawned.stdout.on('data', chunk =>
+            this.emit('stdout', this.#label, parseChunk(chunk)),
+        )
+        spawned.stderr.on('data', chunk =>
+            this.emit('stderr', this.#label, parseChunk(chunk)),
+        )
+        spawned.on('error', e => {
+            if (e.name === 'AbortError') {
+                return
             }
+            const cause =
+                'code' in e && e.code === 'ENOENT'
+                    ? 'program not found'
+                    : e.message
+            this.emit('error', this.#label, cause)
+        })
+        spawned.on('exit', (code, signal) =>
+            this.emit('exit', this.#label, code || signal!),
+        )
+        return spawned
+    }
+}
+
+type SpawnProcess = (
+    program: string,
+    args: Array<string>,
+    env: NodeJS.ProcessEnv | undefined,
+    cwd: string | undefined,
+) => ChildProcessWithoutNullStreams
+
+type KillProcess = (p: ChildProcess) => void
+
+const killProcess: KillProcess =
+    process.platform === 'win32'
+        ? p => execSync(`taskkill /pid ${p.pid} /T /F`)
+        : p => p.kill()
+
+const spawnProcess: SpawnProcess =
+    process.platform === 'win32'
+        ? (
+              path: string,
+              args: Array<string>,
+              env: NodeJS.ProcessEnv | undefined,
+              cwd: string | undefined,
+          ) =>
+              spawn('cmd', ['/c', path, ...args], {
+                  cwd,
+                  env,
+                  detached: false,
+                  shell: false,
+                  windowsHide: true,
+              })
+        : (
+              path: string,
+              args: Array<string>,
+              env: NodeJS.ProcessEnv | undefined,
+              cwd: string | undefined,
+          ) =>
+              spawn(path, args, {
+                  cwd,
+                  env,
+                  detached: false,
+                  shell: false,
+              })
+
+export class DevServices extends EventEmitter<DevServiceEvents> {
+    #running: Array<ManagedService>
+
+    constructor(services: ResolvedDankConfig['services']) {
+        super()
+        this.#running = services ? this.#start(services) : []
+        if (process.platform === 'win32') {
+            process.once('SIGINT', () => process.exit())
         }
-        if (updating.stopping.length) {
-            for (const s of next) {
-                if (
-                    !updating.starting.find(queued => matchingConfig(queued, s))
-                ) {
-                    updating.starting.push(s)
-                }
-            }
-        } else {
-            updating = null
-            for (const s of next) {
-                running.push({ s, process: startService(s) })
-            }
+        process.once('exit', this.shutdown)
+    }
+
+    get httpServices(): Array<HttpService> {
+        return this.#running.map(s => s.httpSpec).filter(http => !!http)
+    }
+
+    shutdown = () => {
+        this.#running.forEach(s => {
+            s.kill()
+            s.removeAllListeners()
+        })
+        this.#running.length = 0
+    }
+
+    update(services: ResolvedDankConfig['services']) {
+        if (!services?.length) {
+            this.shutdown()
+        } else if (
+            !matchingConfigs(
+                this.#running.map(s => s.spec),
+                services,
+            )
+        ) {
+            this.shutdown()
+            this.#running = this.#start(services)
         }
     }
+
+    #start(
+        services: NonNullable<ResolvedDankConfig['services']>,
+    ): Array<ManagedService> {
+        return services.map(spec => {
+            const service = new ManagedService(spec)
+            service.on('error', (label, cause) =>
+                this.emit('error', label, cause),
+            )
+            service.on('exit', (label, code) => this.emit('exit', label, code))
+            service.on('launch', label => this.emit('launch', label))
+            service.on('stdout', (label, output) =>
+                this.emit('stdout', label, output),
+            )
+            service.on('stderr', (label, output) =>
+                this.emit('stderr', label, output),
+            )
+            return service
+        })
+    }
 }
 
-function stopService(s: DevService, process: ChildProcess) {
-    opPrint(s, 'stopping')
-    updating!.stopping.push(s)
-    process.kill()
+function matchingConfigs(
+    a: Array<DevService>,
+    b: NonNullable<ResolvedDankConfig['services']>,
+): boolean {
+    if (a.length !== b.length) {
+        return false
+    }
+    const crossRef = [...a]
+    for (const toFind of b) {
+        const found = crossRef.findIndex(spec => matchingConfig(spec, toFind))
+        if (found === -1) {
+            return false
+        } else {
+            crossRef.splice(found, 1)
+        }
+    }
+    return true
 }
 
 function matchingConfig(a: DevService, b: DevService): boolean {
@@ -138,72 +252,6 @@ function matchingConfig(a: DevService, b: DevService): boolean {
     return true
 }
 
-function startService(s: DevService): ChildProcess {
-    opPrint(s, 'starting')
-    const splitCmdAndArgs = s.command.split(/\s+/)
-    const cmd = splitCmdAndArgs[0]
-    const args = splitCmdAndArgs.length === 1 ? [] : splitCmdAndArgs.slice(1)
-    const spawned = spawn(cmd, args, {
-        cwd: resolveCwd(s.cwd),
-        env: s.env,
-        signal,
-        detached: false,
-        shell: false,
-    })
-
-    const stdoutLabel = logLabel(s.cwd, cmd, args, 32)
-    spawned.stdout.on('data', chunk => printChunk(stdoutLabel, chunk))
-
-    const stderrLabel = logLabel(s.cwd, cmd, args, 31)
-    spawned.stderr.on('data', chunk => printChunk(stderrLabel, chunk))
-
-    spawned.on('error', e => {
-        if (e.name !== 'AbortError') {
-            const cause =
-                'code' in e && e.code === 'ENOENT'
-                    ? 'program not found'
-                    : e.message
-            opPrint(s, 'error: ' + cause)
-        }
-        removeFromRunning(s)
-    })
-
-    spawned.on('exit', () => {
-        opPrint(s, 'exited')
-        removeFromRunning(s)
-        removeFromUpdating(s)
-    })
-    return spawned
-}
-
-function removeFromRunning(s: DevService) {
-    for (let i = 0; i < running.length; i++) {
-        if (matchingConfig(running[i].s, s)) {
-            running.splice(i, 1)
-            return
-        }
-    }
-}
-
-function removeFromUpdating(s: DevService) {
-    if (updating !== null) {
-        for (let i = 0; i < updating.stopping.length; i++) {
-            if (matchingConfig(updating.stopping[i], s)) {
-                updating.stopping.splice(i, 1)
-                if (!updating.stopping.length) {
-                    updating.starting.forEach(startService)
-                    updating = null
-                    return
-                }
-            }
-        }
-    }
-}
-
-function printChunk(label: string, c: Buffer) {
-    for (const l of parseChunk(c)) console.log(label, l)
-}
-
 function parseChunk(c: Buffer): Array<string> {
     return c
         .toString()
@@ -211,34 +259,50 @@ function parseChunk(c: Buffer): Array<string> {
         .split(/\r?\n/)
 }
 
-function resolveCwd(p?: string): string | undefined {
-    if (!p || isAbsolute(p)) {
-        return p
-    } else {
-        return resolve(process.cwd(), p)
+export function parseCommand(command: string): {
+    path: string
+    args: Array<string>
+} {
+    command = command.trimStart()
+    const programSplitIndex = command.indexOf(' ')
+    if (programSplitIndex === -1) {
+        return { path: command.trim(), args: [] }
     }
-}
-
-function opPrint(s: DevService, msg: string) {
-    console.log(opLabel(s), msg)
-}
-
-function opLabel(s: DevService) {
-    return `\`${s.cwd ? s.cwd + ' ' : ''}${s.command}\``
-}
-
-function logLabel(
-    cwd: string | undefined,
-    cmd: string,
-    args: Array<string>,
-    ansiColor: number,
-): string {
-    cwd = !cwd
-        ? './'
-        : cwd.startsWith('/')
-          ? `/.../${basename(cwd)}`
-          : cwd.startsWith('.')
-            ? cwd
-            : `./${cwd}`
-    return `\u001b[${ansiColor}m[\u001b[1m${cmd}\u001b[22m ${args.join(' ')} \u001b[2;3m${cwd}\u001b[22;23m]\u001b[0m`
+    const path = command.substring(0, programSplitIndex)
+    const args: Array<string> = []
+    let argStart = programSplitIndex + 1
+    let withinLiteral: false | "'" | '"' = false
+    for (let i = 0; i < command.length; i++) {
+        const c = command[i]
+        if (!withinLiteral) {
+            if (c === "'" || c === '"') {
+                withinLiteral = c
+                continue
+            }
+            if (c === '\\') {
+                i++
+                continue
+            }
+        }
+        if (withinLiteral) {
+            if (c === withinLiteral) {
+                withinLiteral = false
+                args.push(command.substring(argStart + 1, i))
+                argStart = i + 1
+            }
+            continue
+        }
+        if (c === ' ' && i > argStart) {
+            const maybeArg = command.substring(argStart, i).trim()
+            if (maybeArg.length) {
+                args.push(maybeArg)
+            }
+            argStart = i + 1
+        }
+    }
+    const maybeArg = command.substring(argStart, command.length).trim()
+    if (maybeArg.length) {
+        args.push(maybeArg)
+    }
+    return { path, args }
 }

package/lib/watch.ts

@@ -1,18 +1,49 @@
-import { watch as createWatch } from 'node:fs/promises'
+import {
+    watch as createWatch,
+    type WatchOptionsWithStringEncoding,
+} from 'node:fs/promises'
+
+type WatchCallback = (filename: string) => void
+
+export async function watch(p: string, fire: WatchCallback): Promise<void>
 
 export async function watch(
     p: string,
     signal: AbortSignal,
-    fire: (filename: string) => void,
-) {
+    fire: WatchCallback,
+): Promise<void>
+
+export async function watch(
+    p: string,
+    opts: WatchOptionsWithStringEncoding,
+    fire: WatchCallback,
+): Promise<void>
+
+export async function watch(
+    p: string,
+    signalFireOrOpts:
+        | AbortSignal
+        | WatchCallback
+        | WatchOptionsWithStringEncoding,
+    fireOrUndefined?: WatchCallback,
+): Promise<void> {
+    let opts: WatchOptionsWithStringEncoding | undefined
+    let fire: WatchCallback
+    if (signalFireOrOpts instanceof AbortSignal) {
+        opts = { signal: signalFireOrOpts }
+    } else if (typeof signalFireOrOpts === 'object') {
+        opts = signalFireOrOpts
+    } else {
+        fire = signalFireOrOpts
+    }
+    if (opts && typeof fireOrUndefined === 'function') {
+        fire = fireOrUndefined
+    }
     const delayFire = 90
     const timeout = 100
     let changes: Record<string, number> = {}
     try {
-        for await (const { filename } of createWatch(p, {
-            recursive: true,
-            signal,
-        })) {
+        for await (const { filename } of createWatch(p, opts)) {
             if (filename) {
                 if (!changes[filename]) {
                     const now = Date.now()

package/lib_js/dirs.js

@@ -2,18 +2,18 @@ import { realpath } from "node:fs/promises";
 import { dirname, isAbsolute, join, resolve } from "node:path";
 async function defaultProjectDirs(projectRootAbs) {
   if (!isAbsolute(projectRootAbs)) {
-    throw Error();
+    throw Error("must use an absolute project root path");
+  }
+  if (await realpath(projectRootAbs) !== projectRootAbs) {
+    throw Error("must use a real project root path");
   }
-  const projectResolved = await realpath(projectRootAbs);
   const pages = "pages";
-  const pagesResolved = join(projectResolved, pages);
   return Object.freeze({
     buildRoot: "build",
     buildDist: join("build", "dist"),
     buildWatch: join("build", "watch"),
     pages,
-    pagesResolved,
-    projectResolved,
+    pagesAbs: join(projectRootAbs, pages),
     projectRootAbs,
     public: "public"
   });
@@ -40,7 +40,7 @@ class Resolver {
   }
   // `p` is expected to be a relative path resolvable from the project dir
   isProjectSubpathInPagesDir(p) {
-    return resolve(join(this.#dirs.projectResolved, p)).startsWith(this.#dirs.pagesResolved);
+    return resolve(join(this.#dirs.projectRootAbs, p)).startsWith(this.#dirs.pagesAbs);
   }
   // `p` is expected to be a relative path resolvable from the pages dir
   isPagesSubpathInPagesDir(p) {

package/lib_js/http.js

@@ -4,7 +4,7 @@ import { createServer } from "node:http";
 import { extname, join } from "node:path";
 import { Readable } from "node:stream";
 import mime from "mime";
-function startWebServer(port, flags, dirs, urlRewriteProvider, frontendFetcher, httpServices) {
+function startWebServer(port, flags, dirs, urlRewriteProvider, frontendFetcher, devServices) {
   const serverAddress = "http://localhost:" + port;
   const handler = (req, res) => {
     if (!req.url || !req.method) {
@@ -12,13 +12,13 @@ function startWebServer(port, flags, dirs, urlRewriteProvider, frontendFetcher,
     } else {
       const url = new URL(serverAddress + req.url);
       const headers = convertHeadersToFetch(req.headers);
-      frontendFetcher(url, headers, res, () => onNotFound(req, url, headers, httpServices, flags, dirs, urlRewriteProvider, res));
+      frontendFetcher(url, headers, res, () => onNotFound(req, url, headers, devServices, flags, dirs, urlRewriteProvider, res));
     }
   };
   createServer(flags.logHttp ? createLogWrapper(handler) : handler).listen(port);
   console.log(flags.preview ? "preview" : "dev", `server is live at http://127.0.0.1:${port}`);
 }
-async function onNotFound(req, url, headers, httpServices, flags, dirs, urlRewriteProvider, res) {
+async function onNotFound(req, url, headers, devServices, flags, dirs, urlRewriteProvider, res) {
   if (req.method === "GET" && extname(url.pathname) === "") {
     const urlRewrite = tryUrlRewrites(flags, dirs, urlRewriteProvider.urlRewrites, url);
     if (urlRewrite) {
@@ -26,7 +26,7 @@ async function onNotFound(req, url, headers, httpServices, flags, dirs, urlRewri
       return;
     }
   }
-  const fetchResponse = await tryHttpServices(req, url, headers, httpServices);
+  const fetchResponse = await tryHttpServices(req, url, headers, devServices);
   if (fetchResponse) {
     sendFetchResponse(res, fetchResponse);
   } else {
@@ -46,13 +46,12 @@ function tryUrlRewrites(flags, dirs, urlRewrites, url) {
   const urlRewrite = urlRewrites.find((urlRewrite2) => urlRewrite2.pattern.test(url.pathname));
   return urlRewrite ? join(flags.preview ? dirs.buildDist : dirs.buildWatch, urlRewrite.url, "index.html") : null;
 }
-async function tryHttpServices(req, url, headers, httpServices) {
+async function tryHttpServices(req, url, headers, devServices) {
   if (url.pathname.startsWith("/.well-known/")) {
     return null;
   }
   const body = await collectReqBody(req);
-  const { running } = httpServices;
-  for (const httpService of running) {
+  for (const httpService of devServices.httpServices) {
     const proxyUrl = new URL(url);
     proxyUrl.port = `${httpService.port}`;
     try {
@@ -94,9 +93,9 @@ function createLogWrapper(handler) {
 function createBuiltDistFilesFetcher(dirs, manifest) {
   return (url, _headers, res, notFound) => {
     if (manifest.pageUrls.has(url.pathname)) {
-      streamFile(join(dirs.projectResolved, dirs.buildDist, url.pathname, "index.html"), res);
+      streamFile(join(dirs.projectRootAbs, dirs.buildDist, url.pathname, "index.html"), res);
     } else if (manifest.files.has(url.pathname)) {
-      streamFile(join(dirs.projectResolved, dirs.buildDist, url.pathname), res);
+      streamFile(join(dirs.projectRootAbs, dirs.buildDist, url.pathname), res);
     } else {
       notFound();
     }