room.ts 31.3 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(%s)', 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
   * @private
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
   * @private
353
   */
354
  public isReady (): boolean {
355
    return !!(this.payload)
356 357
  }

358 359 360 361
  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>
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
362 363

  public say (...args: never[]): never
L
lijiarui 已提交
364 365 366

  /**
   * Send message inside Room, if set [replyTo], wechaty will mention the contact as well.
367 368
   * > 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 已提交
369
   *
L
lijiarui 已提交
370 371 372 373 374 375 376 377 378 379 380
   * @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 已提交
381
   * // 1. Send text inside Room
L
lijiarui 已提交
382 383 384
   *
   * await room.say('Hello world!')
   *
L
lijiarui 已提交
385
   * // 2. Send media file inside Room
L
lijiarui 已提交
386 387 388 389 390
   * 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 已提交
391
   *
L
lijiarui 已提交
392
   * // 3. Send Contact Card in a room
L
lijiarui 已提交
393 394 395
   * const contactCard = await bot.Contact.find({name: 'lijiarui'}) // change 'lijiarui' to any of the room member
   * await room.say(contactCard)
   *
L
lijiarui 已提交
396
   * // 4. Send text inside room and mention @mention contact
L
lijiarui 已提交
397
   * const contact = await bot.Contact.find({name: 'lijiarui'}) // change 'lijiarui' to any of the room member
L
lijiarui 已提交
398 399
   * await room.say('Hello world!', contact)
   */
400
  public async say (
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
401 402
    textOrContactOrFile : string | Contact | FileBox,
    mention?            : Contact | Contact[],
403 404
  ): Promise<void> {
    log.verbose('Room', 'say(%s, %s)',
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
405
                                  textOrContactOrFile,
406 407 408 409
                                  Array.isArray(mention)
                                  ? mention.map(c => c.name()).join(', ')
                                  : mention ? mention.name() : '',
                )
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
410
    let text: string
411

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

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

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

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

442
  public emit (event: 'invite',inviter: Contact, invitation: RoomInvitation)                 : boolean
443 444 445 446
  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
447

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

455
  public on (event: 'invite', listener: (this: Room, inviter: Contact, invitation: RoomInvitation) => void)              : this
456 457 458 459
  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 已提交
460 461 462 463 464 465 466 467 468 469

   /**
    * @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 已提交
470
  /**
L
lijiarui 已提交
471 472 473 474 475
   * @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 已提交
476
   */
L
lijiarui 已提交
477

H
hcz 已提交
478
  /**
L
lijiarui 已提交
479 480 481 482 483 484
   * @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 已提交
485 486 487 488
   * 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 已提交
489
   * if (room) {
L
lijiarui 已提交
490
   *   room.on('join', (room, inviteeList, inviter) => {
L
lijiarui 已提交
491
   *     const nameList = inviteeList.map(c => c.name()).join(',')
L
lijiarui 已提交
492
   *     console.log(`Room got new member ${nameList}, invited by ${inviter}`)
L
lijiarui 已提交
493 494 495 496
   *   })
   * }
   *
   * @example <caption>Event:leave </caption>
L
lijiarui 已提交
497 498 499 500
   * 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 已提交
501
   * if (room) {
L
lijiarui 已提交
502
   *   room.on('leave', (room, leaverList) => {
L
lijiarui 已提交
503
   *     const nameList = leaverList.map(c => c.name()).join(',')
L
lijiarui 已提交
504
   *     console.log(`Room lost member ${nameList}`)
L
lijiarui 已提交
505 506 507 508
   *   })
   * }
   *
   * @example <caption>Event:topic </caption>
L
lijiarui 已提交
509 510 511 512
   * 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 已提交
513
   * if (room) {
L
lijiarui 已提交
514
   *   room.on('topic', (room, topic, oldTopic, changer) => {
L
lijiarui 已提交
515
   *     console.log(`Room topic changed from ${oldTopic} to ${topic} by ${changer.name()}`)
L
lijiarui 已提交
516 517 518
   *   })
   * }
   *
H
hcz 已提交
519
   */
520
  public on (event: RoomEventName, listener: (...args: any[]) => any): this {
521
    log.verbose('Room', 'on(%s, %s)', event, typeof listener)
522

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

H
hcz 已提交
527
  /**
L
lijiarui 已提交
528 529
   * Add contact in a room
   *
530 531 532 533 534
   * > 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 已提交
535
   * @param {Contact} contact
L
lijiarui 已提交
536
   * @returns {Promise<void>}
L
lijiarui 已提交
537
   * @example
L
lijiarui 已提交
538 539 540 541 542
   * 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 已提交
543
   * if (room) {
L
lijiarui 已提交
544 545 546 547
   *   try {
   *      await room.add(contact)
   *   } catch(e) {
   *      console.error(e)
L
lijiarui 已提交
548 549
   *   }
   * }
H
hcz 已提交
550
   */
551
  public async add (contact: Contact): Promise<void> {
552
    log.verbose('Room', 'add(%s)', contact)
553
    await this.puppet.roomAdd(this.id, contact.id)
554
  }
555

H
hcz 已提交
556
  /**
L
lijiarui 已提交
557 558
   * Delete a contact from the room
   * It works only when the bot is the owner of the room
559 560 561 562 563 564
   *
   * > 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 已提交
565
   * @param {Contact} contact
L
lijiarui 已提交
566
   * @returns {Promise<void>}
L
lijiarui 已提交
567
   * @example
L
lijiarui 已提交
568 569 570 571 572
   * 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 已提交
573
   * if (room) {
L
lijiarui 已提交
574 575 576 577
   *   try {
   *      await room.del(contact)
   *   } catch(e) {
   *      console.error(e)
L
lijiarui 已提交
578 579
   *   }
   * }
H
hcz 已提交
580
   */
581
  public async del (contact: Contact): Promise<void> {
582
    log.verbose('Room', 'del(%s)', contact)
583
    await this.puppet.roomDel(this.id, contact.id)
584
    // this.delLocal(contact)
585 586
  }

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

590 591 592 593 594 595 596 597 598 599
  //   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 (李卓桓) 已提交
600

H
hcz 已提交
601
  /**
L
lijiarui 已提交
602 603
   * Bot quit the room itself
   *
604 605 606
   * > 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 已提交
607 608 609
   * @returns {Promise<void>}
   * @example
   * await room.quit()
H
hcz 已提交
610
   */
611
  public async quit (): Promise<void> {
612
    log.verbose('Room', 'quit() %s', this)
613
    await this.puppet.roomQuit(this.id)
614
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
615

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

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

    if (typeof newTopic === 'undefined') {
658 659 660 661 662 663 664 665
      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))

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

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

    return future
  }
685

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

L
lijiarui 已提交
689 690 691
  /**
   * SET/GET announce from the room
   * > Tips: It only works when bot is the owner of the room.
692 693
   * >
   * > 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 已提交
694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714
   *
   * @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()}`)
   */
715
  public async announce (text?: string): Promise<void | string> {
716 717 718 719 720
    log.verbose('Room', 'announce(%s)', text ? text : '')

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

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
726
  /**
L
lijiarui 已提交
727
   * Get QR Code of the Room from the room, which can be used as scan and join the room.
728 729
   * > 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 已提交
730
   * @returns {Promise<string>}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
731
   */
732
  public async qrcode (): Promise<string> {
733
    log.verbose('Room', 'qrcode()')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
734 735
    const qrcode = await this.puppet.roomQrcode(this.id)
    return qrcode
736 737
  }

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

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

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

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

    return null
773
  }
774

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

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

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

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

L
lijiarui 已提交
809 810 811
  /**
   * The way to search member by Room.member()
   *
L
lijiarui 已提交
812
   * @typedef    RoomMemberQueryFilter
L
lijiarui 已提交
813 814 815 816 817 818 819 820 821 822 823
   * @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 (李卓桓) 已提交
824
   * - `roomAlias`            the name-string set by user-self in the room, should be called roomAlias
L
lijiarui 已提交
825
   * - `contactAlias`         the name-string set by bot for others, should be called alias, equal to `Contact.alias()`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
826
   * @param {(RoomMemberQueryFilter | string)} query -When use memberAll(name:string), return all matched members, including name, roomAlias, contactAlias
L
lijiarui 已提交
827
   * @returns {Promise<Contact[]>}
L
lijiarui 已提交
828
   */
829
  public async memberAll (
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
830
    query: string | RoomMemberQueryFilter,
831 832
  ): Promise<Contact[]> {
    log.silly('Room', 'memberAll(%s)',
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
833
                      JSON.stringify(query),
834 835
              )

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

    return contactList
840
  }
841

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

L
lijiarui 已提交
845 846 847
  /**
   * Find all contacts in a room, if get many, return the first one.
   *
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
848
   * @param {(RoomMemberQueryFilter | string)} queryArg -When use member(name:string), return all matched members, including name, roomAlias, contactAlias
L
lijiarui 已提交
849
   * @returns {Promise<null | Contact>}
L
lijiarui 已提交
850 851
   *
   * @example <caption>Find member by name</caption>
L
lijiarui 已提交
852 853 854 855
   * 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 已提交
856
   * if (room) {
L
lijiarui 已提交
857
   *   const member = await room.member('lijiarui')             // change 'lijiarui' to any room member in your wechat
L
lijiarui 已提交
858
   *   if (member) {
L
lijiarui 已提交
859
   *     console.log(`wechaty room got the member: ${member.name()}`)
L
lijiarui 已提交
860
   *   } else {
L
lijiarui 已提交
861
   *     console.log(`cannot get member in wechaty room!`)
L
lijiarui 已提交
862 863 864 865
   *   }
   * }
   *
   * @example <caption>Find member by MemberQueryFilter</caption>
L
lijiarui 已提交
866 867 868 869
   * 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 已提交
870
   * if (room) {
L
lijiarui 已提交
871
   *   const member = await room.member({name: 'lijiarui'})        // change 'lijiarui' to any room member in your wechat
L
lijiarui 已提交
872
   *   if (member) {
L
lijiarui 已提交
873
   *     console.log(`wechaty room got the member: ${member.name()}`)
L
lijiarui 已提交
874
   *   } else {
L
lijiarui 已提交
875
   *     console.log(`cannot get member in wechaty room!`)
L
lijiarui 已提交
876 877 878
   *   }
   * }
   */
879
  public async member (
880
    queryArg: string | RoomMemberQueryFilter,
881
  ): Promise<null | Contact> {
882 883 884 885 886 887
    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') {
888
      memberList =  await this.memberAll(queryArg)
889
    } else {
890
      memberList =  await this.memberAll(queryArg)
891 892 893 894 895 896 897 898 899 900 901
    }

    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]
  }
902

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

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

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

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

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

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

L
lijiarui 已提交
945 946
  /**
   * Get room's owner from the room.
947 948
   * > 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 已提交
949
   * @returns {(Contact | null)}
L
lijiarui 已提交
950 951
   * @example
   * const owner = room.owner()
L
lijiarui 已提交
952
   */
953
  public owner (): Contact | null {
954 955
    log.info('Room', 'owner()')

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

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

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

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

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