@juzi/wechaty-puppet-service 1.0.18 → 1.0.21

package/src/client/puppet-service.ts

@@ -68,7 +68,6 @@ import { packageJson }  from '../package-json.js'
 
 import { GrpcManager }  from './grpc-manager.js'
 import { PayloadStore } from './payload-store.js'
-import type { TagIdentifier } from '@juzi/wechaty-puppet/filters'
 
 export type PuppetServiceOptions = PUPPET.PuppetOptions & {
   authority?  : string
@@ -348,6 +347,12 @@ class PuppetService extends PUPPET.Puppet {
       case grpcPuppet.EventType.EVENT_TYPE_SCAN:
         this.emit('scan', JSON.parse(payload) as PUPPET.payloads.EventScan)
         break
+      case grpcPuppet.EventType.EVENT_TYPE_TAG:
+        this.emit('tag', JSON.parse(payload) as PUPPET.payloads.EventTag)
+        break
+      case grpcPuppet.EventType.EVENT_TYPE_TAG_GROUP:
+        this.emit('tag-group', JSON.parse(payload) as PUPPET.payloads.EventTagGroup)
+        break
       case grpcPuppet.EventType.EVENT_TYPE_RESET:
         log.warn('PuppetService', 'onGrpcStreamEvent() got an EventType.EVENT_TYPE_RESET ?')
         // the `reset` event should be dealed not send out
@@ -440,6 +445,8 @@ class PuppetService extends PUPPET.Puppet {
       [PUPPET.types.Dirty.Post]:         async (_: string) => {},
       [PUPPET.types.Dirty.Room]:         async (id: string) => this._payloadStore.room?.delete(id),
       [PUPPET.types.Dirty.RoomMember]:   async (id: string) => this._payloadStore.roomMember?.delete(id),
+      [PUPPET.types.Dirty.Tag]:          async (key: string) => this._payloadStore.tag?.delete(key),
+      [PUPPET.types.Dirty.TagGroup]:     async (id: string) => this._payloadStore.tagGroup?.delete(id),
       [PUPPET.types.Dirty.Unspecified]:  async (id: string) => { throw new Error('Unspecified type with id: ' + id) },
     }
 
@@ -1930,19 +1937,14 @@ class PuppetService extends PUPPET.Puppet {
    */
 
   override async tagContactTagAdd (
-    tags: TagIdentifier[],
+    tagIds: string[],
     contactIds: string[],
   ): Promise<void> {
-    log.verbose('PuppetService', 'tagContactTagAdd(%s, %s)', JSON.stringify(tags), contactIds)
+    log.verbose('PuppetService', 'tagContactTagAdd(%s, %s)', tagIds, contactIds)
 
     const request = new grpcPuppet.TagContactTagAddRequest()
 
-    request.setTagsList(tags.map(t => {
-      const tag = new grpcPuppet.TagIdentifier()
-      t.groupId && tag.setGroupId(t.groupId)
-      tag.setId(t.id)
-      return tag
-    }))
+    request.setTagIdsList(tagIds)
     request.setContactIdsList(contactIds)
 
     await util.promisify(
@@ -1952,19 +1954,14 @@ class PuppetService extends PUPPET.Puppet {
   }
 
   override async tagContactTagRemove (
-    tags: TagIdentifier[],
+    tagIds: string[],
     contactIds: string[],
   ): Promise<void> {
-    log.verbose('PuppetService', 'tagContactTagRemove(%s, %s)', JSON.stringify(tags), contactIds)
+    log.verbose('PuppetService', 'tagContactTagRemove(%s, %s)', tagIds, contactIds)
 
     const request = new grpcPuppet.TagContactTagRemoveRequest()
 
-    request.setTagsList(tags.map(t => {
-      const tag = new grpcPuppet.TagIdentifier()
-      t.groupId && tag.setGroupId(t.groupId)
-      tag.setId(t.id)
-      return tag
-    }))
+    request.setTagIdsList(tagIds)
     request.setContactIdsList(contactIds)
 
     await util.promisify(
@@ -1975,7 +1972,7 @@ class PuppetService extends PUPPET.Puppet {
 
   override async tagGroupAdd (
     tagGroupName: string,
-  ): Promise<PUPPET.payloads.TagGroup | void> {
+  ): Promise<string | void> {
     log.verbose('PuppetService', 'tagGroupAdd(%s)', tagGroupName)
 
     const request = new grpcPuppet.TagGroupAddRequest()
@@ -1987,18 +1984,9 @@ class PuppetService extends PUPPET.Puppet {
         .bind(this.grpcManager.client),
     )(request)
 
-    const grpcPayload = result.getPayload()
-    const id = grpcPayload?.getId()
-    const name = grpcPayload?.getName()
-
-    if (id && name) {
-      const payload: PUPPET.payloads.TagGroup = {
-        id,
-        name,
-      }
-      return payload
-    }
+    const id = result.getTagGroupId()
 
+    return id
   }
 
   override async tagGroupDelete (
@@ -2020,7 +2008,7 @@ class PuppetService extends PUPPET.Puppet {
   override async tagTagAdd (
     tagName: string,
     tagGroupId?: string,
-  ): Promise<PUPPET.payloads.Tag | void> {
+  ): Promise<string | void> {
     log.verbose('PuppetService', 'tagTagAdd(%s, %s)', tagName, tagGroupId)
 
     const request = new grpcPuppet.TagTagAddRequest()
@@ -2035,37 +2023,18 @@ class PuppetService extends PUPPET.Puppet {
         .bind(this.grpcManager.client),
     )(request)
 
-    const grpcPayload = result.getPayload()
-    const id = grpcPayload?.getId()
-    const name = grpcPayload?.getName()
-    const groupId = grpcPayload?.getGroupId()
-    const type = grpcPayload?.getType()
-
-    if (id && name && (typeof type !== 'undefined')) {
-      const payload: PUPPET.payloads.Tag = {
-        id,
-        name,
-        groupId,
-        type,
-      }
-      return payload
-    }
+    const id = result.getTagId()
 
+    return id
   }
 
   override async tagTagDelete (
-    tag: TagIdentifier,
+    tagId: string,
   ): Promise<void> {
-    log.verbose('PuppetService', 'tagTagDelete(%s)', JSON.stringify(tag))
+    log.verbose('PuppetService', 'tagTagDelete(%s)', tagId)
 
     const request = new grpcPuppet.TagTagDeleteRequest()
-    const tagProb = new grpcPuppet.TagIdentifier()
-
-    tag.groupId && tagProb.setGroupId(tag.groupId)
-    tagProb.setId(tag.id)
-    request.setTag(tagProb)
-
-    request.setTag(tagProb)
+    request.setTagId(tagId)
 
     await util.promisify(
       this.grpcManager.client.tagTagDelete
@@ -2074,7 +2043,7 @@ class PuppetService extends PUPPET.Puppet {
 
   }
 
-  override async tagGroupList (): Promise<PUPPET.payloads.TagGroup[]> {
+  override async tagGroupList (): Promise<string[]> {
     log.verbose('PuppetService', 'tagGroupList()')
 
     const request = new grpcPuppet.TagGroupListRequest()
@@ -2084,19 +2053,13 @@ class PuppetService extends PUPPET.Puppet {
         .bind(this.grpcManager.client),
     )(request)
 
-    const grpcPayloads = result.getPayloadsList()
-    const payloads: PUPPET.payloads.TagGroup[] = grpcPayloads.map(payload => {
-      return {
-        id: payload.getId(),
-        name: payload.getName(),
-      } as PUPPET.payloads.TagGroup
-    })
-    return payloads
+    const groupIds = result.getTagGroupIdsList()
+    return groupIds
   }
 
   override async tagGroupTagList (
     tagGroupId?: string,
-  ): Promise<PUPPET.payloads.Tag[]> {
+  ): Promise<string[]> {
     log.verbose('PuppetService', 'tagGroupTagList(%s)', tagGroupId)
 
     const request = new grpcPuppet.TagGroupTagListRequest()
@@ -2109,20 +2072,12 @@ class PuppetService extends PUPPET.Puppet {
         .bind(this.grpcManager.client),
     )(request)
 
-    const grpcPayloads = result.getPayloadsList()
-    const payloads: PUPPET.payloads.Tag[] = grpcPayloads.map(payload => {
-      return {
-        id: payload.getId(),
-        name: payload.getName(),
-        groupId: payload.getGroupId(),
-        type: payload.getType(),
-      } as PUPPET.payloads.Tag
-    })
-    return payloads
+    const tagIds = result.getTagIdsList()
+    return tagIds
   }
 
   override async tagTagList (
-  ): Promise<PUPPET.payloads.Tag[]> {
+  ): Promise<string[]> {
     log.verbose('PuppetService', 'tagTagList()')
 
     const request = new grpcPuppet.TagTagListRequest()
@@ -2132,21 +2087,13 @@ class PuppetService extends PUPPET.Puppet {
         .bind(this.grpcManager.client),
     )(request)
 
-    const grpcPayloads = result.getPayloadsList()
-    const payloads: PUPPET.payloads.Tag[] = grpcPayloads.map(payload => {
-      return {
-        id: payload.getId(),
-        name: payload.getName(),
-        groupId: payload.getGroupId(),
-        type: payload.getType(),
-      } as PUPPET.payloads.Tag
-    })
-    return payloads
+    const tagIds = result.getTagIdsList()
+    return tagIds
   }
 
   override async tagContactTagList (
     contactId: string,
-  ): Promise<PUPPET.payloads.Tag[]> {
+  ): Promise<string[]> {
     log.verbose('PuppetService', 'tagContactTagList(%s)', contactId)
 
     const request = new grpcPuppet.TagContactTagListRequest()
@@ -2157,37 +2104,94 @@ class PuppetService extends PUPPET.Puppet {
         .bind(this.grpcManager.client),
     )(request)
 
-    const grpcPayloads = result.getPayloadsList()
-    const payloads: PUPPET.payloads.Tag[] = grpcPayloads.map(payload => {
-      return {
-        id: payload.getId(),
-        name: payload.getName(),
-        groupId: payload.getGroupId(),
-        type: payload.getType(),
-      } as PUPPET.payloads.Tag
-    })
-    return payloads
+    const tagIds = result.getTagIdsList()
+    return tagIds
   }
 
   override async tagTagContactList (
-    tag: TagIdentifier,
+    tagId: string,
   ): Promise<string[]> {
-    log.verbose('PuppetService', 'tagTagContactList(%s)', JSON.stringify(tag))
+    log.verbose('PuppetService', 'tagTagContactList(%s)', tagId)
 
     const request = new grpcPuppet.TagTagContactListRequest()
-    const tagProb = new grpcPuppet.TagIdentifier()
-
-    tag.groupId && tagProb.setGroupId(tag.groupId)
-    tagProb.setId(tag.id)
-    request.setTag(tagProb)
+    request.setTagId(tagId)
 
     const result = await util.promisify(
       this.grpcManager.client.tagTagContactList
         .bind(this.grpcManager.client),
     )(request)
 
-    const payloads = result.getPayloadsList()
-    return payloads
+    const contactIds = result.getContactIdsList()
+    return contactIds
+  }
+
+  override async tagGroupPayloadPuppet (id: string): Promise<PUPPET.payloads.TagGroup> {
+    log.verbose('PuppetService', 'tagGroupPayload(%s)', id)
+
+    const cachedPayload = await this._payloadStore.tagGroup?.get(id)
+    if (cachedPayload) {
+      log.silly('PuppetService', 'tagGroupPayload(%s) cache HIT', id)
+      return cachedPayload
+    }
+
+    const request = new grpcPuppet.TagGroupPayloadRequest()
+    request.setGroupId(id)
+
+    const response = await util.promisify(
+      this.grpcManager.client.tagGroupPayload
+        .bind(this.grpcManager.client),
+    )(request)
+    const grpcPayload = response.getPayload()
+
+    if (!grpcPayload) {
+      throw new Error(`tagGroup ${id} got no payload!`)
+    }
+
+    const payload: PUPPET.payloads.TagGroup = {
+      id: grpcPayload.getId(),
+      name: grpcPayload.getName(),
+      type: grpcPayload.getType(),
+    }
+
+    await this._payloadStore.tagGroup?.set(id, payload)
+    log.silly('PuppetService', 'tagGroupPayloadPuppet(%s) cache SET', id)
+
+    return payload
+  }
+
+  override async tagPayloadPuppet (tagId: string): Promise<PUPPET.payloads.Tag> {
+    log.verbose('PuppetService', 'tagPayloadPuppet(%s)', tagId)
+
+    const cachedPayload = await this._payloadStore.tag?.get(tagId)
+    if (cachedPayload) {
+      log.silly('PuppetService', 'tagPayloadPuppet(%s) cache HIT', tagId)
+      return cachedPayload
+    }
+
+    const request = new grpcPuppet.TagPayloadRequest()
+    request.setTagId(tagId)
+
+    const response = await util.promisify(
+      this.grpcManager.client.tagPayload
+        .bind(this.grpcManager.client),
+    )(request)
+    const grpcPayload = response.getPayload()
+
+    if (!grpcPayload) {
+      throw new Error(`tag ${tagId} got no payload!`)
+    }
+
+    const payload: PUPPET.payloads.Tag = {
+      id: grpcPayload.getId(),
+      name: grpcPayload.getName(),
+      groupId: grpcPayload.getGroupId(),
+      type: grpcPayload.getType(),
+    }
+
+    await this._payloadStore.tag?.set(tagId, payload)
+    log.silly('PuppetService', 'tagPayloadPuppet(%s) cache SET', tagId)
+
+    return payload
   }
 
   /**

package/src/config.ts

@@ -12,6 +12,7 @@ import {
 } from '@juzi/wechaty-grpc'
 
 const VERSION = packageJson.version || '0.0.0'
+export const FOUR_PER_EM_SPACE = String.fromCharCode(0x2005)
 
 const envVars = {
   ...rootEnvVars,

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.18",
+  "version": "1.0.21",
   "description": "Puppet Service for Wechaty",
   "type": "module",
   "exports": {
@@ -55,6 +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-mock": "^1.0.1",
     "@types/google-protobuf": "^3.15.5",
     "@types/lru-cache": "^7.5.0",
     "@types/semver": "^7.3.9",
@@ -65,14 +66,14 @@ export const packageJson: PackageJson = {
     "temp": "^0.9.4",
     "typed-emitter": "^1.5.0-from-event",
     "utility-types": "^3.10.0",
-    "@juzi/wechaty-puppet-mock": "^1.0.1",
     "why-is-node-running": "^2.2.1"
   },
   "peerDependencies": {
-    "@juzi/wechaty-puppet": "^1.0.15"
+    "@juzi/wechaty-puppet": "^1.0.28"
   },
   "dependencies": {
-    "@juzi/wechaty-grpc": "^1.0.12",
+    "@juzi/wechaty-grpc": "^1.0.17",
+    "@juzi/wechaty-redux": "^1.0.1",
     "clone-class": "^1.1.1",
     "ducks": "^1.0.2",
     "file-box": "^1.5.5",
@@ -83,7 +84,6 @@ export const packageJson: PackageJson = {
     "semver": "^7.3.5",
     "stronger-typed-streams": "^0.2.0",
     "uuid": "^8.3.2",
-    "@juzi/wechaty-redux": "^1.0.1",
     "wechaty-token": "^1.0.6"
   },
   "publishConfig": {

package/src/server/event-stream-manager.ts

@@ -273,6 +273,20 @@ class EventStreamManager {
           this.offCallbackList.push(off)
           break
         }
+        case 'tag': {
+          const listener = (payload: PUPPET.payloads.EventTag) => this.grpcEmit(grpcPuppet.EventType.EVENT_TYPE_TAG, payload)
+          this.puppet.on('tag', listener)
+          const off = () => this.puppet.off('tag', listener)
+          this.offCallbackList.push(off)
+          break
+        }
+        case 'tag-group': {
+          const listener = (payload: PUPPET.payloads.EventTagGroup) => this.grpcEmit(grpcPuppet.EventType.EVENT_TYPE_TAG_GROUP, payload)
+          this.puppet.on('tag-group', listener)
+          const off = () => this.puppet.off('tag-group', listener)
+          this.offCallbackList.push(off)
+          break
+        }
         case 'reset':
           // the `reset` event should be dealed internally, should not send out
           break

package/src/server/puppet-implementation.ts

@@ -1507,16 +1507,10 @@ function puppetImplementation (
       log.verbose('PuppetServiceImpl', 'tagContactTagAdd()')
 
       try {
-        const tagsProb = call.request.getTagsList()
+        const tagIds = call.request.getTagIdsList()
         const contactIds = call.request.getContactIdsList()
-        const tags = tagsProb.map(tagProb => {
-          return {
-            id: tagProb.getId(),
-            groupId: tagProb.getGroupId(),
-          }
-        })
 
-        await puppet.tagContactTagAdd(tags, contactIds)
+        await puppet.tagContactTagAdd(tagIds, contactIds)
 
         return callback(null, new grpcPuppet.TagContactTagAddResponse())
       } catch (e) {
@@ -1528,16 +1522,10 @@ function puppetImplementation (
       log.verbose('PuppetServiceImpl', 'tagContactTagRemove()')
 
       try {
-        const tagsProb = call.request.getTagsList()
+        const tagIds = call.request.getTagIdsList()
         const contactIds = call.request.getContactIdsList()
-        const tags = tagsProb.map(tagProb => {
-          return {
-            id: tagProb.getId(),
-            groupId: tagProb.getGroupId(),
-          }
-        })
 
-        await puppet.tagContactTagRemove(tags, contactIds)
+        await puppet.tagContactTagRemove(tagIds, contactIds)
 
         return callback(null, new grpcPuppet.TagContactTagRemoveResponse())
       } catch (e) {
@@ -1552,13 +1540,11 @@ function puppetImplementation (
         const tagGroupName = call.request.getTagGroupName()
 
         const result = await puppet.tagGroupAdd(tagGroupName)
-        const payload = new grpcPuppet.TagGroupPayload()
+
         const response = new grpcPuppet.TagGroupAddResponse()
 
         if (result) {
-          payload.setId(result.id)
-          payload.setName(result.name)
-          response.setPayload(payload)
+          response.setTagGroupId(result)
         }
 
         return callback(null, response)
@@ -1589,17 +1575,10 @@ function puppetImplementation (
         const tagName = call.request.getTagName()
 
         const result = await puppet.tagTagAdd(tagName, tagGroupId)
-        const payload = new grpcPuppet.TagPayload()
         const response = new grpcPuppet.TagTagAddResponse()
 
         if (result) {
-          payload.setId(result.id)
-          payload.setName(result.name)
-          payload.setType(result.type)
-          if (result.groupId) {
-            payload.setGroupId(result.groupId)
-          }
-          response.setPayload(payload)
+          response.setTagId(result)
         }
 
         return callback(null, response)
@@ -1612,11 +1591,9 @@ function puppetImplementation (
       log.verbose('PuppetServiceImpl', 'tagTagDelete()')
 
       try {
-        const tagProb = call.request.getTag()
-        const id = tagProb?.getId()!
-        const groupId = tagProb?.getGroupId()
+        const tagId = call.request.getTagId()
 
-        await puppet.tagTagDelete({ id, groupId })
+        await puppet.tagTagDelete(tagId)
 
         return callback(null, new grpcPuppet.TagTagDeleteResponse())
       } catch (e) {
@@ -1630,14 +1607,8 @@ function puppetImplementation (
 
       try {
         const result = await puppet.tagGroupList()
-        const payloads = result.map(tagGroup => {
-          const payload = new grpcPuppet.TagGroupPayload()
-          payload.setId(tagGroup.id)
-          payload.setName(tagGroup.name)
-          return payload
-        })
         const response = new grpcPuppet.TagGroupListResponse()
-        response.setPayloadsList(payloads)
+        response.setTagGroupIdsList(result)
 
         return callback(null, response)
       } catch (e) {
@@ -1652,18 +1623,9 @@ function puppetImplementation (
         const tagGroupId = call.request.getTagGroupId()
 
         const result = await puppet.tagGroupTagList(tagGroupId)
-        const payloads = result.map(tag => {
-          const payload = new grpcPuppet.TagPayload()
-          payload.setId(tag.id)
-          payload.setName(tag.name)
-          payload.setType(tag.type)
-          if (tag.groupId) {
-            payload.setGroupId(tag.groupId)
-          }
-          return payload
-        })
+
         const response = new grpcPuppet.TagGroupTagListResponse()
-        response.setPayloadsList(payloads)
+        response.setTagIdsList(result)
 
         return callback(null, response)
       } catch (e) {
@@ -1677,18 +1639,9 @@ function puppetImplementation (
 
       try {
         const result = await puppet.tagTagList()
-        const payloads = result.map(tag => {
-          const payload = new grpcPuppet.TagPayload()
-          payload.setId(tag.id)
-          payload.setName(tag.name)
-          payload.setType(tag.type)
-          if (tag.groupId) {
-            payload.setGroupId(tag.groupId)
-          }
-          return payload
-        })
+
         const response = new grpcPuppet.TagTagListResponse()
-        response.setPayloadsList(payloads)
+        response.setTagIdsList(result)
 
         return callback(null, response)
       } catch (e) {
@@ -1703,18 +1656,9 @@ function puppetImplementation (
         const contactId = call.request.getContactId()
 
         const result = await puppet.tagContactTagList(contactId)
-        const payloads = result.map(tag => {
-          const payload = new grpcPuppet.TagPayload()
-          payload.setId(tag.id)
-          payload.setName(tag.name)
-          payload.setType(tag.type)
-          if (tag.groupId) {
-            payload.setGroupId(tag.groupId)
-          }
-          return payload
-        })
+
         const response = new grpcPuppet.TagContactTagListResponse()
-        response.setPayloadsList(payloads)
+        response.setTagIdsList(result)
 
         return callback(null, response)
       } catch (e) {
@@ -1726,14 +1670,12 @@ function puppetImplementation (
       log.verbose('PuppetServiceImpl', 'tagTagContactList()')
 
       try {
-        const tagProb = call.request.getTag()
-        const id = tagProb?.getId()!
-        const groupId = tagProb?.getGroupId()
+        const tagId = call.request.getTagId()
 
-        const result = await puppet.tagTagContactList({ id, groupId })
+        const result = await puppet.tagTagContactList(tagId)
 
         const response = new grpcPuppet.TagTagContactListResponse()
-        response.setPayloadsList(result)
+        response.setContactIdsList(result)
 
         return callback(null, response)
       } catch (e) {
@@ -1741,6 +1683,51 @@ function puppetImplementation (
       }
     },
 
+    tagGroupPayload: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagGroupPayload()')
+
+      try {
+        const id = call.request.getGroupId()
+
+        const result = await puppet.tagGroupPayloadPuppet(id)
+        const response = new grpcPuppet.TagGroupPayloadResponse()
+        const payload = new grpcPuppet.TagGroupPayload()
+        payload.setId(result.id)
+        payload.setName(result.name)
+        payload.setType(result.type)
+        response.setPayload(payload)
+
+        return callback(null, response)
+
+      } catch (e) {
+        return grpcError('tagTagContactList', e, callback)
+      }
+    },
+
+    tagPayload: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagPayload()')
+
+      try {
+        const tagId = call.request.getTagId()
+
+        const result = await puppet.tagPayloadPuppet(tagId)
+        const response = new grpcPuppet.TagPayloadResponse()
+        const payload = new grpcPuppet.TagPayload()
+        payload.setId(result.id)
+        payload.setName(result.name)
+        payload.setType(result.type)
+        if (result.groupId) {
+          payload.setGroupId(result.groupId)
+        }
+        response.setPayload(payload)
+
+        return callback(null, response)
+
+      } catch (e) {
+        return grpcError('tagTagContactList', e, callback)
+      }
+    },
+
     version: async (call, callback) => {
       log.verbose('PuppetServiceImpl', 'version() v%s', puppet.version())
       void call