wechaty.ts 27.6 KB
Newer Older
1
/**
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
 *   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
9
 *
10
 *       http://www.apache.org/licenses/LICENSE-2.0
11
 *
12 13 14 15 16
 *   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 已提交
17 18
 *
 *  @ignore
19
 */
20 21
import * as cuid    from 'cuid'
import * as os      from 'os'
22
import * as semver  from 'semver'
23

24
import {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
25
  // Constructor,
26
  cloneClass,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
27
  // instanceToClass,
28
}                   from 'clone-class'
29 30 31
import {
  callerResolve,
  hotImport,
32 33
}                   from 'hot-import'
import StateSwitch  from 'state-switch'
34

35
import {
36
  VERSION,
37
  config,
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
38
  log,
39
  Raven,
40
  Sayable,
41 42 43
}                       from './config'
import Profile          from './profile'
import PuppetAccessory  from './puppet-accessory'
M
Mukaiu 已提交
44
import {
45 46 47
  PUPPET_DICT,
  PuppetName,
}                       from './puppet-config'
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
48

49
import {
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
50
  Contact,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
51 52
}                       from './contact'
import {
53
  FriendRequest,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
54 55
}                       from './friend-request'
import {
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
56
  Message,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
57 58
}                       from './message'
import {
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
59
  Room,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
60 61 62
}                       from './room'
import {
  Puppet,
63
}                       from './puppet/'
64 65 66 67 68 69 70 71 72 73 74

export const WECHAT_EVENT_DICT = {
  friend      : 'tbw',
  login       : 'tbw',
  logout      : 'tbw',
  message     : 'tbw',
  'room-join' : 'tbw',
  'room-leave': 'tbw',
  'room-topic': 'tbw',
  scan        : 'tbw',
}
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
75

76 77
export const WECHATY_EVENT_DICT = {
  ...WECHAT_EVENT_DICT,
78 79 80 81
  error     : 'tbw',
  heartbeat : 'tbw',
  start     : 'tbw',
  stop      : 'tbw',
82 83 84 85
}

export type WechatEventName   = keyof typeof WECHAT_EVENT_DICT
export type WechatyEventName  = keyof typeof WECHATY_EVENT_DICT
86 87

export interface WechatyOptions {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
88 89
  puppet?  : PuppetName | Puppet,
  profile? : null | string,
90
}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
91

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
92
/**
L
lijiarui 已提交
93
 * Main bot class.
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
94
 *
L
lijiarui 已提交
95
 * [The World's Shortest ChatBot Code: 6 lines of JavaScript]{@link #wechatyinstance}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
96
 *
L
lijiarui 已提交
97 98 99
 * [Wechaty Starter Project]{@link https://github.com/lijiarui/wechaty-getting-started}
 * @example
 * import { Wechaty } from 'wechaty'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
100
 *
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
101
 */
102
export class Wechaty extends PuppetAccessory implements Sayable {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
103
  /**
104
   * singleton globalInstance
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
105
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
106
   */
107
  private static globalInstance: Wechaty
108

109 110
  private profile: Profile

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
111 112
  /**
   * the state
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
113
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
114
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
115
  private state = new StateSwitch('Wechaty', log)
116

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
117
  /**
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
118
   * the cuid
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
119
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
120
   */
121
  public readonly cuid : string
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
122

123
  // tslint:disable-next-line:variable-name
124
  public readonly Contact       : typeof Contact
125
  // tslint:disable-next-line:variable-name
126
  public readonly FriendRequest : typeof FriendRequest
127
  // tslint:disable-next-line:variable-name
128
  public readonly Message       : typeof Message
129
  // tslint:disable-next-line:variable-name
130
  public readonly Room          : typeof Room
131

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
132
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
133
   * get the singleton instance of Wechaty
L
lijiarui 已提交
134 135 136 137 138 139 140 141 142
   *
   * @example <caption>The World's Shortest ChatBot Code: 6 lines of JavaScript</caption>
   * const { Wechaty } = require('wechaty')
   *
   * Wechaty.instance() // Singleton
   * .on('scan', (url, code) => console.log(`Scan QR Code to login: ${code}\n${url}`))
   * .on('login',       user => console.log(`User ${user} logined`))
   * .on('message',  message => console.log(`Message: ${message}`))
   * .init()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
143
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
144 145 146
  public static instance(
    options?: WechatyOptions,
  ) {
147
    if (options && this.globalInstance) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
148
      throw new Error('instance can be only set once!')
149
    }
150 151
    if (!this.globalInstance) {
      this.globalInstance = new Wechaty(options)
152
    }
153
    return this.globalInstance
154 155
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
156
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
157
   * @public
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
158
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
159
  constructor(
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
160
    private options: WechatyOptions = {},
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
161
  ) {
162
    super()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
163 164
    log.verbose('Wechaty', 'contructor()')

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
165
    options.profile = options.profile === null
166
                      ? null
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
167
                      : (options.profile || config.default.DEFAULT_PROFILE)
168

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
169
    this.profile = new Profile(options.profile)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
170

Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
171
    this.cuid = cuid()
172 173 174 175 176 177 178 179 180 181 182 183 184

    /**
     * Clone Classes for this bot and attach the `puppet` to the Class
     *
     *   https://stackoverflow.com/questions/36886082/abstract-constructor-type-in-typescript
     *   https://github.com/Microsoft/TypeScript/issues/5843#issuecomment-290972055
     *   https://github.com/Microsoft/TypeScript/issues/19197
     */
    // TODO: make Message & Room constructor private???
    this.Contact        = cloneClass(Contact)
    this.FriendRequest  = cloneClass(FriendRequest)
    this.Message        = cloneClass(Message)
    this.Room           = cloneClass(Room)
185 186
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
187
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
188
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
189
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
190 191 192 193 194 195 196 197 198 199 200
  public toString() {
    if (!this.options) {
      return this.constructor.name
    }

    return [
      'Wechaty',
      `<${this.options && this.options.puppet || ''}>`,
      `(${this.profile && this.profile.name   || ''})`,
    ].join('')
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
201

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
202
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
203
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
204
   */
205
  public static version(forceNpm = false): string {
206
    if (!forceNpm) {
207
      const revision = config.gitRevision()
208
      if (revision) {
209
        return `#git[${revision}]`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
210
      }
211
    }
212
    return VERSION
213
  }
214

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
215
 /**
Huan (李卓桓)'s avatar
linting  
Huan (李卓桓) 已提交
216 217 218 219 220 221 222 223 224
  * Return version of Wechaty
  *
  * @param {boolean} [forceNpm=false]  - if set to true, will only return the version in package.json.
  *                                      otherwise will return git commit hash if .git exists.
  * @returns {string}                  - the version number
  * @example
  * console.log(Wechaty.instance().version())       // return '#git[af39df]'
  * console.log(Wechaty.instance().version(true))   // return '0.7.9'
  */
225
  public version(forceNpm = false): string {
226
    return Wechaty.version(forceNpm)
227
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
228

229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
  public emit(event: 'error'      , error: Error)                                                  : boolean
  public emit(event: 'friend'     , request: FriendRequest)                                        : boolean
  public emit(event: 'heartbeat'  , data: any)                                                     : boolean
  public emit(event: 'logout'     , user: Contact)                                                 : boolean
  public emit(event: 'login'      , user: Contact)                                                 : boolean
  public emit(event: 'message'    , message: Message)                                              : boolean
  public emit(event: 'room-join'  , room: Room, inviteeList : Contact[], inviter  : Contact)       : boolean
  public emit(event: 'room-leave' , room: Room, leaverList  : Contact[], remover? : Contact)       : boolean
  public emit(event: 'room-topic' , room: Room, topic: string, oldTopic: string, changer: Contact) : boolean
  public emit(event: 'scan'       , qrCode: string, code: number, data?: string)                   : boolean
  public emit(event: 'start')                                                                      : boolean
  public emit(event: 'stop')                                                                       : boolean

  // guard for the above event: make sure it includes all the possible values
  public emit(event: never, listener: never): never

  public emit(
    event:   WechatyEventName,
    ...args: any[]
  ): boolean {
    return super.emit(event, ...args)
  }

252
  public on(event: 'error'      , listener: string | ((this: Wechaty, error: Error) => void))                                                 : this
253
  public on(event: 'friend'     , listener: string | ((this: Wechaty, request: FriendRequest) => void))                     : this
254 255 256 257 258 259 260
  public on(event: 'heartbeat'  , listener: string | ((this: Wechaty, data: any) => void))                                                    : this
  public on(event: 'logout'     , listener: string | ((this: Wechaty, user: Contact) => void))                                                : this
  public on(event: 'login'      , listener: string | ((this: Wechaty, user: Contact) => void))                                                : this
  public on(event: 'message'    , listener: string | ((this: Wechaty, message: Message) => void))                                             : this
  public on(event: 'room-join'  , listener: string | ((this: Wechaty, room: Room, inviteeList: Contact[],  inviter: Contact) => void))        : this
  public on(event: 'room-leave' , listener: string | ((this: Wechaty, room: Room, leaverList: Contact[], remover?: Contact) => void))         : this
  public on(event: 'room-topic' , listener: string | ((this: Wechaty, room: Room, topic: string, oldTopic: string, changer: Contact) => void)): this
261
  public on(event: 'scan'       , listener: string | ((this: Wechaty, url: string, code: number) => void))                                   : this
262 263
  public on(event: 'start'      , listener: string | ((this: Wechaty) => void))                                                               : this
  public on(event: 'stop'       , listener: string | ((this: Wechaty) => void))                                                               : this
264

265
  // guard for the above event: make sure it includes all the possible values
266
  public on(event: never, listener: never): never
L
lijiarui 已提交
267

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
268
  /**
L
lijiarui 已提交
269 270 271 272 273 274 275 276 277 278 279 280
   * @desc       Wechaty Class Event Type
   * @typedef    WechatyEventName
   * @property   {string}  error      - When the bot get error, there will be a Wechaty error event fired.
   * @property   {string}  login      - After the bot login full successful, the event login will be emitted, with a Contact of current logined user.
   * @property   {string}  logout     - Logout will be emitted when bot detected log out, with a Contact of the current login user.
   * @property   {string}  heartbeat  - Get bot's heartbeat.
   * @property   {string}  friend     - When someone sends you a friend request, there will be a Wechaty friend event fired.
   * @property   {string}  message    - Emit when there's a new message.
   * @property   {string}  room-join  - Emit when anyone join any room.
   * @property   {string}  room-topic - Get topic event, emitted when someone change room topic.
   * @property   {string}  room-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.
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
281
   * @property   {string}  scan       - A scan event will be emitted when the bot needs to show you a QR Code for scanning.
L
lijiarui 已提交
282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301
   */

  /**
   * @desc       Wechaty Class Event Function
   * @typedef    WechatyEventFunction
   * @property   {Function} error           -(this: Wechaty, error: Error) => void callback function
   * @property   {Function} login           -(this: Wechaty, user: Contact)=> void
   * @property   {Function} logout          -(this: Wechaty, user: Contact) => void
   * @property   {Function} scan            -(this: Wechaty, url: string, code: number) => void <br>
   * <ol>
   * <li>URL: {String} the QR code image URL</li>
   * <li>code: {Number} the scan status code. some known status of the code list here is:</li>
   * </ol>
   * <ul>
   * <li>0 initial_</li>
   * <li>200 login confirmed</li>
   * <li>201 scaned, wait for confirm</li>
   * <li>408 waits for scan</li>
   * </ul>
   * @property   {Function} heartbeat       -(this: Wechaty, data: any) => void
302
   * @property   {Function} friend          -(this: Wechaty, request?: FriendRequest) => void
L
lijiarui 已提交
303 304 305 306 307 308 309 310
   * @property   {Function} message         -(this: Wechaty, message: Message) => void
   * @property   {Function} room-join       -(this: Wechaty, room: Room, inviteeList: Contact[],  inviter: Contact) => void
   * @property   {Function} room-topic      -(this: Wechaty, room: Room, topic: string, oldTopic: string, changer: Contact) => void
   * @property   {Function} room-leave      -(this: Wechaty, room: Room, leaverList: Contact[]) => void
   */

  /**
   * @listens Wechaty
311
   * @param   {WechatyEventName}      event      - Emit WechatyEvent
L
lijiarui 已提交
312 313 314
   * @param   {WechatyEventFunction}  listener   - Depends on the WechatyEvent
   * @return  {Wechaty}                          - this for chain
   *
315
   * More Example Gist: [Examples/Friend-Bot]{@link https://github.com/Chatie/wechaty/blob/master/examples/friend-bot.ts}
L
lijiarui 已提交
316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
   *
   * @example <caption>Event:scan </caption>
   * wechaty.on('scan', (url: string, code: number) => {
   *   console.log(`[${code}] Scan ${url} to login.` )
   * })
   *
   * @example <caption>Event:login </caption>
   * bot.on('login', (user: Contact) => {
   *   console.log(`user ${user} login`)
   * })
   *
   * @example <caption>Event:logout </caption>
   * bot.on('logout', (user: Contact) => {
   *   console.log(`user ${user} logout`)
   * })
   *
   * @example <caption>Event:message </caption>
   * wechaty.on('message', (message: Message) => {
   *   console.log(`message ${message} received`)
   * })
   *
   * @example <caption>Event:friend </caption>
338 339 340
   * bot.on('friend', (request: FriendRequest) => {
   *   if(request.type === FriendRequest.Type.RECEIVE){ // 1. receive new friend request from new contact
   *     const contact = request.contact()
L
lijiarui 已提交
341 342 343 344 345 346
   *     let result = await request.accept()
   *       if(result){
   *         console.log(`Request from ${contact.name()} is accept succesfully!`)
   *       } else{
   *         console.log(`Request from ${contact.name()} failed to accept!`)
   *       }
347
   * 	  } else if (request.type === FriendRequest.Type.CONFIRM) { // 2. confirm friend ship
L
lijiarui 已提交
348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367
   *       console.log(`new friendship confirmed with ${contact.name()}`)
   *    }
   *  })
   *
   * @example <caption>Event:room-join </caption>
   * bot.on('room-join', (room: Room, inviteeList: Contact[], inviter: Contact) => {
   *   const nameList = inviteeList.map(c => c.name()).join(',')
   *   console.log(`Room ${room.topic()} got new member ${nameList}, invited by ${inviter}`)
   * })
   *
   * @example <caption>Event:room-leave </caption>
   * bot.on('room-leave', (room: Room, leaverList: Contact[]) => {
   *   const nameList = leaverList.map(c => c.name()).join(',')
   *   console.log(`Room ${room.topic()} lost member ${nameList}`)
   * })
   *
   * @example <caption>Event:room-topic </caption>
   * bot.on('room-topic', (room: Room, topic: string, oldTopic: string, changer: Contact) => {
   *   console.log(`Room ${room.topic()} topic changed from ${oldTopic} to ${topic} by ${changer.name()}`)
   * })
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
368
   */
369
  public on(event: WechatyEventName, listener: string | ((...args: any[]) => any)): this {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
370
    log.verbose('Wechaty', 'on(%s, %s) registered',
371 372 373 374 375 376
                            event,
                            typeof listener === 'string'
                              ? listener
                              : typeof listener,
                )

377 378 379 380
    if (typeof listener === 'function') {
      this.onFunction(event, listener)
    } else {
      this.onModulePath(event, listener)
381
    }
382
    return this
383 384
  }

385
  private onModulePath(event: WechatyEventName, modulePath: string): void {
386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404
    const absoluteFilename = callerResolve(modulePath, __filename)
    log.verbose('Wechaty', 'onModulePath() hotImpor(%s)', absoluteFilename)
    hotImport(absoluteFilename)
      .then((func: Function) => super.on(event, (...args: any[]) => {
        try {
          func.apply(this, args)
        } catch (e) {
          log.error('Wechaty', 'onModulePath(%s, %s) listener exception: %s',
                                event, modulePath, e)
          this.emit('error', e)
        }
      }))
      .catch(e => {
        log.error('Wechaty', 'onModulePath(%s, %s) hotImport() exception: %s',
                              event, modulePath, e)
        this.emit('error', e)
      })
  }

405
  private onFunction(event: WechatyEventName, listener: Function): void {
406 407 408 409 410 411 412 413 414 415 416 417
    log.verbose('Wechaty', 'onFunction(%s)', event)

    super.on(event, (...args: any[]) => {
      try {
        listener.apply(this, args)
      } catch (e) {
        log.error('Wechaty', 'onFunction(%s) listener exception: %s', event, e)
        this.emit('error', e)
      }
    })
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
418
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
419 420 421 422 423 424
   * set Wechaty attach to the puppet,
   * this function should be called before `start()`
   *
   * Will be called from the Puppet in constructor:
   *  When we declare a wechaty without a puppet instance,
   *  the wechaty need to attach to puppet at here.
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
425 426 427 428
   */
  public attach(puppet: Puppet) {
    log.verbose('Wechaty', 'attach(%s) this.options.puppet="%s"',
                            puppet,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
429
                            this.options.puppet && this.options.puppet.toString() || '',
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
430 431 432 433 434 435 436 437 438 439 440 441 442 443
                )

    if (this.options.puppet instanceof Puppet) {
      if (this.options.puppet === puppet) {
        log.silly('Wechaty', 'attach(%s) called again', puppet)
        return
      } else {
        throw new Error('puppet can only be attached once!')
      }
    }

    this.options.puppet = puppet
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
444
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
445
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
446
   */
447
  private initPuppet(): void {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
448
    log.verbose('Wechaty', 'initPuppet()')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
449

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
450
    if (!this.options.puppet) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
451
      log.info('Wechaty', 'initPuppet() using default puppet: %s', config.puppet)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
452
      this.options.puppet  = config.puppet
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
453 454
    }

455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476
    const puppet = this.initPuppetResolver(this.options.puppet)

    if (!this.initPuppetSemverSatisfy(
      puppet.wechatyVersionRange(),
    )) {
      throw new Error(`The Puppet Plugin(${puppet.constructor.name}) `
          + `requires a version range(${puppet.wechatyVersionRange()}) `
          + `that is not satisfying the Wechaty version: ${this.version()}.`,
        )
    }

    this.initPuppetEventBridge(puppet)
    this.initPuppetAccessory(puppet)
  }

  /**
   * Init the Puppet
   */
  private initPuppetResolver(puppet: PuppetName | Puppet): Puppet {
    log.verbose('Wechaty', 'initPuppetResolver(%s)', puppet)

    if (typeof puppet === 'string') {
477
      // tslint:disable-next-line:variable-name
478 479 480 481 482
      const MyPuppet = PUPPET_DICT[puppet]
      if (!MyPuppet) {
        throw new Error('no such puppet: ' + puppet)
      }

483
      const options = {
484 485
        profile:  this.profile,
        wechaty:  this,
486 487
      }

488
      return new MyPuppet(options)
489

490 491
    } else if (puppet instanceof Puppet) {
      return puppet
492 493
    } else {
      throw new Error('unsupported options.puppet!')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
494
    }
495
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
496

497 498 499 500
  /**
   * Plugin Version Range Check
   */
  private initPuppetSemverSatisfy(versionRange: string) {
501
    log.verbose('Wechaty', 'initPuppetSemverSatisfy(%s)', versionRange)
502
    return semver.satisfies(
503
      this.version(true),
504 505 506
      versionRange,
    )
  }
507

508
  private initPuppetEventBridge(puppet: Puppet) {
509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533
    const eventNameList: WechatyEventName[] = Object.keys(WECHATY_EVENT_DICT) as any
    for (const eventName of eventNameList) {
      log.verbose('Wechaty', 'initPuppetEventBridge() puppet.on(%s) registered', eventName)
      // /// e as any ??? Maybe this is a bug of TypeScript v2.5.3
      // puppet.on(event as any, (...args: any[]) => {
      //   this.emit(event, ...args)
      // })

      switch (eventName) {
        case 'error':
          puppet.removeAllListeners('error')
          puppet.on('error', error => {
            this.emit('error', new Error(error))
          })
          break

        case 'heartbeat':
          puppet.removeAllListeners('heartbeat')
          puppet.on('heartbeat', data => {
            this.emit('heartbeat', data)
          })
          break

        case 'start':
        case 'stop':
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
534
          // do not emit 'start'/'stop' again for wechaty
535 536
          break

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
537 538 539 540 541 542 543 544 545 546 547 548 549 550
        // case 'start':
        //   puppet.removeAllListeners('start')
        //   puppet.on('start', () => {
        //     this.emit('start')
        //   } )
        //   break

        // case 'stop':
        //   puppet.removeAllListeners('stop')
        //   puppet.on('stop', () => {
        //     this.emit('stop')
        //   } )
        //   break

551 552
        case 'friend':
          puppet.removeAllListeners('friend')
553 554 555
          puppet.on('friend', async requestId => {
            const request = this.FriendRequest.load(requestId)
            await request.ready()
556
            this.emit('friend', request)
557
            request.contact().emit('friend', request)
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600
          })
          break

        case 'login':
          puppet.removeAllListeners('login')
          puppet.on('login', async contactId => {
            const contact = this.Contact.load(contactId)
            await contact.ready()
            this.emit('login', contact)
          })
          break

        case 'logout':
          puppet.removeAllListeners('logout')
          puppet.on('logout', async contactId => {
            const contact = this.Contact.load(contactId)
            await contact.ready()
            this.emit('logout', contact)
          })
          break

        case 'message':
          puppet.removeAllListeners('message')
          puppet.on('message', async messageId => {
            const msg = this.Message.create(messageId)
            await msg.ready()
            this.emit('message', msg)
          })
          break

        case 'room-join':
          puppet.removeAllListeners('room-join')
          puppet.on('room-join', async (roomId, inviteeIdList, inviterId) => {
            const room = this.Room.load(roomId)
            await room.ready()

            const inviteeList = inviteeIdList.map(id => this.Contact.load(id))
            await Promise.all(inviteeList.map(c => c.ready()))

            const inviter = this.Contact.load(inviterId)
            await inviter.ready()

            this.emit('room-join', room, inviteeList, inviter)
601
            room.emit('join', inviteeList, inviter)
602 603 604 605 606 607 608 609 610 611 612 613 614
          })
          break

        case 'room-leave':
          puppet.removeAllListeners('room-leave')
          puppet.on('room-leave', async (roomId, leaverIdList) => {
            const room = this.Room.load(roomId)
            await room.ready()

            const leaverList = leaverIdList.map(id => this.Contact.load(id))
            await Promise.all(leaverList.map(c => c.ready()))

            this.emit('room-leave', room, leaverList)
615
            room.emit('leave', leaverList)
616 617 618 619 620 621 622 623 624 625 626 627 628
          })
          break

        case 'room-topic':
          puppet.removeAllListeners('room-topic')
          puppet.on('room-topic', async (roomId, topic, oldTopic, changerId) => {
            const room = this.Room.load(roomId)
            await room.ready()

            const changer = this.Contact.load(changerId)
            await changer.ready()

            this.emit('room-topic', room, topic, oldTopic, changer)
629
            room.emit('topic', topic, oldTopic, changer)
630 631 632 633 634 635 636 637 638 639 640 641 642 643
          })
          break

        case 'scan':
          puppet.removeAllListeners('scan')
          puppet.on('scan', async (qrCode, code, data) => {
            this.emit('scan', qrCode, code, data)
          })
          break

        default:
          throw new Error('eventName ' + eventName + 'unsupported!')

      }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
644
    }
645
  }
646

647 648
  private initPuppetAccessory(puppet: Puppet) {
    log.verbose('Wechaty', 'initPuppetAccessory(%s)', puppet)
649

650 651 652 653 654 655
    this.Contact.puppet       = puppet
    this.FriendRequest.puppet = puppet
    this.Message.puppet       = puppet
    this.Room.puppet          = puppet

    this.puppet = puppet
656 657
  }

658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681
  /**
   * Start the bot, return Promise.
   *
   * @returns {Promise<void>}
   * @example
   * await bot.start()
   * // do other stuff with bot here
   */
  public async start(): Promise<void> {
    log.info('Wechaty', 'v%s starting...' , this.version())
    log.verbose('Wechaty', 'puppet: %s'   , this.options.puppet)
    log.verbose('Wechaty', 'profile: %s'  , this.options.profile)
    log.verbose('Wechaty', 'cuid: %s'     , this.cuid)

    if (this.state.on()) {
      log.silly('Wechaty', 'start() on a starting/started instance')
      await this.state.ready()
      log.silly('Wechaty', 'start() state.ready() resolved')
      return
    }

    this.state.on('pending')

    try {
682 683
      await this.profile.load()
      await this.initPuppet()
684 685 686 687

      await this.puppet.start()

    } catch (e) {
688
      // console.log(e)
689 690 691 692 693 694 695 696 697 698 699 700 701
      log.error('Wechaty', 'start() exception: %s', e && e.message)
      Raven.captureException(e)
      throw e
    }

    this.on('heartbeat', () => this.memoryCheck())

    this.state.on(true)
    this.emit('start')

    return
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
702 703 704 705 706 707 708 709
  /**
   * Stop the bot
   *
   * @returns {Promise<void>}
   * @example
   * await bot.stop()
   */
  public async stop(): Promise<void> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
710
    log.verbose('Wechaty', 'stop()')
711

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
712
    if (this.state.off()) {
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
713 714 715
      log.silly('Wechaty', 'stop() on an stopping/stopped instance')
      await this.state.ready('off')
      log.silly('Wechaty', 'stop() state.ready(off) resolved')
716
      return
717
    }
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
718

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
719
    this.state.off('pending')
720

721 722 723 724
    let puppet: Puppet
    try {
      puppet = this.puppet
    } catch (e) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
725
      log.warn('Wechaty', 'stop() without this.puppet')
726
      return
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
727 728
    }

729
    try {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
730
      await puppet.stop()
731
    } catch (e) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
732
      log.error('Wechaty', 'stop() exception: %s', e.message)
733 734
      Raven.captureException(e)
      throw e
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
735
    } finally {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
736
      this.state.off(true)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
737
      this.emit('stop')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
738

739 740
      // MUST use setImmediate at here(the end of this function),
      // because we need to run the micro task registered by the `emit` method
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
741
      setImmediate(() => puppet.removeAllListeners())
742
    }
743
    return
744
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
745

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
746
  /**
L
lijiarui 已提交
747 748 749 750 751
   * Logout the bot
   *
   * @returns {Promise<void>}
   * @example
   * await bot.logout()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
752
   */
753
  public async logout(): Promise<void>  {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
754 755
    log.verbose('Wechaty', 'logout()')

756 757 758 759 760 761 762
    try {
      await this.puppet.logout()
    } catch (e) {
      log.error('Wechaty', 'logout() exception: %s', e.message)
      Raven.captureException(e)
      throw e
    }
763
    return
764
  }
765

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
766 767 768 769 770
  /**
   * Get the logon / logoff state
   *
   * @returns {boolean}
   * @example
771
   * if (bot.logonoff()) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
772 773 774 775 776
   *   console.log('Bot logined')
   * } else {
   *   console.log('Bot not logined')
   * }
   */
777
  public logonoff(): Boolean {
778
    return this.puppet.logonoff()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
779 780
  }

781 782 783 784
  /**
   * @deprecated
   */
  public self(): Contact {
785
    log.warn('Wechaty', 'self() DEPRECATED. use userSelf() instead.')
786 787 788
    return this.userSelf()
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
789
  /**
L
lijiarui 已提交
790 791 792 793
   * Get current user
   *
   * @returns {Contact}
   * @example
794
   * const contact = bot.userSelf()
L
lijiarui 已提交
795
   * console.log(`Bot is ${contact.name()}`)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
796
   */
797
  public userSelf(): Contact {
798 799 800
    const userId = this.puppet.selfId()
    const user = this.Contact.load(userId)
    return user
801
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
802

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
803
  /**
L
lijiarui 已提交
804 805
   * Send message to filehelper
   *
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
806
   * @param {string} text
L
lijiarui 已提交
807
   * @returns {Promise<boolean>}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
808
   */
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
809 810 811
  public async say(text: string): Promise<void> {
    log.verbose('Wechaty', 'say(%s)', text)
    await this.puppet.say(text)
812 813
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
814
  /**
L
lijiarui 已提交
815
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
816
   */
817
  public static async sleep(millisecond: number): Promise<void> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
818
    await new Promise(resolve => {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
819 820 821 822
      setTimeout(resolve, millisecond)
    })
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
823
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
824
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
825
   */
826 827 828 829 830 831 832 833
  public async ding(): Promise<string> {
    try {
      return await this.puppet.ding() // should return 'dong'
    } catch (e) {
      log.error('Wechaty', 'ding() exception: %s', e.message)
      Raven.captureException(e)
      throw e
    }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
834
  }
835 836 837 838 839 840 841 842 843 844 845 846 847 848 849

  /**
   * @private
   */
  private memoryCheck(minMegabyte = 4): void {
    const freeMegabyte = Math.floor(os.freemem() / 1024 / 1024)
    log.silly('Wechaty', 'memoryCheck() free: %d MB, require: %d MB',
                          freeMegabyte, minMegabyte)

    if (freeMegabyte < minMegabyte) {
      const e = new Error(`memory not enough: free ${freeMegabyte} < require ${minMegabyte} MB`)
      log.warn('Wechaty', 'memoryCheck() %s', e.message)
      this.emit('error', e)
    }
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
850 851 852 853 854

  /**
   * @private
   */
  public async reset(reason?: string): Promise<void> {
855
    log.verbose('Wechaty', 'reset() because %s', reason || 'no reason')
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
856 857
    await this.puppet.stop()
    await this.puppet.start()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
858 859 860
    return
  }

861 862 863
  public unref(): void {
    log.warn('Wechaty', 'unref() To Be Implemented. See: https://github.com/Chatie/wechaty/issues/1197')
  }
864
}
865 866

export default Wechaty