✌️
This commit is contained in:
		
							parent
							
								
									0b20c3caa6
								
							
						
					
					
						commit
						0f2b503f2f
					
				
					 2 changed files with 70 additions and 20 deletions
				
			
		| 
						 | 
					@ -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;
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -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: {
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue