@ornncompute/cli 0.2.17 → 0.2.19

package/src/color.mjs

@@ -0,0 +1,71 @@
+// SGR open/close pairs for the four styles the CLI uses.
+const CODES = { red: [31, 39], green: [32, 39], bold: [1, 22], dim: [2, 22] };
+
+function createColors(enabled) {
+  return Object.fromEntries(
+    Object.entries(CODES).map(([name, [open, close]]) => [
+      name,
+      enabled ? (text) => `\x1b[${open}m${text}\x1b[${close}m` : (text) => text,
+    ])
+  );
+}
+
+const NO_COLOR_FLAG = "--no-color";
+
+// Structured-error label prefixes that get dim/bold styling. Matched at the
+// start of a line; the remainder of the line (the value) is left untouched.
+const LABEL_PREFIX_RE = /^(Code|Reason|Next steps?):/;
+
+// `--no-color` is global: strip it anywhere in argv so per-command parsing
+// never sees it as an unknown option.
+export function extractNoColorFlag(argv) {
+  return {
+    argv: argv.filter((arg) => arg !== NO_COLOR_FLAG),
+    noColor: argv.includes(NO_COLOR_FLAG),
+  };
+}
+
+function streamSupportsColor(stream, env, noColorFlag) {
+  if (noColorFlag || Object.hasOwn(env ?? {}, "NO_COLOR")) {
+    return false;
+  }
+  if (Object.hasOwn(env ?? {}, "FORCE_COLOR")) {
+    return String(env.FORCE_COLOR) !== "0";
+  }
+  return Boolean(stream.isTTY);
+}
+
+// stdout and stderr are evaluated independently: one can be a TTY while the
+// other is piped.
+export function attachStreamColors({ env, noColorFlag, stdout, stderr }) {
+  stdout.pc = createColors(streamSupportsColor(stdout, env, noColorFlag));
+  stderr.pc = createColors(streamSupportsColor(stderr, env, noColorFlag));
+}
+
+export function writeColored(stream, variant, text) {
+  const colorize = stream.pc?.[variant];
+  stream.write(`${colorize ? colorize(text) : text}\n`);
+}
+
+// First line is the error summary (red); later `Code:`/`Reason:`/`Next step:`
+// label prefixes are dim/bold, their values left plain.
+export function colorizeErrorMessage(message, stream) {
+  const pc = stream?.pc;
+  if (!pc) {
+    return message;
+  }
+  return message
+    .split("\n")
+    .map((line, index) => {
+      if (index === 0) {
+        return line ? pc.red(line) : line;
+      }
+      const match = line.match(LABEL_PREFIX_RE);
+      if (!match) {
+        return line;
+      }
+      const label = match[0];
+      return `${pc.bold(pc.dim(label))}${line.slice(label.length)}`;
+    })
+    .join("\n");
+}

package/vendor/capabilities/capabilities/listings.d.ts

@@ -44,6 +44,7 @@ export declare const listingCreateInput: z.ZodObject<{
 }, z.core.$strip>;
 export declare const listCatalogTermsCapability: import("../capability.ts").Capability<unknown, unknown>;
 export declare const listListingsCapability: import("../capability.ts").Capability<unknown, unknown>;
+export declare const forecastListingsCapability: import("../capability.ts").Capability<unknown, unknown>;
 export declare const createListingCapability: import("../capability.ts").Capability<unknown, unknown>;
 export declare const listingCapabilities: import("../capability.ts").Capability<unknown, unknown>[];
 /** @deprecated Use createListingCapability. */

package/vendor/capabilities/capabilities/listings.js

@@ -1,6 +1,14 @@
 import { z } from "zod";
 import { defineCapability, preview } from "../capability.js";
 import { bindHardwareFields, HARDWARE_LISTING_FIELDS, HARDWARE_LISTING_FILTERS, loadHardwareCatalog, projectCatalogTerms, } from "../spec-catalog.js";
+function queryPath(path, params) {
+    const parts = [];
+    for (const [key, value] of Object.entries(params)) {
+        if (value?.trim())
+            parts.push(`${encodeURIComponent(key)}=${encodeURIComponent(value.trim())}`);
+    }
+    return parts.length ? `${path}?${parts.join("&")}` : path;
+}
 export const listingCreateInput = z.object({
     operatorId: z.string().optional(),
     operator_id: z.string().optional(),
@@ -46,6 +54,10 @@ export const listingCreateInput = z.object({
     confirm: z.boolean().optional(),
 });
 const catalogTermsInput = z.object({ kind: z.enum(["gpu", "cpu", "node", "link"]).optional() });
+const listingForecastInput = z.object({
+    limit: z.number().int().min(1).max(500).optional(),
+    cursor: z.string().min(1).optional(),
+});
 const slackListingCreateParams = {
     type: "object",
     properties: {
@@ -135,6 +147,23 @@ export const listListingsCapability = defineCapability({
         return ctx.fetch("/listings/available", { method: "GET", body: bound.payload });
     },
 });
+export const forecastListingsCapability = defineCapability({
+    id: "listings.forecast",
+    domain: "listings",
+    roles: ["user"],
+    description: "List the next 48 hourly availability buckets for customer-visible compute listings.",
+    mutation: "read",
+    http: { method: "GET", path: "/v1/cli/compute/listings/forecast" },
+    slack: false,
+    input: listingForecastInput,
+    execute: async (ctx, input) => {
+        const parsed = listingForecastInput.parse(input);
+        return ctx.fetch(queryPath("/v1/cli/compute/listings/forecast", {
+            limit: parsed.limit?.toString(),
+            cursor: parsed.cursor,
+        }), { method: "GET" });
+    },
+});
 /** Commerce parses listing windows as RFC3339 timestamps, not bare dates. */
 function rfc3339(value) {
     const trimmed = value.trim();
@@ -317,6 +346,7 @@ export const createListingCapability = defineCapability({
 export const listingCapabilities = [
     listCatalogTermsCapability,
     listListingsCapability,
+    forecastListingsCapability,
     createListingCapability,
 ];
 /** @deprecated Use createListingCapability. */

package/vendor/capabilities/capability.d.ts

@@ -65,7 +65,7 @@ export type Capability<I = unknown, O = unknown> = CapabilityBindings & {
     execute: (ctx: CapabilityContext, input: I) => Promise<O>;
 };
 export type CapabilitySpec<I = unknown, O = unknown> = Omit<Capability<I, O>, "cli" | "mcp" | "slack"> & {
-    slack?: {
+    slack?: false | {
         parameters?: JsonSchema;
     };
 };

package/vendor/capabilities/capability.js

@@ -31,14 +31,19 @@ export function preview(action, payload) {
 /** Fill CLI/MCP/Slack names from `id`. Slack parameters stay optional. */
 export function defineCapability(spec) {
     const names = namesFromId(spec.id);
+    const { slack, ...rest } = spec;
     return {
-        ...spec,
+        ...rest,
         cli: { command: names.cli },
         mcp: { name: names.mcp },
-        slack: {
-            name: names.slack,
-            ...(spec.slack?.parameters ? { parameters: spec.slack.parameters } : {}),
-        },
+        ...(slack === false
+            ? {}
+            : {
+                slack: {
+                    name: names.slack,
+                    ...(slack?.parameters ? { parameters: slack.parameters } : {}),
+                },
+            }),
     };
 }
 /**

package/vendor/capabilities/index.d.ts

@@ -4,7 +4,7 @@ export { catalog } from "./catalog.ts";
 export { fleetCapabilities, listFacilitiesCapability, listNodesCapability, listOperatorsCapability, nodeExecCapability, nodeOffGridCapability, nodeOnGridCapability, nodeTerminateCapability, } from "./capabilities/fleet.ts";
 export { buildEnrollmentInstallCommand, tokenCapabilities, tokensCreateCapability, tokensListCapability, } from "./capabilities/tokens.ts";
 export { identityCapabilities, statusCapability, whoamiCapability } from "./capabilities/identity.ts";
-export { createListingCapability, listingCreateInput, listCatalogTermsCapability, listingCapabilities, listListingsCapability, proposeListingCapability, } from "./capabilities/listings.ts";
+export { createListingCapability, forecastListingsCapability, listingCreateInput, listCatalogTermsCapability, listingCapabilities, listListingsCapability, proposeListingCapability, } from "./capabilities/listings.ts";
 export { userCapabilities, usersQueryCapability } from "./capabilities/users.ts";
 export { kubernetesCapabilities, slurmCapabilities } from "./capabilities/clusters.ts";
 export { logsLatestCapability, logsTailCapability, observabilityCapabilities, telemetryLatestCapability, telemetryTailCapability, } from "./capabilities/observability.ts";

package/vendor/capabilities/index.js

@@ -4,7 +4,7 @@ export { catalog } from "./catalog.js";
 export { fleetCapabilities, listFacilitiesCapability, listNodesCapability, listOperatorsCapability, nodeExecCapability, nodeOffGridCapability, nodeOnGridCapability, nodeTerminateCapability, } from "./capabilities/fleet.js";
 export { buildEnrollmentInstallCommand, tokenCapabilities, tokensCreateCapability, tokensListCapability, } from "./capabilities/tokens.js";
 export { identityCapabilities, statusCapability, whoamiCapability } from "./capabilities/identity.js";
-export { createListingCapability, listingCreateInput, listCatalogTermsCapability, listingCapabilities, listListingsCapability, proposeListingCapability, } from "./capabilities/listings.js";
+export { createListingCapability, forecastListingsCapability, listingCreateInput, listCatalogTermsCapability, listingCapabilities, listListingsCapability, proposeListingCapability, } from "./capabilities/listings.js";
 export { userCapabilities, usersQueryCapability } from "./capabilities/users.js";
 export { kubernetesCapabilities, slurmCapabilities } from "./capabilities/clusters.js";
 export { logsLatestCapability, logsTailCapability, observabilityCapabilities, telemetryLatestCapability, telemetryTailCapability, } from "./capabilities/observability.js";