wechaty.ts 27.2 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 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 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 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633
    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':
          puppet.removeAllListeners('start')
          puppet.on('start', () => {
            this.emit('start')
          } )
          break

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

        case 'friend':
          puppet.removeAllListeners('friend')
          puppet.on('friend', payload => {
            const request = this.FriendRequest.fromJSON(payload)
            this.emit('friend', request)
          })
          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)
          })
          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)
          })
          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)
          })
          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 (李卓桓) 已提交
634
    }
635
  }
636

637 638
  private initPuppetAccessory(puppet: Puppet) {
    log.verbose('Wechaty', 'initPuppetAccessory(%s)', puppet)
639

640 641 642 643 644 645
    this.Contact.puppet       = puppet
    this.FriendRequest.puppet = puppet
    this.Message.puppet       = puppet
    this.Room.puppet          = puppet

    this.puppet = puppet
646 647
  }

648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
  /**
   * 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 {
672 673
      await this.profile.load()
      await this.initPuppet()
674 675 676 677

      await this.puppet.start()

    } catch (e) {
678
      // console.log(e)
679 680 681 682 683 684 685 686 687 688 689 690 691
      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 (李卓桓) 已提交
692 693 694 695 696 697 698 699
  /**
   * Stop the bot
   *
   * @returns {Promise<void>}
   * @example
   * await bot.stop()
   */
  public async stop(): Promise<void> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
700
    log.verbose('Wechaty', 'stop()')
701

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
702
    if (this.state.off()) {
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
703 704 705
      log.silly('Wechaty', 'stop() on an stopping/stopped instance')
      await this.state.ready('off')
      log.silly('Wechaty', 'stop() state.ready(off) resolved')
706
      return
707
    }
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
708

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

711 712 713 714
    let puppet: Puppet
    try {
      puppet = this.puppet
    } catch (e) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
715
      log.warn('Wechaty', 'stop() without this.puppet')
716
      return
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
717 718
    }

719
    try {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
720
      await puppet.stop()
721
    } catch (e) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
722
      log.error('Wechaty', 'stop() exception: %s', e.message)
723 724
      Raven.captureException(e)
      throw e
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
725
    } finally {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
726
      this.state.off(true)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
727
      this.emit('stop')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
728

729 730
      // 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 (李卓桓) 已提交
731
      setImmediate(() => puppet.removeAllListeners())
732
    }
733
    return
734
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
735

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
736
  /**
L
lijiarui 已提交
737 738 739 740 741
   * Logout the bot
   *
   * @returns {Promise<void>}
   * @example
   * await bot.logout()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
742
   */
743
  public async logout(): Promise<void>  {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
744 745
    log.verbose('Wechaty', 'logout()')

746 747 748 749 750 751 752
    try {
      await this.puppet.logout()
    } catch (e) {
      log.error('Wechaty', 'logout() exception: %s', e.message)
      Raven.captureException(e)
      throw e
    }
753
    return
754
  }
755

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
756 757 758 759 760
  /**
   * Get the logon / logoff state
   *
   * @returns {boolean}
   * @example
761
   * if (bot.logonoff()) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
762 763 764 765 766
   *   console.log('Bot logined')
   * } else {
   *   console.log('Bot not logined')
   * }
   */
767
  public logonoff(): Boolean {
768
    return this.puppet.logonoff()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
769 770
  }

771 772 773 774
  /**
   * @deprecated
   */
  public self(): Contact {
775
    log.warn('Wechaty', 'self() DEPRECATED. use userSelf() instead.')
776 777 778
    return this.userSelf()
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
779
  /**
L
lijiarui 已提交
780 781 782 783
   * Get current user
   *
   * @returns {Contact}
   * @example
784
   * const contact = bot.userSelf()
L
lijiarui 已提交
785
   * console.log(`Bot is ${contact.name()}`)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
786
   */
787
  public userSelf(): Contact {
788 789 790
    const userId = this.puppet.selfId()
    const user = this.Contact.load(userId)
    return user
791
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
792

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
793
  /**
L
lijiarui 已提交
794 795
   * Send message to filehelper
   *
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
796
   * @param {string} text
L
lijiarui 已提交
797
   * @returns {Promise<boolean>}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
798
   */
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
799 800 801
  public async say(text: string): Promise<void> {
    log.verbose('Wechaty', 'say(%s)', text)
    await this.puppet.say(text)
802 803
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
804
  /**
L
lijiarui 已提交
805
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
806
   */
807
  public static async sleep(millisecond: number): Promise<void> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
808
    await new Promise(resolve => {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
809 810 811 812
      setTimeout(resolve, millisecond)
    })
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
813
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
814
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
815
   */
816 817 818 819 820 821 822 823
  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 (李卓桓) 已提交
824
  }
825 826 827 828 829 830 831 832 833 834 835 836 837 838 839

  /**
   * @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 (李卓桓) 已提交
840 841 842 843 844

  /**
   * @private
   */
  public async reset(reason?: string): Promise<void> {
845
    log.verbose('Wechaty', 'reset() because %s', reason || 'no reason')
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
846 847
    await this.puppet.stop()
    await this.puppet.start()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
848 849 850
    return
  }

851 852 853
  public unref(): void {
    log.warn('Wechaty', 'unref() To Be Implemented. See: https://github.com/Chatie/wechaty/issues/1197')
  }
854
}
855 856

export default Wechaty