@dbx-tools/cli-service 0.9.79 → 0.9.95

package/src/icon.ts

@@ -1,7 +1,7 @@
 import { deflateSync } from "node:zlib";
 
 const SIZE = 32;
-const RECTANGLES = [
+const MODEL_PROXY_RECTANGLES = [
   [4, 13, 6, 6],
   [24, 4, 6, 6],
   [24, 13, 6, 6],
@@ -12,35 +12,93 @@ const RECTANGLES = [
   [18, 15, 6, 2],
   [18, 24, 6, 2],
 ] as const;
+const GRAPHITI_BOLT_RECTANGLES = [
+  [10, 5, 10, 2],
+  [17, 7, 2, 2],
+  [15, 9, 7, 2],
+] as const;
+
+/** Plain glyphs available to identify dbx-tools system-tray services. */
+export type ServiceTrayGlyph = "model-proxy" | "graphiti" | "lakebase";
 
-/** Return the dbx-tools service glyph as a systray2 PNG or Windows ICO payload. */
-export function serviceTrayIcon(platform: NodeJS.Platform = process.platform): string {
+/** Return a service glyph as a systray2 PNG or Windows ICO payload. */
+export function serviceTrayIcon(
+  glyph: ServiceTrayGlyph = "model-proxy",
+  platform: NodeJS.Platform = process.platform,
+): string {
   const color =
     platform === "darwin"
       ? ([0x00, 0x00, 0x00, 0xff] as const)
       : ([0xff, 0x36, 0x21, 0xff] as const);
-  const png = encodePng(renderGlyph(color));
+  const png = encodePng(renderGlyph(glyph, color));
   return (platform === "win32" ? encodeIco(png) : png).toString("base64");
 }
 
-function renderGlyph(color: readonly [number, number, number, number]): Buffer {
+function renderGlyph(
+  glyph: ServiceTrayGlyph,
+  color: readonly [number, number, number, number],
+): Buffer {
   const rgba = Buffer.alloc(SIZE * SIZE * 4);
   for (let y = 0; y < SIZE; y += 1) {
     for (let x = 0; x < SIZE; x += 1) {
-      if (
-        !RECTANGLES.some(
-          ([left, top, width, height]) =>
-            x >= left && x < left + width && y >= top && y < top + height,
-        )
-      ) {
-        continue;
-      }
-      rgba.set(color, (y * SIZE + x) * 4);
+      const opacity = glyphOpacity(glyph, x, y);
+      if (opacity === 0) continue;
+      rgba.set([color[0], color[1], color[2], Math.round(color[3] * opacity)], (y * SIZE + x) * 4);
     }
   }
   return rgba;
 }
 
+function glyphOpacity(glyph: ServiceTrayGlyph, x: number, y: number): number {
+  switch (glyph) {
+    case "model-proxy":
+      return modelProxyGlyphContains(x, y) ? 1 : 0;
+    case "graphiti":
+      return graphitiGlyphContains(x, y) ? 1 : 0;
+    case "lakebase":
+      return lakebaseGlyphOpacity(x, y);
+  }
+}
+
+function modelProxyGlyphContains(x: number, y: number): boolean {
+  return MODEL_PROXY_RECTANGLES.some(
+    ([left, top, width, height]) => x >= left && x < left + width && y >= top && y < top + height,
+  );
+}
+
+function graphitiGlyphContains(x: number, y: number): boolean {
+  // Preserve the Graphiti bot mark through its bolt, antennae, outlined head, and ringed eyes.
+  const bolt = GRAPHITI_BOLT_RECTANGLES.some(
+    ([left, top, width, height]) => x >= left && x < left + width && y >= top && y < top + height,
+  );
+  const head =
+    (y >= 10 && y < 13 && x >= 6 && x < 26) ||
+    (y >= 13 && y < 25 && x >= 4 && x < 7) ||
+    (y >= 13 && y < 25 && x >= 25 && x < 28) ||
+    (y >= 24 && y < 27 && x >= 6 && x < 26);
+  const antenna =
+    y >= 3 &&
+    y < 11 &&
+    ((x >= 7 && x < 9) || (x >= 23 && x < 25));
+  return bolt || head || antenna || graphitiEyeContains(x, y, 11) || graphitiEyeContains(x, y, 21);
+}
+
+function graphitiEyeContains(x: number, y: number, centerX: number): boolean {
+  const distance = (x - centerX) ** 2 + (y - 18) ** 2;
+  return distance >= 10 && distance <= 25;
+}
+
+function lakebaseGlyphOpacity(x: number, y: number): number {
+  if (x < 5 || x >= 27) return 0;
+  if (y >= 5 && y < 11) return 0.45;
+
+  // Keep the official mark's light middle band, dark wave channel, and solid lower band.
+  const middleBoundary = 19 - Math.round(Math.sin(((x - 5) / 21) * Math.PI * 2));
+  if (y >= 13 && y < middleBoundary) return 0.45;
+  if (y >= middleBoundary + 3 && y < 27) return 1;
+  return 0;
+}
+
 function encodePng(rgba: Buffer): Buffer {
   const header = Buffer.alloc(13);
   header.writeUInt32BE(SIZE, 0);

package/src/service.ts

@@ -66,6 +66,7 @@ export type CliServicePythonInstaller = (
   packageSpecifiers: readonly string[],
   python: string,
   platform: NodeJS.Platform,
+  offline: boolean,
 ) => Promise<string>;
 
 /** Host runtime overrides for tests, embedded distributions, and nonstandard homes. */
@@ -100,6 +101,10 @@ export interface CliServiceRuntimeOptions {
 export interface CliServiceInstallOptions {
   /** Start the service immediately after writing its configuration and login entry. */
   readonly start?: boolean;
+  /** Install the primary Python package from this local project instead of the registry. */
+  readonly pythonProject?: string;
+  /** Resolve and install Python packages only from the uv cache. */
+  readonly offline?: boolean;
 }
 
 /** Current installation and process state for a system-tray service. */
@@ -124,6 +129,8 @@ export interface CliServiceLifecycle {
   stop(): Promise<void>;
   /** Stop and start the installed service. */
   restart(): Promise<void>;
+  /** Return the managed service process log path. */
+  logPath(): string;
   /** Stop the service and remove its login entry and package-owned state. */
   uninstall(): Promise<void>;
   /** Read the current installation and process state. */
@@ -175,6 +182,9 @@ export class CliService implements CliServiceLifecycle {
 
   /** Install the login entry and optionally start the service. */
   async install(options: CliServiceInstallOptions = {}): Promise<void> {
+    if (options.pythonProject && !this.definition.pythonPackage) {
+      throw new Error(`${this.definition.name} does not define a Python package`);
+    }
     const owner = resolveServicePackage(this.definition.packageName);
     const dependencies = owner.dependencies();
     const external = new Set(externalRuntimePackages(dependencies));
@@ -184,7 +194,7 @@ export class CliService implements CliServiceLifecycle {
       Object.fromEntries(Object.entries(dependencies).filter(([name]) => external.has(name))),
     );
     const paths = resolveServicePaths(this.definition, this.runtime);
-    const installedDefinition = await this.installDefinition(owner, paths.directory);
+    const installedDefinition = await this.installDefinition(owner, paths.directory, options);
     const host = await this.ensureCompiledBinary(
       this.binaryName(installedDefinition, "service"),
       this.hostEntrypoint,
@@ -260,6 +270,11 @@ export class CliService implements CliServiceLifecycle {
     await this.start();
   }
 
+  /** Return the managed service process log path. */
+  logPath(): string {
+    return resolveServicePaths(this.definition, this.runtime).processLog;
+  }
+
   /** Stop the service and remove its login entry and package-owned state. */
   async uninstall(): Promise<void> {
     const paths = resolveServicePaths(this.definition, this.runtime);
@@ -295,6 +310,7 @@ export class CliService implements CliServiceLifecycle {
   private async installDefinition(
     owner: ServicePackage,
     directory: string,
+    options: CliServiceInstallOptions,
   ): Promise<CliServiceDefinition> {
     const definition = CliServiceDefinitionSchema.parse({
       ...this.definition,
@@ -309,12 +325,16 @@ export class CliService implements CliServiceLifecycle {
       : undefined;
     const pythonEnvironment: Record<string, string> = {};
     if (pythonPackage) {
+      const primarySpecifier = options.pythonProject
+        ? _localPythonSpecifier(options.pythonProject, pythonPackage.name)
+        : `${pythonPackage.name}==${pythonPackage.version}`;
       pythonEnvironment.PYTHON = await this.pythonInstaller(
         this.uvExecutable,
         join(directory, "python"),
-        [`${pythonPackage.name}==${pythonPackage.version}`, ...pythonPackage.dependencies],
+        [primarySpecifier, ...pythonPackage.dependencies],
         pythonPackage.python,
         this.runtime.platform,
+        options.offline ?? false,
       );
     }
     const command = definition.command
@@ -429,6 +449,16 @@ export class CliService implements CliServiceLifecycle {
   }
 }
 
+/** Preserve a package's extras while replacing its registry name with a local project path. */
+function _localPythonSpecifier(project: string, packageName: string): string {
+  const directory = resolve(project);
+  if (!existsSync(join(directory, "pyproject.toml"))) {
+    throw new Error(`Local Python project has no pyproject.toml: ${directory}`);
+  }
+  const extras = /(\[[^\]]+\])$/.exec(packageName)?.[1] ?? "";
+  return `${directory}${extras}`;
+}
+
 function resolveHostEntrypoint(): string {
   const current = fileURLToPath(import.meta.url);
   const source =