@wictorwilen/cocogen 1.0.0 → 1.0.11

package/dist/init/templates/ts/src/cli.ts.ejs

@@ -1,30 +1,56 @@
+/**
+ * Connector CLI for provisioning, ingestion, and deletion.
+ */
 import "dotenv/config";
 
 import { Command } from "commander";
 import { ClientSecretCredential } from "@azure/identity";
 
-import type { Item } from "./schema/model.js";
+import type { <%= itemTypeName %> } from "./<%= schemaFolderName %>/model.js";
 import {
   contentCategory,
+  connectionName as defaultConnectionName,
   connectionId as defaultConnectionId,
   connectionDescription as defaultConnectionDescription,
   profileSourceWebUrl as defaultProfileSourceWebUrl,
   profileSourceDisplayName as defaultProfileSourceDisplayName,
   profileSourcePriority as defaultProfileSourcePriority,
   schemaPayload
-} from "./schema/index.js";
-import { getItemId, toExternalItem } from "./schema/itemPayload.js";
+} from "./<%= schemaFolderName %>/index.js";
+import { getItemId, toExternalItem } from "./<%= schemaFolderName %>/itemPayload.js";
 import { CsvItemSource } from "./datasource/csvItemSource.js";
+import { ConnectorCore } from "./core/connectorCore.js";
 
 const GRAPH_BASE_URL = <%= JSON.stringify(graphBaseUrl) %>;
-const PROFILE_SOURCE_URL_PREFIX = `${GRAPH_BASE_URL}/admin/people/profileSources`;
 
+const useColor = !process.env.NO_COLOR;
+const color = {
+  cyan: (value: string) => (useColor ? `\u001b[36m${value}\u001b[0m` : value),
+  green: (value: string) => (useColor ? `\u001b[32m${value}\u001b[0m` : value),
+  yellow: (value: string) => (useColor ? `\u001b[33m${value}\u001b[0m` : value),
+  dim: (value: string) => (useColor ? `\u001b[2m${value}\u001b[0m` : value),
+};
+
+function printBanner(): void {
+  const title = color.cyan(`✨ 🥥 <%= itemTypeName %> connector CLI 🥥 ✨`);
+  const subtitle = color.dim("Provision • Ingest • Delete");
+  console.log(`${title}\n${subtitle}`);
+}
+
+printBanner();
+
+/**
+ * Resolve a required environment variable or throw a friendly error.
+ */
 function requiredEnv(name: string): string {
   const value = process.env[name];
   if (!value) throw new Error(`Missing env var: ${name}`);
   return value;
 }
 
+/**
+ * Resolve the connection ID from env or schema defaults.
+ */
 function resolveConnectionId(): string {
   const value = process.env.CONNECTION_ID ?? defaultConnectionId;
   if (!value) throw new Error("Missing env var: CONNECTION_ID");
@@ -32,7 +58,9 @@ function resolveConnectionId(): string {
 }
 
 function resolveConnectionName(): string {
-  return requiredEnv("CONNECTION_NAME");
+  const value = process.env.CONNECTION_NAME ?? defaultConnectionName;
+  if (!value) throw new Error("Missing env var: CONNECTION_NAME");
+  return value;
 }
 
 function resolveConnectionDescription(): string {
@@ -53,7 +81,7 @@ function resolveProfileSourcePriority(): "first" | "last" {
   if (value !== "first" && value !== "last") {
     throw new Error("Invalid PROFILE_SOURCE_PRIORITY: expected 'first' or 'last'");
   }
-  return value;
+  return value as "first" | "last";
 }
 
 async function getAccessToken(): Promise<string> {
@@ -67,173 +95,56 @@ async function getAccessToken(): Promise<string> {
   return token.token;
 }
 
-async function graphRequest(method: string, url: string, body?: unknown): Promise<Response> {
-  const token = await getAccessToken();
-  return fetch(url, {
-    method,
-    headers: {
-      Authorization: `Bearer ${token}`,
-      "Content-Type": "application/json"
-    },
-    body: body ? JSON.stringify(body) : undefined
+/**
+ * Build a reusable connector core instance.
+ */
+function buildConnectorCore(): ConnectorCore<<%= itemTypeName %>> {
+  const profileSource = <%- isPeopleConnector
+    ? `{
+    webUrl: resolveProfileSourceWebUrl(),
+    displayName: process.env.PROFILE_SOURCE_DISPLAY_NAME ?? defaultProfileSourceDisplayName ?? requiredEnv("CONNECTION_NAME"),
+    priority: resolveProfileSourcePriority()
+  }`
+    : "undefined" %>;
+
+  return new ConnectorCore<<%= itemTypeName %>>({
+    graphBaseUrl: GRAPH_BASE_URL,
+    contentCategory,
+    schemaPayload,
+    getAccessToken,
+    getItemId,
+    toExternalItem,
+    profileSource
   });
 }
 
-<% if (isPeopleConnector) { -%>
-async function listProfilePropertySettings(): Promise<Array<{ id: string; prioritizedSourceUrls: string[] }>> {
-  const res = await graphRequest("GET", `${GRAPH_BASE_URL}/admin/people/profilePropertySettings`);
-  if (!res.ok) {
-    const text = await res.text();
-    throw new Error(`Failed to list profile property settings (HTTP ${res.status}): ${text}`);
-  }
-  const json = (await res.json()) as { value?: Array<{ id?: string; prioritizedSourceUrls?: string[] }> };
-  return (json.value ?? [])
-    .filter((entry) => typeof entry.id === "string")
-    .map((entry) => ({
-      id: entry.id as string,
-      prioritizedSourceUrls: Array.isArray(entry.prioritizedSourceUrls) ? entry.prioritizedSourceUrls : [],
-    }));
-}
-
-async function registerProfileSource(connectionId: string): Promise<void> {
-  const webUrl = resolveProfileSourceWebUrl();
-  const displayName =
-    process.env.PROFILE_SOURCE_DISPLAY_NAME ?? defaultProfileSourceDisplayName ?? requiredEnv("CONNECTION_NAME");
-  const priority = resolveProfileSourcePriority();
-
-  const payload: any = {
-    sourceId: connectionId,
-    displayName,
-    webUrl
-  };
-
-  const create = await graphRequest("POST", `${GRAPH_BASE_URL}/admin/people/profileSources`, payload);
-  if (!create.ok && create.status !== 409) {
-    const text = await create.text();
-    throw new Error(`Failed to register profile source (HTTP ${create.status}): ${text}`);
-  }
-
-  const sourceUrl = `${PROFILE_SOURCE_URL_PREFIX}(sourceId='${connectionId}')`;
-  const settings = await listProfilePropertySettings();
-  for (const setting of settings) {
-    const existing = setting.prioritizedSourceUrls.filter((value) => value !== sourceUrl);
-    const updated = priority === "first" ? [sourceUrl, ...existing] : [...existing, sourceUrl];
-    const res = await graphRequest(
-      "PATCH",
-      `${GRAPH_BASE_URL}/admin/people/profilePropertySettings/${setting.id}`,
-      {
-        "@odata.type": "#microsoft.graph.profilePropertySetting",
-        prioritizedSourceUrls: updated
-      }
-    );
-    if (!res.ok) {
-      const text = await res.text();
-      throw new Error(`Failed to update profile property setting ${setting.id} (HTTP ${res.status}): ${text}`);
-    }
-  }
-}
-
-async function unregisterProfileSource(connectionId: string): Promise<void> {
-  const sourceUrl = `${PROFILE_SOURCE_URL_PREFIX}(sourceId='${connectionId}')`;
-  const settings = await listProfilePropertySettings();
-  for (const setting of settings) {
-    const updated = setting.prioritizedSourceUrls.filter((value) => value !== sourceUrl);
-    const res = await graphRequest(
-      "PATCH",
-      `${GRAPH_BASE_URL}/admin/people/profilePropertySettings/${setting.id}`,
-      {
-        "@odata.type": "#microsoft.graph.profilePropertySetting",
-        prioritizedSourceUrls: updated
-      }
-    );
-    if (!res.ok) {
-      const text = await res.text();
-      throw new Error(`Failed to update profile property setting ${setting.id} (HTTP ${res.status}): ${text}`);
-    }
-  }
-
-  const res = await graphRequest(
-    "DELETE",
-    `${PROFILE_SOURCE_URL_PREFIX}(sourceId='${connectionId}')`
-  );
-  if (!res.ok && res.status !== 404) {
-    const text = await res.text();
-    throw new Error(`Failed to delete profile source (HTTP ${res.status}): ${text}`);
-  }
-}
-<% } -%>
-
-async function ensureConnection(connectionId: string): Promise<void> {
-  const name = resolveConnectionName();
-  const description = resolveConnectionDescription();
-
-  const payload: any = {
-    id: connectionId,
-    name,
-    description
-  };
-
-  if (contentCategory) payload.contentCategory = contentCategory;
-
-  const createUrl = `${GRAPH_BASE_URL}/external/connections`;
-  const create = await graphRequest("POST", createUrl, payload);
-  if (!create.ok) {
-    if (create.status !== 409) {
-      const text = await create.text();
-      throw new Error(`Failed to create connection (HTTP ${create.status}): ${text}`);
-    }
-  }
-}
-
-async function patchSchema(connectionId: string): Promise<void> {
-  const schemaUrl = `${GRAPH_BASE_URL}/external/connections/${connectionId}/schema`;
-  const res = await graphRequest("PATCH", schemaUrl, schemaPayload);
-  if (!res.ok) {
-    const text = await res.text();
-    throw new Error(`Failed to patch schema (HTTP ${res.status}): ${text}`);
-  }
-}
-
+/**
+ * Provision connection + schema and (optionally) profile source.
+ */
 async function provision(): Promise<void> {
   const connectionId = resolveConnectionId();
-  await ensureConnection(connectionId);
-  await patchSchema(connectionId);
-<% if (isPeopleConnector) { -%>
-  await registerProfileSource(connectionId);
-<% } -%>
+  const core = buildConnectorCore();
+  await core.provision({
+    connectionId,
+    connectionName: resolveConnectionName(),
+    connectionDescription: resolveConnectionDescription()
+  });
   console.log("ok: provisioned");
 }
 
+/**
+ * Delete the external connection.
+ */
 async function deleteConnection(): Promise<void> {
   const connectionId = resolveConnectionId();
-<% if (isPeopleConnector) { -%>
-  await unregisterProfileSource(connectionId);
-<% } -%>
-  const res = await graphRequest("DELETE", `${GRAPH_BASE_URL}/external/connections/${connectionId}`);
-  if (!res.ok && res.status !== 404) {
-    const text = await res.text();
-    throw new Error(`Failed to delete connection (HTTP ${res.status}): ${text}`);
-  }
+  const core = buildConnectorCore();
+  await core.deleteConnection(connectionId);
   console.log("ok: deleted");
 }
 
-async function putItem(connectionId: string, item: Item, verbose: boolean): Promise<void> {
-  const itemId = getItemId(item);
-  const url = `${GRAPH_BASE_URL}/external/connections/${connectionId}/items/${encodeURIComponent(itemId)}`;
-
-  const payload = toExternalItem(item);
-  if (verbose) {
-    console.log("verbose: PUT", url);
-    console.log("verbose: payload", JSON.stringify(payload, null, 2));
-  }
-
-  const res = await graphRequest("PUT", url, payload);
-  if (!res.ok) {
-    const text = await res.text();
-    throw new Error(`Failed to ingest item '${itemId}' (HTTP ${res.status}): ${text}`);
-  }
-}
-
+/**
+ * Ingest items from the configured datasource.
+ */
 async function ingest(options: {
   csvPath?: string;
   dryRun?: boolean;
@@ -243,18 +154,15 @@ async function ingest(options: {
   const connectionId = options.dryRun ? "dry-run" : resolveConnectionId();
   // Swap this for any ItemSource implementation (API, DB, queue, etc.).
   const source = new CsvItemSource(options.csvPath ?? process.env.CSV_PATH ?? "data.csv");
-  let count = 0;
-  for await (const item of source.getItems()) {
-    if (options.limit && count >= options.limit) break;
-    if (!options.dryRun) {
-      await putItem(connectionId, item, Boolean(options.verbose));
-    } else if (options.verbose) {
-      console.log("verbose: DRY RUN item", JSON.stringify(toExternalItem(item), null, 2));
-    }
-    count++;
-  }
-
-  console.log("ok: ingested " + count + " item(s)");
+  const core = buildConnectorCore();
+  await core.ingest({
+    source,
+    connectionId,
+    dryRun: options.dryRun,
+    limit: options.limit,
+    verbose: options.verbose,
+    toExternalItem
+  });
 }
 
 const program = new Command();
@@ -285,7 +193,11 @@ program
 program
   .command("register-profile-source")
   .description("Register the connection as a profile source (people connectors)")
-  .action(() => registerProfileSource(resolveConnectionId()));
+  .action(async () => {
+    const connectionId = resolveConnectionId();
+    const core = buildConnectorCore();
+    await core.registerProfileSource(connectionId);
+  });
 <% } -%>
 
 program