septum 0.1.3 → 0.1.5

package/src/mcp/schemas.ts

@@ -1,238 +1,578 @@
-import { z } from "zod";
-
-function withAliases(raw: unknown, aliases: Record<string, string>): unknown {
-  if (typeof raw !== "object" || raw === null || Array.isArray(raw)) return raw;
-  const obj = { ...(raw as Record<string, unknown>) };
-  for (const [alias, target] of Object.entries(aliases)) {
-    if (obj[alias] !== undefined && obj[target] === undefined) {
-      obj[target] = obj[alias];
-    }
-  }
-  return obj;
-}
-
-function normalizeBoundaryArgs(raw: unknown): unknown {
-  if (typeof raw !== "object" || raw === null || Array.isArray(raw)) return raw;
-  const obj = { ...(raw as Record<string, unknown>) };
-  const aliases: Record<string, string> = {
-    filePath: "file_path",
-    file: "file_path",
-    path: "file_path",
-    filePaths: "file_paths",
-    files: "file_paths",
-    paths: "file_paths",
-    proposedImports: "proposed_imports",
-    imports: "proposed_imports",
-    featureKey: "feature_key",
-    feature: "feature_key",
-  };
-  for (const [alias, target] of Object.entries(aliases)) {
-    if (obj[alias] !== undefined && obj[target] === undefined) {
-      obj[target] = obj[alias];
-    }
-  }
-  if (Array.isArray(obj.file_path) && !obj.file_paths) {
-    obj.file_paths = obj.file_path;
-    delete obj.file_path;
-  }
-  if (typeof obj.file_paths === "string" && !obj.file_path) {
-    obj.file_path = obj.file_paths;
-    delete obj.file_paths;
-  }
-  return obj;
-}
-
-export const GetDomainCatalogSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      domain_name: "domain",
-      domainName: "domain",
-      name: "domain",
-      archetype_filter: "archetype",
-      archetypeFilter: "archetype",
-      type: "archetype",
-      workspace: "workspace_path",
-      workspacePath: "workspace_path",
-      project_root: "workspace_path",
-      target_path: "workspace_path",
-      path: "workspace_path",
-    }),
-  z.object({
-    domain: z.string().optional(),
-    archetype: z.string().optional(),
-    workspace_path: z.string().optional(),
-  })
-);
-export type GetDomainCatalogArgs = z.infer<typeof GetDomainCatalogSchema>;
-
-export const GetFeatureContextSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      feature_key: "feature",
-      featureKey: "feature",
-      name: "feature",
-      workspace: "workspace_path",
-      workspacePath: "workspace_path",
-    }),
-  z.object({
-    feature: z.string().min(1, "feature is required"),
-    workspace_path: z.string().optional(),
-  })
-);
-export type GetFeatureContextArgs = z.infer<typeof GetFeatureContextSchema>;
-
-export const LocateSymbolSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      symbol: "query",
-      name: "query",
-      symbol_name: "query",
-      symbolName: "query",
-      target: "query",
-      domain_name: "domain",
-      domainName: "domain",
-      max: "limit",
-      workspace: "workspace_path",
-      workspacePath: "workspace_path",
-      project_root: "workspace_path",
-      target_path: "workspace_path",
-      path: "workspace_path",
-    }),
-  z.object({
-    query: z.string().min(1, "query is required"),
-    domain: z.string().optional(),
-    limit: z.number().int().positive().optional().default(10),
-    workspace_path: z.string().optional(),
-  })
-);
-export type LocateSymbolArgs = z.infer<typeof LocateSymbolSchema>;
-
-export const GetSymbolSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      query: "symbol",
-      name: "symbol",
-      symbol_name: "symbol",
-      symbolName: "symbol",
-      target: "symbol",
-      domain_name: "domain",
-      domainName: "domain",
-      includeDependencies: "include_dependencies",
-      include_deps: "include_dependencies",
-      includeDeps: "include_dependencies",
-      dependencies: "include_dependencies",
-      deps: "include_dependencies",
-    }),
-  z.object({
-    symbol: z.string().min(1, "symbol is required"),
-    domain: z.string().optional(),
-    include_dependencies: z.boolean().optional().default(true),
-  })
-);
-export type GetSymbolArgs = z.infer<typeof GetSymbolSchema>;
-
-export const GetSymbolImpactSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      query: "symbol",
-      name: "symbol",
-      symbol_name: "symbol",
-      symbolName: "symbol",
-      file_path: "symbol",
-      filePath: "symbol",
-      file: "symbol",
-      path: "symbol",
-      target: "symbol",
-    }),
-  z.object({
-    symbol: z.string().min(1, "symbol is required"),
-  })
-);
-export type GetSymbolImpactArgs = z.infer<typeof GetSymbolImpactSchema>;
-
-export const TraceVerticalSliceSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      route: "query",
-      uri: "query",
-      route_uri: "query",
-      routeUri: "query",
-      symbol: "query",
-      name: "query",
-      feature: "query",
-      feature_key: "query",
-      featureKey: "query",
-      maxDepth: "max_depth",
-      depth: "max_depth",
-    }),
-  z.object({
-    query: z.string().min(1, "query is required"),
-    max_depth: z.number().int().positive().optional().default(5),
-  })
-);
-export type TraceVerticalSliceArgs = z.infer<typeof TraceVerticalSliceSchema>;
-
-export const CheckBoundarySchema = z.preprocess(
-  normalizeBoundaryArgs,
-  z.object({
-    file_path: z.string().optional(),
-    file_paths: z.array(z.string()).optional(),
-    proposed_imports: z.array(z.string()).default([]),
-    feature_key: z.string().optional(),
-  })
-);
-export type CheckBoundaryArgs = z.infer<typeof CheckBoundarySchema>;
-
-export const GetSymbolHotspotsSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      domain_name: "domain",
-      domainName: "domain",
-      max: "limit",
-    }),
-  z.object({
-    domain: z.string().optional(),
-    limit: z.number().int().positive().optional().default(10),
-  })
-);
-export type GetSymbolHotspotsArgs = z.infer<typeof GetSymbolHotspotsSchema>;
-
-export const RegisterDomainSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      domain: "name",
-      domain_name: "name",
-      domainName: "name",
-      root_path: "root",
-      rootPath: "root",
-      path: "root",
-      allowedDependencies: "allowed_dependencies",
-      allowed: "allowed_dependencies",
-      forbiddenDependencies: "forbidden_dependencies",
-      forbidden: "forbidden_dependencies",
-      archetype: "archetypes",
-      ingestNow: "ingest_now",
-    }),
-  z.object({
-    name: z.string().min(1, "Domain name is required"),
-    root: z.string().min(1, "Domain root path is required"),
-    description: z.string().optional(),
-    allowed_dependencies: z.array(z.string()).optional(),
-    forbidden_dependencies: z.array(z.string()).optional(),
-    archetypes: z.record(z.string()).optional(),
-    ingest_now: z.boolean().optional().default(false),
-  })
-);
-export type RegisterDomainArgs = z.infer<typeof RegisterDomainSchema>;
-
-export const AutoDiscoverDomainsSchema = z.preprocess(
-  (args) =>
-    withAliases(args, {
-      save: "persist",
-    }),
-  z.object({
-    persist: z.boolean().optional().default(true),
-  })
-);
-export type AutoDiscoverDomainsArgs = z.infer<typeof AutoDiscoverDomainsSchema>;
-
+import { z } from "zod";
+
+function withAliases(raw: unknown, aliases: Record<string, string>): unknown {
+  if (typeof raw !== "object" || raw === null || Array.isArray(raw)) return raw;
+  const obj = { ...(raw as Record<string, unknown>) };
+  for (const [alias, target] of Object.entries(aliases)) {
+    if (obj[alias] !== undefined && obj[target] === undefined) {
+      obj[target] = obj[alias];
+    }
+  }
+  return obj;
+}
+
+export function extractSingleStringFallback(
+  raw: unknown,
+  targetField: string,
+  ignoredKeys: string[] = ["workspace_path", "workspacePath", "workspace", "project_root", "projectRoot"]
+): unknown {
+  if (typeof raw === "string") {
+    return { [targetField]: raw };
+  }
+  if (Array.isArray(raw) && raw.length === 1 && typeof raw[0] === "string") {
+    return { [targetField]: raw[0] };
+  }
+  if (typeof raw === "object" && raw !== null && !Array.isArray(raw)) {
+    const obj = { ...(raw as Record<string, unknown>) };
+    if (obj[targetField] !== undefined && typeof obj[targetField] === "string" && (obj[targetField] as string).trim() !== "") {
+      return obj;
+    }
+    const nonIgnoredEntries = Object.entries(obj).filter(
+      ([key, val]) => !ignoredKeys.includes(key) && typeof val === "string" && (val as string).trim() !== ""
+    );
+    if (nonIgnoredEntries.length === 1) {
+      obj[targetField] = nonIgnoredEntries[0][1];
+    }
+    return obj;
+  }
+  return raw;
+}
+
+export function normalizeFeatureContextArgs(raw: unknown): unknown {
+  const coerced = extractSingleStringFallback(raw, "feature_key");
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = { ...(coerced as Record<string, unknown>) };
+
+  const aliases: Record<string, string> = {
+    feature: "feature_key",
+    featureKey: "feature_key",
+    feature_name: "feature_key",
+    featureName: "feature_key",
+    feat: "feature_key",
+    name: "feature_key",
+    key: "feature_key",
+    task: "feature_key",
+    task_name: "feature_key",
+    target: "feature_key",
+    context: "feature_key",
+    slug: "feature_key",
+    workspace: "workspace_path",
+    workspacePath: "workspace_path",
+    project_root: "workspace_path",
+    projectRoot: "workspace_path",
+  };
+
+  for (const [alias, target] of Object.entries(aliases)) {
+    if (obj[alias] !== undefined && obj[target] === undefined) {
+      obj[target] = obj[alias];
+    }
+  }
+
+  // Populate feature as backward-compatible alias
+  if (obj.feature_key !== undefined && obj.feature === undefined) {
+    obj.feature = obj.feature_key;
+  } else if (obj.feature !== undefined && obj.feature_key === undefined) {
+    obj.feature_key = obj.feature;
+  }
+
+  return obj;
+}
+
+function normalizeBoundaryArgs(raw: unknown): unknown {
+  if (Array.isArray(raw) && raw.length > 1 && raw.every((item) => typeof item === "string")) {
+    raw = { file_paths: raw };
+  }
+  const coerced = extractSingleStringFallback(raw, "file_path");
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = { ...(coerced as Record<string, unknown>) };
+  const aliases: Record<string, string> = {
+    filePath: "file_path",
+    file: "file_path",
+    path: "file_path",
+    from_file: "file_path",
+    fromFile: "file_path",
+    source_file: "file_path",
+    sourceFile: "file_path",
+    filePaths: "file_paths",
+    files: "file_paths",
+    paths: "file_paths",
+    proposedImports: "proposed_imports",
+    imports: "proposed_imports",
+    featureKey: "feature_key",
+    feature: "feature_key",
+    feature_name: "feature_key",
+    featureName: "feature_key",
+    name: "feature_key",
+    key: "feature_key",
+  };
+  for (const [alias, target] of Object.entries(aliases)) {
+    if (obj[alias] !== undefined && obj[target] === undefined) {
+      obj[target] = obj[alias];
+    }
+  }
+
+  if (obj.feature_key !== undefined && obj.feature === undefined) {
+    obj.feature = obj.feature_key;
+  } else if (obj.feature !== undefined && obj.feature_key === undefined) {
+    obj.feature_key = obj.feature;
+  }
+
+  if (obj.to_file || obj.toFile) {
+    const rawTo = obj.to_file ?? obj.toFile;
+    const toFiles = Array.isArray(rawTo) ? rawTo.map(String) : [String(rawTo)];
+    if (!obj.proposed_imports) {
+      obj.proposed_imports = toFiles;
+    } else if (Array.isArray(obj.proposed_imports)) {
+      obj.proposed_imports = [...obj.proposed_imports, ...toFiles];
+    }
+  }
+  if (Array.isArray(obj.file_path) && !obj.file_paths) {
+    obj.file_paths = obj.file_path;
+    delete obj.file_path;
+  }
+  if (typeof obj.file_paths === "string" && !obj.file_path) {
+    obj.file_path = obj.file_paths;
+    delete obj.file_paths;
+  }
+  return obj;
+}
+
+export function normalizeDomainCatalogArgs(raw: unknown): unknown {
+  const coerced = extractSingleStringFallback(raw, "domain");
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = withAliases(coerced, {
+    domain_name: "domain",
+    domainName: "domain",
+    name: "domain",
+    archetype_filter: "archetype",
+    archetypeFilter: "archetype",
+    type: "archetype",
+    workspace: "workspace_path",
+    workspacePath: "workspace_path",
+    project_root: "workspace_path",
+    target_path: "workspace_path",
+    path: "workspace_path",
+  }) as Record<string, unknown>;
+  if (obj.domain !== undefined && obj.name === undefined) {
+    obj.name = obj.domain;
+  } else if (obj.name !== undefined && obj.domain === undefined) {
+    obj.domain = obj.name;
+  }
+  return obj;
+}
+
+export const GetDomainCatalogSchema = z.preprocess(
+  normalizeDomainCatalogArgs,
+  z.object({
+    domain: z.string().optional(),
+    name: z.string().optional(),
+    archetype: z.string().optional(),
+    workspace_path: z.string().optional(),
+  })
+);
+export type GetDomainCatalogArgs = z.infer<typeof GetDomainCatalogSchema>;
+
+export const GetFeatureContextSchema = z.preprocess(
+  normalizeFeatureContextArgs,
+  z.object({
+    feature_key: z.string().optional(),
+    feature: z.string().optional(),
+    workspace_path: z.string().optional(),
+  })
+);
+export type GetFeatureContextArgs = z.infer<typeof GetFeatureContextSchema>;
+
+export function normalizeLocateSymbolArgs(raw: unknown): unknown {
+  const coerced = extractSingleStringFallback(raw, "query");
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = withAliases(coerced, {
+    symbol: "query",
+    name: "query",
+    symbol_name: "query",
+    symbolName: "query",
+    target: "query",
+    domain_name: "domain",
+    domainName: "domain",
+    max: "limit",
+    workspace: "workspace_path",
+    workspacePath: "workspace_path",
+    project_root: "workspace_path",
+    target_path: "workspace_path",
+    path: "workspace_path",
+  }) as Record<string, unknown>;
+  if (obj.query !== undefined && obj.symbol === undefined) {
+    obj.symbol = obj.query;
+  } else if (obj.symbol !== undefined && obj.query === undefined) {
+    obj.query = obj.symbol;
+  }
+  return obj;
+}
+
+export const LocateSymbolSchema = z.preprocess(
+  normalizeLocateSymbolArgs,
+  z.object({
+    query: z.string().min(1, "query is required"),
+    symbol: z.string().optional(),
+    domain: z.string().optional(),
+    limit: z.number().int().positive().optional().default(10),
+    workspace_path: z.string().optional(),
+  })
+);
+export type LocateSymbolArgs = z.infer<typeof LocateSymbolSchema>;
+
+export function normalizeGetSymbolArgs(raw: unknown): unknown {
+  const coerced = extractSingleStringFallback(raw, "symbol");
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = withAliases(coerced, {
+    query: "symbol",
+    name: "symbol",
+    symbol_name: "symbol",
+    symbolName: "symbol",
+    target: "symbol",
+    domain_name: "domain",
+    domainName: "domain",
+    includeDependencies: "include_dependencies",
+    include_deps: "include_dependencies",
+    includeDeps: "include_dependencies",
+    dependencies: "include_dependencies",
+    deps: "include_dependencies",
+  }) as Record<string, unknown>;
+  if (obj.symbol !== undefined && obj.query === undefined) {
+    obj.query = obj.symbol;
+  } else if (obj.query !== undefined && obj.symbol === undefined) {
+    obj.symbol = obj.query;
+  }
+  return obj;
+}
+
+export const GetSymbolSchema = z.preprocess(
+  normalizeGetSymbolArgs,
+  z.object({
+    symbol: z.string().min(1, "symbol is required"),
+    query: z.string().optional(),
+    domain: z.string().optional(),
+    include_dependencies: z.boolean().optional().default(true),
+  })
+);
+export type GetSymbolArgs = z.infer<typeof GetSymbolSchema>;
+
+export function normalizeGetSymbolImpactArgs(raw: unknown): unknown {
+  const coerced = extractSingleStringFallback(raw, "symbol");
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = withAliases(coerced, {
+    query: "symbol",
+    name: "symbol",
+    symbol_name: "symbol",
+    symbolName: "symbol",
+    file_path: "symbol",
+    filePath: "symbol",
+    file: "symbol",
+    path: "symbol",
+    target: "symbol",
+  }) as Record<string, unknown>;
+  if (obj.symbol !== undefined && obj.query === undefined) {
+    obj.query = obj.symbol;
+  } else if (obj.query !== undefined && obj.symbol === undefined) {
+    obj.symbol = obj.query;
+  }
+  return obj;
+}
+
+export const GetSymbolImpactSchema = z.preprocess(
+  normalizeGetSymbolImpactArgs,
+  z.object({
+    symbol: z.string().min(1, "symbol is required"),
+    query: z.string().optional(),
+  })
+);
+export type GetSymbolImpactArgs = z.infer<typeof GetSymbolImpactSchema>;
+
+export function normalizeTraceVerticalSliceArgs(raw: unknown): unknown {
+  const coerced = extractSingleStringFallback(raw, "query");
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = withAliases(coerced, {
+    route: "query",
+    uri: "query",
+    route_uri: "query",
+    routeUri: "query",
+    entry_file: "query",
+    entryFile: "query",
+    file_path: "query",
+    filePath: "query",
+    file: "query",
+    path: "query",
+    symbol: "query",
+    name: "query",
+    feature: "query",
+    feature_key: "query",
+    featureKey: "query",
+    maxDepth: "max_depth",
+    depth: "max_depth",
+  }) as Record<string, unknown>;
+  if (obj.query !== undefined && obj.symbol === undefined) {
+    obj.symbol = obj.query;
+  } else if (obj.symbol !== undefined && obj.query === undefined) {
+    obj.query = obj.symbol;
+  }
+  return obj;
+}
+
+export const TraceVerticalSliceSchema = z.preprocess(
+  normalizeTraceVerticalSliceArgs,
+  z.object({
+    query: z.string().min(1, "query is required"),
+    symbol: z.string().optional(),
+    max_depth: z.number().int().positive().optional().default(5),
+  })
+);
+export type TraceVerticalSliceArgs = z.infer<typeof TraceVerticalSliceSchema>;
+
+export const CheckBoundarySchema = z.preprocess(
+  normalizeBoundaryArgs,
+  z.object({
+    file_path: z.string().optional(),
+    file_paths: z.array(z.string()).optional(),
+    proposed_imports: z.array(z.string()).default([]),
+    feature_key: z.string().optional(),
+  })
+);
+export type CheckBoundaryArgs = z.infer<typeof CheckBoundarySchema>;
+
+export const GetSymbolHotspotsSchema = z.preprocess(
+  (args) =>
+    withAliases(args, {
+      domain_name: "domain",
+      domainName: "domain",
+      max: "limit",
+      nesting: "min_nesting",
+      minNesting: "min_nesting",
+      lines: "min_lines",
+      minLines: "min_lines",
+      sortBy: "sort_by",
+    }),
+  z.object({
+    min_lines: z.number().int().nonnegative().optional(),
+    min_nesting: z.number().int().positive().optional(),
+    kind: z.string().optional(),
+    domain: z.string().optional(),
+    limit: z.number().int().positive().optional().default(10),
+    sort_by: z.enum(["lines", "nesting", "risk_score"]).optional(),
+  })
+);
+export type GetSymbolHotspotsArgs = z.infer<typeof GetSymbolHotspotsSchema>;
+
+export function normalizeRegisterDomainArgs(raw: unknown): unknown {
+  const coerced = extractSingleStringFallback(raw, "domain", [
+    "root",
+    "root_path",
+    "rootPath",
+    "path",
+    "root_dir",
+    "workspace_path",
+    "workspacePath",
+    "workspace",
+    "project_root",
+    "projectRoot",
+  ]);
+  if (typeof coerced !== "object" || coerced === null || Array.isArray(coerced)) return coerced;
+  const obj = withAliases(coerced, {
+    name: "domain",
+    domain_name: "domain",
+    domainName: "domain",
+    root_path: "root",
+    rootPath: "root",
+    path: "root",
+    allowedDependencies: "allowed_dependencies",
+    allowed: "allowed_dependencies",
+    forbiddenDependencies: "forbidden_dependencies",
+    forbidden: "forbidden_dependencies",
+    archetype: "archetypes",
+    ingestNow: "ingest_now",
+  }) as Record<string, unknown>;
+  if (obj.domain !== undefined && obj.name === undefined) {
+    obj.name = obj.domain;
+  } else if (obj.name !== undefined && obj.domain === undefined) {
+    obj.domain = obj.name;
+  }
+  return obj;
+}
+
+export const RegisterDomainSchema = z.preprocess(
+  normalizeRegisterDomainArgs,
+  z.object({
+    domain: z.string().min(1, "Domain name/identifier is required"),
+    name: z.string().optional(),
+    root: z.string().min(1, "Domain root path is required"),
+    description: z.string().optional(),
+    allowed_dependencies: z.array(z.string()).optional(),
+    forbidden_dependencies: z.array(z.string()).optional(),
+    archetypes: z.record(z.string()).optional(),
+    ingest_now: z.boolean().optional().default(false),
+  })
+);
+export type RegisterDomainArgs = z.infer<typeof RegisterDomainSchema>;
+
+export function normalizeRegisterFeatureArgs(raw: unknown): unknown {
+  if (typeof raw !== "object" || raw === null || Array.isArray(raw)) return raw;
+  const obj = { ...(raw as Record<string, unknown>) };
+
+  const aliases: Record<string, string> = {
+    feature: "feature_key",
+    featureKey: "feature_key",
+    feature_name: "feature_key",
+    featureName: "feature_key",
+    feat: "feature_key",
+    name: "feature_key",
+    key: "feature_key",
+    domain_name: "domain",
+    domainName: "domain",
+    touchpoints: "allowed_touchpoints",
+    allowedTouchpoints: "allowed_touchpoints",
+    touchpoint: "allowed_touchpoints",
+    files: "allowed_touchpoints",
+    allowed_files: "allowed_touchpoints",
+    reuseSymbols: "reuse_symbols",
+    reuse: "reuse_symbols",
+    symbols: "reuse_symbols",
+    inputContract: "input_contract",
+    outputContract: "output_contract",
+    persistToConfig: "persist_to_config",
+    persist: "persist_to_config",
+    workspace: "workspace_path",
+    workspacePath: "workspace_path",
+    project_root: "workspace_path",
+    projectRoot: "workspace_path",
+  };
+
+  for (const [alias, target] of Object.entries(aliases)) {
+    if (obj[alias] !== undefined && obj[target] === undefined) {
+      obj[target] = obj[alias];
+    }
+  }
+
+  // Symmetrical contract: feature_key <-> feature
+  if (obj.feature_key !== undefined && obj.feature === undefined) {
+    obj.feature = obj.feature_key;
+  } else if (obj.feature !== undefined && obj.feature_key === undefined) {
+    obj.feature_key = obj.feature;
+  }
+
+  // Symmetrical contract: domain <-> domain_name
+  if (obj.domain !== undefined && obj.domain_name === undefined) {
+    obj.domain_name = obj.domain;
+  } else if (obj.domain_name !== undefined && obj.domain === undefined) {
+    obj.domain = obj.domain_name;
+  }
+
+  // Array coercion for allowed_touchpoints
+  if (typeof obj.allowed_touchpoints === "string") {
+    obj.allowed_touchpoints = [obj.allowed_touchpoints];
+  } else if (!Array.isArray(obj.allowed_touchpoints) && obj.allowed_touchpoints !== undefined) {
+    obj.allowed_touchpoints = [];
+  }
+
+  // Array coercion for reuse_symbols
+  if (typeof obj.reuse_symbols === "string") {
+    obj.reuse_symbols = [obj.reuse_symbols];
+  }
+
+  return obj;
+}
+
+export const RegisterFeatureSchema = z.preprocess(
+  normalizeRegisterFeatureArgs,
+  z.object({
+    feature_key: z.string().min(1, "Feature key is required"),
+    feature: z.string().optional(),
+    domain: z.string().min(1, "Domain name is required"),
+    domain_name: z.string().optional(),
+    description: z.string().optional(),
+    allowed_touchpoints: z.array(z.string()).default([]),
+    reuse_symbols: z.array(z.string()).optional(),
+    input_contract: z.record(z.unknown()).optional(),
+    output_contract: z.record(z.unknown()).optional(),
+    persist_to_config: z.boolean().optional().default(false),
+    workspace_path: z.string().optional(),
+  })
+);
+export type RegisterFeatureArgs = z.infer<typeof RegisterFeatureSchema>;
+
+export const AutoDiscoverDomainsSchema = z.preprocess(
+  (args) =>
+    withAliases(args, {
+      save: "persist",
+    }),
+  z.object({
+    persist: z.boolean().optional().default(true),
+  })
+);
+export type AutoDiscoverDomainsArgs = z.infer<typeof AutoDiscoverDomainsSchema>;
+
+function normalizeTopologyArgs(raw: unknown): unknown {
+  if (typeof raw !== "object" || raw === null || Array.isArray(raw)) return raw;
+  const obj = { ...(raw as Record<string, unknown>) };
+  if (obj.resolve_node !== undefined && obj.resolveNode === undefined) {
+    obj.resolveNode = obj.resolve_node;
+  }
+  if (obj.resolve_layer !== undefined && obj.resolve === undefined) {
+    obj.resolve = obj.resolve_layer;
+  }
+  if (obj.resolveLayer !== undefined && obj.resolve === undefined) {
+    obj.resolve = obj.resolveLayer;
+  }
+  if (typeof obj.resolveNode === "object" && obj.resolveNode !== null) {
+    obj.resolveNode = withAliases(obj.resolveNode, {
+      canonicalTag: "canonical_tag",
+      domainOrIp: "domain_or_ip",
+      connectedTo: "connected_to",
+      nodeId: "id",
+      node_id: "id",
+      nodeName: "name",
+      node_name: "name",
+    });
+  }
+  return obj;
+}
+
+export const GetEnvironmentTopologySchema = z.preprocess(
+  (args) => normalizeTopologyArgs(args),
+  z.object({
+    resolve: z
+      .object({
+        layer: z.enum(["edge", "gateway", "host", "runtime", "storage", "telemetry"]),
+        platform: z.string().min(1, "platform name cannot be empty"),
+        constraints: z.array(z.string()).optional(),
+      })
+      .optional(),
+    resolveNode: z
+      .object({
+        id: z.string().min(1, "Node id cannot be empty"),
+        layer: z.enum(["edge", "gateway", "host", "runtime", "storage", "telemetry"]),
+        name: z.string().min(1, "Node name cannot be empty"),
+        platform: z.string().min(1, "Platform cannot be empty"),
+        canonical_tag: z
+          .enum([
+            "shared_hosting",
+            "vps",
+            "docker",
+            "serverless",
+            "kubernetes",
+            "managed_db",
+            "cdn_edge",
+            "paas",
+            "static_cdn",
+            "unspecified",
+          ])
+          .optional(),
+        domain_or_ip: z.string().optional(),
+        role: z.string().optional(),
+        constraints: z.array(z.string()).optional(),
+        connected_to: z.array(z.string()).optional(),
+      })
+      .optional(),
+    refresh: z.boolean().optional().default(false),
+  })
+);
+export type GetEnvironmentTopologyArgs = z.infer<typeof GetEnvironmentTopologySchema>;
+
+