wechaty.ts 30.9 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 22
import cuid    from 'cuid'
import os      from 'os'
import semver  from 'semver'
23

24
import {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
25
  // Constructor,
26
  cloneClass,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
27
  // instanceToClass,
28
}                   from 'clone-class'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
29 30 31
import {
  FileBox,
}                   from 'file-box'
32 33 34
import {
  callerResolve,
  hotImport,
35
}                   from 'hot-import'
36 37 38 39 40 41
import {
  StateSwitch,
}                   from 'state-switch'
import {
  MemoryCard,
}                   from 'memory-card'
42

43 44 45
import {
  Accessory,
}                       from './accessory'
46
import {
47
  VERSION,
48
  config,
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
49
  log,
50
  Raven,
51
  Sayable,
52
}                       from './config'
53

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
54 55 56
import {
  Io,
}                       from './io'
M
Mukaiu 已提交
57
import {
58 59 60
  PUPPET_DICT,
  PuppetName,
}                       from './puppet-config'
61
import {
62 63
  Puppet,
  PuppetOptions,
64
}                       from './puppet/'
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
65

66
import {
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
67
  Contact,
68
  ContactSelf,
69
  Friendship,
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
70 71
  Message,
  Room,
72
}                       from './user/'
73

74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
import {
  CHAT_EVENT_DICT,
  PUPPET_EVENT_DICT,
  PuppetEventName,
  // ChatEventName,
}                       from './puppet/schemas/puppet'

// 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 (李卓桓) 已提交
91

92
export const WECHATY_EVENT_DICT = {
93
  ...CHAT_EVENT_DICT,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
94
  dong      : 'tbw',
95 96 97 98
  error     : 'tbw',
  heartbeat : 'tbw',
  start     : 'tbw',
  stop      : 'tbw',
99 100 101
}

export type WechatyEventName  = keyof typeof WECHATY_EVENT_DICT
102 103

export interface WechatyOptions {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
104 105 106 107
  profile?     : null | string,         // Wechaty Name
  puppet?      : PuppetName | Puppet,   // Puppet name or instance
  puppetToken? : string,                // Puppet TOKEN
  ioToken?     : string,                // Io TOKEN
108
}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
109

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
110
/**
L
lijiarui 已提交
111
 * Main bot class.
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
112
 *
L
lijiarui 已提交
113
 * [The World's Shortest ChatBot Code: 6 lines of JavaScript]{@link #wechatyinstance}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
114
 *
L
lijiarui 已提交
115 116 117
 * [Wechaty Starter Project]{@link https://github.com/lijiarui/wechaty-getting-started}
 * @example
 * import { Wechaty } from 'wechaty'
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
118
 *
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
119
 */
120
export class Wechaty extends Accessory implements Sayable {
121 122 123

  public readonly state  : StateSwitch

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
124
  /**
125
   * singleton globalInstance
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
126
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
127
   */
128
  private static globalInstance: Wechaty
129

130
  private readonly memory : MemoryCard
131

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
132 133
  private lifeTimer? : NodeJS.Timer
  private io?        : Io
134

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
135
  /**
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
136
   * the cuid
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
137
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
138
   */
139
  public readonly id : string
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
140

141
  // tslint:disable-next-line:variable-name
142
  public readonly Contact       : typeof Contact
143
  // tslint:disable-next-line:variable-name
144 145
  public readonly ContactSelf   : typeof ContactSelf
  // tslint:disable-next-line:variable-name
146
  public readonly Friendship    : typeof Friendship
147
  // tslint:disable-next-line:variable-name
148
  public readonly Message       : typeof Message
149
  // tslint:disable-next-line:variable-name
150
  public readonly Room          : typeof Room
151

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
152
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
153
   * get the singleton instance of Wechaty
L
lijiarui 已提交
154 155 156 157 158 159 160 161 162
   *
   * @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 (李卓桓) 已提交
163
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
164 165 166
  public static instance(
    options?: WechatyOptions,
  ) {
167
    if (options && this.globalInstance) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
168
      throw new Error('instance can be only inited once by options!')
169
    }
170 171
    if (!this.globalInstance) {
      this.globalInstance = new Wechaty(options)
172
    }
173
    return this.globalInstance
174 175
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
176
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
177
   * @public
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
178
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
179
  constructor(
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
180
    private options: WechatyOptions = {},
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
181
  ) {
182
    super()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
183 184
    log.verbose('Wechaty', 'contructor()')

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
185
    options.profile = options.profile === null
186
                      ? null
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
187
                      : (options.profile || config.default.DEFAULT_PROFILE)
188

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
189
    this.memory = new MemoryCard(options.profile || undefined)
190 191
    this.state  = new StateSwitch('Wechaty', log)
    this.id     = cuid()
192 193 194 195 196 197 198 199 200 201

    /**
     * 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)
202
    this.ContactSelf    = cloneClass(ContactSelf)
203
    this.Friendship  = cloneClass(Friendship)
204 205
    this.Message        = cloneClass(Message)
    this.Room           = cloneClass(Room)
206 207
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
208
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
209
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
210
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
211 212 213 214 215 216
  public toString() {
    if (!this.options) {
      return this.constructor.name
    }

    return [
217 218
      'Wechaty#',
      this.id,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
219
      `<${this.options && this.options.puppet || ''}>`,
220
      `(${this.memory  && this.memory.name    || ''})`,
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
221 222
    ].join('')
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
223

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
224
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
225
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
226
   */
227
  public static version(forceNpm = false): string {
228
    if (!forceNpm) {
229
      const revision = config.gitRevision()
230
      if (revision) {
231
        return `#git[${revision}]`
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
232
      }
233
    }
234
    return VERSION
235
  }
236

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
237
 /**
Huan (李卓桓)'s avatar
linting  
Huan (李卓桓) 已提交
238 239 240 241 242 243 244 245 246
  * 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'
  */
247
  public version(forceNpm = false): string {
248
    return Wechaty.version(forceNpm)
249
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
250

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
251
  public emit(event: 'dong'       , data?: string)                                                    : boolean
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
252
  public emit(event: 'error'      , error: Error)                                                     : boolean
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
253
  public emit(event: 'friendship' , friendship: Friendship)                                           : boolean
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
254 255 256 257 258 259 260 261 262 263
  public emit(event: 'heartbeat'  , data: any)                                                        : boolean
  public emit(event: 'logout'     , user: ContactSelf)                                                : boolean
  public emit(event: 'login'      , user: ContactSelf)                                                : 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, newTopic: string, oldTopic: string, changer: Contact) : boolean
  public emit(event: 'scan'       , qrcode: string, status: number, data?: string)                    : boolean
  public emit(event: 'start')                                                                         : boolean
  public emit(event: 'stop')                                                                          : boolean
264 265 266 267 268 269 270 271 272 273 274

  // 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)
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
275
  public on(event: 'dong'       , listener: string | ((this: Wechaty, data?: string) => void))                                                     : this
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
276
  public on(event: 'error'      , listener: string | ((this: Wechaty, error: Error) => void))                                                     : this
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
277
  public on(event: 'friendship' , listener: string | ((this: Wechaty, friendship: Friendship) => void))                                           : this
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
278 279 280 281 282 283 284 285 286 287
  public on(event: 'heartbeat'  , listener: string | ((this: Wechaty, data: any) => void))                                                        : this
  public on(event: 'logout'     , listener: string | ((this: Wechaty, user: ContactSelf) => void))                                                : this
  public on(event: 'login'      , listener: string | ((this: Wechaty, user: ContactSelf) => 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, newTopic: string, oldTopic: string, changer: Contact) => void)) : this
  public on(event: 'scan'       , listener: string | ((this: Wechaty, qrcode: string, status: number, data?: string) => void))                    : this
  public on(event: 'start'      , listener: string | ((this: Wechaty) => void))                                                                   : this
  public on(event: 'stop'       , listener: string | ((this: Wechaty) => void))                                                                   : this
288

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

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
292
  /**
L
lijiarui 已提交
293 294 295 296 297 298 299 300 301 302 303 304
   * @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 (李卓桓) 已提交
305
   * @property   {string}  scan       - A scan event will be emitted when the bot needs to show you a QR Code for scanning.
L
lijiarui 已提交
306 307 308 309 310 311
   */

  /**
   * @desc       Wechaty Class Event Function
   * @typedef    WechatyEventFunction
   * @property   {Function} error           -(this: Wechaty, error: Error) => void callback function
312 313
   * @property   {Function} login           -(this: Wechaty, user: ContactSelf)=> void
   * @property   {Function} logout          -(this: Wechaty, user: ContactSelf) => void
L
lijiarui 已提交
314 315 316 317 318 319 320 321 322 323 324 325
   * @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
326
   * @property   {Function} friend          -(this: Wechaty, request?: Friendship) => void
L
lijiarui 已提交
327 328
   * @property   {Function} message         -(this: Wechaty, message: Message) => void
   * @property   {Function} room-join       -(this: Wechaty, room: Room, inviteeList: Contact[],  inviter: Contact) => void
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
329
   * @property   {Function} room-topic      -(this: Wechaty, room: Room, newTopic: string, oldTopic: string, changer: Contact) => void
L
lijiarui 已提交
330 331 332 333 334
   * @property   {Function} room-leave      -(this: Wechaty, room: Room, leaverList: Contact[]) => void
   */

  /**
   * @listens Wechaty
335
   * @param   {WechatyEventName}      event      - Emit WechatyEvent
L
lijiarui 已提交
336 337 338
   * @param   {WechatyEventFunction}  listener   - Depends on the WechatyEvent
   * @return  {Wechaty}                          - this for chain
   *
339
   * More Example Gist: [Examples/Friend-Bot]{@link https://github.com/Chatie/wechaty/blob/master/examples/friend-bot.ts}
L
lijiarui 已提交
340 341 342 343 344 345 346
   *
   * @example <caption>Event:scan </caption>
   * wechaty.on('scan', (url: string, code: number) => {
   *   console.log(`[${code}] Scan ${url} to login.` )
   * })
   *
   * @example <caption>Event:login </caption>
347
   * bot.on('login', (user: ContactSelf) => {
L
lijiarui 已提交
348 349 350 351
   *   console.log(`user ${user} login`)
   * })
   *
   * @example <caption>Event:logout </caption>
352
   * bot.on('logout', (user: ContactSelf) => {
L
lijiarui 已提交
353 354 355 356 357 358 359 360 361
   *   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>
362 363
   * bot.on('friend', (request: Friendship) => {
   *   if(request.type === Friendship.Type.RECEIVE){ // 1. receive new friend request from new contact
364
   *     const contact = request.contact()
L
lijiarui 已提交
365 366 367 368 369 370
   *     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!`)
   *       }
371
   * 	  } else if (request.type === Friendship.Type.CONFIRM) { // 2. confirm friend ship
L
lijiarui 已提交
372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391
   *       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 (李卓桓) 已提交
392
   */
393
  public on(event: WechatyEventName, listener: string | ((...args: any[]) => any)): this {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
394
    log.verbose('Wechaty', 'on(%s, %s) registered',
395 396 397 398 399 400
                            event,
                            typeof listener === 'string'
                              ? listener
                              : typeof listener,
                )

401 402
    // DEPRECATED for 'friend' event
    if (event as any === 'friend') {
403
      log.warn('Wechaty', `on('friend', contact, friendRequest) is DEPRECATED. use on('friendship', friendship) instead`)
404 405 406
      if (typeof listener === 'function') {
        const oldListener = listener
        listener = (...args: any[]) => {
407
          log.warn('Wechaty', `on('friend', contact, friendRequest) is DEPRECATED. use on('friendship', friendship) instead`)
408 409 410 411 412
          oldListener.apply(this, args)
        }
      }
    }

413
    if (typeof listener === 'function') {
414
      this.addListenerFunction(event, listener)
415
    } else {
416
      this.addListenerModuleFile(event, listener)
417
    }
418
    return this
419 420
  }

421
  private addListenerModuleFile(event: WechatyEventName, modulePath: string): void {
422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440
    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)
      })
  }

441
  private addListenerFunction(event: WechatyEventName, listener: Function): void {
442 443 444 445 446 447 448 449 450 451 452 453
    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 (李卓桓) 已提交
454 455 456
  private initPuppet(): void {
    log.verbose('Wechaty', 'initPuppet(%s)', this.options.puppet)

457 458 459 460 461 462 463 464 465 466 467 468
    let inited = false
    try {
      inited = !!this.puppet
    } catch (e) {
      inited = false
    }

    if (inited) {
      log.verbose('Wechaty', 'initPuppet(%s) had already been inited, no need to init twice', this.options.puppet)
      return
    }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
469 470 471 472
    const puppet = this.initPuppetResolver(this.options.puppet)

    this.initPuppetVersionSatisfy(puppet)
    this.initPuppetEventBridge(puppet)
473

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
474 475
    this.initPuppetAccessory(puppet)
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
476

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
477
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
478
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
479
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
480 481
  private initPuppetVersionSatisfy(puppet: Puppet): void {
    log.verbose('Wechaty', 'initPuppetVersionSatisfy(%s)', puppet)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
482

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
483
    if (this.initPuppetSemverSatisfy(
484 485
      puppet.wechatyVersionRange(),
    )) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
486
      return
487 488
    }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
489 490 491 492
    throw new Error(`The Puppet Plugin(${puppet.constructor.name}) `
      + `requires a version range(${puppet.wechatyVersionRange()}) `
      + `that is not satisfying the Wechaty version: ${this.version()}.`,
    )
493 494 495 496 497
  }

  /**
   * Init the Puppet
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
498
  private initPuppetResolver(puppet?: PuppetName | Puppet): Puppet {
499 500
    log.verbose('Wechaty', 'initPuppetResolver(%s)', puppet)

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
501
    if (!puppet) {
502
      log.info('Wechaty', 'initPuppet() using puppet: %s', config.puppet)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
503 504 505
      puppet  = config.puppet
    }

506
    if (typeof puppet === 'string') {
507
      // tslint:disable-next-line:variable-name
508 509 510 511 512
      const MyPuppet = PUPPET_DICT[puppet]
      if (!MyPuppet) {
        throw new Error('no such puppet: ' + puppet)
      }

513
      const options: PuppetOptions = {
514
        memory : this.memory,
515
        token  : this.options.puppetToken,
516 517
      }

518
      return new MyPuppet(options)
519

520 521
    } else if (puppet instanceof Puppet) {
      return puppet
522 523
    } else {
      throw new Error('unsupported options.puppet!')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
524
    }
525
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
526

527 528 529 530
  /**
   * Plugin Version Range Check
   */
  private initPuppetSemverSatisfy(versionRange: string) {
531
    log.verbose('Wechaty', 'initPuppetSemverSatisfy(%s)', versionRange)
532
    return semver.satisfies(
533
      this.version(true),
534 535 536
      versionRange,
    )
  }
537

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
538
  protected initPuppetEventBridge(puppet: Puppet) {
539
    const eventNameList: PuppetEventName[] = Object.keys(PUPPET_EVENT_DICT) as any
540 541 542 543 544 545 546 547 548 549 550 551 552 553 554
    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

555
        case 'watchdog':
556
          puppet.removeAllListeners('heartbeat')
557 558 559 560
          puppet.on('watchdog', data => {
            /**
             * Use `watchdog` event from Puppet to `heartbeat` Wechaty.
             */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
561
            // TODO: use a throttle queue to prevent beat too fast.
562 563 564 565 566 567
            this.emit('heartbeat', data)
          })
          break

        case 'start':
        case 'stop':
568 569 570
          // do not emit 'start'/'stop' again for wechaty:
          // because both puppet & wechaty should have their own
          // `start`/`stop` event seprately
571 572
          break

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
573 574 575 576 577 578 579 580 581 582 583 584 585 586
        // case 'start':
        //   puppet.removeAllListeners('start')
        //   puppet.on('start', () => {
        //     this.emit('start')
        //   } )
        //   break

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

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
587 588 589 590 591 592 593 594 595 596 597
        case 'friendship':
          puppet.removeAllListeners('friendship')
          puppet.on('friendship', async friendshipId => {
            const friendship = this.Friendship.load(friendshipId)
            await friendship.ready()
            this.emit('friendship', friendship)
            friendship.contact().emit('friendship', friendship)

            // support deprecated event name: friend.
            // Huan LI 201806
            this.emit('friend' as any, friendship as any)
598 599 600 601 602 603
          })
          break

        case 'login':
          puppet.removeAllListeners('login')
          puppet.on('login', async contactId => {
604
            const contact = this.ContactSelf.load(contactId)
605 606 607 608 609 610 611 612
            await contact.ready()
            this.emit('login', contact)
          })
          break

        case 'logout':
          puppet.removeAllListeners('logout')
          puppet.on('logout', async contactId => {
613
            const contact = this.ContactSelf.load(contactId)
614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640
            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)
641
            room.emit('join', inviteeList, inviter)
642 643 644 645 646
          })
          break

        case 'room-leave':
          puppet.removeAllListeners('room-leave')
647
          puppet.on('room-leave', async (roomId, leaverIdList, removerId) => {
648 649 650 651 652 653
            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()))

654 655 656 657 658 659 660
            let remover: undefined | Contact = undefined
            if (removerId) {
              remover = this.Contact.load(removerId)
              await remover.ready()
            }

            this.emit('room-leave', room, leaverList, remover)
661
            room.emit('leave', leaverList, remover)
662 663 664 665 666
          })
          break

        case 'room-topic':
          puppet.removeAllListeners('room-topic')
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
667
          puppet.on('room-topic', async (roomId, newTopic, oldTopic, changerId) => {
668 669 670 671 672 673
            const room = this.Room.load(roomId)
            await room.ready()

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

Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
674 675
            this.emit('room-topic', room, newTopic, oldTopic, changer)
            room.emit('topic', newTopic, oldTopic, changer)
676 677 678 679 680
          })
          break

        case 'scan':
          puppet.removeAllListeners('scan')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
681 682
          puppet.on('scan', async (qrcode, status, data) => {
            this.emit('scan', qrcode, status, data)
683 684 685
          })
          break

686 687 688
        case 'watchdog':
          break

689 690 691 692
        default:
          throw new Error('eventName ' + eventName + 'unsupported!')

      }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
693
    }
694
  }
695

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
696
  protected initPuppetAccessory(puppet: Puppet) {
697
    log.verbose('Wechaty', 'initAccessory(%s)', puppet)
698

699
    /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
700
     * 1. Set Wechaty
701
     */
702 703 704 705 706
    this.Contact.wechaty     = this
    this.ContactSelf.wechaty = this
    this.Friendship.wechaty  = this
    this.Message.wechaty     = this
    this.Room.wechaty        = this
707

708
    /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
709
     * 2. Set Puppet
710
     */
711 712 713 714 715
    this.Contact.puppet     = puppet
    this.ContactSelf.puppet = puppet
    this.Friendship.puppet  = puppet
    this.Message.puppet     = puppet
    this.Room.puppet        = puppet
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
716 717

    this.puppet               = puppet
718 719
  }

720 721 722 723 724 725 726 727 728
  /**
   * Start the bot, return Promise.
   *
   * @returns {Promise<void>}
   * @example
   * await bot.start()
   * // do other stuff with bot here
   */
  public async start(): Promise<void> {
729
    log.info('Wechaty', 'start() v%s is starting...' , this.version())
730 731
    log.verbose('Wechaty', 'puppet: %s'   , this.options.puppet)
    log.verbose('Wechaty', 'profile: %s'  , this.options.profile)
732
    log.verbose('Wechaty', 'id: %s'       , this.id)
733 734 735

    if (this.state.on()) {
      log.silly('Wechaty', 'start() on a starting/started instance')
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
736
      await this.state.ready('on')
737 738 739 740
      log.silly('Wechaty', 'start() state.ready() resolved')
      return
    }

741 742 743 744
    if (this.lifeTimer) {
      throw new Error('start() lifeTimer exist')
    }

745 746 747
    this.state.on('pending')

    try {
748
      await this.memory.load()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
749 750

      this.initPuppet()
751 752
      await this.puppet.start()

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
753 754 755 756 757 758 759 760
      if (this.options.ioToken) {
        this.io = new Io({
          token   : this.options.ioToken,
          wechaty : this,
        })
        await this.io.start()
      }

761
    } catch (e) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
762
      console.error(e)
763 764
      log.error('Wechaty', 'start() exception: %s', e && e.message)
      Raven.captureException(e)
765 766 767 768 769 770 771 772 773 774 775
      this.emit('error', e)

      try {
        await this.stop()
      } catch (e) {
        log.error('Wechaty', 'start() stop() exception: %s', e && e.message)
        Raven.captureException(e)
        this.emit('error', e)
      } finally {
        return
      }
776 777 778 779
    }

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

780 781 782 783
    this.lifeTimer = setInterval(() => {
      log.silly('Wechaty', 'start() setInterval() this timer is to keep Wechaty running...')
    }, 1000 * 60 * 60)

784 785 786 787 788 789
    this.state.on(true)
    this.emit('start')

    return
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
790 791 792 793 794 795 796 797
  /**
   * Stop the bot
   *
   * @returns {Promise<void>}
   * @example
   * await bot.stop()
   */
  public async stop(): Promise<void> {
798
    log.info('Wechaty', 'stop() v%s is stoping ...' , this.version())
799

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
800
    if (this.state.off()) {
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
801 802 803
      log.silly('Wechaty', 'stop() on an stopping/stopped instance')
      await this.state.ready('off')
      log.silly('Wechaty', 'stop() state.ready(off) resolved')
804
      return
805
    }
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
806

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
807
    this.state.off('pending')
808
    await this.memory.save()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
809

810 811 812 813 814
    if (this.lifeTimer) {
      clearInterval(this.lifeTimer)
      this.lifeTimer = undefined
    }

815
    try {
816
      await this.puppet.stop()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
817 818 819 820 821 822

      if (this.io) {
        await this.io.stop()
        this.io = undefined
      }

823
    } catch (e) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
824
      log.error('Wechaty', 'stop() exception: %s', e.message)
825
      Raven.captureException(e)
826
      this.emit('error', e)
827
    }
828 829 830 831 832 833 834 835 836 837

    this.state.off(true)
    this.emit('stop')

    /**
     * MUST use setImmediate at here(the end of this function),
     * because we need to run the micro task registered by the `emit` method
     */
    setImmediate(() => this.puppet.removeAllListeners())

838
    return
839
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
840

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
841
  /**
L
lijiarui 已提交
842 843 844 845 846
   * Logout the bot
   *
   * @returns {Promise<void>}
   * @example
   * await bot.logout()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
847
   */
848
  public async logout(): Promise<void>  {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
849 850
    log.verbose('Wechaty', 'logout()')

851 852 853 854 855 856 857
    try {
      await this.puppet.logout()
    } catch (e) {
      log.error('Wechaty', 'logout() exception: %s', e.message)
      Raven.captureException(e)
      throw e
    }
858
    return
859
  }
860

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
861 862 863 864 865
  /**
   * Get the logon / logoff state
   *
   * @returns {boolean}
   * @example
866
   * if (bot.logonoff()) {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
867 868 869 870 871
   *   console.log('Bot logined')
   * } else {
   *   console.log('Bot not logined')
   * }
   */
872
  public logonoff(): Boolean {
873
    return this.puppet.logonoff()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
874 875
  }

876 877 878 879
  /**
   * @deprecated
   */
  public self(): Contact {
880
    log.warn('Wechaty', 'self() DEPRECATED. use userSelf() instead.')
881 882 883
    return this.userSelf()
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
884
  /**
L
lijiarui 已提交
885 886 887 888
   * Get current user
   *
   * @returns {Contact}
   * @example
889
   * const contact = bot.userSelf()
L
lijiarui 已提交
890
   * console.log(`Bot is ${contact.name()}`)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
891
   */
892
  public userSelf(): Contact {
893
    const userId = this.puppet.selfId()
894
    const user = this.ContactSelf.load(userId)
895
    return user
896
  }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
897

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
898
  /**
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
899
   * Send message to userSelf
L
lijiarui 已提交
900
   *
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
901
   * @param {string} textOrContactOrFile
L
lijiarui 已提交
902
   * @returns {Promise<boolean>}
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
903
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
904 905 906 907 908 909 910 911 912 913 914 915 916
  public async say(textOrContactOrFile: string | Contact | FileBox): Promise<void> {
    log.verbose('Wechaty', 'say(%s)', textOrContactOrFile)

    // Make Typescript Happy:
    if (typeof textOrContactOrFile === 'string') {
      await this.userSelf().say(textOrContactOrFile)
    } else if (textOrContactOrFile instanceof Contact) {
      await this.userSelf().say(textOrContactOrFile)
    } else if (textOrContactOrFile instanceof FileBox) {
      await this.userSelf().say(textOrContactOrFile)
    } else {
      throw new Error('unsupported')
    }
917 918
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
919
  /**
L
lijiarui 已提交
920
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
921
   */
922
  public static async sleep(millisecond: number): Promise<void> {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
923
    await new Promise(resolve => {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
924 925 926 927
      setTimeout(resolve, millisecond)
    })
  }

Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
928
  /**
Huan (李卓桓)'s avatar
jsdoc  
Huan (李卓桓) 已提交
929
   * @private
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
930
   */
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
931 932 933
  public ding(data?: string): void {
    log.silly('Wechaty', 'ding(%s)', data || '')

934
    try {
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
935
      this.puppet.ding(data)
936 937 938
    } catch (e) {
      log.error('Wechaty', 'ding() exception: %s', e.message)
      Raven.captureException(e)
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
939
      throw e
940
    }
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
941
  }
942 943 944 945 946 947 948 949 950 951 952 953 954 955 956

  /**
   * @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 (李卓桓) 已提交
957 958 959 960 961

  /**
   * @private
   */
  public async reset(reason?: string): Promise<void> {
962
    log.verbose('Wechaty', 'reset() because %s', reason || 'no reason')
Huan (李卓桓)'s avatar
wip...  
Huan (李卓桓) 已提交
963 964
    await this.puppet.stop()
    await this.puppet.start()
Huan (李卓桓)'s avatar
Huan (李卓桓) 已提交
965 966 967
    return
  }

968 969 970
  public unref(): void {
    log.warn('Wechaty', 'unref() To Be Implemented. See: https://github.com/Chatie/wechaty/issues/1197')
  }
971
}
972 973

export default Wechaty