@dickpy/dsh-imagegen 1.0.1 → 1.0.3

package/src/routes.ts

@@ -11,7 +11,8 @@ import type { WebRoute } from '@deepseek-ai/dsh-host-webserver'
 import { SettingsConflictError, settingsNamespace, type SettingsDescriptor } from '@deepseek-ai/dsh-settings'
 import { generateImage, type UpstreamConfig } from './engine.ts'
 import { appendHistory, clearHistory, listHistory, readHistoryImage, removeHistory } from './history-store.ts'
-import { GENERATE_API, HISTORY_API, IMAGEGEN_SETTINGS_NAMESPACE, SETTINGS_API, type GeneratedImage, type GenerateRequest, type HistoryEntry, type HistoryEntryInput } from './protocol.ts'
+import { checkForUpdate, CURRENT_VERSION, installUpdate } from './updater.ts'
+import { GENERATE_API, HISTORY_API, IMAGEGEN_SETTINGS_NAMESPACE, SETTINGS_API, UPDATE_API, type GeneratedImage, type GenerateRequest, type HistoryEntry, type HistoryEntryInput } from './protocol.ts'
 
 /** Cap on JSON request bodies (settings ops and generate payloads are small). */
 const MAX_JSON_BODY_BYTES = 24 * 1024 * 1024
@@ -304,6 +305,44 @@ export function makeRoutes(deps: ImageGenRoutesDeps): WebRoute[] {
         }
       },
     },
+    // ----------------------------------------------- update check
+    {
+      kind: 'exact',
+      path: UPDATE_API.check,
+      handler: async (req, res) => {
+        if (!guard(req, res, 'POST')) return
+        try {
+          writeJson(res, 200, { ok: true, update: await checkForUpdate() })
+        } catch (error) {
+          writeJson(res, 200, { ok: false, code: 'update-check-failed', message: messageOf(error) })
+        }
+      },
+    },
+    // ----------------------------------------------- update apply
+    {
+      kind: 'exact',
+      path: UPDATE_API.apply,
+      handler: async (req, res) => {
+        if (!guard(req, res, 'POST')) return
+        const body = await readJsonBody(req)
+        const version = body !== undefined && typeof body.version === 'string' ? body.version.trim() : ''
+        if (version === '') {
+          writeJson(res, 200, { ok: false, code: 'bad-request', message: 'update version is required' })
+          return
+        }
+        try {
+          const latest = await checkForUpdate()
+          if (!latest.updateAvailable || latest.latestVersion !== version) {
+            writeJson(res, 200, { ok: false, code: 'update-not-available', message: `version ${version} is not the latest available release` })
+            return
+          }
+          await installUpdate(version)
+          writeJson(res, 200, { ok: true, currentVersion: CURRENT_VERSION, updatedVersion: version, restartRequired: true })
+        } catch (error) {
+          writeJson(res, 200, { ok: false, code: 'update-failed', message: messageOf(error) })
+        }
+      },
+    },
     // ----------------------------------------------------- history list
     {
       kind: 'exact',

package/src/updater.ts

@@ -0,0 +1,117 @@
+/** GitHub Release discovery and explicit, user-triggered plugin updates. */
+
+import { spawn, type ChildProcess } from 'node:child_process'
+import { PLUGIN_VERSION } from './protocol.ts'
+
+/** Keep this in sync with package.json for each published release. */
+export const CURRENT_VERSION = PLUGIN_VERSION
+export const PACKAGE_NAME = '@dickpy/dsh-imagegen'
+export const RELEASES_URL = 'https://api.github.com/repos/dickpy/dsh-imagegen/releases/latest'
+
+const CHECK_TIMEOUT_MS = 10_000
+const CACHE_TTL_MS = 15 * 60_000
+
+export interface UpdateInfo {
+  currentVersion: string
+  latestVersion: string
+  updateAvailable: boolean
+  releaseUrl: string
+  publishedAt?: string
+}
+
+interface GitHubRelease {
+  tag_name?: unknown
+  html_url?: unknown
+  published_at?: unknown
+  draft?: unknown
+  prerelease?: unknown
+}
+
+let cached: { expiresAt: number; value: UpdateInfo } | undefined
+
+/** Compare stable semver triples; returns positive when `left` is newer. */
+export function compareVersions(left: string, right: string): number {
+  const parse = (value: string): [number, number, number] => {
+    const match = /^v?(\d+)\.(\d+)\.(\d+)/.exec(value.trim())
+    if (match === null) return [0, 0, 0]
+    return [Number(match[1]), Number(match[2]), Number(match[3])]
+  }
+  const a = parse(left)
+  const b = parse(right)
+  return a[0] - b[0] || a[1] - b[1] || a[2] - b[2]
+}
+
+function normalizedReleaseVersion(tag: unknown): string | undefined {
+  if (typeof tag !== 'string' || !/^v?\d+\.\d+\.\d+(?:-[0-9A-Za-z.-]+)?$/.test(tag.trim())) return undefined
+  return tag.trim().replace(/^v/, '')
+}
+
+/** Read the latest stable GitHub Release, with a short host-side cache. */
+export async function checkForUpdate(fetchFn: typeof fetch = fetch, now = Date.now()): Promise<UpdateInfo> {
+  if (cached !== undefined && cached.expiresAt > now) return cached.value
+  const response = await fetchFn(RELEASES_URL, {
+    headers: {
+      accept: 'application/vnd.github+json',
+      'user-agent': 'dsh-imagegen-update-check',
+    },
+    signal: AbortSignal.timeout(CHECK_TIMEOUT_MS),
+  })
+  if (!response.ok) throw new Error(`GitHub Releases returned HTTP ${response.status}`)
+  const payload: unknown = await response.json()
+  if (payload === null || typeof payload !== 'object') throw new Error('GitHub Releases returned malformed JSON')
+  const release = payload as GitHubRelease
+  if (release.draft === true || release.prerelease === true) throw new Error('latest GitHub Release is not stable')
+  const latestVersion = normalizedReleaseVersion(release.tag_name)
+  if (latestVersion === undefined) throw new Error('latest GitHub Release has an invalid version tag')
+  const releaseUrl = typeof release.html_url === 'string' ? release.html_url : 'https://github.com/dickpy/dsh-imagegen/releases'
+  const value: UpdateInfo = {
+    currentVersion: CURRENT_VERSION,
+    latestVersion,
+    updateAvailable: compareVersions(latestVersion, CURRENT_VERSION) > 0,
+    releaseUrl,
+    ...typeof release.published_at === 'string' ? { publishedAt: release.published_at } : {},
+  }
+  cached = { expiresAt: now + CACHE_TTL_MS, value }
+  return value
+}
+
+/** Resolve the profile that launched the current DSH process. */
+export function profileFromProcess(argv: readonly string[] = process.argv, env: NodeJS.ProcessEnv = process.env): string {
+  const envProfile = env.DSH_PROFILE?.trim()
+  if (envProfile !== undefined && /^[a-zA-Z0-9_-]+$/.test(envProfile)) return envProfile
+  const profileIndex = argv.indexOf('--profile')
+  const explicit = profileIndex >= 0 ? argv[profileIndex + 1]?.trim() : undefined
+  if (explicit !== undefined && /^[a-zA-Z0-9_-]+$/.test(explicit)) return explicit
+  if (argv.includes('web')) return 'web'
+  return 'web'
+}
+
+/** Run the same official command documented for plugin installation. */
+export function installUpdate(
+  version: string,
+  spawnFn: typeof spawn = spawn,
+  argv: readonly string[] = process.argv,
+  env: NodeJS.ProcessEnv = process.env,
+): Promise<void> {
+  if (!/^\d+\.\d+\.\d+(?:-[0-9A-Za-z.-]+)?$/.test(version)) {
+    return Promise.reject(new Error('invalid update version'))
+  }
+  const profile = profileFromProcess(argv, env)
+  const command = process.platform === 'win32' ? 'dsh.cmd' : 'dsh'
+  const child = spawnFn(command, ['plugin', '--profile', profile, 'add', `${PACKAGE_NAME}@${version}`], {
+    shell: process.platform === 'win32',
+    stdio: 'ignore',
+  }) as ChildProcess
+  return new Promise((resolve, reject) => {
+    child.once('error', reject)
+    child.once('exit', (code, signal) => {
+      if (code === 0) resolve()
+      else reject(new Error(signal === null ? `plugin update exited with code ${code ?? 'unknown'}` : `plugin update terminated by ${signal}`))
+    })
+  })
+}
+
+/** Test helper: clear the host-side Release cache. */
+export function clearUpdateCache(): void {
+  cached = undefined
+}