t3-code-android-nightly/.repos/alchemy-effect/packages/alchemy/test/GCP/AndroidDeviceProvisioning/common.ts

import * as androiddeviceprovisioning from "@distilled.cloud/gcp/androiddeviceprovisioning_v1";
import * as Effect from "effect/Effect";
import { MinimumLogLevel } from "effect/References";
import * as Schedule from "effect/Schedule";

export const logLevel = Effect.provideService(
  MinimumLogLevel,
  process.env.DEBUG ? "Debug" : "Info",
);

const lastSegment = (value: string) => {
  const trimmed = value.replace(/\/+$/, "");
  const parts = trimmed.split("/");
  return parts[parts.length - 1] || trimmed;
};

export const toCustomerName = (value: string) => {
  const trimmed = value.replace(/\/+$/, "").trim();
  if (trimmed.length === 0) return "";
  const idx = trimmed.lastIndexOf("/customers/");
  if (idx >= 0) {
    const id = trimmed.slice(idx + "/customers/".length).split("/")[0] ?? "";
    return id.length > 0 ? `customers/${id}` : "";
  }
  if (trimmed.startsWith("customers/")) {
    const id = trimmed.slice("customers/".length).split("/")[0] ?? "";
    return id.length > 0 ? `customers/${id}` : trimmed;
  }
  return `customers/${lastSegment(trimmed)}`;
};

export const customerName = (() => {
  const raw =
    process.env.GCP_ANDROIDDEVICEPROVISIONING_CUSTOMER?.trim() ||
    process.env.GCP_ANDROIDDEVICEPROVISIONING_CUSTOMER_ID?.trim();
  return raw ? toCustomerName(raw) : undefined;
})();

// Needs a zero-touch customer the credentials administer
// (GCP_ANDROIDDEVICEPROVISIONING_CUSTOMER); zero-touch is a partner/reseller
// portal with no self-serve customer for a service account.
export const runLifecycle = !!customerName;

export const probeParent = "customers/0";
export const probeName = `${probeParent}/configurations/0`;

const ANDROID_DEVICE_POLICY_PACKAGE = "com.google.android.apps.work.clouddpc";

export const resolveDpc = (parent: string) =>
  androiddeviceprovisioning.listCustomersDpcs({ parent }).pipe(
    Effect.map((page) => {
      const dpcs = page.dpcs ?? [];
      const preferred = dpcs.find(
        (dpc) => dpc.packageName === ANDROID_DEVICE_POLICY_PACKAGE && dpc.name,
      );
      return preferred?.name ?? dpcs.find((dpc) => dpc.name)?.name;
    }),
    Effect.catchTag("NotFound", () => Effect.succeed(undefined)),
    Effect.map((name) => name ?? `${parent}/dpcs/alchemy-missing`),
  );

export const waitUntilGone = (name: string) =>
  androiddeviceprovisioning.getCustomersConfigurations({ name }).pipe(
    Effect.as("found" as const),
    Effect.catchTag("NotFound", () => Effect.succeed("gone" as const)),
    Effect.repeat({
      schedule: Schedule.spaced("1 second"),
      until: (status) => status === "gone",
      times: 10,
    }),
  );