@sanity/workbench 0.1.0-alpha.12 → 0.1.0-alpha.13

package/src/system/root.machine.ts

@@ -0,0 +1,165 @@
+import { createSanityInstance, type SanityInstance } from "@sanity/sdk";
+import { raise, sendTo, setup, type ActorOptions } from "xstate";
+
+import { authLogic } from "./auth.machine";
+import { inspect } from "./inspect";
+import {
+  telemetryLogic,
+  type WorkbenchUserProperties,
+} from "./telemetry.machine";
+
+type OSInput = WorkbenchUserProperties;
+type OSContext = {
+  instance: SanityInstance;
+  userProperties: WorkbenchUserProperties;
+};
+
+/**
+ * The base inputs for the OS machine.
+ * @internal
+ */
+export interface OSBaseInput extends Pick<OSContext, "instance"> {}
+
+/**
+ * The sanity OS machine, responsible for managing the global state of the OS.
+ * @public
+ * @example
+ * ```ts
+ * import { os, createOSOptions } from "@sanity/workbench/system";
+ * import { useActor } from "@xstate/react";
+ *
+ * const [state, send] = useActor(os, createOSOptions());
+ * ```
+ */
+export const os = setup({
+  types: {
+    input: {} as OSInput,
+    context: {} as OSContext,
+    events: {} as
+      | { type: "boot.auth.ready" }
+      | { type: "boot.auth.failed" }
+      | { type: "boot.telemetry.ready" },
+    // https://github.com/statelyai/xstate/issues/5515
+    children: {} as {
+      auth: "auth";
+      telemetry: "telemetry";
+    },
+  },
+  actors: {
+    auth: authLogic,
+    telemetry: telemetryLogic,
+  },
+  guards: {
+    hasTag: (_, params: { hasTag: boolean }) => params.hasTag,
+  },
+  actions: {
+    raiseAuthReady: raise({ type: "boot.auth.ready" }),
+    raiseAuthFailed: raise({ type: "boot.auth.failed" }),
+    raiseTelemetryReady: raise({
+      type: "boot.telemetry.ready",
+    }),
+    startTelemetry: sendTo("telemetry", {
+      type: "telemetry.start",
+    }),
+  },
+}).createMachine({
+  id: "os",
+  context: ({ input }) => ({
+    instance: createSanityInstance(),
+    userProperties: {
+      version: input.version,
+      organizationId: input.organizationId,
+      environment: input.environment,
+      userAgent: input.userAgent,
+    },
+  }),
+  initial: "booting",
+  invoke: [
+    {
+      id: "auth",
+      systemId: "auth",
+      src: "auth",
+      input: ({ context }) => ({ instance: context.instance }),
+      onSnapshot: [
+        {
+          guard: {
+            type: "hasTag",
+            params: ({ event }) => ({
+              hasTag: event.snapshot.hasTag("authenticated"),
+            }),
+          },
+          actions: [{ type: "raiseAuthReady" }],
+        },
+        {
+          guard: {
+            type: "hasTag",
+            params: ({ event }) => ({
+              hasTag: event.snapshot.hasTag("error"),
+            }),
+          },
+          actions: [{ type: "raiseAuthFailed" }],
+        },
+      ],
+    },
+    {
+      id: "telemetry",
+      systemId: "telemetry",
+      src: "telemetry",
+      input: ({ context }) => ({
+        instance: context.instance,
+        ...context.userProperties,
+      }),
+      onSnapshot: [
+        {
+          guard: {
+            type: "hasTag",
+            params: ({ event }) => ({
+              hasTag: event.snapshot.hasTag("telemetry-resolved"),
+            }),
+          },
+          actions: [{ type: "raiseTelemetryReady" }],
+        },
+      ],
+    },
+  ],
+  states: {
+    booting: {
+      initial: "auth",
+      states: {
+        auth: {
+          on: {
+            "boot.auth.ready": {
+              target: "telemetry",
+              actions: [{ type: "startTelemetry" }],
+            },
+            "boot.auth.failed": { target: "error" },
+          },
+        },
+        telemetry: {
+          on: {
+            "boot.telemetry.ready": { target: "done" },
+          },
+        },
+        error: {},
+        done: { type: "final" },
+      },
+      onDone: { target: "running" },
+    },
+    running: {
+      type: "parallel",
+    },
+  },
+});
+
+/**
+ * Creates a set of default options for the OS machine, including passing
+ * the sanity configuration for the internal instance.
+ * @public
+ */
+export function createOSOptions(input: OSInput) {
+  return {
+    id: "os",
+    input,
+    inspect,
+  } satisfies ActorOptions<typeof os>;
+}

package/src/system/telemetry.machine.ts

@@ -0,0 +1,179 @@
+import { getClient, type SanityInstance } from "@sanity/sdk";
+import {
+  type ConsentStatus,
+  createBatchedStore,
+  createSessionId,
+  type TelemetryEvent,
+  type TelemetryStore,
+} from "@sanity/telemetry";
+import { assign, fromPromise, setup } from "xstate";
+
+import type { OSBaseInput } from "./root.machine";
+
+/**
+ * @public
+ */
+export type WorkbenchUserProperties = {
+  version: string;
+  organizationId: string;
+  environment: string;
+  userAgent: string;
+};
+
+/**
+ * @internal
+ */
+export interface TelemetryInput extends OSBaseInput, WorkbenchUserProperties {}
+
+/**
+ * TODO: this shouldn't be unique to the telemetry machine,
+ * remove this and set it globally with a single client actor.
+ */
+const TELEMETRY_API_VERSION = "2024-11-12";
+/**
+ * 30 seconds, in milliseconds, is the default flush interval
+ * for the telemetry store.
+ */
+const FLUSH_INTERVAL_MS = 30_000;
+
+type ConsentResult = { status: ConsentStatus };
+
+const checkConsentLogic = fromPromise<
+  ConsentResult,
+  { instance: SanityInstance }
+>(async ({ input, signal }) => {
+  try {
+    const client = getClient(input.instance, {
+      apiVersion: TELEMETRY_API_VERSION,
+    });
+    return await client.request<ConsentResult>({
+      uri: "/intake/telemetry-status",
+      tag: "telemetry-consent",
+      signal,
+    });
+  } catch {
+    return { status: "undetermined" } satisfies ConsentResult;
+  }
+});
+
+type TelemetryContext = {
+  instance: SanityInstance;
+  store: TelemetryStore<WorkbenchUserProperties> | null;
+  userProperties: WorkbenchUserProperties;
+};
+
+type TelemetryMachineEvent =
+  | { type: "telemetry.start" }
+  | { type: "telemetry.stop" };
+
+export const telemetryLogic = setup({
+  types: {
+    input: {} as TelemetryInput,
+    context: {} as TelemetryContext,
+    events: {} as TelemetryMachineEvent,
+  },
+  actors: {
+    checkConsent: checkConsentLogic,
+  },
+  guards: {
+    isConsentGranted: (_, params: { status: ConsentStatus }) =>
+      params.status === "granted",
+  },
+  actions: {
+    createStore: assign({
+      store: ({ context }) => {
+        const sessionId = createSessionId();
+        const client = getClient(context.instance, {
+          apiVersion: TELEMETRY_API_VERSION,
+        });
+        const store = createBatchedStore<WorkbenchUserProperties>(sessionId, {
+          flushInterval: FLUSH_INTERVAL_MS,
+          resolveConsent: () =>
+            client.request<{ status: ConsentStatus }>({
+              uri: "/intake/telemetry-status",
+              tag: "telemetry-consent",
+            }),
+          sendEvents: (batch: TelemetryEvent[]) =>
+            client.request({
+              uri: "/intake/batch",
+              method: "POST",
+              body: { batch },
+              tag: "telemetry.batch",
+            }),
+          sendBeacon: (batch: TelemetryEvent[]) => {
+            if (typeof navigator === "undefined") {
+              return false;
+            }
+            return navigator.sendBeacon(
+              client.getUrl("/intake/batch"),
+              JSON.stringify({ batch }),
+            );
+          },
+        });
+        store.logger.updateUserProperties(context.userProperties);
+        return store;
+      },
+    }),
+    teardownStore: ({ context }) => {
+      context.store?.end();
+    },
+  },
+}).createMachine({
+  id: "telemetry",
+  initial: "idle",
+  context: ({ input }) => ({
+    instance: input.instance,
+    store: null,
+    userProperties: {
+      version: input.version,
+      organizationId: input.organizationId,
+      environment: input.environment,
+      userAgent: input.userAgent,
+    },
+  }),
+  states: {
+    idle: {
+      on: {
+        "telemetry.start": {
+          target: "checkingConsent",
+        },
+      },
+    },
+    checkingConsent: {
+      invoke: {
+        src: "checkConsent",
+        input: ({ context }) => ({
+          instance: context.instance,
+        }),
+        onDone: [
+          {
+            guard: {
+              type: "isConsentGranted",
+              params: ({ event }) => ({
+                status: event.output.status,
+              }),
+            },
+            actions: [{ type: "createStore" }],
+            target: "active",
+          },
+          {
+            target: "denied",
+          },
+        ],
+      },
+    },
+    active: {
+      tags: ["telemetry-resolved"],
+      exit: [{ type: "teardownStore" }],
+      on: {
+        "telemetry.stop": { target: "stopped" },
+      },
+    },
+    stopped: {
+      type: "final",
+    },
+    denied: {
+      tags: ["telemetry-resolved"],
+    },
+  },
+});