forked from FoundKeyGang/FoundKey
✌️
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…
Reference in a new issue