summaryrefslogtreecommitdiff
path: root/src/api/bot
diff options
context:
space:
mode:
authorAkihiko Odaki <nekomanma@pixiv.co.jp>2018-03-29 01:20:40 +0900
committerAkihiko Odaki <nekomanma@pixiv.co.jp>2018-03-29 01:54:41 +0900
commit90f8fe7e538bb7e52d2558152a0390e693f39b11 (patch)
tree0f830887053c8f352b1cd0c13ca715fd14c1f030 /src/api/bot
parentImplement remote account resolution (diff)
downloadsharkey-90f8fe7e538bb7e52d2558152a0390e693f39b11.tar.gz
sharkey-90f8fe7e538bb7e52d2558152a0390e693f39b11.tar.bz2
sharkey-90f8fe7e538bb7e52d2558152a0390e693f39b11.zip
Introduce processor
Diffstat (limited to 'src/api/bot')
-rw-r--r--src/api/bot/core.ts438
-rw-r--r--src/api/bot/interfaces/line.ts238
2 files changed, 0 insertions, 676 deletions
diff --git a/src/api/bot/core.ts b/src/api/bot/core.ts
deleted file mode 100644
index 77a68aaee6..0000000000
--- a/src/api/bot/core.ts
+++ /dev/null
@@ -1,438 +0,0 @@
-import * as EventEmitter from 'events';
-import * as bcrypt from 'bcryptjs';
-
-import User, { ILocalAccount, IUser, init as initUser } from '../models/user';
-
-import getPostSummary from '../../common/get-post-summary';
-import getUserSummary from '../../common/user/get-summary';
-import parseAcct from '../../common/user/parse-acct';
-import getNotificationSummary from '../../common/get-notification-summary';
-
-const hmm = [
- '?',
- 'ふぅ~む...?',
- 'ちょっと何言ってるかわからないです',
- '「ヘルプ」と言うと利用可能な操作が確認できますよ'
-];
-
-/**
- * Botの頭脳
- */
-export default class BotCore extends EventEmitter {
- public user: IUser = null;
-
- private context: Context = null;
-
- constructor(user?: IUser) {
- super();
-
- this.user = user;
- }
-
- public clearContext() {
- this.setContext(null);
- }
-
- public setContext(context: Context) {
- this.context = context;
- this.emit('updated');
-
- if (context) {
- context.on('updated', () => {
- this.emit('updated');
- });
- }
- }
-
- public export() {
- return {
- user: this.user,
- context: this.context ? this.context.export() : null
- };
- }
-
- protected _import(data) {
- this.user = data.user ? initUser(data.user) : null;
- this.setContext(data.context ? Context.import(this, data.context) : null);
- }
-
- public static import(data) {
- const bot = new BotCore();
- bot._import(data);
- return bot;
- }
-
- public async q(query: string): Promise<string> {
- 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';
-
- case 'help':
- case 'ヘルプ':
- return '利用可能なコマンド一覧です:\n' +
- 'help: これです\n' +
- 'me: アカウント情報を見ます\n' +
- 'login, signin: サインインします\n' +
- 'logout, signout: サインアウトします\n' +
- 'post: 投稿します\n' +
- 'tl: タイムラインを見ます\n' +
- 'no: 通知を見ます\n' +
- '@<ユーザー名>: ユーザーを表示します\n' +
- '\n' +
- 'タイムラインや通知を見た後、「次」というとさらに遡ることができます。';
-
- case 'me':
- return this.user ? `${this.user.name}としてサインインしています。\n\n${getUserSummary(this.user)}` : 'サインインしていません';
-
- case 'login':
- case 'signin':
- case 'ログイン':
- case 'サインイン':
- if (this.user != null) return '既にサインインしていますよ!';
- this.setContext(new SigninContext(this));
- return await this.context.greet();
-
- case 'logout':
- case 'signout':
- case 'ログアウト':
- case 'サインアウト':
- if (this.user == null) return '今はサインインしてないですよ!';
- this.signout();
- return 'ご利用ありがとうございました <3';
-
- case 'post':
- case '投稿':
- if (this.user == null) return 'まずサインインしてください。';
- this.setContext(new PostContext(this));
- return await this.context.greet();
-
- case 'tl':
- case 'タイムライン':
- if (this.user == null) return 'まずサインインしてください。';
- this.setContext(new TlContext(this));
- return await this.context.greet();
-
- case 'no':
- case 'notifications':
- case '通知':
- if (this.user == null) return 'まずサインインしてください。';
- this.setContext(new NotificationsContext(this));
- return await this.context.greet();
-
- case 'guessing-game':
- case '数当てゲーム':
- this.setContext(new GuessingGameContext(this));
- return await this.context.greet();
-
- default:
- return hmm[Math.floor(Math.random() * hmm.length)];
- }
- }
-
- public signin(user: IUser) {
- this.user = user;
- this.emit('signin', user);
- this.emit('updated');
- }
-
- public signout() {
- const user = this.user;
- this.user = null;
- this.emit('signout', user);
- this.emit('updated');
- }
-
- public async refreshUser() {
- this.user = await User.findOne({
- _id: this.user._id
- }, {
- fields: {
- data: false
- }
- });
-
- this.emit('updated');
- }
-
- public async showUserCommand(q: string): Promise<string> {
- try {
- const user = await require('../endpoints/users/show')(parseAcct(q.substr(1)), this.user);
-
- const text = getUserSummary(user);
-
- return text;
- } catch (e) {
- return `問題が発生したようです...: ${e}`;
- }
- }
-}
-
-abstract class Context extends EventEmitter {
- protected bot: BotCore;
-
- public abstract async greet(): Promise<string>;
- public abstract async q(query: string): Promise<string>;
- public abstract export(): any;
-
- constructor(bot: BotCore) {
- super();
- this.bot = bot;
- }
-
- public static import(bot: BotCore, data: any) {
- if (data.type == 'guessing-game') return GuessingGameContext.import(bot, data.content);
- if (data.type == 'post') return PostContext.import(bot, data.content);
- if (data.type == 'tl') return TlContext.import(bot, data.content);
- if (data.type == 'notifications') return NotificationsContext.import(bot, data.content);
- if (data.type == 'signin') return SigninContext.import(bot, data.content);
- return null;
- }
-}
-
-class SigninContext extends Context {
- private temporaryUser: IUser = null;
-
- public async greet(): Promise<string> {
- return 'まずユーザー名を教えてください:';
- }
-
- public async q(query: string): Promise<string> {
- if (this.temporaryUser == null) {
- // Fetch user
- const user: IUser = await User.findOne({
- username_lower: query.toLowerCase(),
- host: null
- }, {
- fields: {
- data: false
- }
- });
-
- if (user === null) {
- return `${query}というユーザーは存在しませんでした... もう一度教えてください:`;
- } else {
- this.temporaryUser = user;
- this.emit('updated');
- return `パスワードを教えてください:`;
- }
- } else {
- // Compare password
- const same = await bcrypt.compare(query, (this.temporaryUser.account as ILocalAccount).password);
-
- if (same) {
- this.bot.signin(this.temporaryUser);
- this.bot.clearContext();
- return `${this.temporaryUser.name}さん、おかえりなさい!`;
- } else {
- return `パスワードが違います... もう一度教えてください:`;
- }
- }
- }
-
- public export() {
- return {
- type: 'signin',
- content: {
- temporaryUser: this.temporaryUser
- }
- };
- }
-
- public static import(bot: BotCore, data: any) {
- const context = new SigninContext(bot);
- context.temporaryUser = data.temporaryUser;
- return context;
- }
-}
-
-class PostContext extends Context {
- public async greet(): Promise<string> {
- return '内容:';
- }
-
- public async q(query: string): Promise<string> {
- await require('../endpoints/posts/create')({
- text: query
- }, this.bot.user);
- this.bot.clearContext();
- return '投稿しましたよ!';
- }
-
- public export() {
- return {
- type: 'post'
- };
- }
-
- public static import(bot: BotCore, data: any) {
- const context = new PostContext(bot);
- return context;
- }
-}
-
-class TlContext extends Context {
- private next: string = null;
-
- public async greet(): Promise<string> {
- return await this.getTl();
- }
-
- public async q(query: string): Promise<string> {
- if (query == '次') {
- return await this.getTl();
- } else {
- this.bot.clearContext();
- return await this.bot.q(query);
- }
- }
-
- private async getTl() {
- const tl = await require('../endpoints/posts/timeline')({
- limit: 5,
- until_id: this.next ? this.next : undefined
- }, this.bot.user);
-
- if (tl.length > 0) {
- this.next = tl[tl.length - 1].id;
- this.emit('updated');
-
- const text = tl
- .map(post => `${post.user.name}\n「${getPostSummary(post)}」`)
- .join('\n-----\n');
-
- return text;
- } else {
- return 'タイムラインに表示するものがありません...';
- }
- }
-
- public export() {
- return {
- type: 'tl',
- content: {
- next: this.next,
- }
- };
- }
-
- public static import(bot: BotCore, data: any) {
- const context = new TlContext(bot);
- context.next = data.next;
- return context;
- }
-}
-
-class NotificationsContext extends Context {
- private next: string = null;
-
- public async greet(): Promise<string> {
- return await this.getNotifications();
- }
-
- public async q(query: string): Promise<string> {
- if (query == '次') {
- return await this.getNotifications();
- } else {
- this.bot.clearContext();
- return await this.bot.q(query);
- }
- }
-
- private async getNotifications() {
- const notifications = await require('../endpoints/i/notifications')({
- limit: 5,
- until_id: this.next ? this.next : undefined
- }, this.bot.user);
-
- if (notifications.length > 0) {
- this.next = notifications[notifications.length - 1].id;
- this.emit('updated');
-
- const text = notifications
- .map(notification => getNotificationSummary(notification))
- .join('\n-----\n');
-
- return text;
- } else {
- return '通知はありません';
- }
- }
-
- public export() {
- return {
- type: 'notifications',
- content: {
- next: this.next,
- }
- };
- }
-
- public static import(bot: BotCore, data: any) {
- const context = new NotificationsContext(bot);
- context.next = data.next;
- return context;
- }
-}
-
-class GuessingGameContext extends Context {
- private secret: number;
- private history: number[] = [];
-
- public async greet(): Promise<string> {
- this.secret = Math.floor(Math.random() * 100);
- this.emit('updated');
- return '0~100の秘密の数を当ててみてください:';
- }
-
- public async q(query: string): Promise<string> {
- if (query == 'やめる') {
- this.bot.clearContext();
- return 'やめました。';
- }
-
- const guess = parseInt(query, 10);
-
- if (isNaN(guess)) {
- return '整数で推測してください。「やめる」と言うとゲームをやめます。';
- }
-
- const firsttime = this.history.indexOf(guess) === -1;
-
- this.history.push(guess);
- this.emit('updated');
-
- if (this.secret < guess) {
- return firsttime ? `${guess}よりも小さいですね` : `もう一度言いますが${guess}より小さいですよ`;
- } else if (this.secret > guess) {
- return firsttime ? `${guess}よりも大きいですね` : `もう一度言いますが${guess}より大きいですよ`;
- } else {
- this.bot.clearContext();
- return `正解です🎉 (${this.history.length}回目で当てました)`;
- }
- }
-
- public export() {
- return {
- type: 'guessing-game',
- content: {
- secret: this.secret,
- history: this.history
- }
- };
- }
-
- public static import(bot: BotCore, data: any) {
- const context = new GuessingGameContext(bot);
- context.secret = data.secret;
- context.history = data.history;
- return context;
- }
-}
diff --git a/src/api/bot/interfaces/line.ts b/src/api/bot/interfaces/line.ts
deleted file mode 100644
index 8036b2fde4..0000000000
--- a/src/api/bot/interfaces/line.ts
+++ /dev/null
@@ -1,238 +0,0 @@
-import * as EventEmitter from 'events';
-import * as express from 'express';
-import * as request from 'request';
-import * as crypto from 'crypto';
-import User from '../../models/user';
-import config from '../../../conf';
-import BotCore from '../core';
-import _redis from '../../../db/redis';
-import prominence = require('prominence');
-import getAcct from '../../../common/user/get-acct';
-import parseAcct from '../../../common/user/parse-acct';
-import getPostSummary from '../../../common/get-post-summary';
-
-const redis = prominence(_redis);
-
-// SEE: https://developers.line.me/media/messaging-api/messages/sticker_list.pdf
-const stickers = [
- '297',
- '298',
- '299',
- '300',
- '301',
- '302',
- '303',
- '304',
- '305',
- '306',
- '307'
-];
-
-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<void> {
- this.replyToken = ev.replyToken;
-
- switch (ev.type) {
- // メッセージ
- case 'message':
- switch (ev.message.type) {
- // テキスト
- case 'text':
- const res = await this.q(ev.message.text);
- if (res == null) return;
- // 返信
- this.reply([{
- type: 'text',
- text: res
- }]);
- break;
-
- // スタンプ
- case 'sticker':
- // スタンプで返信
- this.reply([{
- type: 'sticker',
- packageId: '4',
- stickerId: stickers[Math.floor(Math.random() * stickers.length)]
- }]);
- break;
- }
- break;
-
- // postback
- case 'postback':
- const data = ev.postback.data;
- const cmd = data.split('|')[0];
- const arg = data.split('|')[1];
- switch (cmd) {
- case 'showtl':
- this.showUserTimelinePostback(arg);
- break;
- }
- break;
- }
- }
-
- 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')(parseAcct(q.substr(1)), this.user);
-
- const acct = getAcct(user);
- const actions = [];
-
- actions.push({
- type: 'postback',
- label: 'タイムラインを見る',
- data: `showtl|${user.id}`
- });
-
- if (user.account.twitter) {
- actions.push({
- type: 'uri',
- label: 'Twitterアカウントを見る',
- uri: `https://twitter.com/${user.account.twitter.screen_name}`
- });
- }
-
- actions.push({
- type: 'uri',
- label: 'Webで見る',
- uri: `${config.url}/@${acct}`
- });
-
- this.reply([{
- type: 'template',
- altText: await super.showUserCommand(q),
- template: {
- type: 'buttons',
- thumbnailImageUrl: `${user.avatar_url}?thumbnail&size=1024`,
- title: `${user.name} (@${acct})`,
- text: user.description || '(no description)',
- actions: actions
- }
- }]);
-
- return null;
- }
-
- public async showUserTimelinePostback(userId: string) {
- const tl = await require('../../endpoints/users/posts')({
- user_id: userId,
- limit: 5
- }, this.user);
-
- const text = `${tl[0].user.name}さんのタイムラインはこちらです:\n\n` + tl
- .map(post => getPostSummary(post))
- .join('\n-----\n');
-
- this.reply([{
- type: 'text',
- text: text
- }]);
- }
-}
-
-module.exports = async (app: express.Application) => {
- if (config.line_bot == null) return;
-
- const handler = new EventEmitter();
-
- handler.on('event', async (ev) => {
-
- const sourceId = ev.source.userId;
- const sessionId = `line-bot-sessions:${sourceId}`;
-
- const session = await redis.get(sessionId);
- let bot: LineBot;
-
- if (session == null) {
- const user = await User.findOne({
- host: null,
- 'account.line': {
- user_id: sourceId
- }
- });
-
- bot = new LineBot(user);
-
- bot.on('signin', user => {
- User.update(user._id, {
- $set: {
- 'account.line': {
- user_id: sourceId
- }
- }
- });
- });
-
- bot.on('signout', user => {
- User.update(user._id, {
- $set: {
- 'account.line': {
- user_id: null
- }
- }
- });
- });
-
- redis.set(sessionId, JSON.stringify(bot.export()));
- } else {
- bot = LineBot.import(JSON.parse(session));
- }
-
- bot.on('updated', () => {
- redis.set(sessionId, JSON.stringify(bot.export()));
- });
-
- if (session != null) bot.refreshUser();
-
- bot.react(ev);
- });
-
- app.post('/hooks/line', (req, res, next) => {
- // req.headers['x-line-signature'] は常に string ですが、型定義の都合上
- // string | string[] になっているので string を明示しています
- const sig1 = req.headers['x-line-signature'] as string;
-
- const hash = crypto.createHmac('SHA256', config.line_bot.channel_secret)
- .update((req as any).rawBody);
-
- const sig2 = hash.digest('base64');
-
- // シグネチャ比較
- if (sig1 === sig2) {
- req.body.events.forEach(ev => {
- handler.emit('event', ev);
- });
-
- res.sendStatus(200);
- } else {
- res.sendStatus(400);
- }
- });
-};