room.ts 31.1 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 42
import {
  Contact,
}               from './contact'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
43

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
44 45 46 47 48 49
export const ROOM_EVENT_DICT = {
  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 89 90 91 92 93
    log.verbose('Room', 'create(%s, %s)', contactList.join(','), topic)

    if (!contactList || !Array.isArray(contactList)) {
      throw new Error('contactList not found')
    }

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

  /**
L
lijiarui 已提交
106
   * The filter to find the room:  {topic: string | RegExp}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
107
   *
L
lijiarui 已提交
108 109 110 111 112 113
   * @typedef    RoomQueryFilter
   * @property   {string} topic
   */

  /**
   * Find room by by filter: {topic: string | RegExp}, return all the matched room
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
114 115 116 117
   * @static
   * @param {RoomQueryFilter} [query]
   * @returns {Promise<Room[]>}
   * @example
L
lijiarui 已提交
118 119 120 121 122
   * 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 (李卓桓) 已提交
123
   */
124
  public static async findAll<T extends typeof Room> (
125 126
    this  : T,
    query : RoomQueryFilter = { topic: /.*/ },
127
  ): Promise<Array<T['prototype']>> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
128
    log.verbose('Room', 'findAll()', JSON.stringify(query))
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
129 130 131 132 133 134

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

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

      return roomList

    } catch (e) {
      log.verbose('Room', 'findAll() rejected: %s', e.message)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
153
      console.error(e)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
154 155 156 157 158 159 160 161 162 163
      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 已提交
164 165 166 167 168 169
   * @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 (李卓桓) 已提交
170
   */
171

172
  public static async find<T extends typeof Room> (
173
    this  : T,
174
    query : string | RoomQueryFilter,
175
  ): Promise<T['prototype'] | null> {
176 177 178 179 180
    log.verbose('Room', 'find(%s)', JSON.stringify(query))

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

    const roomList = await this.findAll(query)
183
    if (!roomList) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
184 185
      return null
    }
186 187 188 189 190 191 192 193 194 195 196
    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]
197
      // use puppet.roomValidate() to confirm double confirm that this roomId is valid.
198 199
      // https://github.com/lijiarui/wechaty-puppet-padchat/issues/64
      // https://github.com/Chatie/wechaty/issues/1345
200
      const valid = await this.puppet.roomValidate(room.id)
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215
      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 (李卓桓) 已提交
216 217 218 219
  }

  /**
   * @private
220
   * About the Generic: https://stackoverflow.com/q/43003970/1123955
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
221
   */
L
lijiarui 已提交
222 223 224 225 226 227
  /**
   * 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.
   *
228
   * 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 已提交
229 230 231 232 233 234 235 236 237
   * @static
   * @param {string} id
   * @returns {Room}
   * @example
   * const bot = new Wechaty()
   * await bot.start()
   * // after logged in...
   * const room = bot.Room.load('roomId')
   */
238
  public static load<T extends typeof Room> (
239 240
    this : T,
    id   : string,
241
  ): T['prototype'] {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
242 243 244 245
    if (!this.pool) {
      this.pool = new Map<string, Room>()
    }

246 247 248
    const existingRoom = this.pool.get(id)
    if (existingRoom) {
      return existingRoom
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
249
    }
250

251 252
    const newRoom = new (this as any)(id) as Room
    // newRoom.payload = this.puppet.cacheRoomPayload.get(id)
253

254 255
    this.pool.set(id, newRoom)
    return newRoom
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
256 257
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
258
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
259
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
260 261 262 263 264
   *
   * Instance Properties
   *
   *
   */
265
  protected get payload (): undefined | RoomPayload {
266 267 268 269
    if (!this.id) {
      return undefined
    }

270 271 272
    const readyPayload = this.puppet.roomPayloadCache(this.id)
    return readyPayload
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
273

274 275
  public readonly id: string  // Room Id

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

285 286
    this.id = id

287 288 289 290 291 292 293 294 295 296 297
    // 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!')
    }

298
  }
299

H
hcz 已提交
300 301 302
  /**
   * @private
   */
303
  public toString () {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
304
    if (this.payload && this.payload.topic) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
305
      return `Room<${this.payload.topic}>`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
306
    }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
307
    return `Room<${this.id || ''}>`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
308
  }
L
lijiarui 已提交
309

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

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

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

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

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

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

  /**
   * @private
   */
351
  public isReady (): boolean {
352
    return !!(this.payload)
353 354
  }

355 356 357 358 359
  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 已提交
360 361 362

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

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

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
410
    if (typeof textOrContactOrFile === 'string') {
411 412

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

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

438 439 440 441
  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
442

443
  public emit (
444
    event:   RoomEventName,
445
    ...args: any[]
446 447 448 449
  ): boolean {
    return super.emit(event, ...args)
  }

450 451 452 453
  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 已提交
454 455 456 457 458 459 460 461 462 463

   /**
    * @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 已提交
464
  /**
L
lijiarui 已提交
465 466 467 468 469
   * @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 已提交
470
   */
L
lijiarui 已提交
471

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

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
517
    super.on(event, listener) // Room is `Sayable`
518
    return this
519 520
  }

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

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

581 582
  // private delLocal(contact: Contact): void {
  //   log.verbose('Room', 'delLocal(%s)', contact)
583

584 585 586 587 588 589 590 591 592 593
  //   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 (李卓桓) 已提交
594

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

610 611
  public async topic ()                : Promise<string>
  public async topic (newTopic: string): Promise<void>
612

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

    if (typeof newTopic === 'undefined') {
652 653 654 655 656 657 658 659
      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))

660
        let defaultTopic = memberList[0] && memberList[0].name() || ''
661 662 663 664 665
        for (let i = 1; i < 3 && memberList[i]; i++) {
          defaultTopic += ',' + memberList[i].name()
        }
        return defaultTopic
      }
666 667 668
    }

    const future = this.puppet
669
        .roomTopic(this.id, newTopic)
670 671 672 673 674 675 676 677 678
        .catch(e => {
          log.warn('Room', 'topic(newTopic=%s) exception: %s',
                            newTopic, e && e.message || e,
                  )
          Raven.captureException(e)
        })

    return future
  }
679

680 681
  public async announce ()             : Promise<string>
  public async announce (text: string) : Promise<void>
682

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

    if (text) {
      await this.puppet.roomAnnounce(this.id, text)
    } else {
715 716
      const announcement = await this.puppet.roomAnnounce(this.id)
      return announcement
717 718 719
    }
  }

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

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

H
hcz 已提交
753
  /**
L
lijiarui 已提交
754 755
   * Same as function alias
   * @param {Contact} contact
L
lijiarui 已提交
756
   * @returns {Promise<string | null>}
H
hcz 已提交
757
   */
758
  public async roomAlias (contact: Contact): Promise<null | string> {
759

760
    const memberPayload = await this.puppet.roomMemberPayload(this.id, contact.id)
761 762 763

    if (memberPayload && memberPayload.roomAlias) {
      return memberPayload.roomAlias
764
    }
765 766

    return null
767
  }
768

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

    if (!memberIdList) {
792 793
      return false
    }
794 795 796 797

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

800 801
  public async memberAll (name: string)                  : Promise<Contact[]>
  public async memberAll (filter: RoomMemberQueryFilter) : Promise<Contact[]>
802

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

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
830 831 832 833
    const contactIdList = await this.puppet.roomMemberSearch(this.id, query)
    const contactList   = contactIdList.map(id => this.wechaty.Contact.load(id))

    return contactList
834
  }
835

836 837
  public async member (name  : string)               : Promise<null | Contact>
  public async member (filter: RoomMemberQueryFilter): Promise<null | Contact>
838

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

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

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

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

909
    if (!memberIdList) {
910 911 912
      log.warn('Room', 'memberList() not ready')
      return []
    }
913 914 915 916

    const contactList = memberIdList.map(
      id => this.wechaty.Contact.load(id),
    )
917
    return contactList
918
  }
919

L
lijiarui 已提交
920
  /**
L
lijiarui 已提交
921
   * @ignore
L
lijiarui 已提交
922
   */
923
  public async refresh (): Promise<void> {
924 925
    return this.sync()
  }
L
lijiarui 已提交
926

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

L
lijiarui 已提交
938 939
  /**
   * Get room's owner from the room.
940 941
   * > 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 已提交
942
   * @returns {(Contact | null)}
L
lijiarui 已提交
943 944
   * @example
   * const owner = room.owner()
L
lijiarui 已提交
945
   */
946
  public owner (): Contact | null {
947 948
    log.info('Room', 'owner()')

949 950 951 952 953
    const ownerId = this.payload && this.payload.ownerId
    if (!ownerId) {
      return null
    }

954
    const owner = this.wechaty.Contact.load(ownerId)
955
    return owner
956
  }
L
lijiarui 已提交
957

958
  public async avatar (): Promise<FileBox> {
959 960 961 962 963
    log.verbose('Room', 'avatar()')

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

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
964
}
Huan (李卓桓)'s avatar
merge  
Huan (李卓桓) 已提交
965 966

export default Room