const { NlpManager, ConversationContext } = require('node-nlp')
const readline = require('readline')
const ri = readline.createInterface({ input: process.stdin, output: process.stdout, terminal: false })

const languages = ['en']
const m = new NlpManager({ languages })

m.addNamedEntityText('action', 'kick', languages, ['kick', 'to kick'])
m.addNamedEntityText('action', 'ban', languages, ['ban', 'to ban'])
m.addRegexEntity('member', languages, /@[0-9]+/g)

m.addDocument('en', 'I want to %action% %member%', 'member.act')
m.addDocument('en', 'I wanna %action% %member%', 'member.act')
m.addDocument('en', '%action% %member%', 'member.act')

m.addAnswer('en', 'member.act', 'confirm {{action}} on {{member}}?')

m.slotManager.addSlot('member.act', 'member', true, {
  en: 'who should I {{action}}?',
})
m.slotManager.addSlot('member.act', 'action', true, {
  en: 'what shoud I do to {{member}}?',
})

async function main() {
  await m.train()
  m.save()
  console.log('you may type now')
  const context = new ConversationContext()
  ri.on('line', async (line) => {
    if (line == 'quit') process.exit()
    console.log('> ' + line)
    try {
      const response = await m.process(null, line, context)
      console.log('BOT:', JSON.stringify(response, null, 2))
    } catch (err) {
      console.error(err)
      process.exit()
    }
  })
}

main()