@brantrusnak/openclaw-omadeus 1.0.2 → 1.0.4

package/src/onboarding.ts

@@ -8,29 +8,16 @@ import { listMemberChannelViews } from "./api/channel.api.js";
 import { authenticate } from "./auth.js";
 import { getOmadeusChannelConfig, resolveOmadeusAccount } from "./config.js";
 import { OMADEUS_CAS_URL, OMADEUS_MAESTRO_URL } from "./defaults.js";
+import { formatMemberLabel } from "./member-resolve.js";
 import type {
   OmadeusChannelConfig,
   OmadeusChannelView,
   OmadeusInboundEntityKind,
   OmadeusOrganizationMember,
 } from "./types.js";
+import { OMADEUS_INBOUND_ENTITY_KINDS } from "./types.js";
 
 const channel = "omadeus" as const;
-const DONE = "__done__";
-const ALL_ENTITY_KINDS: OmadeusInboundEntityKind[] = [
-  "task",
-  "nugget",
-  "project",
-  "release",
-  "sprint",
-  "summary",
-  "client",
-  "folder",
-];
-
-type CoreConfig = OpenClawConfig & {
-  channels?: { omadeus?: OmadeusChannelConfig };
-};
 
 type SelectOption = {
   value: string;
@@ -38,23 +25,20 @@ type SelectOption = {
   hint?: string;
 };
 
-type MultiSelectPrompter = {
-  multiSelect?: (args: {
-    message: string;
-    options: SelectOption[];
-    initialValues?: string[];
-    initialValue?: string[];
-  }) => Promise<string[]>;
-  multiselect?: (args: {
-    message: string;
-    options: SelectOption[];
-    initialValues?: string[];
-    initialValue?: string[];
-  }) => Promise<string[]>;
-};
-
-function getOmadeusSection(cfg: OpenClawConfig): OmadeusChannelConfig | undefined {
-  return getOmadeusChannelConfig(cfg as CoreConfig);
+function formatAuthError(err: unknown): string {
+  if (!(err instanceof Error)) return String(err);
+  const parts = [err.message];
+  const { cause } = err;
+  if (cause instanceof Error) {
+    parts.push(cause.message);
+    const code = (cause as Error & { code?: unknown }).code;
+    if (typeof code === "string" && code) {
+      parts.push(`(${code})`);
+    }
+  } else if (typeof cause === "string" && cause.trim()) {
+    parts.push(cause);
+  }
+  return parts.join(" — ");
 }
 
 async function noteOmadeusAuthHelp(prompter: WizardPrompter): Promise<void> {
@@ -80,35 +64,32 @@ async function promptOrganizationId(params: {
 }): Promise<number> {
   const { prompter, maestroUrl, email, existing } = params;
 
-  // Try to list organizations from the API
-  if (maestroUrl && email) {
-    try {
-      const orgs = await listOrganizations({ maestroUrl, email });
-      if (orgs.length > 0) {
-        if (orgs.length === 1) {
-          await prompter.note(
-            `Found organization: ${orgs[0]!.title} (${orgs[0]!.id})`,
-            "Omadeus organization",
-          );
-          return orgs[0]!.id;
-        }
-        const choice = await prompter.select({
-          message: "Select organization",
-          options: orgs.map((org) => ({
-            value: String(org.id),
-            label: `${org.title} (${org.membersCount} members)`,
-            hint: `ID: ${org.id}`,
-          })),
-          initialValue: existing ? String(existing) : String(orgs[0]!.id),
-        });
-        return Number(choice);
+  try {
+    const orgs = await listOrganizations({ maestroUrl, email });
+    if (orgs.length > 0) {
+      if (orgs.length === 1) {
+        await prompter.note(
+          `Found organization: ${orgs[0]!.title} (${orgs[0]!.id})`,
+          "Omadeus organization",
+        );
+        return orgs[0]!.id;
       }
-    } catch {
-      await prompter.note(
-        "Could not fetch organizations from the API. Enter the ID manually.",
-        "Omadeus organization",
-      );
+      const choice = await prompter.select({
+        message: "Select organization",
+        options: orgs.map((org) => ({
+          value: String(org.id),
+          label: `${org.title} (${org.membersCount} members)`,
+          hint: `ID: ${org.id}`,
+        })),
+        initialValue: existing ? String(existing) : String(orgs[0]!.id),
+      });
+      return Number(choice);
     }
+  } catch {
+    await prompter.note(
+      "Could not fetch organizations from the API. Enter the ID manually.",
+      "Omadeus organization",
+    );
   }
 
   const raw = await prompter.text({
@@ -140,8 +121,21 @@ async function promptChannelSelection(params: {
     take: 100,
   });
   if (channels.length === 0) {
-    throw new Error("No channels found for this account.");
+    await prompter.note(
+      "No channels found for this account. Channel listening will stay disabled.",
+      "Omadeus channels",
+    );
+    return [];
+  }
+
+  const listenToChannels = await prompter.confirm({
+    message: "Listen for messages in Omadeus channels?",
+    initialValue: (existingChannelViewIds?.length ?? 0) > 0,
+  });
+  if (!listenToChannels) {
+    return [];
   }
+
   const selected = await promptMultiSelect({
     prompter,
     message: "Which channels should OpenClaw listen to?",
@@ -155,27 +149,9 @@ async function promptChannelSelection(params: {
     initialValues:
       existingChannelViewIds && existingChannelViewIds.length > 0
         ? existingChannelViewIds.map(String)
-        : [String(channels[0]!.id)],
+        : undefined,
   });
-  const chosen = channels.filter((item) => selected.includes(String(item.id)));
-  if (chosen.length === 0) {
-    throw new Error("At least one channel must be selected.");
-  }
-  return chosen;
-}
-
-function memberLabel(member: OmadeusOrganizationMember): string {
-  const fullName = `${member.firstName ?? ""} ${member.lastName ?? ""}`.trim();
-  if (member.title?.trim()) {
-    return member.title.trim();
-  }
-  if (fullName) {
-    return fullName;
-  }
-  if (member.email?.trim()) {
-    return member.email.trim();
-  }
-  return `Member ${member.referenceId}`;
+  return channels.filter((item) => selected.includes(String(item.id)));
 }
 
 function memberHint(member: OmadeusOrganizationMember): string | undefined {
@@ -190,40 +166,11 @@ async function promptMultiSelect(params: {
   options: SelectOption[];
   initialValues?: string[];
 }): Promise<string[]> {
-  const multi = params.prompter as unknown as MultiSelectPrompter;
-  const runMulti = multi.multiSelect ?? multi.multiselect;
-  if (runMulti) {
-    return runMulti({
-      message: params.message,
-      options: params.options,
-      initialValues: params.initialValues,
-      initialValue: params.initialValues,
-    });
-  }
-
-  const selected = new Set(params.initialValues ?? []);
-  while (true) {
-    const next = await params.prompter.select({
-      message: `${params.message} (${selected.size} selected)`,
-      options: [
-        { value: DONE, label: selected.size > 0 ? "Done" : "Done (select none)" },
-        ...params.options.map((option) => ({
-          ...option,
-          label: selected.has(option.value) ? `[selected] ${option.label}` : option.label,
-        })),
-      ],
-      initialValue: DONE,
-    });
-    const value = String(next);
-    if (value === DONE) {
-      return [...selected];
-    }
-    if (selected.has(value)) {
-      selected.delete(value);
-    } else {
-      selected.add(value);
-    }
-  }
+  return params.prompter.multiselect({
+    message: params.message,
+    options: params.options,
+    initialValues: params.initialValues,
+  });
 }
 
 async function loadSelectableMembers(params: {
@@ -241,13 +188,13 @@ async function loadSelectableMembers(params: {
     })
   )
     .filter((member) => member.isSystem !== true && !excluded.has(member.referenceId))
-    .sort((a, b) => memberLabel(a).localeCompare(memberLabel(b)));
+    .sort((a, b) => formatMemberLabel(a).localeCompare(formatMemberLabel(b)));
 }
 
 function memberOptions(members: OmadeusOrganizationMember[]): SelectOption[] {
   return members.map((member) => ({
     value: String(member.referenceId),
-    label: memberLabel(member),
+    label: formatMemberLabel(member),
     hint: memberHint(member),
   }));
 }
@@ -258,6 +205,27 @@ function readReferenceIds(values: string[]): number[] {
     .filter((value) => Number.isInteger(value) && value > 0);
 }
 
+async function promptCredentials(
+  prompter: WizardPrompter,
+  existing: { email?: string; password?: string },
+): Promise<{ email: string; password: string }> {
+  const email = String(
+    await prompter.text({
+      message: "Omadeus username (email)",
+      initialValue: existing.email,
+      validate: (value) => (String(value ?? "").trim() ? undefined : "Required"),
+    }),
+  ).trim();
+  const password = String(
+    await prompter.text({
+      message: "Omadeus password",
+      sensitive: true,
+      validate: (value) => (String(value ?? "").trim() ? undefined : "Required"),
+    }),
+  ).trim();
+  return { email, password };
+}
+
 async function promptSenderAllowlist(params: {
   prompter: WizardPrompter;
   message: string;
@@ -277,7 +245,7 @@ async function promptSenderAllowlist(params: {
     ],
     initialValue: existingReferenceIds && existingReferenceIds.length > 0 ? "specific" : "all",
   });
-  if (String(mode) === "all") {
+  if (mode === "all") {
     return undefined;
   }
 
@@ -297,16 +265,17 @@ async function promptEntityKindSelection(params: {
   const selected = await promptMultiSelect({
     prompter: params.prompter,
     message: "Which entity room types should OpenClaw listen to?",
-    options: ALL_ENTITY_KINDS.map((kind) => ({
+    options: OMADEUS_INBOUND_ENTITY_KINDS.map((kind) => ({
       value: kind,
       label: kind,
     })),
     initialValues:
       params.existingKinds && params.existingKinds.length > 0
         ? params.existingKinds
-        : ALL_ENTITY_KINDS,
+        : [...OMADEUS_INBOUND_ENTITY_KINDS],
   });
-  return ALL_ENTITY_KINDS.filter((kind) => selected.includes(kind));
+  const selectedSet = new Set(selected);
+  return OMADEUS_INBOUND_ENTITY_KINDS.filter((kind) => selectedSet.has(kind));
 }
 
 export const omadeusSetupWizard: ChannelSetupWizard = {
@@ -343,7 +312,7 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
   credentials: [],
   finalize: async ({ cfg, prompter }) => {
     const account = resolveOmadeusAccount({ cfg });
-    const section = getOmadeusSection(cfg) ?? {};
+    const section = getOmadeusChannelConfig(cfg) ?? {};
     let next = cfg;
 
     if (account.credentialSource === "none") {
@@ -356,21 +325,10 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
     const casUrl = OMADEUS_CAS_URL;
     const maestroUrl = OMADEUS_MAESTRO_URL;
 
-    let email = String(
-      await prompter.text({
-        message: "Omadeus username (email)",
-        initialValue: section.email ?? envEmail,
-        validate: (value) => (String(value ?? "").trim() ? undefined : "Required"),
-      }),
-    ).trim();
-
-    let password = String(
-      await prompter.text({
-        message: "Omadeus password",
-        initialValue: section.password ?? envPassword,
-        validate: (value) => (String(value ?? "").trim() ? undefined : "Required"),
-      }),
-    ).trim();
+    let { email, password } = await promptCredentials(prompter, {
+      email: section.email ?? envEmail,
+      password: section.password ?? envPassword,
+    });
 
     const organizationId = await promptOrganizationId({
       prompter,
@@ -379,7 +337,6 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
       existing: section.organizationId,
     });
 
-    // Verify the full auth flow before saving
     let sessionToken: string | undefined;
     let selfReferenceId: number | undefined;
     while (true) {
@@ -396,8 +353,10 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
         await prompter.note(`Authenticated as ${payload.email}`, "Omadeus authentication");
         break;
       } catch (err) {
-        const msg = err instanceof Error ? err.message : String(err);
-        await prompter.note(`Authentication failed: ${msg}`, "Omadeus authentication");
+        await prompter.note(
+          `Authentication failed: ${formatAuthError(err)}`,
+          "Omadeus authentication",
+        );
         const retry = await prompter.confirm({
           message: "Re-enter email/password and try again?",
           initialValue: true,
@@ -409,20 +368,7 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
           );
           break;
         }
-        email = String(
-          await prompter.text({
-            message: "Omadeus email",
-            initialValue: email,
-            validate: (value) => (String(value ?? "").trim() ? undefined : "Required"),
-          }),
-        ).trim();
-        password = String(
-          await prompter.text({
-            message: "Omadeus password",
-            initialValue: password,
-            validate: (value) => (String(value ?? "").trim() ? undefined : "Required"),
-          }),
-        ).trim();
+        ({ email, password } = await promptCredentials(prompter, { email, password }));
       }
     }
 
@@ -457,12 +403,15 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
       existingChannelViewIds: existingInbound?.channels?.allowedChannelViewIds,
     });
 
-    const channelSenderIds = await promptSenderAllowlist({
-      prompter,
-      message: "Which users can trigger OpenClaw from allowed channels?",
-      members,
-      existingReferenceIds: existingInbound?.channels?.allowedSenderReferenceIds,
-    });
+    const channelSenderIds =
+      selectedChannels.length > 0
+        ? await promptSenderAllowlist({
+            prompter,
+            message: "Which users can trigger OpenClaw from allowed channels?",
+            members,
+            existingReferenceIds: existingInbound?.channels?.allowedSenderReferenceIds,
+          })
+        : undefined;
 
     const entityKinds = await promptEntityKindSelection({
       prompter,
@@ -495,11 +444,16 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
     const entityKindSummary =
       entityKinds.length > 0 ? entityKinds.join(", ") : "none (entity rooms disabled)";
 
+    const channelSummary =
+      selectedChannels.length > 0
+        ? `- Channels "${channelTitles}": rooms ${channelRoomIds.join(", ") || "(no room ids)"} from ${senderSummary(channelSenderIds)}; @mention not required in those rooms.`
+        : "- Channels: disabled (none selected).";
+
     await prompter.note(
       [
         `Inbound policy (Jaguar chat):`,
         `- Direct messages: enabled for ${senderSummary(directSenderIds)} (no @mention required).`,
-        `- Channels "${channelTitles}": rooms ${channelRoomIds.join(", ") || "(no room ids)"} from ${senderSummary(channelSenderIds)}; @mention not required in those rooms.`,
+        channelSummary,
         `- Entity rooms (${entityKindSummary}): ${senderSummary(entitySenderIds)}; @mention required.`,
       ].join("\n"),
       "Omadeus inbound policy",
@@ -516,7 +470,7 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
           email,
           password,
           organizationId,
-          ...(sessionToken ? { sessionToken } : {}),
+          sessionToken,
           inbound: {
             version: 1,
             direct: {
@@ -525,7 +479,7 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
               requireMention: "never",
             },
             channels: {
-              enabled: true,
+              enabled: selectedChannels.length > 0,
               allowedRoomIds: channelRoomIds,
               allowedChannelViewIds: channelViewIds,
               ...(channelSenderIds ? { allowedSenderReferenceIds: channelSenderIds } : {}),
@@ -548,7 +502,7 @@ export const omadeusSetupWizard: ChannelSetupWizard = {
     ...cfg,
     channels: {
       ...cfg.channels,
-      omadeus: { ...getOmadeusSection(cfg), enabled: false },
+      omadeus: { ...getOmadeusChannelConfig(cfg), enabled: false },
     },
   }),
 };

package/src/setup-core.ts

@@ -35,12 +35,21 @@ export const omadeusSetupAdapter: ChannelSetupAdapter = {
     const password = input.password?.trim() || undefined;
     const organizationId = readSetupNumberField(rawInput, "organizationId");
 
+    const channelsRecord = cfg.channels as Record<string, unknown> | undefined;
+    const omadeusExisting = channelsRecord?.["omadeus"];
+    const omadeusPrevious =
+      omadeusExisting !== null &&
+      typeof omadeusExisting === "object" &&
+      !Array.isArray(omadeusExisting)
+        ? (omadeusExisting as Record<string, unknown>)
+        : {};
+
     return {
       ...cfg,
       channels: {
         ...cfg.channels,
         omadeus: {
-          ...(cfg.channels as Record<string, unknown>)?.["omadeus"],
+          ...omadeusPrevious,
           enabled: true,
           ...(casUrl ? { casUrl } : {}),
           ...(maestroUrl ? { maestroUrl } : {}),

package/src/socket/socket.ts

@@ -31,10 +31,12 @@ const HEARTBEAT_MISSED_MAX = 5;
 const KEEP_ALIVE_CONTENT = "keep-alive";
 const KEEP_ALIVE_ACTION = "answer";
 
-function isKeepAliveMessage(data: Record<string, unknown>): boolean {
-  const content = (data as { content?: unknown }).content;
-  const payloadData = (data as { data?: unknown }).data;
-  return content === KEEP_ALIVE_CONTENT || payloadData === KEEP_ALIVE_CONTENT;
+function isServerKeepAlive(data: Record<string, unknown>): boolean {
+  return (data as { content?: unknown }).content === KEEP_ALIVE_CONTENT;
+}
+
+function isClientKeepAlive(data: Record<string, unknown>): boolean {
+  return (data as { data?: unknown }).data === KEEP_ALIVE_CONTENT;
 }
 
 export function createOmadeusSocketClient(opts: OmadeusSocketOptions): OmadeusSocketClient {
@@ -87,7 +89,7 @@ export function createOmadeusSocketClient(opts: OmadeusSocketOptions): OmadeusSo
       return;
     }
     heartbeatMissCount += 1;
-    ws.send(JSON.stringify({ data: KEEP_ALIVE_CONTENT, action: KEEP_ALIVE_ACTION }));
+    sendKeepAliveFrame();
 
     if (heartbeatMissCount >= HEARTBEAT_MISSED_MAX) {
       log?.warn(
@@ -104,6 +106,12 @@ export function createOmadeusSocketClient(opts: OmadeusSocketOptions): OmadeusSo
     }, HEARTBEAT_INTERVAL_MS);
   }
 
+  function sendKeepAliveFrame() {
+    if (ws?.readyState === WebSocket.OPEN) {
+      ws.send(JSON.stringify({ data: KEEP_ALIVE_CONTENT, action: KEEP_ALIVE_ACTION }));
+    }
+  }
+
   function connect() {
     if (ws) {
       ws.removeAllListeners();
@@ -144,19 +152,24 @@ export function createOmadeusSocketClient(opts: OmadeusSocketOptions): OmadeusSo
         const data = JSON.parse(String(raw)) as Record<string, unknown>;
 
         const action = (data as { action?: unknown }).action;
-        if (isKeepAliveMessage(data) && action === KEEP_ALIVE_ACTION) {
+        if (isServerKeepAlive(data) && action === KEEP_ALIVE_ACTION) {
           resetHeartbeat();
           return;
         }
 
-        // If backend sends heartbeat pings, answer them immediately.
-        if (isKeepAliveMessage(data) && action === "heartbeat") {
-          if (ws?.readyState === WebSocket.OPEN) {
-            ws.send(JSON.stringify({ data: KEEP_ALIVE_CONTENT, action: KEEP_ALIVE_ACTION }));
-          }
+        if (isClientKeepAlive(data) && action === KEEP_ALIVE_ACTION) {
+          resetHeartbeat();
+          return;
+        }
+
+        // If backend sends a heartbeat ping, answer it immediately.
+        if (isServerKeepAlive(data) && action === "heartbeat") {
+          resetHeartbeat();
+          sendKeepAliveFrame();
           return;
         }
 
+        resetHeartbeat();
         onEvent?.(data);
       } catch {
         log?.warn(`${logPrefix} unparseable message: ${String(raw).slice(0, 200)}`);