room.ts 31.4 KB
Newer Older
1
/**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
2
 *   Wechaty - https://github.com/chatie/wechaty
3
 *
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
4
 *   @copyright 2016-2018 Huan LI <zixia@zixia.net>
5 6 7 8 9 10 11 12 13 14 15 16 17
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
L
lijiarui 已提交
18
 *   @ignore
19
 */
20 21 22
import {
  instanceToClass,
}                   from 'clone-class'
23 24 25
import {
  FileBox,
}                   from 'file-box'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
26

27 28 29
import {
  Accessory,
}               from '../accessory'
30
import {
31
  // config,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
32
  FOUR_PER_EM_SPACE,
33 34
  log,
  Raven,
35
}                       from '../config'
36 37 38
import {
  Sayable,
}             from '../types'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
39

40 41
import { Contact }        from './contact'
import { RoomInvitation } from './room-invitation'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
42

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
43
export const ROOM_EVENT_DICT = {
44
  invite: 'tbw',
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
45 46 47 48 49
  join: 'tbw',
  leave: 'tbw',
  topic: 'tbw',
}
export type RoomEventName = keyof typeof ROOM_EVENT_DICT
50

51 52 53 54
import {
  RoomMemberQueryFilter,
  RoomPayload,
  RoomQueryFilter,
55
}                         from 'wechaty-puppet'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
56

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
57
/**
L
lijiarui 已提交
58
 * All wechat rooms(groups) will be encapsulated as a Room.
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
59
 *
L
lijiarui 已提交
60
 * [Examples/Room-Bot]{@link https://github.com/Chatie/wechaty/blob/1523c5e02be46ebe2cc172a744b2fbe53351540e/examples/room-bot.ts}
61 62 63
 *
 * @property {string}  id               - Get Room id.
 * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
64
 */
65
export class Room extends Accessory implements Sayable {
66

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
67
  protected static pool: Map<string, Room>
68

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
  /**
   * Create a new room.
   *
   * @static
   * @param {Contact[]} contactList
   * @param {string} [topic]
   * @returns {Promise<Room>}
   * @example <caption>Creat a room with 'lijiarui' and 'juxiaomi', the room topic is 'ding - created'</caption>
   * const helperContactA = await Contact.find({ name: 'lijiarui' })  // change 'lijiarui' to any contact in your wechat
   * const helperContactB = await Contact.find({ name: 'juxiaomi' })  // change 'juxiaomi' to any contact in your wechat
   * const contactList = [helperContactA, helperContactB]
   * console.log('Bot', 'contactList: %s', contactList.join(','))
   * const room = await Room.create(contactList, 'ding')
   * console.log('Bot', 'createDingRoom() new ding room created: %s', room)
   * await room.topic('ding - created')
   * await room.say('ding - created')
   */
86
  public static async create (contactList: Contact[], topic?: string): Promise<Room> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
87 88
    log.verbose('Room', 'create(%s, %s)', contactList.join(','), topic)

89 90 91
    if (   !contactList
        || !Array.isArray(contactList)
    ) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
92 93 94
      throw new Error('contactList not found')
    }

95 96 97 98
    if (contactList.length < 2) {
      throw new Error('contactList need at least 2 contact to create a new room')
    }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
99
    try {
100 101 102
      const contactIdList = contactList.map(contact => contact.id)
      const roomId = await this.puppet.roomCreate(contactIdList, topic)
      const room = this.load(roomId)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
103 104 105 106 107 108 109 110 111
      return room
    } catch (e) {
      log.error('Room', 'create() exception: %s', e && e.stack || e.message || e)
      Raven.captureException(e)
      throw e
    }
  }

  /**
L
lijiarui 已提交
112
   * The filter to find the room:  {topic: string | RegExp}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
113
   *
L
lijiarui 已提交
114 115 116 117 118 119
   * @typedef    RoomQueryFilter
   * @property   {string} topic
   */

  /**
   * Find room by by filter: {topic: string | RegExp}, return all the matched room
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
120 121 122 123
   * @static
   * @param {RoomQueryFilter} [query]
   * @returns {Promise<Room[]>}
   * @example
L
lijiarui 已提交
124 125 126 127 128
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in
   * const roomList = await bot.Room.findAll()                    // get the room list of the bot
   * const roomList = await bot.Room.findAll({topic: 'wechaty'})  // find all of the rooms with name 'wechaty'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
129
   */
130
  public static async findAll<T extends typeof Room> (
131 132
    this  : T,
    query : RoomQueryFilter = { topic: /.*/ },
133
  ): Promise<Array<T['prototype']>> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
134
    log.verbose('Room', 'findAll()', JSON.stringify(query))
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
135 136 137 138 139 140

    if (!query.topic) {
      throw new Error('topicFilter not found')
    }

    try {
141
      const roomIdList = await this.puppet.roomSearch(query)
142
      const roomList = roomIdList.map(id => this.load(id))
143 144 145 146 147 148 149 150 151 152 153
      await Promise.all(
        roomList.map(
          room => {
            try {
              return room.ready()
            } catch (e) {
              return {} as any
            }
          },
        ),
      )
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
154 155 156 157 158

      return roomList

    } catch (e) {
      log.verbose('Room', 'findAll() rejected: %s', e.message)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
159
      console.error(e)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
160 161 162 163 164 165 166 167 168 169
      Raven.captureException(e)
      return [] as Room[] // fail safe
    }
  }

  /**
   * Try to find a room by filter: {topic: string | RegExp}. If get many, return the first one.
   *
   * @param {RoomQueryFilter} query
   * @returns {Promise<Room | null>} If can find the room, return Room, or return null
L
lijiarui 已提交
170 171 172 173 174 175
   * @example
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const roomList = await bot.Room.find()
   * const roomList = await bot.Room.find({topic: 'wechaty'})
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
176
   */
177

178
  public static async find<T extends typeof Room> (
179
    this  : T,
180
    query : string | RoomQueryFilter,
181
  ): Promise<T['prototype'] | null> {
182 183 184 185 186
    log.verbose('Room', 'find(%s)', JSON.stringify(query))

    if (typeof query === 'string') {
      query = { topic: query }
    }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
187 188

    const roomList = await this.findAll(query)
189
    if (!roomList) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
190 191
      return null
    }
192 193 194 195 196 197 198 199 200 201 202
    if (roomList.length < 1) {
      return null
    }

    if (roomList.length > 1) {
      log.warn('Room', 'find() got more than one(%d) result', roomList.length)
    }

    let n = 0
    for (n = 0; n < roomList.length; n++) {
      const room = roomList[n]
203
      // use puppet.roomValidate() to confirm double confirm that this roomId is valid.
204 205
      // https://github.com/lijiarui/wechaty-puppet-padchat/issues/64
      // https://github.com/Chatie/wechaty/issues/1345
206
      const valid = await this.puppet.roomValidate(room.id)
207 208 209 210 211 212 213 214 215 216 217 218 219 220 221
      if (valid) {
        log.verbose('Room', 'find() confirm room[#%d] with id=%d is vlaid result, return it.',
                            n,
                            room.id,
                  )
        return room
      } else {
        log.verbose('Room', 'find() confirm room[#%d] with id=%d is INVALID result, try next',
                            n,
                            room.id,
                    )
      }
    }
    log.warn('Room', 'find() got %d rooms but no one is valid.', roomList.length)
    return null
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
222 223 224 225
  }

  /**
   * @private
226
   * About the Generic: https://stackoverflow.com/q/43003970/1123955
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
227
   */
L
lijiarui 已提交
228 229 230 231 232 233
  /**
   * Load room by topic. <br>
   * > Tips: For Web solution, it cannot get the unique topic id,
   * but for other solutions besides web,
   * we can get unique and permanent topic id.
   *
234
   * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
L
lijiarui 已提交
235 236 237 238 239 240 241 242 243
   * @static
   * @param {string} id
   * @returns {Room}
   * @example
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = bot.Room.load('roomId')
   */
244
  public static load<T extends typeof Room> (
245 246
    this : T,
    id   : string,
247
  ): T['prototype'] {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
248 249 250 251
    if (!this.pool) {
      this.pool = new Map<string, Room>()
    }

252 253 254
    const existingRoom = this.pool.get(id)
    if (existingRoom) {
      return existingRoom
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
255
    }
256

257
    const newRoom = new (this as any)(id) as Room
258

259 260
    this.pool.set(id, newRoom)
    return newRoom
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
261 262
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
263
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
264
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
265 266 267 268 269
   *
   * Instance Properties
   *
   *
   */
270
  protected payload?: RoomPayload
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
271

272 273
  public readonly id: string  // Room Id

H
hcz 已提交
274 275 276
  /**
   * @private
   */
277
  constructor (
278
    id: string,
279
  ) {
280
    super()
281
    log.silly('Room', `constructor(${id})`)
282

283 284
    this.id = id

285 286 287 288 289 290 291 292 293 294 295
    // tslint:disable-next-line:variable-name
    const MyClass = instanceToClass(this, Room)

    if (MyClass === Room) {
      throw new Error('Room class can not be instanciated directly! See: https://github.com/Chatie/wechaty/issues/1217')
    }

    if (!this.puppet) {
      throw new Error('Room class can not be instanciated without a puppet!')
    }

296
  }
297

H
hcz 已提交
298 299 300
  /**
   * @private
   */
301
  public toString () {
302 303
    if (!this.payload) {
      return this.constructor.name
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
304
    }
305 306

    return `Room<${this.payload.topic || 'loadind...'}>`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
307
  }
L
lijiarui 已提交
308

309
  public async *[Symbol.asyncIterator] (): AsyncIterableIterator<Contact> {
310
    const memberList = await this.memberList()
311 312 313 314 315
    for (const contact of memberList) {
      yield contact
    }
  }

L
lijiarui 已提交
316
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
317
   * @hidden
L
lijiarui 已提交
318
   */
319
  public async ready (
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
320
    dirty = false,
321
  ): Promise<void> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
322
    log.verbose('Room', 'ready()')
323

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
324
    if (!dirty && this.isReady()) {
325 326 327
      return
    }

328 329 330
    if (dirty) {
      await this.puppet.roomPayloadDirty(this.id)
    }
331 332 333 334 335
    this.payload = await this.puppet.roomPayload(this.id)

    if (!this.payload) {
      throw new Error('ready() no payload')
    }
336 337 338

    const memberIdList = await this.puppet.roomMemberList(this.id)

339
    await Promise.all(
340 341
      memberIdList
        .map(id => this.wechaty.Contact.load(id))
342 343 344 345 346 347
        .map(contact => {
          contact.ready()
            .catch(() => {
              //
            })
        }),
348 349 350 351
    )
  }

  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
352
   * @hidden
353
   */
354
  public isReady (): boolean {
355
    return !!(this.payload)
356 357
  }

358 359 360 361 362
  public say (text: string)                     : Promise<void>
  public say (text: string, mention: Contact)   : Promise<void>
  public say (text: string, mention: Contact[]) : Promise<void>
  public say (file: FileBox)                    : Promise<void>
  public say (text: never, ...args: never[])    : never
L
lijiarui 已提交
363 364 365

  /**
   * Send message inside Room, if set [replyTo], wechaty will mention the contact as well.
366 367
   * > Tips:
   * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
L
lijiarui 已提交
368
   *
L
lijiarui 已提交
369 370 371 372 373 374 375 376 377 378 379
   * @param {(string | Contact | FileBox)} textOrContactOrFile - Send `text` or `media file` inside Room. <br>
   * You can use {@link https://www.npmjs.com/package/file-box|FileBox} to send file
   * @param {(Contact | Contact[])} [mention] - Optional parameter, send content inside Room, and mention @replyTo contact or contactList.
   * @returns {Promise<void>}
   *
   * @example
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'wechaty'})
   *
L
lijiarui 已提交
380
   * // 1. Send text inside Room
L
lijiarui 已提交
381 382 383
   *
   * await room.say('Hello world!')
   *
L
lijiarui 已提交
384
   * // 2. Send media file inside Room
L
lijiarui 已提交
385 386 387 388 389
   * import { FileBox }  from 'file-box'
   * const fileBox1 = FileBox.fromUrl('https://chatie.io/wechaty/images/bot-qr-code.png')
   * const fileBox2 = FileBox.fromLocal('/tmp/text.txt')
   * await room.say(fileBox1)
   * await room.say(fileBox2)
L
lijiarui 已提交
390
   *
L
lijiarui 已提交
391
   * // 3. Send Contact Card in a room
L
lijiarui 已提交
392 393 394
   * const contactCard = await bot.Contact.find({name: 'lijiarui'}) // change 'lijiarui' to any of the room member
   * await room.say(contactCard)
   *
L
lijiarui 已提交
395
   * // 4. Send text inside room and mention @mention contact
L
lijiarui 已提交
396
   * const contact = await bot.Contact.find({name: 'lijiarui'}) // change 'lijiarui' to any of the room member
L
lijiarui 已提交
397 398
   * await room.say('Hello world!', contact)
   */
399
  public async say (
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
400 401
    textOrContactOrFile : string | Contact | FileBox,
    mention?            : Contact | Contact[],
402 403
  ): Promise<void> {
    log.verbose('Room', 'say(%s, %s)',
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
404
                                  textOrContactOrFile,
405 406 407 408
                                  Array.isArray(mention)
                                  ? mention.map(c => c.name()).join(', ')
                                  : mention ? mention.name() : '',
                )
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
409
    let text: string
410

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
411 412
    const replyToList: Contact[] = [].concat(mention as any || [])

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
413
    if (typeof textOrContactOrFile === 'string') {
414 415

      if (replyToList.length > 0) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
416 417 418
        // const AT_SEPRATOR = String.fromCharCode(8197)
        const AT_SEPRATOR = FOUR_PER_EM_SPACE

419
        const mentionList = replyToList.map(c => '@' + c.name()).join(AT_SEPRATOR)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
420
        text = mentionList + ' ' + textOrContactOrFile
421
      } else {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
422
        text = textOrContactOrFile
423
      }
424
      await this.puppet.messageSendText({
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
425
        contactId : replyToList.length && replyToList[0].id || undefined,
426
        roomId    : this.id,
427
      }, text)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
428
    } else if (textOrContactOrFile instanceof FileBox) {
429 430
      await this.puppet.messageSendFile({
        roomId: this.id,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
431 432 433 434 435
      }, textOrContactOrFile)
    } else if (textOrContactOrFile instanceof Contact) {
      await this.puppet.messageSendContact({
        roomId: this.id,
      }, textOrContactOrFile.id)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
436 437
    } else {
      throw new Error('arg unsupported')
438 439
    }
  }
L
lijiarui 已提交
440

441
  public emit (event: 'invite',inviter: Contact, invitation: RoomInvitation)                 : boolean
442 443 444 445
  public emit (event: 'leave', leaverList:   Contact[],  remover?: Contact)                    : boolean
  public emit (event: 'join' , inviteeList:  Contact[] , inviter:  Contact)                    : boolean
  public emit (event: 'topic', topic:        string,     oldTopic: string,   changer: Contact) : boolean
  public emit (event: never, ...args: never[]): never
446

447
  public emit (
448
    event:   RoomEventName,
449
    ...args: any[]
450 451 452 453
  ): boolean {
    return super.emit(event, ...args)
  }

454
  public on (event: 'invite', listener: (this: Room, inviter: Contact, invitation: RoomInvitation) => void)              : this
455 456 457 458
  public on (event: 'leave', listener: (this: Room, leaverList:  Contact[], remover?: Contact) => void)                  : this
  public on (event: 'join' , listener: (this: Room, inviteeList: Contact[], inviter:  Contact) => void)                  : this
  public on (event: 'topic', listener: (this: Room, topic:       string,    oldTopic: string, changer: Contact) => void) : this
  public on (event: never,   ...args: never[])                                                                           : never
L
lijiarui 已提交
459 460 461 462 463 464 465 466 467 468

   /**
    * @desc       Room Class Event Type
    * @typedef    RoomEventName
    * @property   {string}  join  - Emit when anyone join any room.
    * @property   {string}  topic - Get topic event, emitted when someone change room topic.
    * @property   {string}  leave - Emit when anyone leave the room.<br>
    *                               If someone leaves the room by themselves, wechat will not notice other people in the room, so the bot will never get the "leave" event.
    */

H
hcz 已提交
469
  /**
L
lijiarui 已提交
470 471 472 473 474
   * @desc       Room Class Event Function
   * @typedef    RoomEventFunction
   * @property   {Function} room-join       - (this: Room, inviteeList: Contact[] , inviter: Contact)  => void
   * @property   {Function} room-topic      - (this: Room, topic: string, oldTopic: string, changer: Contact) => void
   * @property   {Function} room-leave      - (this: Room, leaver: Contact) => void
H
hcz 已提交
475
   */
L
lijiarui 已提交
476

H
hcz 已提交
477
  /**
L
lijiarui 已提交
478 479 480 481 482 483
   * @listens Room
   * @param   {RoomEventName}      event      - Emit WechatyEvent
   * @param   {RoomEventFunction}  listener   - Depends on the WechatyEvent
   * @return  {this}                          - this for chain
   *
   * @example <caption>Event:join </caption>
L
lijiarui 已提交
484 485 486 487
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'topic of your room'}) // change `event-room` to any room topic in your wechat
L
lijiarui 已提交
488 489 490
   * if (room) {
   *   room.on('join', (room: Room, inviteeList: Contact[], inviter: Contact) => {
   *     const nameList = inviteeList.map(c => c.name()).join(',')
L
lijiarui 已提交
491
   *     console.log(`Room got new member ${nameList}, invited by ${inviter}`)
L
lijiarui 已提交
492 493 494 495
   *   })
   * }
   *
   * @example <caption>Event:leave </caption>
L
lijiarui 已提交
496 497 498 499
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'topic of your room'}) // change `event-room` to any room topic in your wechat
L
lijiarui 已提交
500 501 502
   * if (room) {
   *   room.on('leave', (room: Room, leaverList: Contact[]) => {
   *     const nameList = leaverList.map(c => c.name()).join(',')
L
lijiarui 已提交
503
   *     console.log(`Room lost member ${nameList}`)
L
lijiarui 已提交
504 505 506 507
   *   })
   * }
   *
   * @example <caption>Event:topic </caption>
L
lijiarui 已提交
508 509 510 511
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'topic of your room'}) // change `event-room` to any room topic in your wechat
L
lijiarui 已提交
512 513
   * if (room) {
   *   room.on('topic', (room: Room, topic: string, oldTopic: string, changer: Contact) => {
L
lijiarui 已提交
514
   *     console.log(`Room topic changed from ${oldTopic} to ${topic} by ${changer.name()}`)
L
lijiarui 已提交
515 516 517
   *   })
   * }
   *
H
hcz 已提交
518
   */
519
  public on (event: RoomEventName, listener: (...args: any[]) => any): this {
520
    log.verbose('Room', 'on(%s, %s)', event, typeof listener)
521

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
522
    super.on(event, listener) // Room is `Sayable`
523
    return this
524 525
  }

H
hcz 已提交
526
  /**
L
lijiarui 已提交
527 528
   * Add contact in a room
   *
529 530 531 532 533
   * > Tips:
   * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
   * >
   * > see {@link https://github.com/Chatie/wechaty/issues/1441|Web version of WeChat closed group interface}
   *
L
lijiarui 已提交
534
   * @param {Contact} contact
L
lijiarui 已提交
535
   * @returns {Promise<void>}
L
lijiarui 已提交
536
   * @example
L
lijiarui 已提交
537 538 539 540 541
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const contact = await bot.Contact.find({name: 'lijiarui'}) // change 'lijiarui' to any contact in your wechat
   * const room = await bot.Room.find({topic: 'wechat'})        // change 'wechat' to any room topic in your wechat
L
lijiarui 已提交
542
   * if (room) {
L
lijiarui 已提交
543 544 545 546
   *   try {
   *      await room.add(contact)
   *   } catch(e) {
   *      console.error(e)
L
lijiarui 已提交
547 548
   *   }
   * }
H
hcz 已提交
549
   */
550
  public async add (contact: Contact): Promise<void> {
551
    log.verbose('Room', 'add(%s)', contact)
552
    await this.puppet.roomAdd(this.id, contact.id)
553
  }
554

H
hcz 已提交
555
  /**
L
lijiarui 已提交
556 557
   * Delete a contact from the room
   * It works only when the bot is the owner of the room
558 559 560 561 562 563
   *
   * > Tips:
   * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
   * >
   * > see {@link https://github.com/Chatie/wechaty/issues/1441|Web version of WeChat closed group interface}
   *
L
lijiarui 已提交
564
   * @param {Contact} contact
L
lijiarui 已提交
565
   * @returns {Promise<void>}
L
lijiarui 已提交
566
   * @example
L
lijiarui 已提交
567 568 569 570 571
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'wechat'})          // change 'wechat' to any room topic in your wechat
   * const contact = await bot.Contact.find({name: 'lijiarui'})   // change 'lijiarui' to any room member in the room you just set
L
lijiarui 已提交
572
   * if (room) {
L
lijiarui 已提交
573 574 575 576
   *   try {
   *      await room.del(contact)
   *   } catch(e) {
   *      console.error(e)
L
lijiarui 已提交
577 578
   *   }
   * }
H
hcz 已提交
579
   */
580
  public async del (contact: Contact): Promise<void> {
581
    log.verbose('Room', 'del(%s)', contact)
582
    await this.puppet.roomDel(this.id, contact.id)
583
    // this.delLocal(contact)
584 585
  }

586 587
  // private delLocal(contact: Contact): void {
  //   log.verbose('Room', 'delLocal(%s)', contact)
588

589 590 591 592 593 594 595 596 597 598
  //   const memberIdList = this.payload && this.payload.memberIdList
  //   if (memberIdList && memberIdList.length > 0) {
  //     for (let i = 0; i < memberIdList.length; i++) {
  //       if (memberIdList[i] === contact.id) {
  //         memberIdList.splice(i, 1)
  //         break
  //       }
  //     }
  //   }
  // }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
599

H
hcz 已提交
600
  /**
L
lijiarui 已提交
601 602
   * Bot quit the room itself
   *
603 604 605
   * > Tips:
   * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
   *
L
lijiarui 已提交
606 607 608
   * @returns {Promise<void>}
   * @example
   * await room.quit()
H
hcz 已提交
609
   */
610
  public async quit (): Promise<void> {
611
    log.verbose('Room', 'quit() %s', this)
612
    await this.puppet.roomQuit(this.id)
613
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
614

615 616
  public async topic ()                : Promise<string>
  public async topic (newTopic: string): Promise<void>
617

L
lijiarui 已提交
618 619 620 621
  /**
   * SET/GET topic from the room
   *
   * @param {string} [newTopic] If set this para, it will change room topic.
622
   * @returns {Promise<string | void>}
L
lijiarui 已提交
623 624
   *
   * @example <caption>When you say anything in a room, it will get room topic. </caption>
L
lijiarui 已提交
625
   * const bot = new Wechaty()
L
lijiarui 已提交
626 627 628 629
   * bot
   * .on('message', async m => {
   *   const room = m.room()
   *   if (room) {
630
   *     const topic = await room.topic()
L
lijiarui 已提交
631 632 633
   *     console.log(`room topic is : ${topic}`)
   *   }
   * })
L
lijiarui 已提交
634
   * .start()
L
lijiarui 已提交
635 636
   *
   * @example <caption>When you say anything in a room, it will change room topic. </caption>
L
lijiarui 已提交
637
   * const bot = new Wechaty()
L
lijiarui 已提交
638 639 640 641
   * bot
   * .on('message', async m => {
   *   const room = m.room()
   *   if (room) {
L
lijiarui 已提交
642 643
   *     const oldTopic = await room.topic()
   *     await room.topic('change topic to wechaty!')
L
lijiarui 已提交
644 645 646
   *     console.log(`room topic change from ${oldTopic} to ${room.topic()}`)
   *   }
   * })
L
lijiarui 已提交
647
   * .start()
L
lijiarui 已提交
648
   */
649
  public async topic (newTopic?: string): Promise<void | string> {
650 651 652
    log.verbose('Room', 'topic(%s)', newTopic ? newTopic : '')
    if (!this.isReady()) {
      log.warn('Room', 'topic() room not ready')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
653
      throw new Error('not ready')
654 655 656
    }

    if (typeof newTopic === 'undefined') {
657 658 659 660 661 662 663 664
      if (this.payload && this.payload.topic) {
        return this.payload.topic
      } else {
        const memberIdList = await this.puppet.roomMemberList(this.id)
        const memberList = memberIdList
                            .filter(id => id !== this.puppet.selfId())
                            .map(id => this.wechaty.Contact.load(id))

665
        let defaultTopic = memberList[0] && memberList[0].name() || ''
666 667 668 669 670
        for (let i = 1; i < 3 && memberList[i]; i++) {
          defaultTopic += ',' + memberList[i].name()
        }
        return defaultTopic
      }
671 672 673
    }

    const future = this.puppet
674
        .roomTopic(this.id, newTopic)
675 676 677 678 679 680 681 682 683
        .catch(e => {
          log.warn('Room', 'topic(newTopic=%s) exception: %s',
                            newTopic, e && e.message || e,
                  )
          Raven.captureException(e)
        })

    return future
  }
684

685 686
  public async announce ()             : Promise<string>
  public async announce (text: string) : Promise<void>
687

L
lijiarui 已提交
688 689 690
  /**
   * SET/GET announce from the room
   * > Tips: It only works when bot is the owner of the room.
691 692
   * >
   * > This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
L
lijiarui 已提交
693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713
   *
   * @param {string} [text] If set this para, it will change room announce.
   * @returns {(Promise<void | string>)}
   *
   * @example <caption>When you say anything in a room, it will get room announce. </caption>
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'your room'})
   * const announce = await room.announce()
   * console.log(`room announce is : ${announce}`)
   *
   * @example <caption>When you say anything in a room, it will change room announce. </caption>
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'your room'})
   * const oldAnnounce = await room.announce()
   * await room.announce('change announce to wechaty!')
   * console.log(`room announce change from ${oldAnnounce} to ${room.announce()}`)
   */
714
  public async announce (text?: string): Promise<void | string> {
715 716 717 718 719
    log.verbose('Room', 'announce(%s)', text ? text : '')

    if (text) {
      await this.puppet.roomAnnounce(this.id, text)
    } else {
720 721
      const announcement = await this.puppet.roomAnnounce(this.id)
      return announcement
722 723 724
    }
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
725
  /**
L
lijiarui 已提交
726
   * Get QR Code of the Room from the room, which can be used as scan and join the room.
727 728
   * > Tips:
   * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
L
lijiarui 已提交
729
   * @returns {Promise<string>}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
730
   */
731
  public async qrcode (): Promise<string> {
732
    log.verbose('Room', 'qrcode()')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
733 734
    const qrcode = await this.puppet.roomQrcode(this.id)
    return qrcode
735 736
  }

L
lijiarui 已提交
737
  /**
L
lijiarui 已提交
738
   * Return contact's roomAlias in the room, the same as roomAlias
L
lijiarui 已提交
739
   * @param {Contact} contact
L
lijiarui 已提交
740
   * @returns {Promise<string | null>} - If a contact has an alias in room, return string, otherwise return null
L
lijiarui 已提交
741
   * @example
L
lijiarui 已提交
742
   * const bot = new Wechaty()
L
lijiarui 已提交
743 744 745 746 747
   * bot
   * .on('message', async m => {
   *   const room = m.room()
   *   const contact = m.from()
   *   if (room) {
L
lijiarui 已提交
748
   *     const alias = await room.alias(contact)
L
lijiarui 已提交
749 750 751
   *     console.log(`${contact.name()} alias is ${alias}`)
   *   }
   * })
L
lijiarui 已提交
752
   * .start()
L
lijiarui 已提交
753
   */
754
  public async alias (contact: Contact): Promise<null | string> {
755 756
    return this.roomAlias(contact)
  }
757

H
hcz 已提交
758
  /**
L
lijiarui 已提交
759 760
   * Same as function alias
   * @param {Contact} contact
L
lijiarui 已提交
761
   * @returns {Promise<string | null>}
H
hcz 已提交
762
   */
763
  public async roomAlias (contact: Contact): Promise<null | string> {
764

765
    const memberPayload = await this.puppet.roomMemberPayload(this.id, contact.id)
766 767 768

    if (memberPayload && memberPayload.roomAlias) {
      return memberPayload.roomAlias
769
    }
770 771

    return null
772
  }
773

H
hcz 已提交
774
  /**
775
   * Check if the room has member `contact`, the return is a Promise and must be `await`-ed
L
lijiarui 已提交
776 777
   *
   * @param {Contact} contact
L
lijiarui 已提交
778
   * @returns {Promise<boolean>} Return `true` if has contact, else return `false`.
L
lijiarui 已提交
779
   * @example <caption>Check whether 'lijiarui' is in the room 'wechaty'</caption>
L
lijiarui 已提交
780 781 782 783 784
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const contact = await bot.Contact.find({name: 'lijiarui'})   // change 'lijiarui' to any of contact in your wechat
   * const room = await bot.Room.find({topic: 'wechaty'})         // change 'wechaty' to any of the room in your wechat
L
lijiarui 已提交
785
   * if (contact && room) {
786
   *   if (await room.has(contact)) {
L
lijiarui 已提交
787
   *     console.log(`${contact.name()} is in the room wechaty!`)
L
lijiarui 已提交
788
   *   } else {
L
lijiarui 已提交
789
   *     console.log(`${contact.name()} is not in the room wechaty!`)
L
lijiarui 已提交
790 791
   *   }
   * }
H
hcz 已提交
792
   */
793
  public async has (contact: Contact): Promise<boolean> {
794 795 796
    const memberIdList = await this.puppet.roomMemberList(this.id)

    if (!memberIdList) {
797 798
      return false
    }
799 800 801 802

    return memberIdList
            .filter(id => id === contact.id)
            .length > 0
803
  }
L
lijiarui 已提交
804

805 806
  public async memberAll (name: string)                  : Promise<Contact[]>
  public async memberAll (filter: RoomMemberQueryFilter) : Promise<Contact[]>
807

L
lijiarui 已提交
808 809 810
  /**
   * The way to search member by Room.member()
   *
L
lijiarui 已提交
811
   * @typedef    RoomMemberQueryFilter
L
lijiarui 已提交
812 813 814 815 816 817 818 819 820 821 822
   * @property   {string} name            -Find the contact by wechat name in a room, equal to `Contact.name()`.
   * @property   {string} roomAlias       -Find the contact by alias set by the bot for others in a room.
   * @property   {string} contactAlias    -Find the contact by alias set by the contact out of a room, equal to `Contact.alias()`.
   * [More Detail]{@link https://github.com/Chatie/wechaty/issues/365}
   */

  /**
   * Find all contacts in a room
   *
   * #### definition
   * - `name`                 the name-string set by user-self, should be called name, equal to `Contact.name()`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
823
   * - `roomAlias`            the name-string set by user-self in the room, should be called roomAlias
L
lijiarui 已提交
824
   * - `contactAlias`         the name-string set by bot for others, should be called alias, equal to `Contact.alias()`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
825
   * @param {(RoomMemberQueryFilter | string)} query -When use memberAll(name:string), return all matched members, including name, roomAlias, contactAlias
L
lijiarui 已提交
826
   * @returns {Promise<Contact[]>}
L
lijiarui 已提交
827
   */
828
  public async memberAll (
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
829
    query: string | RoomMemberQueryFilter,
830 831
  ): Promise<Contact[]> {
    log.silly('Room', 'memberAll(%s)',
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
832
                      JSON.stringify(query),
833 834
              )

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
835 836 837 838
    const contactIdList = await this.puppet.roomMemberSearch(this.id, query)
    const contactList   = contactIdList.map(id => this.wechaty.Contact.load(id))

    return contactList
839
  }
840

841 842
  public async member (name  : string)               : Promise<null | Contact>
  public async member (filter: RoomMemberQueryFilter): Promise<null | Contact>
843

L
lijiarui 已提交
844 845 846
  /**
   * Find all contacts in a room, if get many, return the first one.
   *
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
847
   * @param {(RoomMemberQueryFilter | string)} queryArg -When use member(name:string), return all matched members, including name, roomAlias, contactAlias
L
lijiarui 已提交
848
   * @returns {Promise<null | Contact>}
L
lijiarui 已提交
849 850
   *
   * @example <caption>Find member by name</caption>
L
lijiarui 已提交
851 852 853 854
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'wechaty'})           // change 'wechaty' to any room name in your wechat
L
lijiarui 已提交
855
   * if (room) {
L
lijiarui 已提交
856
   *   const member = await room.member('lijiarui')             // change 'lijiarui' to any room member in your wechat
L
lijiarui 已提交
857
   *   if (member) {
L
lijiarui 已提交
858
   *     console.log(`wechaty room got the member: ${member.name()}`)
L
lijiarui 已提交
859
   *   } else {
L
lijiarui 已提交
860
   *     console.log(`cannot get member in wechaty room!`)
L
lijiarui 已提交
861 862 863 864
   *   }
   * }
   *
   * @example <caption>Find member by MemberQueryFilter</caption>
L
lijiarui 已提交
865 866 867 868
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = await bot.Room.find({topic: 'wechaty'})          // change 'wechaty' to any room name in your wechat
L
lijiarui 已提交
869
   * if (room) {
L
lijiarui 已提交
870
   *   const member = await room.member({name: 'lijiarui'})        // change 'lijiarui' to any room member in your wechat
L
lijiarui 已提交
871
   *   if (member) {
L
lijiarui 已提交
872
   *     console.log(`wechaty room got the member: ${member.name()}`)
L
lijiarui 已提交
873
   *   } else {
L
lijiarui 已提交
874
   *     console.log(`cannot get member in wechaty room!`)
L
lijiarui 已提交
875 876 877
   *   }
   * }
   */
878
  public async member (
879
    queryArg: string | RoomMemberQueryFilter,
880
  ): Promise<null | Contact> {
881 882 883 884 885 886
    log.verbose('Room', 'member(%s)', JSON.stringify(queryArg))

    let memberList: Contact[]
    // ISSUE #622
    // error TS2345: Argument of type 'string | MemberQueryFilter' is not assignable to parameter of type 'MemberQueryFilter' #622
    if (typeof queryArg === 'string') {
887
      memberList =  await this.memberAll(queryArg)
888
    } else {
889
      memberList =  await this.memberAll(queryArg)
890 891 892 893 894 895 896 897 898 899 900
    }

    if (!memberList || !memberList.length) {
      return null
    }

    if (memberList.length > 1) {
      log.warn('Room', 'member(%s) get %d contacts, use the first one by default', JSON.stringify(queryArg), memberList.length)
    }
    return memberList[0]
  }
901

H
hcz 已提交
902
  /**
L
lijiarui 已提交
903 904
   * Get all room member from the room
   *
L
lijiarui 已提交
905 906 907
   * @returns {Promise<Contact[]>}
   * @example
   * await room.memberList()
H
hcz 已提交
908
   */
909
  public async memberList (): Promise<Contact[]> {
910 911 912
    log.verbose('Room', 'memberList()')

    const memberIdList = await this.puppet.roomMemberList(this.id)
913

914
    if (!memberIdList) {
915 916 917
      log.warn('Room', 'memberList() not ready')
      return []
    }
918 919 920 921

    const contactList = memberIdList.map(
      id => this.wechaty.Contact.load(id),
    )
922
    return contactList
923
  }
924

L
lijiarui 已提交
925
  /**
L
lijiarui 已提交
926
   * @ignore
L
lijiarui 已提交
927
   */
928
  public async refresh (): Promise<void> {
929 930
    return this.sync()
  }
L
lijiarui 已提交
931

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
932
  /**
L
lijiarui 已提交
933
   * Force reload data for Room, Sync data for Room
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
934 935
   *
   * @returns {Promise<void>}
L
lijiarui 已提交
936 937
   * @example
   * await room.sync()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
938
   */
939
  public async sync (): Promise<void> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
940
    await this.ready(true)
941
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
942

L
lijiarui 已提交
943 944
  /**
   * Get room's owner from the room.
945 946
   * > Tips:
   * This function is depending on the Puppet Implementation, see [puppet-compatible-table](https://github.com/Chatie/wechaty/wiki/Puppet#3-puppet-compatible-table)
L
lijiarui 已提交
947
   * @returns {(Contact | null)}
L
lijiarui 已提交
948 949
   * @example
   * const owner = room.owner()
L
lijiarui 已提交
950
   */
951
  public owner (): Contact | null {
952 953
    log.info('Room', 'owner()')

954 955 956 957 958
    const ownerId = this.payload && this.payload.ownerId
    if (!ownerId) {
      return null
    }

959
    const owner = this.wechaty.Contact.load(ownerId)
960
    return owner
961
  }
L
lijiarui 已提交
962

963
  public async avatar (): Promise<FileBox> {
964 965 966 967 968
    log.verbose('Room', 'avatar()')

    return this.puppet.roomAvatar(this.id)
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
969
}