@sundaeswap/sprinkles 0.6.0 → 0.7.0

package/src/Sprinkle/index.ts

@@ -19,6 +19,7 @@ import {
 } from "./prompts.js";
 import colors from "yoctocolors-cjs";
 import { type TSchema, Type, OptionalKind } from "@sinclair/typebox";
+import { Value } from "@sinclair/typebox/value";
 import * as fs from "fs";
 import * as path from "path";
 export * from "@sinclair/typebox";
@@ -68,7 +69,10 @@ import {
   isImport,
   isArray,
   isBigInt,
+  isBoolean,
+  isInteger,
   isLiteral,
+  isNumber,
   isObject,
   isRef,
   isString,
@@ -87,7 +91,10 @@ export {
   isImport,
   isArray,
   isBigInt,
+  isBoolean,
+  isInteger,
   isLiteral,
+  isNumber,
   isObject,
   isRef,
   isString,
@@ -145,6 +152,47 @@ import {
 import { promptObject, promptArray } from "./menus/index.js";
 import { formatPath } from "./utils/formatting.js";
 
+// Import and re-export action system
+import {
+  ActionRegistry,
+  ActionError,
+  executeAction,
+  detectMode,
+  parseCliArgs,
+  generateActionHelp,
+  camelToKebab,
+  kebabToCamel,
+  coerceValue,
+  parseArgvWithSchema,
+  generateAppHelp,
+  runCli,
+  runMcp,
+  promptAndExecute,
+} from "./actions/index.js";
+import type { AnyAction, IActionContext, IActionResult } from "./actions/index.js";
+export type { IAction, IActionContext, IActionResult, IActionSuccess, IActionFailure, AnyAction } from "./actions/index.js";
+export {
+  ActionRegistry,
+  ActionError,
+  executeAction,
+  detectMode,
+  parseCliArgs,
+  generateActionHelp,
+  camelToKebab,
+  kebabToCamel,
+  coerceValue,
+  parseArgvWithSchema,
+  generateAppHelp,
+  runCli,
+  typeboxToJsonSchema,
+  coerceMcpInput,
+  getMcpSdk,
+  createMcpServer,
+  runMcp,
+  getBuiltinActions,
+  promptAndExecute,
+} from "./actions/index.js";
+
 export interface IMenuAction<S extends TSchema> {
   title: string;
   action: (sprinkle: Sprinkle<S>) => Promise<Sprinkle<S> | void>;
@@ -164,11 +212,13 @@ export class Sprinkle<S extends TSchema> {
   options: ISprinkleOptions;
   profileId: string = "";
   profileMeta: IProfileMeta = { name: "", createdAt: "", updatedAt: "" };
+  actionRegistry: ActionRegistry<S>;
 
   constructor(type: S, storagePath: string, options?: ISprinkleOptions) {
     this.type = type;
     this.storagePath = storagePath;
     this.options = options ?? {};
+    this.actionRegistry = new ActionRegistry<S>();
   }
 
   // --- Current Profile Accessor ---
@@ -590,8 +640,15 @@ export class Sprinkle<S extends TSchema> {
           {
             title: "Edit settings",
             action: async () => {
-              this.settings = await this.EditStruct(this.type, this.settings);
-              this.saveSettings();
+              try {
+                this.settings = await this.EditStruct(this.type, this.settings);
+                this.saveSettings();
+              } catch (e) {
+                if (e instanceof UserCancelledError) {
+                  return; // User cancelled, return to menu
+                }
+                throw e;
+              }
             },
           },
           {
@@ -763,6 +820,104 @@ export class Sprinkle<S extends TSchema> {
     return maskSensitiveFields(this.settings, this.type);
   }
 
+  // --- Non-interactive profile management (for CLI/MCP actions) ---
+
+  /**
+   * Look up a profile entry by its ID without loading it.
+   * Returns undefined if no profile with the given ID exists.
+   */
+  getProfileById(id: string): IProfileEntry | undefined {
+    return this.scanProfiles().find((p) => p.id === id);
+  }
+
+  /**
+   * Create a new profile file without interactive prompts.
+   * Does NOT switch the active profile.
+   *
+   * @throws ActionError with code DUPLICATE_PROFILE if a profile with the same name already exists.
+   * @returns The created IProfileEntry
+   */
+  async createProfileNonInteractive(
+    name: string,
+    description?: string,
+    initialSettings?: Partial<TExact<S>>,
+  ): Promise<IProfileEntry> {
+    const profiles = this.scanProfiles();
+    const nameLower = name.toLowerCase();
+    const duplicate = profiles.find(
+      (p) => p.meta.name.toLowerCase() === nameLower,
+    );
+    if (duplicate) {
+      throw new ActionError(
+        `A profile named "${name}" already exists.`,
+        "DUPLICATE_PROFILE",
+        { existingId: duplicate.id },
+      );
+    }
+
+    const profilesDir = Sprinkle.profilesDir(this.storagePath);
+    if (!fs.existsSync(profilesDir)) {
+      fs.mkdirSync(profilesDir, { recursive: true });
+    }
+
+    const baseId = Sprinkle.sanitizeProfileId(name);
+    const id = Sprinkle.findAvailableId(profilesDir, baseId);
+    const now = new Date().toISOString();
+    const meta: IProfileMeta = { name, description, createdAt: now, updatedAt: now };
+
+    const profileData = {
+      meta,
+      settings: initialSettings ?? {},
+      defaults: {},
+    };
+
+    fs.writeFileSync(
+      path.join(profilesDir, `${id}.json`),
+      JSON.stringify(profileData, bigIntReplacer, 2),
+      "utf-8",
+    );
+
+    return { id, meta };
+  }
+
+  /**
+   * Delete a profile file by ID.
+   *
+   * @throws ActionError with code PROFILE_NOT_FOUND if no profile with the given ID exists.
+   * @throws ActionError with code CANNOT_DELETE_ONLY_PROFILE if this is the only profile.
+   * @throws ActionError with code CANNOT_DELETE_ACTIVE_PROFILE if this is the currently active profile.
+   */
+  deleteProfileById(id: string): void {
+    const profiles = this.scanProfiles();
+    const profile = profiles.find((p) => p.id === id);
+
+    if (!profile) {
+      throw new ActionError(
+        `Profile "${id}" not found.`,
+        "PROFILE_NOT_FOUND",
+        { id },
+      );
+    }
+
+    if (profiles.length === 1) {
+      throw new ActionError(
+        "Cannot delete the only profile.",
+        "CANNOT_DELETE_ONLY_PROFILE",
+        { id },
+      );
+    }
+
+    if (id === this.profileId) {
+      throw new ActionError(
+        "Cannot delete the active profile. Switch to a different profile first.",
+        "CANNOT_DELETE_ACTIVE_PROFILE",
+        { id },
+      );
+    }
+
+    fs.unlinkSync(Sprinkle.profilePath(this.storagePath, id));
+  }
+
   async TxDialog<P extends Provider, W extends Wallet>(
     blaze: Blaze<P, W>,
     tx: Core.Transaction,
@@ -1137,7 +1292,40 @@ export class Sprinkle<S extends TSchema> {
         throw new UserCancelledError();
       }
       const selection = selectionResult as TSchema;
-      return this._fillInStruct(selection, path, defs) as Promise<TExact<U>>;
+      // Determine if the provided default value matches the selected variant.
+      // For discriminated unions (objects with literal fields like `type`), check
+      // if the literal field values in the selected variant match those in `def`.
+      // For non-discriminated unions, fall back to structural matching with Value.Check.
+      let matchedDef: unknown = undefined;
+      if (def !== undefined) {
+        if (isObject(selection)) {
+          // Check if all literal fields in the selected variant match def
+          const literalFields = Object.entries(selection.properties ?? {}).filter(
+            ([, fieldSchema]) => isLiteral(fieldSchema as TSchema),
+          );
+          if (literalFields.length > 0) {
+            const defRecord = def as Record<string, unknown>;
+            const allLiteralsMatch = literalFields.every(
+              ([fieldName, fieldSchema]) =>
+                defRecord[fieldName] === (fieldSchema as unknown as { const: unknown }).const,
+            );
+            if (allLiteralsMatch) {
+              matchedDef = def;
+            }
+          } else {
+            // No literal discriminators - use structural check
+            if (Value.Check(selection, def)) {
+              matchedDef = def;
+            }
+          }
+        } else {
+          // Non-object variant - use structural check
+          if (Value.Check(selection, def)) {
+            matchedDef = def;
+          }
+        }
+      }
+      return this._fillInStruct(selection, path, defs, matchedDef as TExact<typeof selection>) as Promise<TExact<U>>;
     }
 
     if (isString(type)) {
@@ -1335,4 +1523,208 @@ export class Sprinkle<S extends TSchema> {
 
     return def;
   }
+
+  // --- Action System ---
+
+  /**
+   * Register an action on this Sprinkle instance.
+   * Delegates to the internal ActionRegistry with name and schema validation.
+   */
+  registerAction(action: AnyAction<S>): void {
+    this.actionRegistry.register(action);
+  }
+
+  /**
+   * Retrieve a registered action by name.
+   * @returns The action, or undefined if not registered
+   */
+  getAction(name: string): AnyAction<S> | undefined {
+    return this.actionRegistry.get(name);
+  }
+
+  /**
+   * List all registered actions.
+   */
+  listActions(): AnyAction<S>[] {
+    return this.actionRegistry.list();
+  }
+
+  /**
+   * List all registered actions grouped by category.
+   * Uncategorized actions are grouped under "default".
+   */
+  listActionsByCategory(): Map<string, AnyAction<S>[]> {
+    return this.actionRegistry.listByCategory();
+  }
+
+  /**
+   * Execute a registered action by name with raw (unvalidated) input.
+   * Input is validated against the action's inputSchema before execution.
+   *
+   * @throws Error if the action name is not registered
+   */
+  async runAction(
+    name: string,
+    input: unknown,
+  ): Promise<IActionResult<unknown>> {
+    const action = this.actionRegistry.get(name);
+    if (!action) {
+      throw new Error(
+        `Action "${name}" is not registered. Available actions: ${this.actionRegistry.list().map((a) => a.name).join(", ") || "(none)"}`,
+      );
+    }
+
+    const context: IActionContext<S> = {
+      sprinkle: this,
+      settings: this.settings,
+    };
+
+    return executeAction(action, input, context);
+  }
+
+  /**
+   * Non-interactive profile initialization for CLI/MCP modes.
+   * Resolves a profile without any user prompts.
+   *
+   * Resolution order:
+   * 1. If profileName provided, load that profile directly
+   * 2. If exactly one profile exists, auto-select it
+   * 3. If multiple profiles exist and no name given, throw with available names
+   * 4. If no profiles exist, throw instructing user to run in interactive mode
+   */
+  private async initNonInteractive(profileName?: string): Promise<void> {
+    await this.migrateIfNeeded();
+
+    if (profileName) {
+      await this.loadProfile(profileName);
+      return;
+    }
+
+    const profiles = this.scanProfiles();
+
+    if (profiles.length === 0) {
+      throw new Error(
+        "No profiles found. Run in interactive mode to create one.",
+      );
+    }
+
+    if (profiles.length === 1) {
+      await this.loadProfile(profiles[0]!.id);
+      return;
+    }
+
+    // Multiple profiles without a --profile flag
+    const names = profiles.map((p) => `"${p.id}" (${p.meta.name})`).join(", ");
+    throw new Error(
+      `Multiple profiles found. Specify one with --profile <id>. Available profiles: ${names}`,
+    );
+  }
+
+  /**
+   * Static entry point that detects mode and runs accordingly.
+   *
+   * Modes:
+   * - "tui"  -- Interactive TUI menu (requires menu option)
+   * - "cli"  -- CLI action execution (parses argv for action name and args)
+   * - "mcp"  -- MCP protocol mode (not yet implemented)
+   * - "help" -- Print available actions and exit
+   */
+  static async run<S extends TSchema>(opts: {
+    type: S;
+    storagePath: string;
+    menu?: IMenu<S>;
+    actions?: AnyAction<S>[];
+    options?: ISprinkleOptions;
+    argv?: string[];
+  }): Promise<void> {
+    const argv = opts.argv ?? process.argv.slice(2);
+    const mode = detectMode(argv);
+
+    if (mode === "tui") {
+      if (!opts.menu) {
+        throw new Error(
+          "TUI mode requires a menu. Provide a menu option or pass --help to see available actions.",
+        );
+      }
+      const sprinkle = await Sprinkle.New(
+        opts.type,
+        opts.storagePath,
+        opts.options,
+      );
+      // Register any provided actions
+      for (const action of opts.actions ?? []) {
+        sprinkle.registerAction(action);
+      }
+      await sprinkle.showMenu(opts.menu);
+      return;
+    }
+
+    if (mode === "help") {
+      const sprinkle = new Sprinkle(opts.type, opts.storagePath, opts.options);
+      for (const action of opts.actions ?? []) {
+        sprinkle.registerAction(action);
+      }
+      console.log(generateAppHelp(sprinkle.listActions()));
+      return;
+    }
+
+    if (mode === "cli") {
+      const { actionName, args } = parseCliArgs(argv);
+
+      // Extract --profile flag from args if present
+      const profileName =
+        typeof args["profile"] === "string" ? args["profile"] : undefined;
+
+      const sprinkle = new Sprinkle(opts.type, opts.storagePath, opts.options);
+      for (const action of opts.actions ?? []) {
+        sprinkle.registerAction(action);
+      }
+
+      // Handle action-specific --help BEFORE profile initialization
+      // (help should work without a profile)
+      if (args["help"] === true) {
+        const action = sprinkle.getAction(actionName);
+        if (action) {
+          console.log(generateActionHelp(action));
+        } else {
+          console.log(`Unknown action "${actionName}". Run --help to see available actions.`);
+        }
+        return;
+      }
+
+      await sprinkle.initNonInteractive(profileName);
+
+      // Remove internal flags from args before passing to action
+      const actionArgs = { ...args };
+      delete actionArgs["profile"];
+      delete actionArgs["help"];
+
+      // Build context and delegate to runCli for proper error routing
+      const context = {
+        sprinkle,
+        settings: sprinkle.settings,
+      };
+      await runCli(sprinkle, actionName, actionArgs, context);
+      return;
+    }
+
+    if (mode === "mcp") {
+      // Derive a server name from the storage path basename, falling back to a
+      // sensible default.
+      const serverName = path.basename(opts.storagePath) || "sprinkle-mcp";
+
+      const sprinkle = new Sprinkle(opts.type, opts.storagePath, opts.options);
+      for (const action of opts.actions ?? []) {
+        sprinkle.registerAction(action);
+      }
+
+      // Read profile name from environment variable (MCP clients have no
+      // interactive terminal to prompt for profile selection).
+      const profileName = process.env["SPRINKLE_PROFILE"];
+
+      await sprinkle.initNonInteractive(profileName);
+      await runMcp(sprinkle, serverName);
+      return;
+    }
+  }
 }