feat: add orderBy support to chat.getMessages

gemini-extension.json

@@ -1,13 +1,15 @@
 {
   "name": "google-workspace",
-  "version": "DEV",
+  "version": "0.0.2",
   "contextFileName": "workspace-server${/}WORKSPACE-Context.md",
   "mcpServers": {
     "google-workspace": {
       "program": "./workspace-server/dist/index.js",
       "command": "node",
-      "args": ["scripts${/}start.js"],
+      "args": [
+        "scripts${/}start.js"
+      ],
       "cwd": "${extensionPath}"
     }
   }
-}
+}

workspace-server/src/__tests__/services/ChatService.test.ts

@@ -354,6 +354,33 @@ describe('ChatService', () => {
       const response = JSON.parse(result.content[0].text);
       expect(response.messages).toEqual(mockMessages);
     });
+
+    it('should pass orderBy to the API', async () => {
+      const mockMessages = [
+        { name: 'spaces/space1/messages/msg1', text: 'Hello' },
+      ];
+
+      mockChatAPI.spaces.messages.list.mockResolvedValue({
+        data: {
+          messages: mockMessages,
+        },
+      });
+
+      const result = await chatService.getMessages({
+        spaceName: 'spaces/space1',
+        orderBy: 'createTime desc',
+      });
+
+      expect(mockChatAPI.spaces.messages.list).toHaveBeenCalledWith({
+        parent: 'spaces/space1',
+        pageSize: undefined,
+        pageToken: undefined,
+        orderBy: 'createTime desc',
+      });
+
+      const response = JSON.parse(result.content[0].text);
+      expect(response.messages).toEqual(mockMessages);
+    });
   });
 
   describe('sendDm', () => {

workspace-server/src/index.ts

@@ -470,6 +470,7 @@ async function main() {
                 unreadOnly: z.boolean().optional().describe('Whether to return only unread messages.'),
                 pageSize: z.number().optional().describe('The maximum number of messages to return.'),
                 pageToken: z.string().optional().describe('The token for the next page of results.'),
+                orderBy: z.string().optional().describe('The order to list messages in (e.g., "createTime desc").'),
             }
         },
         chatService.getMessages

workspace-server/src/services/ChatService.ts

@@ -176,7 +176,7 @@ export class ChatService {
         }
     }
 
-    public getMessages = async ({ spaceName, unreadOnly, pageSize, pageToken }: { spaceName: string, unreadOnly?: boolean, pageSize?: number, pageToken?: string }) => {
+    public getMessages = async ({ spaceName, unreadOnly, pageSize, pageToken, orderBy }: { spaceName: string, unreadOnly?: boolean, pageSize?: number, pageToken?: string, orderBy?: string }) => {
         logToFile(`Listing messages for space: ${spaceName}`);
         try {
             const chat = await this.getChatClient();
@@ -207,7 +207,7 @@ export class ChatService {
                     logToFile(`No last read time found for user in space: ${spaceName}`);
                     // This can happen if the user has never read messages in the space.
                     // In this case, all messages are unread.
-                    const res = await chat.spaces.messages.list({ parent: spaceName, pageSize, pageToken });
+                    const res = await chat.spaces.messages.list({ parent: spaceName, pageSize, pageToken, orderBy });
                     const messages = res.data.messages || [];
                     logToFile(`Successfully listed ${messages.length} unread messages for space: ${spaceName}`);
                     return { content: [{ type: "text" as const, text: JSON.stringify({ messages, nextPageToken: res.data.nextPageToken }) }] };
@@ -218,6 +218,7 @@ export class ChatService {
                     filter: `createTime > "${lastReadTime}"`,
                     pageSize,
                     pageToken,
+                    orderBy,
                 });
 
                 const messages = res.data.messages || [];
@@ -234,6 +235,7 @@ export class ChatService {
                     parent: spaceName,
                     pageSize,
                     pageToken,
+                    orderBy,
                 });
                 const messages = res.data.messages || [];
                 logToFile(`Successfully listed ${messages.length} messages for space: ${spaceName}`);