This commit is contained in:
syuilo 2017-10-07 05:50:01 +09:00
parent 0b20c3caa6
commit 0f2b503f2f
2 changed files with 70 additions and 20 deletions

View file

@ -14,6 +14,31 @@ export default class BotCore extends EventEmitter {
this.user = user; this.user = user;
} }
private setContect(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
};
}
public static import(data) {
const core = new BotCore();
core.user = data.user;
core.setContect(data.context ? Context.import(core, data.context) : null);
return core;
}
public async q(query: string): Promise<string> { public async q(query: string): Promise<string> {
if (this.context != null) { if (this.context != null) {
return await this.context.q(query); return await this.context.q(query);
@ -22,9 +47,11 @@ export default class BotCore extends EventEmitter {
switch (query) { switch (query) {
case 'ping': case 'ping':
return 'PONG'; return 'PONG';
case 'me':
return this.user ? `${this.user.name}としてサインインしています` : 'サインインしていません';
case 'ログイン': case 'ログイン':
case 'サインイン': case 'サインイン':
this.context = new SigninContext(this); this.setContect(new SigninContext(this));
return await this.context.greet(); return await this.context.greet();
default: default:
return '?'; return '?';
@ -34,18 +61,26 @@ export default class BotCore extends EventEmitter {
public setUser(user: IUser) { public setUser(user: IUser) {
this.user = user; this.user = user;
this.emit('set-user', user); this.emit('set-user', user);
this.emit('updated');
} }
} }
abstract class Context { abstract class Context extends EventEmitter {
protected core: BotCore; protected core: BotCore;
public abstract async greet(): Promise<string>; public abstract async greet(): Promise<string>;
public abstract async q(query: string): Promise<string>; public abstract async q(query: string): Promise<string>;
public abstract export(): any;
constructor(core: BotCore) { constructor(core: BotCore) {
super();
this.core = core; this.core = core;
} }
public static import(core: BotCore, data: any) {
if (data.type == 'signin') return SigninContext.import(core, data.content);
return null;
}
} }
class SigninContext extends Context { class SigninContext extends Context {
@ -71,6 +106,7 @@ class SigninContext extends Context {
return `${query}というユーザーは存在しませんでした... もう一度教えてください:`; return `${query}というユーザーは存在しませんでした... もう一度教えてください:`;
} else { } else {
this.temporaryUser = user; this.temporaryUser = user;
this.emit('updated');
return `パスワードを教えてください:`; return `パスワードを教えてください:`;
} }
} else { } else {
@ -85,4 +121,16 @@ class SigninContext extends Context {
} }
} }
} }
public export() {
return {
temporaryUser: this.temporaryUser
};
}
public static import(core: BotCore, data: any) {
const context = new SigninContext(core);
context.temporaryUser = data.temporaryUser;
return context;
}
} }

View file

@ -5,11 +5,10 @@ import * as crypto from 'crypto';
import User from '../../models/user'; import User from '../../models/user';
import config from '../../../conf'; import config from '../../../conf';
import BotCore from '../core'; import BotCore from '../core';
import _redis from '../../../db/redis';
import prominence = require('prominence');
const sessions: Array<{ const redis = prominence(_redis);
sourceId: string;
core: BotCore;
}> = [];
module.exports = async (app: express.Application) => { module.exports = async (app: express.Application) => {
if (config.line_bot == null) return; if (config.line_bot == null) return;
@ -21,22 +20,23 @@ module.exports = async (app: express.Application) => {
if (ev.message.type !== 'text') return; if (ev.message.type !== 'text') return;
const sourceId = ev.source.userId; const sourceId = ev.source.userId;
let session = sessions.find(s => s.sourceId === sourceId); const sessionId = `line-bot-sessions:${sourceId}`;
if (!session) { const _session = await redis.get(sessionId);
let session: BotCore;
if (_session == null) {
const user = await User.findOne({ const user = await User.findOne({
line: { line: {
user_id: sourceId user_id: sourceId
} }
}); });
let core: BotCore;
if (user) { if (user) {
core = new BotCore(user); session = new BotCore(user);
} else { } else {
core = new BotCore(); session = new BotCore();
core.on('set-user', user => { session.on('set-user', user => {
User.update(user._id, { User.update(user._id, {
$set: { $set: {
line: { line: {
@ -47,16 +47,18 @@ module.exports = async (app: express.Application) => {
}); });
} }
session = { redis.set(sessionId, JSON.stringify(session.export()));
sourceId: sourceId, } else {
core: core session = BotCore.import(JSON.parse(_session));
};
sessions.push(session);
} }
const res = await session.core.q(ev.message.text); session.on('updated', () => {
redis.set(sessionId, JSON.stringify(session.export()));
});
const res = await session.q(ev.message.text);
// 返信
request.post({ request.post({
url: 'https://api.line.me/v2/bot/message/reply', url: 'https://api.line.me/v2/bot/message/reply',
headers: { headers: {