@juzi/wechaty-puppet-service 1.0.129 → 1.0.131

package/dist/esm/src/server/puppet-implementation.js

@@ -1,3 +1,4 @@
+import { inspect } from 'util';
 import { chunkDecoder, chunkEncoder, StringValue, grpc, puppet as grpcPuppet, } from '@juzi/wechaty-grpc';
 import * as PUPPET from '@juzi/wechaty-puppet';
 import { timeoutPromise } from 'gerror';
@@ -6,9 +7,27 @@ import { normalizeFileBoxUuid, } from '../file-box-helper/mod.js';
 import { log } from '../config.js';
 import { grpcError } from './grpc-error.js';
 import { EventStreamManager } from './event-stream-manager.js';
-import { OptionalBooleanUnwrapper, OptionalBooleanWrapper, callRecordPayloadToPb, channelPayloadToPb, channelPbToPayload, chatHistoryPayloadToPb, postPbToPayload, urlLinkPayloadToPb, channelCardPayloadToPb } from '../utils/pb-payload-helper.js';
+import { OptionalBooleanUnwrapper, OptionalBooleanWrapper, callRecordPayloadToPb, channelPayloadToPb, channelPbToPayload, chatHistoryPayloadToPb, postPbToPayload, urlLinkPayloadToPb, channelCardPayloadToPb, roomPayloadToPb, roomMemberLoadStateToPb, roomMemberSnapshotToPb, validateRoomMemberLoadState, validateRoomMemberSnapshot, validateRoomMemberRevision } from '../utils/pb-payload-helper.js';
 import { grpcCallTypeToPuppetMedia, puppetCallMediaTypeToGrpc } from '../utils/call-media-mapping.js';
 import { TextContentType } from '@juzi/wechaty-puppet/types';
+function roomMemberRpcError(method, error, callback) {
+    log.error('PuppetServiceImpl', 'roomMemberRpcError() %s() rejection: %s', method, inspect(error));
+    if (typeof error === 'object' && error !== null && 'code' in error) {
+        const status = error;
+        const code = status['code'];
+        if (typeof code === 'number' && Number.isInteger(code) && code > 0 && code <= 16) {
+            const details = typeof status['details'] === 'string' ? status['details']
+                : error instanceof Error ? error.message : 'Room member RPC failed';
+            callback({ code, details }, null);
+            return;
+        }
+    }
+    if (error instanceof Error && /unsupported/i.test(error.message)) {
+        callback({ code: grpc.status.UNIMPLEMENTED, details: error.message }, null);
+        return;
+    }
+    grpcError(method, error, callback);
+}
 const setGrpcBatchResults = (response, ResultClass, payload) => {
     response.setResultsList(payload.results.map(result => {
         const pbResult = new ResultClass();
@@ -1572,6 +1591,76 @@ function puppetImplementation(puppet, FileBoxUuid) {
                 return grpcError('roomList', e, callback);
             }
         },
+        roomMemberSync: async (call, callback) => {
+            try {
+                const roomId = call.request.getId();
+                if (!roomId) {
+                    return callback({ code: grpc.status.INVALID_ARGUMENT, details: 'roomId is required' }, null);
+                }
+                const options = call.request.getForce() ? { force: true } : undefined;
+                if (typeof puppet.roomMemberSync !== 'function') {
+                    throw new Error('roomMemberSync unsupported by adapter');
+                }
+                const state = await puppet.roomMemberSync(roomId, options);
+                validateRoomMemberLoadState(state, roomId);
+                return callback(null, roomMemberLoadStateToPb(state));
+            }
+            catch (error) {
+                return roomMemberRpcError('roomMemberSync', error, callback);
+            }
+        },
+        roomMemberLoadState: async (call, callback) => {
+            try {
+                const roomId = call.request.getId();
+                if (!roomId) {
+                    return callback({ code: grpc.status.INVALID_ARGUMENT, details: 'roomId is required' }, null);
+                }
+                if (typeof puppet.roomMemberLoadState !== 'function') {
+                    throw new Error('roomMemberLoadState unsupported by adapter');
+                }
+                const state = await puppet.roomMemberLoadState(roomId);
+                validateRoomMemberLoadState(state, roomId);
+                return callback(null, roomMemberLoadStateToPb(state));
+            }
+            catch (error) {
+                return roomMemberRpcError('roomMemberLoadState', error, callback);
+            }
+        },
+        roomMemberSnapshot: async (call, callback) => {
+            try {
+                const roomId = call.request.getId();
+                const revision = call.request.getRevision();
+                const pageSize = call.request.getPageSize();
+                try {
+                    if (!roomId) {
+                        throw new Error('roomId is required');
+                    }
+                    validateRoomMemberRevision(revision);
+                    if (!Number.isInteger(pageSize) || pageSize < 0 || pageSize > 500) {
+                        throw new Error('pageSize must be an integer between 0 and 500');
+                    }
+                }
+                catch (error) {
+                    return callback({ code: grpc.status.INVALID_ARGUMENT, details: error.message }, null);
+                }
+                if (typeof puppet.roomMemberSnapshot !== 'function') {
+                    throw new Error('roomMemberSnapshot unsupported by adapter');
+                }
+                const snapshot = await puppet.roomMemberSnapshot(roomId, revision, {
+                    pageToken: call.request.getPageToken(), pageSize,
+                });
+                try {
+                    validateRoomMemberSnapshot(snapshot, roomId, revision);
+                }
+                catch (error) {
+                    return callback({ code: grpc.status.FAILED_PRECONDITION, details: error.message }, null);
+                }
+                return callback(null, roomMemberSnapshotToPb(snapshot));
+            }
+            catch (error) {
+                return roomMemberRpcError('roomMemberSnapshot', error, callback);
+            }
+        },
         roomMemberList: async (call, callback) => {
             log.verbose('PuppetServiceImpl', 'roomMemberList()');
             try {
@@ -1640,20 +1729,7 @@ function puppetImplementation(puppet, FileBoxUuid) {
             try {
                 const roomId = call.request.getId();
                 const payload = await puppet.roomPayload(roomId);
-                const response = new grpcPuppet.RoomPayloadResponse();
-                response.setAdminIdsList(payload.adminIdList);
-                response.setAvatar(payload.avatar || '');
-                response.setHandle(payload.handle || '');
-                response.setId(payload.id);
-                response.setMemberIdsList(payload.memberIdList);
-                response.setOwnerId(payload.ownerId || '');
-                response.setTopic(payload.topic);
-                response.setAdditionalInfo(payload.additionalInfo || '');
-                response.setRoomRemark(payload.remark || '');
-                response.setExternal(!!payload.external);
-                if (payload.createTime) {
-                    response.setCreateTime(timestampFromMilliseconds(payload.createTime));
-                }
+                const response = roomPayloadToPb(payload);
                 return callback(null, response);
             }
             catch (e) {
@@ -1668,20 +1744,7 @@ function puppetImplementation(puppet, FileBoxUuid) {
                 const response = new grpcPuppet.BatchRoomPayloadResponse();
                 const payloads = [];
                 for (const [_, payload] of payloadMap.entries()) {
-                    const pb = new grpcPuppet.RoomPayloadResponse();
-                    pb.setAdminIdsList(payload.adminIdList);
-                    pb.setAvatar(payload.avatar || '');
-                    pb.setHandle(payload.handle || '');
-                    pb.setId(payload.id);
-                    pb.setMemberIdsList(payload.memberIdList);
-                    pb.setOwnerId(payload.ownerId || '');
-                    pb.setTopic(payload.topic);
-                    pb.setAdditionalInfo(payload.additionalInfo || '');
-                    pb.setRoomRemark(payload.remark || '');
-                    pb.setExternal(!!payload.external);
-                    if (payload.createTime) {
-                        pb.setCreateTime(timestampFromMilliseconds(payload.createTime));
-                    }
+                    const pb = roomPayloadToPb(payload);
                     payloads.push(pb);
                 }
                 response.setRoomPayloadsList(payloads);
@@ -2403,6 +2466,50 @@ function puppetImplementation(puppet, FileBoxUuid) {
                 return grpcError('applyRoomAntiSpamStrategy', e, callback);
             }
         },
+        orgBroadcastPayload: async (call, callback) => {
+            log.verbose('PuppetServiceImpl', 'orgBroadcastPayload()');
+            try {
+                const orgBroadcastId = call.request.getId();
+                const payload = await puppet.orgBroadcastPayload(orgBroadcastId);
+                const response = new grpcPuppet.OrgBroadcastPayloadResponse();
+                response.setId(payload.id);
+                response.setSendType(payload.sendType);
+                response.setConversationType(payload.conversationType);
+                response.setCreatorId(payload.creatorId ?? '');
+                response.setExecTime(payload.execTime);
+                response.setStatus(payload.status);
+                response.setCanCancel(payload.canCancel);
+                response.setAllowSelect(payload.allowSelect);
+                response.setSent(payload.sent);
+                response.setTotalCount(payload.totalCount);
+                response.setSentCount(payload.sentCount);
+                response.setContentListJson(payload.contentListJson);
+                response.setTargetsList(payload.targets.map(target => {
+                    const targetPb = new grpcPuppet.OrgBroadcastTarget();
+                    targetPb.setContactId(target.contactId ?? '');
+                    targetPb.setRoomId(target.roomId ?? '');
+                    targetPb.setStatus(target.status);
+                    return targetPb;
+                }));
+                return callback(null, response);
+            }
+            catch (e) {
+                return grpcError('orgBroadcastPayload', e, callback);
+            }
+        },
+        orgBroadcastExecute: async (call, callback) => {
+            log.verbose('PuppetServiceImpl', 'orgBroadcastExecute()');
+            try {
+                const orgBroadcastId = call.request.getId();
+                const targetIds = call.request.getTargetIdsList();
+                // an empty target_ids on the wire means all targets of the org broadcast
+                await puppet.orgBroadcastExecute(orgBroadcastId, targetIds.length ? targetIds : undefined);
+                return callback(null, new grpcPuppet.OrgBroadcastExecuteResponse());
+            }
+            catch (e) {
+                return grpcError('orgBroadcastExecute', e, callback);
+            }
+        },
         download: async (call) => {
             log.verbose('PuppetServiceImpl', 'download()');
             try {