From 96b6ef4d9ba73dc04eb601ffedee52d6b6ab580a Mon Sep 17 00:00:00 2001 From: syuilo Date: Sat, 7 Oct 2017 18:30:04 +0900 Subject: [PATCH] :v: --- src/api/bot/core.ts | 72 ++++++++++++++-------- src/api/bot/interfaces/line.ts | 105 +++++++++++++++++++++++---------- 2 files changed, 122 insertions(+), 55 deletions(-) diff --git a/src/api/bot/core.ts b/src/api/bot/core.ts index 4970f24469..4e168a6054 100644 --- a/src/api/bot/core.ts +++ b/src/api/bot/core.ts @@ -43,18 +43,26 @@ export default class BotCore extends EventEmitter { }; } - public static import(data) { - const core = new BotCore(); - core.user = data.user ? initUser(data.user) : null; - core.setContext(data.context ? Context.import(core, data.context) : null); - return core; + protected _import(data) { + this.user = data.user ? initUser(data.user) : null; + this.setContext(data.context ? Context.import(this, data.context) : null); } - public async q(query: string): Promise { + public static import(data) { + const bot = new BotCore(); + bot._import(data); + return bot; + } + + public async q(query: string): Promise { if (this.context != null) { return await this.context.q(query); } + if (/^@[a-zA-Z0-9-]+$/.test(query)) { + return await this.showUserCommand(query); + } + switch (query) { case 'ping': return 'PONG'; @@ -67,7 +75,8 @@ export default class BotCore extends EventEmitter { 'login, signin: サインインします\n' + 'logout, signout: サインアウトします\n' + 'post: 投稿します\n' + - 'tl: タイムラインを見ます\n'; + 'tl: タイムラインを見ます\n' + + '@<ユーザー名>: ユーザーを表示します'; case 'me': return this.user ? `${this.user.name}としてサインインしています。\n\n${getUserSummary(this.user)}` : 'サインインしていません'; @@ -76,6 +85,7 @@ export default class BotCore extends EventEmitter { case 'signin': case 'ログイン': case 'サインイン': + if (this.user != null) return '既にサインインしていますよ!'; this.setContext(new SigninContext(this)); return await this.context.greet(); @@ -95,9 +105,9 @@ export default class BotCore extends EventEmitter { case 'tl': case 'タイムライン': - return await this.getTl(); + return await this.tlCommand(); - default: + default: return '?'; } } @@ -115,7 +125,7 @@ export default class BotCore extends EventEmitter { this.emit('updated'); } - public async getTl() { + public async tlCommand(): Promise { if (this.user == null) return 'まずサインインしてください。'; const tl = await require('../endpoints/posts/timeline')({ @@ -128,23 +138,37 @@ export default class BotCore extends EventEmitter { return text; } + + public async showUserCommand(q: string): Promise { + try { + const user = await require('../endpoints/users/show')({ + username: q.substr(1) + }, this.user); + + const text = getUserSummary(user); + + return text; + } catch (e) { + return `問題が発生したようです...: ${e}`; + } + } } abstract class Context extends EventEmitter { - protected core: BotCore; + protected bot: BotCore; public abstract async greet(): Promise; public abstract async q(query: string): Promise; public abstract export(): any; - constructor(core: BotCore) { + constructor(bot: BotCore) { super(); - this.core = core; + this.bot = bot; } - public static import(core: BotCore, data: any) { - if (data.type == 'post') return PostContext.import(core, data.content); - if (data.type == 'signin') return SigninContext.import(core, data.content); + public static import(bot: BotCore, data: any) { + if (data.type == 'post') return PostContext.import(bot, data.content); + if (data.type == 'signin') return SigninContext.import(bot, data.content); return null; } } @@ -179,8 +203,8 @@ class SigninContext extends Context { const same = bcrypt.compareSync(query, this.temporaryUser.password); if (same) { - this.core.signin(this.temporaryUser); - this.core.clearContext(); + this.bot.signin(this.temporaryUser); + this.bot.clearContext(); return `${this.temporaryUser.name}さん、おかえりなさい!`; } else { return `パスワードが違います... もう一度教えてください:`; @@ -197,8 +221,8 @@ class SigninContext extends Context { }; } - public static import(core: BotCore, data: any) { - const context = new SigninContext(core); + public static import(bot: BotCore, data: any) { + const context = new SigninContext(bot); context.temporaryUser = data.temporaryUser; return context; } @@ -212,8 +236,8 @@ class PostContext extends Context { public async q(query: string): Promise { await require('../endpoints/posts/create')({ text: query - }, this.core.user); - this.core.clearContext(); + }, this.bot.user); + this.bot.clearContext(); return '投稿しましたよ!'; } @@ -223,8 +247,8 @@ class PostContext extends Context { }; } - public static import(core: BotCore, data: any) { - const context = new PostContext(core); + public static import(bot: BotCore, data: any) { + const context = new PostContext(bot); return context; } } diff --git a/src/api/bot/interfaces/line.ts b/src/api/bot/interfaces/line.ts index 437f29cb3c..03dc2a85ba 100644 --- a/src/api/bot/interfaces/line.ts +++ b/src/api/bot/interfaces/line.ts @@ -10,20 +10,83 @@ import prominence = require('prominence'); const redis = prominence(_redis); +class LineBot extends BotCore { + private replyToken: string; + + private reply(messages: any[]) { + request.post({ + url: 'https://api.line.me/v2/bot/message/reply', + headers: { + 'Authorization': `Bearer ${config.line_bot.channel_access_token}` + }, + json: { + replyToken: this.replyToken, + messages: messages + } + }, (err, res, body) => { + if (err) { + console.error(err); + return; + } + }); + } + + public async react(ev: any): Promise { + // テキスト以外(スタンプなど)は無視 + if (ev.message.type !== 'text') return; + + const res = await this.q(ev.message.text); + + if (res == null) return; + + // 返信 + this.reply([{ + type: 'text', + text: res + }]); + } + + public static import(data) { + const bot = new LineBot(); + bot._import(data); + return bot; + } + + public async showUserCommand(q: string) { + const user = await require('../endpoints/users/show')({ + username: q.substr(1) + }, this.user); + + this.reply([{ + type: 'template', + altText: await super.showUserCommand(q), + template: { + type: 'buttons', + thumbnailImageUrl: `${user.avatar_url}?thumbnail&size=1024`, + title: `${user.name} (@${user.username})`, + text: user.description || '(no description)', + actions: [{ + type: 'uri', + label: 'Webで見る', + uri: `${config.url}/${user.username}` + }] + } + }]); + } +} + module.exports = async (app: express.Application) => { if (config.line_bot == null) return; const handler = new EventEmitter(); handler.on('message', async (ev) => { - // テキスト以外(スタンプなど)は無視 - if (ev.message.type !== 'text') return; const sourceId = ev.source.userId; const sessionId = `line-bot-sessions:${sourceId}`; const _session = await redis.get(sessionId); - let session: BotCore; + let bot: LineBot; if (_session == null) { const user = await User.findOne({ @@ -32,9 +95,9 @@ module.exports = async (app: express.Application) => { } }); - session = new BotCore(user); + bot = new LineBot(user); - session.on('signin', user => { + bot.on('signin', user => { User.update(user._id, { $set: { line: { @@ -44,7 +107,7 @@ module.exports = async (app: express.Application) => { }); }); - session.on('signout', user => { + bot.on('signout', user => { User.update(user._id, { $set: { line: { @@ -54,36 +117,16 @@ module.exports = async (app: express.Application) => { }); }); - redis.set(sessionId, JSON.stringify(session.export())); + redis.set(sessionId, JSON.stringify(bot.export())); } else { - session = BotCore.import(JSON.parse(_session)); + bot = LineBot.import(JSON.parse(_session)); } - session.on('updated', () => { - redis.set(sessionId, JSON.stringify(session.export())); + bot.on('updated', () => { + redis.set(sessionId, JSON.stringify(bot.export())); }); - const res = await session.q(ev.message.text); - - // 返信 - request.post({ - url: 'https://api.line.me/v2/bot/message/reply', - headers: { - 'Authorization': `Bearer ${config.line_bot.channel_access_token}` - }, - json: { - replyToken: ev.replyToken, - messages: [{ - type: 'text', - text: res - }] - } - }, (err, res, body) => { - if (err) { - console.error(err); - return; - } - }); + bot.react(ev); }); app.post('/hooks/line', (req, res, next) => {