getting-started/examples/advanced/busy-bot.js

#!/usr/bin/env node
/**
 *   Wechaty - https://github.com/chatie/wechaty
 *
 *   @copyright 2016-2018 Huan LI <zixia@zixia.net>
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 */
import qrTerm from 'qrcode-terminal'

import {
  IoClient,
  Wechaty,
  config,
  log,
}             from 'wechaty'

console.log(`
=============== Powered by Wechaty ===============
-------- https://github.com/Chatie/wechaty --------

I'm the BUSY BOT, I can do auto response message for you when you are BUSY.

Send command to FileHelper to:

1. '#busy' - set busy mode ON
2. '#busy I'm busy' - set busy mode ON and set a Auto Reply Message
3. '#free' - set busy mode OFF
4. '#status' - check the current Busy Mode and Auto Reply Message.

Loading... please wait for QrCode Image Url and then scan to login.
`)

let bot

const token = config.token

if (token) {
  log.info('Wechaty', 'TOKEN: %s', token)

  bot = Wechaty.instance({ profile: token })
  const ioClient = new IoClient({
    token,
    wechaty: bot,
  })

  ioClient.start().catch(e => {
    log.error('Wechaty', 'IoClient.init() exception: %s', e)
    bot.emit('error', e)
  })
} else {
  log.verbose('Wechaty', 'TOKEN: N/A')
  bot = Wechaty.instance()
}

bot
.on('scan', (qrcode, status) => {
  qrTerm.generate(qrcode, { small: true })
  console.log(`${status}: ${qrcode} - Scan QR Code of the url to login:`)
})
.on('logout'	, user => log.info('Bot', `${user.name()} logouted`))
.on('error'   , e => log.info('Bot', 'error: %s', e))

.on('login', async function(user) {
  const msg = `${user.name()} logined`

  log.info('Bot', msg)
  await this.say(msg)

})

/**
 * Global Event: message
 */

let busyIndicator    = false
let busyAnnouncement = `Automatic Reply: I cannot read your message because I'm busy now, will talk to you when I get back.`

bot.on('message', async function(msg) {
  log.info('Bot', '(message) %s', msg)

  const filehelper = bot.Contact.load('filehelper')

  const sender   = msg.from()
  const receiver = msg.to()
  const text     = msg.text()
  const room     = msg.room()

  // if (msg.age() > 60) {
  //   log.info('Bot', 'on(message) skip age(%d) > 60 seconds: %s', msg.age(), msg)
  //   return
  // }

  if (!sender || !receiver) {
    return
  }

  if (receiver.id === 'filehelper') {
    if (text === '#status') {
      await filehelper.say('in busy mode: ' + busyIndicator)
      await filehelper.say('auto reply: ' + busyAnnouncement)

    } else if (text === '#free') {
      busyIndicator = false
      await filehelper.say('auto reply stopped.')

    } else if (/^#busy/i.test(text)) {

      busyIndicator = true
      await filehelper.say('in busy mode: ' + 'ON')

      const matches = text.match(/^#busy (.+)$/i)
      if (!matches || !matches[1]) {
        await filehelper.say('auto reply message: "' + busyAnnouncement + '"')

      } else {
        busyAnnouncement = matches[1]
        await filehelper.say('set auto reply to: "' + busyAnnouncement + '"')

      }
    }

    return
  }

  if (sender.type() !== bot.Contact.Type.Personal) {
    return
  }

  if (!busyIndicator) {
    return  // free
  }

  if (msg.self()) {
    return
  }

  /**
   * 1. Send busy anoncement to contact
   */
  if (!room) {
    await msg.say(busyAnnouncement)
    return
  }

  /**
   * 2. If there's someone mentioned me in a room,
   *  then send busy annoncement to room and mention the contact who mentioned me.
   */
  const contactList = await msg.mention()
  const contactIdList = contactList.map(c => c.id)
  if (contactIdList.includes(this.userSelf().id)) {
    await msg.say(busyAnnouncement, sender)
  }

})

bot.start()
.catch(e => console.error(e))