twitbot-cli/src/index.js

import clor from 'clor'
import loudRejection from 'loud-rejection'
import TwitBot from 'twitbot-core'
import * as util from 'twitbot-util'
import debug from 'debug'

const log = debug('twitbot:cli')
loudRejection()

export default function (cmd, extra, version) {
	switch (cmd) {
		case 'profile' :
			(async() => {
				try {
					const answers = await util.prompt('config')
					const result = await util.settingsSave(`users:${answers.username}`, answers)
					console.log(`  ${clor.green(result)}`)
				} catch (err) {
					log(err)
				}
			})()
			break
		case 'search':
			(async() => {
				if (util.checkUser()) {
					try {
						const answers = await util.prompt('search')
						const confd = util.getUser(answers.select_account)
						const T = new TwitBot(confd)
						const msg = util.spinnerMsg(`Twitbot working.. `)
						const twetList = await T.extra.fullSearch({q: answers.keyword, takip_sayi: answers.takip_sayi, lang: confd.lang})
						let favoriteList = null
						let userList = null
						const blocks = await T.extra.fullBlocks()

						const checkBlock = util.notActionBlocks(blocks)
						const checkName = util.notActionHimself(answers.select_account)

						const check = twetList.filter(twet => checkName(twet) !== false).filter(twet => checkBlock(twet) !== false)

						userList = check.map(twet => twet.user.id_str)
						favoriteList = check.map(twet => twet.id_str)

						if (answers.takip === 'Yes') {
							userList = await T.extra.fullUserFollow(userList)
							userList = (userList.length - userList.filter(item => item.errors).length)
						}

						if (answers.favorite === 'Yes') {
							favoriteList	= await T.extra.fullFavoriteList(favoriteList)
							favoriteList = (favoriteList.length - favoriteList.filter(item => item.errors).length)
						}
						clearInterval(msg)
						console.log(`  ${clor.green('Transactions completed.')}`)
						if (answers.takip === 'Yes') {
							console.log(`  ${clor.yellow(`${userList}  people were followed.`)}`) // check userList error
						}
						if (answers.favorite === 'Yes') {
							console.log(`  ${clor.yellow(`${favoriteList}  tweets favorite added.`)}`)  // check favoriteList error
						}
					} catch (err) {
						log(err)
					}
				} else {
					console.log(`${clor.red('First, you must define an account  [ > twitbot new]')}`)
				}
			})()
			break
		case 'unfavorite' :
			(async() => {
				if (util.checkUser()) {
					try {
						const {select_account} = await util.prompt('accountlist')
						const confd = util.getUser(select_account)
						const T = new TwitBot(confd)
						const msg = util.spinnerMsg(`Twitbot working.. `)
						const favoriteList = await T.extra.fullFavorites()
						await T.extra.fullDestoryFavorite(favoriteList.map(item => item.id_str))
						clearInterval(msg)
						console.log(`  ${clor.green('All favorites removed')}`)
					} catch (err) {
						log(err)
					}
				} else {
					console.log(`${clor.red('First, you must define an account  [ > twitbot new]')}`)
				}
			})()
			break
		case 'flush' :
			(async() => {
				if (util.checkUser()) {
					const users = util.getUsers()
					let answers = null
					try {
						if (extra.noPrompt) {
							answers = extra
						} else {
							answers = await util.prompt('accountlist')
						}
						delete users[answers.select_account]
						const msg = await util.settingsSave(`users`, users)
						console.log(`  ${clor.green(msg)}`)
					} catch (err) {
						log(err)
					}
				} else {
					console.log(`${clor.red('First, you must define an account  [ > twitbot new]')}`)
				}
			})()
			break
		case 'message':
			(async() => {
				if (util.checkUser()) {
					try {
						const answers = await util.prompt('messagelist')
						const confd = util.getUser(answers.select_account)
						const T = new TwitBot(confd)
						const msg = util.spinnerMsg(`Twitbot working.. `)
						if (answers.send === 'Yes' && answers.message) {
							const followersList = await T.extra.fullFollowers()
							let list = followersList.ids
							if (answers.limit) {
								list = list.slice(0, answers.limit)
							}
							const msgCount = await T.extra.fullUserMessage(list, answers.message).length
							clearInterval(msg)
							console.log(`  ${clor.green(`Persons posted ${msgCount} messages `)}`)
						} else {
							console.log(`${clor.red('Come back when he decided to take a message..')}`)
						}
					} catch (err) {
						log(err)
					}
				} else {
					console.log(`${clor.red('First, you must define an account  [ > twitbot new]')}`)
				}
			})()
			break
		case 'unfollow' :
			(async() => {
				if (util.checkUser()) {
					try {
						const answers = await util.prompt('unfollowlist')
						const confd = util.getUser(answers.select_account)
						const T = new TwitBot(confd)
						if (answers.type === 'Yes') {
							const msg = util.spinnerMsg(`Twitbot working.. `)
							const fullListids = await T.extra.fullFollowers()
							const result = await T.extra.fullUserDestroy(fullListids)
							clearInterval(msg)
							console.log(`${clor.red(`${result.length} users unfollowed..`)}`)
						}
					} catch (err) {
						log(err)
					}
				} else {
					console.log(`${clor.red('First, you must define an account  [ > twitbot new]')}`)
				}
			})()
			break
		case 'watch' :
			(async() => {
				if (util.checkUser()) {
					try {
						let query = null
						let confd = null
						let takip = null
						let favorite = null
						let username = null
						const actionList = []
						const actionBlacklist = []
						if (extra.noPrompt || extra.username || extra.query) {
							confd = util.getUser(extra.username)
							query = extra.query
							username = extra.username
							if (extra.takip) {
								takip = extra.takip
							}
							if (extra.favorite) {
								favorite = extra.favorite
							}
						} else {
							const answers = await util.prompt('watch')
							confd = util.getUser(answers.select_account)
							query = answers.keywords
							username = answers.select_account
							takip = answers.takip
							favorite = answers.takip
						}

						if (extra.src) {
							const extraMiddleware = require(extra.src)
							if (extraMiddleware.use) {
								actionList.push(extraMiddleware.use())
							}

							if (extraMiddleware.blacklist) {
								actionBlacklist.push(extraMiddleware.blacklist())
							}
						}

						if (takip !== null && takip === 'Yes') {
							actionList.push(util.actionUserFollow())
						}

						if (favorite !== null && favorite === 'Yes') {
							actionList.push(util.actionFavorite())
						}

						const T = new TwitBot(confd)

						const keywords = query

						const blocks = await T.extra.fullBlocks()
						actionBlacklist.push(util.notActionHimself(username))
						actionBlacklist.push(util.notActionBlocks(blocks))

						const stream = T.tweetStream({track: keywords, language: confd.lang})
						const msg = util.spinnerMsg(`Twitbot working.. `)
						stream.on('tweet', twet => {
							util.control(twet, actionBlacklist).then(result => {
								if (result.indexOf(false) === -1) {
									util.action(twet, extra, T, actionList)
								}
							}).catch(err => {
								// Houston we have a problem
								log(err)
							})
						})

						stream.on('disconnect', disconnectMessage => {
							clearInterval(msg)
							console.log(disconnectMessage)
							stream.start()
						})
					} catch (err) {
						log(err)
					}
				} else {
					console.log(`${clor.red('First, you must define an account  [ > twitbot new]')}`)
				}
			})()
			break
		case 'use' :
			(async() => {
				if (util.checkUser()) {
					try {
						let answers = null
						if (extra.src && extra.account) {
							answers = extra
							const args = extra.filter(arg => ['src', 'account'].indexOf(arg) === -1)
							const confd = util.getUser(answers.account)
							const T = new TwitBot(confd)

							const extraMiddleware = require(extra.src)

							if (extraMiddleware.use) {
								let middleware = extraMiddleware.use()
								middleware = middleware.call(T, args)
							}
						} else {
							console.log(`${clor.red(' --src=~/hello  --account=johndoe ')}`)
						}
					} catch (err) {
						log(err)
					}
				} else {
					console.log(`${clor.red('First, you must define an account  [ > twitbot new]')}`)
				}
			})()
			break
		case 'version':
			console.log(version)
			break
		case 'help':
			console.log(util.help())
			break
		default:
			console.log(util.help())
	}
}