summaryrefslogtreecommitdiff
path: root/src/api/bot/core.ts
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/core.ts
parentImplement remote account resolution (diff)
downloadsharkey-90f8fe7e538bb7e52d2558152a0390e693f39b11.tar.gz
sharkey-90f8fe7e538bb7e52d2558152a0390e693f39b11.tar.bz2
sharkey-90f8fe7e538bb7e52d2558152a0390e693f39b11.zip
Introduce processor
Diffstat (limited to 'src/api/bot/core.ts')
-rw-r--r--src/api/bot/core.ts438
1 files changed, 0 insertions, 438 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;
- }
-}