@juzi/wechaty-puppet-service 1.0.106 → 1.0.108

package/src/client/puppet-service.ts

@@ -50,7 +50,7 @@ import { packageJson }  from '../package-json.js'
 
 import { GrpcManager }  from './grpc-manager.js'
 import { PayloadStore } from './payload-store.js'
-import { OptionalBooleanUnwrapper, OptionalBooleanWrapper, callRecordPbToPayload, channelPayloadToPb, channelPbToPayload, chatHistoryPbToPayload, postPayloadToPb, roomMemberPbToPayload, urlLinkPbToPayload } from '../utils/pb-payload-helper.js'
+import { OptionalBooleanUnwrapper, OptionalBooleanWrapper, callRecordPbToPayload, channelPayloadToPb, channelPbToPayload, chatHistoryPbToPayload, contactPbToPayload, postPayloadToPb, roomMemberPbToPayload, urlLinkPbToPayload } from '../utils/pb-payload-helper.js'
 import type { MessageBroadcastTargets } from '@juzi/wechaty-puppet/dist/esm/src/schemas/message.js'
 import { timeoutPromise } from 'gerror'
 import { BooleanIndicator } from 'state-switch'
@@ -763,6 +763,49 @@ class PuppetService extends PUPPET.Puppet {
     return payload
   }
 
+  override async batchContactRawPayload (contactIdList: string[]): Promise<Map<string, PUPPET.payloads.Contact>> {
+    log.verbose('PuppetService', 'batchContactRawPayload(%s)', contactIdList)
+
+    const result = new Map<string, PUPPET.payloads.Contact>()
+    const contactIdSet = new Set<string>(contactIdList)
+    const needGetSet = new Set<string>()
+    for (const contactId of contactIdSet) {
+      const cachedContactPayload = await this._payloadStore.contact?.get(contactId)
+      if (cachedContactPayload) {
+        result.set(contactId, cachedContactPayload)
+      } else {
+        needGetSet.add(contactId)
+      }
+    }
+
+    if (needGetSet.size > 0) {
+      try {
+        const request = new grpcPuppet.BatchContactPayloadRequest()
+        request.setIdsList(Array.from(needGetSet))
+
+        const response = await util.promisify(
+          this.grpcManager.client.batchContactPayload
+            .bind(this.grpcManager.client),
+        )(request)
+
+        const payloads = response.getContactPayloadsList()
+        for (const payload of payloads) {
+          const contactId = payload.getId()
+          const puppetPayload = contactPbToPayload(payload)
+          result.set(contactId, puppetPayload)
+          await this._payloadStore.contact?.set(contactId, puppetPayload)
+        }
+      } catch (e) {
+        log.error('PuppetService', 'batchContactRawPayload(%s, %s) error: %s, use one by one method', contactIdList, needGetSet, e)
+        for (const contactId of needGetSet) {
+          const payload = await this.contactRawPayload(contactId)
+          result.set(contactId, payload)
+        }
+      }
+    }
+    return result
+  }
+
   override async contactPayloadModify (contactId: string, payload: Partial<PUPPET.payloads.Contact>): Promise<void> {
     log.verbose('PuppetService', 'contactPayloadModify(%s, %s)', contactId, JSON.stringify(payload))
 
@@ -1973,10 +2016,12 @@ class PuppetService extends PUPPET.Puppet {
             .bind(this.grpcManager.client),
         )(request)
 
-        const map = response.getMembersMap()
-        for (const [ contactId, payload ] of Object.entries(map)) {
-          result.set(contactId, roomMemberPbToPayload(payload))
-          cachedPayload[contactId] = payload
+        const payloads = response.getMemberPayloadsList()
+        for (const payload of payloads) {
+          const contactId = payload.getId()
+          const puppetPayload = roomMemberPbToPayload(payload)
+          result.set(contactId, puppetPayload)
+          cachedPayload[contactId] = puppetPayload
         }
         await this._payloadStore.roomMember?.set(roomId, cachedPayload)
       } catch (e) {

package/src/config.ts

@@ -24,8 +24,8 @@ const envVars = {
  */
 const GRPC_OPTIONS = {
   // https://github.com/wechaty/wechaty-puppet-service/issues/86
-  'grpc.max_receive_message_length': 1024 * 1024 * 20,
-  'grpc.max_send_message_length': 1024 * 1024 * 20,
+  'grpc.max_receive_message_length': 1024 * 1024 * 80,
+  'grpc.max_send_message_length': 1024 * 1024 * 80,
 }
 
 export const NO_LOG_EVENTS: grpcPuppet.EventTypeMap[keyof grpcPuppet.EventTypeMap][] = [

package/src/package-json.ts

@@ -4,7 +4,7 @@
 import type { PackageJson } from 'type-fest'
 export const packageJson: PackageJson = {
   "name": "@juzi/wechaty-puppet-service",
-  "version": "1.0.106",
+  "version": "1.0.108",
   "description": "Puppet Service for Wechaty",
   "type": "module",
   "exports": {
@@ -55,7 +55,7 @@ export const packageJson: PackageJson = {
     "@chatie/eslint-config": "^1.0.4",
     "@chatie/semver": "^0.4.7",
     "@chatie/tsconfig": "^4.6.3",
-    "@juzi/wechaty-puppet": "^1.0.104",
+    "@juzi/wechaty-puppet": "^1.0.106",
     "@juzi/wechaty-puppet-mock": "^1.0.1",
     "@swc/core": "1.3.39",
     "@types/google-protobuf": "^3.15.5",
@@ -77,7 +77,7 @@ export const packageJson: PackageJson = {
     "@juzi/wechaty-puppet": "^1.0.84"
   },
   "dependencies": {
-    "@juzi/wechaty-grpc": "^1.0.92",
+    "@juzi/wechaty-grpc": "^1.0.94",
     "clone-class": "^1.1.1",
     "ducks": "^1.0.2",
     "file-box": "^1.5.5",

package/src/server/puppet-implementation.ts

@@ -293,6 +293,57 @@ function puppetImplementation (
       }
     },
 
+    batchContactPayload: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'batchContactPayload()')
+
+      try {
+        const contactIdList = call.request.getIdsList()
+
+        const payloadMap = await puppet.batchContactPayload(contactIdList)
+
+        const response = new grpcPuppet.BatchContactPayloadResponse()
+
+        const payloads: grpcPuppet.ContactPayloadResponse[] = []
+        for (const [ _, payload ] of payloadMap.entries()) {
+          const pb = new grpcPuppet.ContactPayloadResponse()
+          pb.setAddress(payload.address || '')
+          pb.setAlias(payload.alias || '')
+          pb.setAvatar(payload.avatar)
+          pb.setCity(payload.city || '')
+          pb.setFriend(payload.friend || false)
+          pb.setGender(payload.gender)
+          pb.setId(payload.id)
+          pb.setName(payload.name)
+          pb.setProvince(payload.province || '')
+          pb.setSignature(payload.signature || '')
+          pb.setStar(payload.star || false)
+          pb.setType(payload.type)
+          /**
+           * @deprecated `payload.weixin` will be removed in v2.0
+           *  @link https://github.com/wechaty/grpc/issues/174
+           */
+          pb.setWeixin(payload.handle || payload.weixin || '')
+          pb.setPhonesList(payload.phone)
+          pb.setCoworker(payload.coworker || false)
+          pb.setCorporation(payload.corporation || '')
+          pb.setTitle(payload.title || '')
+          pb.setDescription(payload.description || '')
+          pb.setAdditionalInfo(payload.additionalInfo || '')
+          pb.setTagIdsList(payload.tags || [])
+          pb.setRealName(payload.realName || '')
+          pb.setAka(payload.aka || '')
+          payloads.push(pb)
+        }
+
+        response.setContactPayloadsList(payloads)
+
+        return callback(null, response)
+
+      } catch (e) {
+        return grpcError('batchContactPayload', e, callback)
+      }
+    },
+
     contactPayloadModify: async (call, callback) => {
       log.verbose('PuppetServiceImpl', 'contactPayloadModify()')
 
@@ -1663,8 +1714,9 @@ function puppetImplementation (
         const payloadMap = await puppet.batchRoomMemberPayload(roomId, contactIdList)
 
         const response = new grpcPuppet.BatchRoomMemberPayloadResponse()
-        const responseMap = response.getMembersMap()
-        for (const [ contactId, payload ] of payloadMap.entries()) {
+
+        const payloads: grpcPuppet.RoomMemberPayloadResponse[] = []
+        for (const [ _, payload ] of payloadMap.entries()) {
           const pb = new grpcPuppet.RoomMemberPayloadResponse()
           pb.setAvatar(payload.avatar)
           pb.setId(payload.id)
@@ -1674,9 +1726,11 @@ function puppetImplementation (
           pb.setAdditionalInfo(payload.additionalInfo || '')
           pb.setJoinScene(payload.joinScene || PUPPET.types.RoomMemberJoinScene.Unknown)
           pb.setJoinTime(payload.joinTime || 0)
-          responseMap.set(contactId, pb)
+          payloads.push(pb)
         }
 
+        response.setMemberPayloadsList(payloads)
+
         return callback(null, response)
 
       } catch (e) {

package/src/utils/pb-payload-helper.ts

@@ -79,6 +79,8 @@ export const channelPayloadToPb = (grpcPuppet: grpcPuppet, channelPayload: PUPPE
   if (channelPayload.url) { pbChannelPayload.setUrl(channelPayload.url) }
   if (channelPayload.objectId) { pbChannelPayload.setObjectId(channelPayload.objectId) }
   if (channelPayload.objectNonceId) { pbChannelPayload.setObjectNonceId(channelPayload.objectNonceId) }
+  if (channelPayload.authIconUrl) { pbChannelPayload.setAuthIconUrl(channelPayload.authIconUrl) }
+  if (channelPayload.authJob) { pbChannelPayload.setAuthJob(channelPayload.authJob) }
   return pbChannelPayload
 }
 
@@ -389,3 +391,40 @@ export const roomMemberPbToPayload = (pb: puppet.RoomMemberPayloadResponse) => {
 
   return payload
 }
+
+export const contactPbToPayload = (pb: puppet.ContactPayloadResponse) => {
+  const payload: PUPPET.payloads.Contact = {
+    address     : pb.getAddress(),
+    alias       : pb.getAlias(),
+    avatar      : pb.getAvatar(),
+    city        : pb.getCity(),
+    corporation : pb.getCorporation(),
+    coworker    : pb.getCoworker(),
+    description : pb.getDescription(),
+    friend      : pb.getFriend(),
+    gender      : pb.getGender() as number,
+    /**
+     * Huan(202203): rename `getWeixin()` to `getHandle()` in v2.0.0
+     *  @link https://github.com/wechaty/grpc/issues/174
+     */
+    handle      : pb.getWeixin(),
+    id          : pb.getId(),
+    name        : pb.getName(),
+    phone       : pb.getPhonesList(),
+    province    : pb.getProvince(),
+    signature   : pb.getSignature(),
+    star        : pb.getStar(),
+    title       : pb.getTitle(),
+    type        : pb.getType() as number,
+    /**
+     * `weixin` is deprecated, will be removed after Dec 31, 2022
+     * use `handle` instead.
+     */
+    weixin        : pb.getWeixin(),
+    additionalInfo: pb.getAdditionalInfo(),
+    tags          : pb.getTagIdsList(),
+    realName      : pb.getRealName(),
+    aka           : pb.getAka(),
+  }
+  return payload
+}