@juzi/wechaty-puppet-service 1.0.12 → 1.0.16

package/src/server/puppet-implementation.ts

@@ -1492,101 +1492,235 @@ function puppetImplementation (
       }
     },
 
-    tagContactAdd: async (call, callback) => {
-      log.verbose('PuppetServiceImpl', 'tagContactAdd()')
+    /**
+     *
+     * tag section
+     *
+     */
+
+    tagContactTagAdd: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagContactTagAdd()')
 
       try {
-        const tagId = call.request.getId()
+        const tagGroupId = call.request.getTagGroupId()
+        const tagId = call.request.getTagId()
         const contactId = call.request.getContactId()
 
-        await puppet.tagContactAdd(tagId, contactId)
+        await puppet.tagContactTagAdd(tagGroupId, tagId, contactId)
+
+        return callback(null, new grpcPuppet.TagContactTagAddResponse())
+      } catch (e) {
+        return grpcError('tagContactTagAdd', e, callback)
+      }
+    },
+
+    tagContactTagRemove: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagContactTagRemove()')
 
-        return callback(null, new grpcPuppet.TagContactAddResponse())
+      try {
+        const tagGroupId = call.request.getTagGroupId()
+        const tagId = call.request.getTagId()
+        const contactId = call.request.getContactId()
 
+        await puppet.tagContactTagRemove(tagGroupId, tagId, contactId)
+
+        return callback(null, new grpcPuppet.TagContactTagRemoveResponse())
       } catch (e) {
-        return grpcError('tagContactAdd', e, callback)
+        return grpcError('tagContactTagRemove', e, callback)
       }
     },
 
-    tagContactDelete: async (call, callback) => {
-      log.verbose('PuppetServiceImpl', 'tagContactDelete()')
+    tagGroupAdd: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagGroupAdd()')
 
       try {
-        const tagId = call.request.getId()
+        const tagGroupName = call.request.getTagGroupName()
 
-        await puppet.tagContactDelete(tagId)
+        const result = await puppet.tagGroupAdd(tagGroupName)
+        const payload = new grpcPuppet.TagGroupPayload()
+        const response = new grpcPuppet.TagGroupAddResponse()
 
-        return callback(null, new grpcPuppet.TagContactDeleteResponse())
+        if (result) {
+          payload.setId(result.id)
+          payload.setName(result.name)
+          response.setPayload(payload)
+        }
 
+        return callback(null, response)
       } catch (e) {
-        return grpcError('tagContactDelete', e, callback)
+        return grpcError('tagGroupAdd', e, callback)
       }
     },
 
-    tagContactList: async (call, callback) => {
-      log.verbose('PuppetServiceImpl', 'tagContactList()')
+    tagGroupDelete: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagGroupDelete()')
 
       try {
-        const contactId = call.request.getContactId()
+        const tagGroupId = call.request.getTagGroupId()
 
-        /**
-         * for a specific contact
-         */
-        if (contactId) {
-          const tagIdList = await puppet.tagContactList(contactId)
+        await puppet.tagGroupDelete(tagGroupId)
+
+        return callback(null, new grpcPuppet.TagGroupDeleteResponse())
+      } catch (e) {
+        return grpcError('tagGroupDelete', e, callback)
+      }
+    },
 
-          const response = new grpcPuppet.TagContactListResponse()
-          response.setIdsList(tagIdList)
+    tagTagAdd: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagTagAdd()')
 
-          return callback(null, new grpcPuppet.TagContactListResponse())
+      try {
+        const tagGroupId = call.request.getTagGroupId()
+        const tagName = call.request.getTagName()
+
+        const result = await puppet.tagTagAdd(tagGroupId, tagName)
+        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)
         }
 
-        {
-          /**
-            * Huan(202110): Deprecated: will be removed after Dec 31, 2022
-            */
-          const contactIdWrapper = call.request.getContactIdStringValueDeprecated()
+        return callback(null, response)
+      } catch (e) {
+        return grpcError('tagTagAdd', e, callback)
+      }
+    },
 
-          if (contactIdWrapper) {
-            const contactId = contactIdWrapper.getValue()
+    tagTagDelete: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagTagDelete()')
 
-            const tagIdList = await puppet.tagContactList(contactId)
+      try {
+        const tagGroupId = call.request.getTagGroupId()
+        const tagId = call.request.getTagId()
 
-            const response = new grpcPuppet.TagContactListResponse()
-            response.setIdsList(tagIdList)
+        await puppet.tagTagDelete(tagGroupId, tagId)
 
-            return callback(null, new grpcPuppet.TagContactListResponse())
-          }
-        }
+        return callback(null, new grpcPuppet.TagTagDeleteResponse())
+      } catch (e) {
+        return grpcError('tagTagDelete', e, callback)
+      }
+    },
 
-        /**
-         * get all tags for all contact
-         */
-        const tagIdList = await puppet.tagContactList()
+    tagGroupList: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagGroupList()')
+      void call
 
-        const response = new grpcPuppet.TagContactListResponse()
-        response.setIdsList(tagIdList)
+      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)
 
         return callback(null, response)
+      } catch (e) {
+        return grpcError('tagGroupList', e, callback)
+      }
+    },
+
+    tagGroupTagList: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagGroupTagList()')
+
+      try {
+        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.TagTagListResponse()
+        response.setPayloadsList(payloads)
+
+        return callback(null, response)
+      } catch (e) {
+        return grpcError('tagTagList', e, callback)
+      }
+    },
 
+    tagTagList: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagTagList()')
+      void call
+
+      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)
+
+        return callback(null, response)
       } catch (e) {
-        return grpcError('tagContactList', e, callback)
+        return grpcError('tagTagList', e, callback)
       }
     },
 
-    tagContactRemove: async (call, callback) => {
-      log.verbose('PuppetServiceImpl', 'tagContactRemove()')
+    tagContactTagList: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagContactTagList()')
 
       try {
-        const tagId = call.request.getId()
         const contactId = call.request.getContactId()
 
-        await puppet.tagContactRemove(tagId, contactId)
+        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)
 
-        return callback(null, new grpcPuppet.TagContactRemoveResponse())
+        return callback(null, response)
+      } catch (e) {
+        return grpcError('tagContactTagList', e, callback)
+      }
+    },
+
+    tagTagContactList: async (call, callback) => {
+      log.verbose('PuppetServiceImpl', 'tagTagContactList()')
+
+      try {
+        const tagGroupId = call.request.getTagGroupId()
+        const tagId = call.request.getTagId()
+
+        const result = await puppet.tagTagContactList(tagGroupId, tagId)
 
+        const response = new grpcPuppet.TagTagContactListResponse()
+        response.setPayloadsList(result)
+
+        return callback(null, response)
       } catch (e) {
-        return grpcError('tagContactRemove', e, callback)
+        return grpcError('tagTagContactList', e, callback)
       }
     },