@fatdoge/wtree 0.1.0

package/api/ui/startUiDev.ts

@@ -0,0 +1,82 @@
+import os from 'node:os'
+import path from 'node:path'
+import fs from 'node:fs'
+import { fileURLToPath } from 'node:url'
+import { createServer } from 'vite'
+import type { ViteDevServer } from 'vite'
+import { createApiApp } from '../createApiApp.js'
+import { openPath } from '../core/open.js'
+
+function findUiRoot(fromDir: string) {
+  let cur = fromDir
+  for (let i = 0; i < 10; i += 1) {
+    const indexHtml = path.join(cur, 'index.html')
+    const viteConfig = path.join(cur, 'vite.config.ts')
+    if (exists(indexHtml) && exists(viteConfig)) return cur
+    const parent = path.dirname(cur)
+    if (parent === cur) break
+    cur = parent
+  }
+  return fromDir
+}
+
+function exists(p: string) {
+  return fs.existsSync(p)
+}
+
+export type UiDevHandle = {
+  uiUrl: string
+  close: () => Promise<void>
+}
+
+export async function startUiDevServer(options: {
+  repoRoot: string
+  uiPort?: number
+  open?: boolean
+}) : Promise<UiDevHandle> {
+  const repoRoot = options.repoRoot
+  const open = options.open !== false
+  const uiPort = Number.isFinite(options.uiPort) ? (options.uiPort as number) : 5173
+
+  const apiApp = createApiApp(() => repoRoot)
+  const apiServer = apiApp.listen(0, '127.0.0.1')
+  await new Promise<void>((resolve) => apiServer.once('listening', () => resolve()))
+  const apiAddress = apiServer.address()
+  const apiPort = typeof apiAddress === 'object' && apiAddress ? apiAddress.port : 0
+  process.env.WTUI_API_URL = `http://127.0.0.1:${apiPort}`
+  process.env.VITE_WTUI_API_URL = process.env.WTUI_API_URL
+
+  const here = path.dirname(fileURLToPath(import.meta.url))
+  const uiRoot = findUiRoot(path.resolve(here, '..', '..'))
+  const prevCwd = process.cwd()
+  process.chdir(uiRoot)
+
+  const vite: ViteDevServer = await createServer({
+    root: uiRoot,
+    configFile: path.join(uiRoot, 'vite.config.ts'),
+    cacheDir: path.join(os.tmpdir(), 'wtui-vite-cache'),
+    server: {
+      host: '127.0.0.1',
+      port: uiPort,
+      strictPort: false,
+    },
+    clearScreen: false,
+    appType: 'spa',
+  })
+
+  await vite.listen()
+  const url = vite.resolvedUrls?.local?.[0] || `http://127.0.0.1:${uiPort}/`
+
+  if (open) {
+    openPath(url)
+  }
+
+  return {
+    uiUrl: url,
+    close: async () => {
+      await vite.close()
+      await new Promise<void>((resolve) => apiServer.close(() => resolve()))
+      process.chdir(prevCwd)
+    },
+  }
+}