@amerged/ohmyhost-mcp 0.1.20 → 0.1.22

package/src/apps/product-cli/command.ts

@@ -127,6 +127,11 @@ export type ProductCliCommand =
       readonly organizationId: string;
       readonly credentialStore: "native";
     }
+  | {
+      readonly kind: "referral-link";
+      readonly organizationId: string;
+      readonly credentialStore: "native";
+    }
   | {
       readonly kind: "credits-balance";
       readonly organizationId: string;
@@ -256,6 +261,18 @@ export type ProductCliCommand =
       readonly projectId: string;
       readonly credentialStore: "native";
     }
+  | {
+      readonly kind: "project-flag-status";
+      readonly projectId: string;
+      readonly credentialStore: "native";
+    }
+  | {
+      readonly kind: "project-flag-set";
+      readonly projectId: string;
+      readonly enabled: boolean;
+      readonly idempotencyKey: string;
+      readonly credentialStore: "native";
+    }
   | {
       readonly kind: "database-write";
       readonly projectId: string;
@@ -549,6 +566,7 @@ export type RecognizedCommandLabel =
   | "version"
   | "whoami"
   | "credits account"
+  | "referral link"
   | "credits balance"
   | "credits usage"
   | "budget get"
@@ -577,6 +595,8 @@ export type RecognizedCommandLabel =
   | "project dev-share rotate"
   | "project dev-share revoke"
   | "project dev-access mode"
+  | "project flag status"
+  | "project flag set"
   | "project handle check"
   | "project handle set"
   | "database write"
@@ -656,6 +676,7 @@ export const recognizedCommandLabel = (argv: readonly string[]): RecognizedComma
     if (value === "github" && values[0] === "status") return "github status";
     if (value === "operation" && values[0] === "get") return "operation get";
     if (value === "credits" && values[0] === "account") return "credits account";
+    if (value === "referral" && values[0] === "link") return "referral link";
     if (value === "credits" && values[0] === "balance") return "credits balance";
     if (value === "billing" && values[0] === "recharge" && values[1] === "get")
       return "billing recharge get";
@@ -701,6 +722,10 @@ export const recognizedCommandLabel = (argv: readonly string[]): RecognizedComma
       return "project dev-access mode";
     if (value === "project" && values[0] === "dev-access" && values[1] === "create")
       return "project dev-access create";
+    if (value === "project" && values[0] === "flag" && values[1] === "status")
+      return "project flag status";
+    if (value === "project" && values[0] === "flag" && values[1] === "set")
+      return "project flag set";
     if (value === "database" && values[0] === "compute" && values[1] === "set")
       return "database compute set";
     if (value === "database" && values[0] === "compute" && values[1] === "get")
@@ -1080,6 +1105,12 @@ export const parseProductCliCommand = (argv: readonly string[]): ProductCliComma
     const options = parseOptions(values, ["--organization"]);
     return { kind: "credits-account", organizationId: options["--organization"], credentialStore };
   }
+  if (first === "referral" && values[0] === "link") {
+    values.shift();
+    if (yes || wait || follow || stdin) throw new InvalidCommandError();
+    const options = parseOptions(values, ["--organization"]);
+    return { kind: "referral-link", organizationId: options["--organization"], credentialStore };
+  }
   if (first === "credits" && values.shift() === "balance") {
     if (yes || wait || follow || stdin) throw new InvalidCommandError();
     const options = parseOptions(values, ["--organization"]);
@@ -1266,6 +1297,27 @@ export const parseProductCliCommand = (argv: readonly string[]): ProductCliComma
         credentialStore,
       };
     }
+    if (subcommand === "flag") {
+      const action = values.shift();
+      if (yes || wait || follow || stdin) throw new InvalidCommandError();
+      if (action === "status") {
+        const options = parseOptions(values, ["--project"]);
+        return { kind: "project-flag-status", projectId: options["--project"], credentialStore };
+      }
+      if (action === "set") {
+        const options = parseOptions(values, ["--project", "--enabled", "--idempotency-key"]);
+        if (options["--enabled"] !== "true" && options["--enabled"] !== "false")
+          throw new InvalidCommandError();
+        return {
+          kind: "project-flag-set",
+          projectId: options["--project"],
+          enabled: options["--enabled"] === "true",
+          idempotencyKey: options["--idempotency-key"],
+          credentialStore,
+        };
+      }
+      throw new InvalidCommandError();
+    }
     if (subcommand === "dev-access" && values.shift() === "create") {
       if (yes || wait || follow || stdin) throw new InvalidCommandError();
       const options = parseOptions(values, ["--project"]);

package/src/apps/product-cli/index.ts

@@ -14,6 +14,8 @@ export {
   parseCurrentIdentity,
   parseDevAccessTicket,
   parseDevAccessState,
+  parsePoweredByFlag,
+  parseOrganizationReferral,
   parseCloudflareDnsAuthorization,
   parseCloudflareDnsAuthorizationStatus,
   parsePaidDomain,

package/src/apps/product-cli/product-api.ts

@@ -45,6 +45,7 @@ import {
   type BillingPortal,
   getOrganizationCredits,
   getOrganizationAccount,
+  getOrganizationReferral,
   type OrganizationAccount,
   getOrganizationCreditUsage,
   type OrganizationCreditUsage,
@@ -61,6 +62,8 @@ import {
   rotateProjectDevShareLink,
   revokeProjectDevShareLink,
   setProjectDevAccessMode,
+  getProjectPoweredByFlag,
+  setProjectPoweredByFlag,
   createDeployment,
   createProject,
   submitFeedback,
@@ -193,6 +196,9 @@ export interface ProductApi {
   }): Promise<import("@ohmyhost/sdk-ts").BillingRecharge>;
   createBillingPortal(organizationId: string): Promise<BillingPortal>;
   getOrganizationAccount(organizationId: string): Promise<OrganizationAccount>;
+  getOrganizationReferral(
+    organizationId: string,
+  ): Promise<import("@ohmyhost/sdk-ts").OrganizationReferral>;
   getOrganizationCredits(organizationId: string): Promise<OrganizationCredits>;
   getOrganizationCreditUsage(input: {
     organizationId: string;
@@ -277,6 +283,12 @@ export interface ProductApi {
     mode: "protected" | "public",
     idempotencyKey: string,
   ): Promise<import("@ohmyhost/sdk-ts").DevAccessState>;
+  getPoweredByFlag(projectId: string): Promise<import("@ohmyhost/sdk-ts").PoweredByFlag>;
+  setPoweredByFlag(
+    projectId: string,
+    enabled: boolean,
+    idempotencyKey: string,
+  ): Promise<import("@ohmyhost/sdk-ts").PoweredByFlag>;
   submitFeedback(input: {
     report: import("@ohmyhost/sdk-ts").FeedbackSubmission;
     idempotencyKey: string;
@@ -489,6 +501,9 @@ export class GeneratedSdkProductApi implements ProductApi {
     return parseAccountProfile(await getAccountProfile({ client: this.#client }));
   }
 
+  public getOrganizationReferral(organizationId: string) {
+    return getOrganizationReferral({ organization_id: organizationId }, { client: this.#client });
+  }
   public getOrganizationAccount(organizationId: string) {
     return getOrganizationAccount({ organization_id: organizationId }, { client: this.#client });
   }
@@ -790,6 +805,19 @@ export class GeneratedSdkProductApi implements ProductApi {
       { client: this.#client },
     );
   }
+  public async getPoweredByFlag(projectId: string) {
+    return getProjectPoweredByFlag({ project_id: projectId }, { client: this.#client });
+  }
+  public async setPoweredByFlag(projectId: string, enabled: boolean, idempotencyKey: string) {
+    return setProjectPoweredByFlag(
+      {
+        project_id: projectId,
+        "Idempotency-Key": idempotencyKey,
+        setPoweredByFlagRequest: { enabled },
+      },
+      { client: this.#client },
+    );
+  }
   public async createDevAccessTicket(projectId: string): Promise<DevAccessTicket> {
     return createProjectDevAccessTicket({ project_id: projectId }, { client: this.#client });
   }

package/src/apps/product-cli/runtime-contract.ts

@@ -273,6 +273,12 @@ export const PROBLEM_POLICY = {
     retryable: false,
     action: "Finish database provisioning, then request the export again. No backup was accepted.",
   },
+  powered_by_flag_required: {
+    status: 409,
+    retryable: false,
+    action:
+      "Keep the flag on, remove the custom domain with 'ohmyhost domain paid delete', or buy Paid, then switch the flag off.",
+  },
   rate_limited: { status: 429, retryable: true, action: "Retry after the server delay." },
   insufficient_organization_credits: {
     status: 402,
@@ -289,7 +295,7 @@ export const PROBLEM_POLICY = {
     status: 402,
     retryable: false,
     action:
-      "Ask the organization Owner to activate Paid through billing checkout, then retry. Top-ups and signup credits do not activate Paid.",
+      "Ask the organization Owner to activate Paid through billing checkout, then retry. Top-ups and signup credits do not activate Paid. For a website domain alone, a Free workspace may instead show the powered-by flag on the project.",
   },
   interactive_login_required: {
     status: 403,
@@ -740,6 +746,39 @@ export const parseDevAccessState = (value: unknown): PublicDevAccessState => {
   return input as unknown as PublicDevAccessState;
 };
 
+export interface PublicPoweredByFlag {
+  readonly project_id: string;
+  readonly enabled: boolean;
+  readonly enabled_at: string | null;
+}
+
+export const parsePoweredByFlag = (value: unknown): PublicPoweredByFlag => {
+  const input = exactRecord(value, ["project_id", "enabled", "enabled_at"]);
+  if (
+    !isUlid(input["project_id"]) ||
+    typeof input["enabled"] !== "boolean" ||
+    (input["enabled"] ? !isIsoInstant(input["enabled_at"]) : input["enabled_at"] !== null)
+  )
+    throw new ResponseContractError();
+  return input as unknown as PublicPoweredByFlag;
+};
+
+export interface PublicOrganizationReferral {
+  readonly organization_id: string;
+  readonly referral_url: string;
+}
+
+export const parseOrganizationReferral = (value: unknown): PublicOrganizationReferral => {
+  const input = exactRecord(value, ["organization_id", "referral_url"]);
+  if (
+    !isUlid(input["organization_id"]) ||
+    input["referral_url"] !==
+      `https://ohmyho.st/?r=ref-${String(input["organization_id"]).toLowerCase()}`
+  )
+    throw new ResponseContractError();
+  return input as unknown as PublicOrganizationReferral;
+};
+
 export const parseDevAccessTicket = (value: unknown): PublicDevAccessTicket => {
   const input = exactRecord(value, [
     "ticket_id",
@@ -1141,6 +1180,8 @@ const parseProjectDeployment = (value: unknown): void => {
 
 export interface PublicOperation {
   readonly blocking_operation_id?: string;
+  /** The deployment this operation created; its diagnostics explain a failure. */
+  readonly deployment_id?: string;
   readonly progress?: OperationDeploymentProgress;
   readonly id: string;
   readonly state: "queued" | "running" | "succeeded" | "failed" | "cancelled";
@@ -1580,6 +1621,7 @@ export const parseSafeProblem = (value: unknown): SafeProblem => {
 export const parseOperation = (value: unknown): PublicOperation => {
   const input = exactRecord(value, [
     "blocking_operation_id",
+    "deployment_id",
     "id",
     "state",
     "created_at",
@@ -1595,9 +1637,11 @@ export const parseOperation = (value: unknown): PublicOperation => {
   const updatedAt = input["updated_at"];
   const result = input["result"];
   const blockingOperationId = input["blocking_operation_id"];
+  const deploymentId = input["deployment_id"];
   if (
-    blockingOperationId !== undefined &&
-    (state !== "queued" || !isUlid(blockingOperationId) || blockingOperationId === id)
+    (blockingOperationId !== undefined &&
+      (state !== "queued" || !isUlid(blockingOperationId) || blockingOperationId === id)) ||
+    (deploymentId !== undefined && !isUlid(deploymentId))
   )
     throw new ResponseContractError();
   if (
@@ -1659,6 +1703,7 @@ export const parseOperation = (value: unknown): PublicOperation => {
     ...(blockingOperationId === undefined
       ? {}
       : { blocking_operation_id: blockingOperationId as string }),
+    ...(deploymentId === undefined ? {} : { deployment_id: deploymentId as string }),
     ...(progress === undefined ? {} : { progress }),
     ...(result === undefined ? {} : { result }),
     ...(error === undefined ? {} : { error }),