agent-messenger 2.6.1 → 2.6.3

package/src/platforms/webex/commands/snapshot.test.ts

@@ -11,24 +11,19 @@ mock.module('@/shared/utils/error-handler', () => ({
 
 const mockSpaces = [
   { id: 'space-1', title: 'General', type: 'group', isLocked: false, lastActivity: '2024-01-15T00:00:00.000Z', created: '2024-01-01T00:00:00.000Z', creatorId: 'person-1' },
+  { id: 'space-2', title: 'Random', type: 'group', isLocked: false, lastActivity: '2024-01-14T00:00:00.000Z', created: '2024-01-01T00:00:00.000Z', creatorId: 'person-1' },
 ]
 
-const mockMessages = [
-  { id: 'msg-1', roomId: 'space-1', roomType: 'group', text: 'Hello', personId: 'person-1', personEmail: 'alice@example.com', created: '2024-01-15T00:00:00.000Z' },
-]
-
-const mockMembers = [
+const mockMyMemberships = [
   { id: 'mem-1', roomId: 'space-1', personId: 'person-1', personEmail: 'alice@example.com', personDisplayName: 'Alice', isModerator: true, created: '2024-01-01T00:00:00.000Z' },
 ]
 
 const mockListSpaces = mock(() => Promise.resolve(mockSpaces as any))
-const mockListMessages = mock(() => Promise.resolve(mockMessages as any))
-const mockListMemberships = mock(() => Promise.resolve(mockMembers as any))
+const mockListMyMemberships = mock(() => Promise.resolve(mockMyMemberships as any))
 
 const mockClient = {
   listSpaces: mockListSpaces,
-  listMessages: mockListMessages,
-  listMemberships: mockListMemberships,
+  listMyMemberships: mockListMyMemberships,
 }
 
 const mockLogin = mock(() => Promise.resolve(mockClient))
@@ -46,8 +41,7 @@ describe('snapshot command', () => {
 
   beforeEach(() => {
     mockListSpaces.mockReset().mockImplementation(() => Promise.resolve(mockSpaces as any))
-    mockListMessages.mockReset().mockImplementation(() => Promise.resolve(mockMessages as any))
-    mockListMemberships.mockReset().mockImplementation(() => Promise.resolve(mockMembers as any))
+    mockListMyMemberships.mockReset().mockImplementation(() => Promise.resolve(mockMyMemberships as any))
     mockLogin.mockReset().mockImplementation(() => Promise.resolve(mockClient))
     mockHandleError.mockReset().mockImplementation((err: Error) => {
       throw err
@@ -60,40 +54,24 @@ describe('snapshot command', () => {
     consoleSpy.mockRestore()
   })
 
-  test('full snapshot includes spaces, recent_messages, members', async () => {
+  test('returns spaces with id, title, type, lastActivity', async () => {
     await snapshotAction({})
 
     expect(consoleSpy).toHaveBeenCalled()
     const output = JSON.parse(consoleSpy.mock.calls[consoleSpy.mock.calls.length - 1][0])
-    expect(output.spaces).toBeDefined()
+    expect(output.spaces).toHaveLength(1)
     expect(output.spaces[0].id).toBe('space-1')
     expect(output.spaces[0].title).toBe('General')
-    expect(output.recent_messages).toBeDefined()
-    expect(output.recent_messages[0].id).toBe('msg-1')
-    expect(output.recent_messages[0].author).toBe('alice@example.com')
-    expect(output.members).toBeDefined()
-    expect(output.members[0].personEmail).toBe('alice@example.com')
-  })
-
-  test('--spaces-only includes only spaces (no messages, no members)', async () => {
-    await snapshotAction({ spacesOnly: true })
-
-    expect(consoleSpy).toHaveBeenCalled()
-    const output = JSON.parse(consoleSpy.mock.calls[consoleSpy.mock.calls.length - 1][0])
-    expect(output.spaces).toBeDefined()
-    expect(output.recent_messages).toBeUndefined()
-    expect(output.members).toBeUndefined()
+    expect(output.spaces[0].type).toBe('group')
+    expect(output.spaces[0].lastActivity).toBe('2024-01-15T00:00:00.000Z')
   })
 
-  test('--members-only includes only members (no spaces, no messages)', async () => {
-    await snapshotAction({ membersOnly: true })
+  test('filters spaces to only those in my memberships', async () => {
+    await snapshotAction({})
 
-    expect(consoleSpy).toHaveBeenCalled()
     const output = JSON.parse(consoleSpy.mock.calls[consoleSpy.mock.calls.length - 1][0])
-    expect(output.spaces).toBeUndefined()
-    expect(output.recent_messages).toBeUndefined()
-    expect(output.members).toBeDefined()
-    expect(output.members[0].personEmail).toBe('alice@example.com')
+    expect(output.spaces).toHaveLength(1)
+    expect(output.spaces[0].id).toBe('space-1')
   })
 
   test('not authenticated outputs error', async () => {
@@ -105,10 +83,4 @@ describe('snapshot command', () => {
 
     expect(mockHandleError).toHaveBeenCalledWith(expect.any(WebexError))
   })
-
-  test('passes limit option to listMessages', async () => {
-    await snapshotAction({ limit: 5 })
-
-    expect(mockListMessages).toHaveBeenCalledWith('space-1', { max: 5 })
-  })
 })

package/src/platforms/webex/commands/snapshot.ts

@@ -1,72 +1,27 @@
 import { Command } from 'commander'
-import { parallelMap } from '@/shared/utils/concurrency'
 import { handleError } from '@/shared/utils/error-handler'
 import { formatOutput } from '@/shared/utils/output'
 import { WebexClient } from '../client'
-import type { WebexSpace } from '../types'
 
 export async function snapshotAction(options: {
-  spacesOnly?: boolean
-  membersOnly?: boolean
-  limit?: number
   pretty?: boolean
 }): Promise<void> {
   try {
     const client = await new WebexClient().login()
-    const messageLimit = options.limit || 20
-    const snapshot: Record<string, unknown> = {}
 
-    if (!options.membersOnly) {
-      const spaces = await client.listSpaces({ max: 50 })
-      snapshot.spaces = spaces.map((s) => ({
+    const myMemberships = await client.listMyMemberships({ max: 100 })
+    const myRoomIds = new Set(myMemberships.map((m) => m.roomId))
+
+    const allSpaces = await client.listSpaces({ max: 100 })
+    const spaces = allSpaces.filter((s) => myRoomIds.has(s.id))
+
+    const snapshot = {
+      spaces: spaces.map((s) => ({
         id: s.id,
         title: s.title,
         type: s.type,
         lastActivity: s.lastActivity,
-      }))
-
-      if (!options.spacesOnly) {
-        const spaceMessages = await parallelMap(
-          spaces,
-          async (space: WebexSpace) => {
-            const messages = await client.listMessages(space.id, { max: messageLimit })
-            return messages.map((msg) => ({
-              ...msg,
-              space_title: space.title,
-            }))
-          },
-          5,
-        )
-
-        snapshot.recent_messages = spaceMessages.flat().map((msg) => ({
-          space_id: msg.roomId,
-          space_title: msg.space_title,
-          id: msg.id,
-          author: msg.personEmail,
-          text: msg.text || msg.markdown || '',
-          created: msg.created,
-        }))
-      }
-    }
-
-    if (!options.spacesOnly) {
-      // Get members for the first few spaces (avoid massive API calls)
-      const spaces = await client.listSpaces({ max: 10 })
-      const spaceMembers = await parallelMap(
-        spaces,
-        async (space: WebexSpace) => {
-          const members = await client.listMemberships(space.id, { max: 100 })
-          return members.map((m) => ({
-            space_id: space.id,
-            space_title: space.title,
-            personEmail: m.personEmail,
-            personDisplayName: m.personDisplayName,
-            isModerator: m.isModerator,
-          }))
-        },
-        5,
-      )
-      snapshot.members = spaceMembers.flat()
+      })),
     }
 
     console.log(formatOutput(snapshot, options.pretty))
@@ -76,16 +31,10 @@ export async function snapshotAction(options: {
 }
 
 export const snapshotCommand = new Command('snapshot')
-  .description('Get comprehensive workspace state for AI agents')
-  .option('--spaces-only', 'Include only spaces (exclude messages and members)')
-  .option('--members-only', 'Include only members (exclude spaces and messages)')
-  .option('--limit <n>', 'Number of recent messages per space (default: 20)', '20')
+  .description('Get workspace spaces overview for AI agents')
   .option('--pretty', 'Pretty print JSON output')
   .action(async (options) => {
     await snapshotAction({
-      spacesOnly: options.spacesOnly,
-      membersOnly: options.membersOnly,
-      limit: parseInt(options.limit, 10),
       pretty: options.pretty,
     })
   })