@eide/foir-cli 0.14.1 → 0.15.1

package/dist/cli.js

@@ -2099,8 +2099,10 @@ import {
   ForceAssignExperimentRequestSchema,
   RemoveExperimentAssignmentRequestSchema,
   ApplyExperimentWinnerRequestSchema,
-  GetCustomerAssignmentsRequestSchema,
-  ExperimentVariantSchema
+  GetAssignmentsRequestSchema,
+  ListExperimentDeclarationsRequestSchema,
+  ExperimentVariantSchema,
+  ExperimentGoalSchema
 } from "@eide/foir-proto-ts/experiments/v1/experiments_pb";
 import { ExperimentStatus as ExperimentStatus2 } from "@eide/foir-proto-ts/experiments/v1/experiments_pb";
 var STATUS_TO_PROTO = {
@@ -2156,6 +2158,14 @@ function createExperimentsMethods(client) {
               name: v.name,
               percent: v.percent
             })
+          ) ?? [],
+          assignmentPersistence: params.assignmentPersistence,
+          goals: params.goals?.map(
+            (g) => create7(ExperimentGoalSchema, {
+              key: g.key,
+              name: g.name,
+              description: g.description
+            })
           ) ?? []
         })
       );
@@ -2175,7 +2185,16 @@ function createExperimentsMethods(client) {
               name: v.name,
               percent: v.percent
             })
-          )
+          ),
+          assignmentPersistence: params.assignmentPersistence,
+          goals: params.goals?.map(
+            (g) => create7(ExperimentGoalSchema, {
+              key: g.key,
+              name: g.name,
+              description: g.description
+            })
+          ) ?? [],
+          goalsClear: params.goalsClear ?? false
         })
       );
       return resp.experiment ?? null;
@@ -2221,30 +2240,41 @@ function createExperimentsMethods(client) {
       return resp.experiment ?? null;
     },
     // ── Assignments ──────────────────────────────────────────
-    async forceAssignExperiment(customerId, experimentId, variantKey) {
+    async forceAssignExperiment(identity, experimentId, variantKey) {
       const resp = await client.forceAssignExperiment(
         create7(ForceAssignExperimentRequestSchema, {
-          customerId,
+          customerId: identity.customerId,
+          viewerKey: identity.viewerKey,
           experimentId,
           variantKey
         })
       );
       return resp.assignment ?? null;
     },
-    async removeExperimentAssignment(customerId, experimentId) {
+    async removeExperimentAssignment(identity, experimentId) {
       const resp = await client.removeExperimentAssignment(
         create7(RemoveExperimentAssignmentRequestSchema, {
-          customerId,
+          customerId: identity.customerId,
+          viewerKey: identity.viewerKey,
           experimentId
         })
       );
       return resp.success;
     },
-    async getCustomerAssignments(customerId) {
-      const resp = await client.getCustomerAssignments(
-        create7(GetCustomerAssignmentsRequestSchema, { customerId })
+    async getAssignments(identity) {
+      const resp = await client.getAssignments(
+        create7(GetAssignmentsRequestSchema, {
+          customerId: identity.customerId,
+          viewerKey: identity.viewerKey
+        })
       );
       return resp.assignments ?? [];
+    },
+    async listExperimentDeclarations() {
+      const resp = await client.listExperimentDeclarations(
+        create7(ListExperimentDeclarationsRequestSchema, {})
+      );
+      return resp.declarations ?? [];
     }
   };
 }
@@ -7188,7 +7218,8 @@ function registerSegmentsCommands(program2, globalOpts) {
 // src/commands/experiments.ts
 import {
   ExperimentSchema,
-  ExperimentStatsSchema
+  ExperimentStatsSchema,
+  ExperimentAssignmentSchema
 } from "@eide/foir-proto-ts/experiments/v1/experiments_pb";
 function registerExperimentsCommands(program2, globalOpts) {
   const experiments = program2.command("experiments").description("Manage experiments");
@@ -7336,6 +7367,104 @@ function registerExperimentsCommands(program2, globalOpts) {
       formatOutputProto(ExperimentStatsSchema, result, opts);
     })
   );
+  experiments.command("force-assign <experimentId>").description(
+    "Force a known identity into a specific variant (admin override)."
+  ).requiredOption("--variant <key>", "Variant key to assign").option("--customer-id <id>", "Customer identity to bucket").option(
+    "--viewer-key <key>",
+    "Anonymous viewer identity (cookie/localStorage value) to bucket"
+  ).action(
+    withErrorHandler(
+      globalOpts,
+      async (experimentId, cmdOpts) => {
+        const opts = globalOpts();
+        if (!cmdOpts.customerId && !cmdOpts.viewerKey) {
+          throw new Error(
+            "Provide exactly one of --customer-id or --viewer-key."
+          );
+        }
+        if (cmdOpts.customerId && cmdOpts.viewerKey) {
+          throw new Error(
+            "Provide exactly one of --customer-id or --viewer-key, not both."
+          );
+        }
+        const client = await createPlatformClient(opts);
+        const result = await client.experiments.forceAssignExperiment(
+          { customerId: cmdOpts.customerId, viewerKey: cmdOpts.viewerKey },
+          experimentId,
+          cmdOpts.variant
+        );
+        if (!result) throw new Error("Force-assign returned no assignment.");
+        formatOutputProto(ExperimentAssignmentSchema, result, opts);
+        if (!(opts.json || opts.jsonl || opts.quiet))
+          success(
+            `Assigned ${cmdOpts.customerId ? "customer" : "viewer"} ${cmdOpts.customerId ?? cmdOpts.viewerKey} \u2192 ${cmdOpts.variant}`
+          );
+      }
+    )
+  );
+  experiments.command("remove-assignment <experimentId>").description("Remove the assignment for a known identity.").option("--customer-id <id>", "Customer identity to clear").option("--viewer-key <key>", "Anonymous viewer identity to clear").action(
+    withErrorHandler(
+      globalOpts,
+      async (experimentId, cmdOpts) => {
+        const opts = globalOpts();
+        if (!cmdOpts.customerId && !cmdOpts.viewerKey) {
+          throw new Error(
+            "Provide exactly one of --customer-id or --viewer-key."
+          );
+        }
+        if (cmdOpts.customerId && cmdOpts.viewerKey) {
+          throw new Error(
+            "Provide exactly one of --customer-id or --viewer-key, not both."
+          );
+        }
+        const client = await createPlatformClient(opts);
+        const ok = await client.experiments.removeExperimentAssignment(
+          { customerId: cmdOpts.customerId, viewerKey: cmdOpts.viewerKey },
+          experimentId
+        );
+        if (opts.json || opts.jsonl)
+          formatOutput({ removed: ok, experimentId }, opts);
+        else if (ok) success("Assignment removed");
+        else success("No matching assignment found");
+      }
+    )
+  );
+  experiments.command("assignments").description("List assignments for a single identity across experiments.").option("--customer-id <id>", "Customer identity to look up").option("--viewer-key <key>", "Anonymous viewer identity to look up").action(
+    withErrorHandler(
+      globalOpts,
+      async (cmdOpts) => {
+        const opts = globalOpts();
+        if (!cmdOpts.customerId && !cmdOpts.viewerKey) {
+          throw new Error(
+            "Provide exactly one of --customer-id or --viewer-key."
+          );
+        }
+        if (cmdOpts.customerId && cmdOpts.viewerKey) {
+          throw new Error(
+            "Provide exactly one of --customer-id or --viewer-key, not both."
+          );
+        }
+        const client = await createPlatformClient(opts);
+        const data = await client.experiments.getAssignments({
+          customerId: cmdOpts.customerId,
+          viewerKey: cmdOpts.viewerKey
+        });
+        formatListProto(ExperimentAssignmentSchema, data, opts, {
+          columns: [
+            { key: "experimentKey", header: "Experiment", width: 24 },
+            { key: "variantKey", header: "Variant", width: 16 },
+            { key: "assignmentMethod", header: "Method", width: 10 },
+            {
+              key: "assignedAt",
+              header: "Assigned",
+              width: 12,
+              format: (v) => timeAgo(v)
+            }
+          ]
+        });
+      }
+    )
+  );
 }
 
 // src/commands/schedules.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eide/foir-cli",
-  "version": "0.14.1",
+  "version": "0.15.1",
   "description": "Universal platform CLI for Foir platform",
   "type": "module",
   "publishConfig": {
@@ -50,7 +50,7 @@
     "@bufbuild/protovalidate": "^1.1.1",
     "@connectrpc/connect": "^2.0.0",
     "@connectrpc/connect-node": "^2.0.0",
-    "@eide/foir-proto-ts": "^0.40.0",
+    "@eide/foir-proto-ts": "^0.43.0",
     "chalk": "^5.3.0",
     "commander": "^12.1.0",
     "dotenv": "^16.4.5",