This commit is contained in:
syuilo 2017-10-07 18:30:04 +09:00
parent d1f5d62251
commit 96b6ef4d9b
2 changed files with 122 additions and 55 deletions

View file

@ -43,18 +43,26 @@ export default class BotCore extends EventEmitter {
};
}
public static import(data) {
const core = new BotCore();
core.user = data.user ? initUser(data.user) : null;
core.setContext(data.context ? Context.import(core, data.context) : null);
return core;
protected _import(data) {
this.user = data.user ? initUser(data.user) : null;
this.setContext(data.context ? Context.import(this, data.context) : null);
}
public async q(query: string): Promise<string> {
public static import(data) {
const bot = new BotCore();
bot._import(data);
return bot;
}
public async q(query: string): Promise<string | void> {
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';
@ -67,7 +75,8 @@ export default class BotCore extends EventEmitter {
'login, signin: サインインします\n' +
'logout, signout: サインアウトします\n' +
'post: 投稿します\n' +
'tl: タイムラインを見ます\n';
'tl: タイムラインを見ます\n' +
'@<ユーザー名>: ユーザーを表示します';
case 'me':
return this.user ? `${this.user.name}としてサインインしています。\n\n${getUserSummary(this.user)}` : 'サインインしていません';
@ -76,6 +85,7 @@ export default class BotCore extends EventEmitter {
case 'signin':
case 'ログイン':
case 'サインイン':
if (this.user != null) return '既にサインインしていますよ!';
this.setContext(new SigninContext(this));
return await this.context.greet();
@ -95,9 +105,9 @@ export default class BotCore extends EventEmitter {
case 'tl':
case 'タイムライン':
return await this.getTl();
return await this.tlCommand();
default:
default:
return '?';
}
}
@ -115,7 +125,7 @@ export default class BotCore extends EventEmitter {
this.emit('updated');
}
public async getTl() {
public async tlCommand(): Promise<string | void> {
if (this.user == null) return 'まずサインインしてください。';
const tl = await require('../endpoints/posts/timeline')({
@ -128,23 +138,37 @@ export default class BotCore extends EventEmitter {
return text;
}
public async showUserCommand(q: string): Promise<string | void> {
try {
const user = await require('../endpoints/users/show')({
username: q.substr(1)
}, this.user);
const text = getUserSummary(user);
return text;
} catch (e) {
return `問題が発生したようです...: ${e}`;
}
}
}
abstract class Context extends EventEmitter {
protected core: BotCore;
protected bot: BotCore;
public abstract async greet(): Promise<string>;
public abstract async q(query: string): Promise<string>;
public abstract export(): any;
constructor(core: BotCore) {
constructor(bot: BotCore) {
super();
this.core = core;
this.bot = bot;
}
public static import(core: BotCore, data: any) {
if (data.type == 'post') return PostContext.import(core, data.content);
if (data.type == 'signin') return SigninContext.import(core, data.content);
public static import(bot: BotCore, data: any) {
if (data.type == 'post') return PostContext.import(bot, data.content);
if (data.type == 'signin') return SigninContext.import(bot, data.content);
return null;
}
}
@ -179,8 +203,8 @@ class SigninContext extends Context {
const same = bcrypt.compareSync(query, this.temporaryUser.password);
if (same) {
this.core.signin(this.temporaryUser);
this.core.clearContext();
this.bot.signin(this.temporaryUser);
this.bot.clearContext();
return `${this.temporaryUser.name}さん、おかえりなさい!`;
} else {
return `パスワードが違います... もう一度教えてください:`;
@ -197,8 +221,8 @@ class SigninContext extends Context {
};
}
public static import(core: BotCore, data: any) {
const context = new SigninContext(core);
public static import(bot: BotCore, data: any) {
const context = new SigninContext(bot);
context.temporaryUser = data.temporaryUser;
return context;
}
@ -212,8 +236,8 @@ class PostContext extends Context {
public async q(query: string): Promise<string> {
await require('../endpoints/posts/create')({
text: query
}, this.core.user);
this.core.clearContext();
}, this.bot.user);
this.bot.clearContext();
return '投稿しましたよ!';
}
@ -223,8 +247,8 @@ class PostContext extends Context {
};
}
public static import(core: BotCore, data: any) {
const context = new PostContext(core);
public static import(bot: BotCore, data: any) {
const context = new PostContext(bot);
return context;
}
}

View file

@ -10,20 +10,83 @@ import prominence = require('prominence');
const redis = prominence(_redis);
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> {
// テキスト以外(スタンプなど)は無視
if (ev.message.type !== 'text') return;
const res = await this.q(ev.message.text);
if (res == null) return;
// 返信
this.reply([{
type: 'text',
text: res
}]);
}
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')({
username: q.substr(1)
}, this.user);
this.reply([{
type: 'template',
altText: await super.showUserCommand(q),
template: {
type: 'buttons',
thumbnailImageUrl: `${user.avatar_url}?thumbnail&size=1024`,
title: `${user.name} (@${user.username})`,
text: user.description || '(no description)',
actions: [{
type: 'uri',
label: 'Webで見る',
uri: `${config.url}/${user.username}`
}]
}
}]);
}
}
module.exports = async (app: express.Application) => {
if (config.line_bot == null) return;
const handler = new EventEmitter();
handler.on('message', async (ev) => {
// テキスト以外(スタンプなど)は無視
if (ev.message.type !== 'text') return;
const sourceId = ev.source.userId;
const sessionId = `line-bot-sessions:${sourceId}`;
const _session = await redis.get(sessionId);
let session: BotCore;
let bot: LineBot;
if (_session == null) {
const user = await User.findOne({
@ -32,9 +95,9 @@ module.exports = async (app: express.Application) => {
}
});
session = new BotCore(user);
bot = new LineBot(user);
session.on('signin', user => {
bot.on('signin', user => {
User.update(user._id, {
$set: {
line: {
@ -44,7 +107,7 @@ module.exports = async (app: express.Application) => {
});
});
session.on('signout', user => {
bot.on('signout', user => {
User.update(user._id, {
$set: {
line: {
@ -54,36 +117,16 @@ module.exports = async (app: express.Application) => {
});
});
redis.set(sessionId, JSON.stringify(session.export()));
redis.set(sessionId, JSON.stringify(bot.export()));
} else {
session = BotCore.import(JSON.parse(_session));
bot = LineBot.import(JSON.parse(_session));
}
session.on('updated', () => {
redis.set(sessionId, JSON.stringify(session.export()));
bot.on('updated', () => {
redis.set(sessionId, JSON.stringify(bot.export()));
});
const res = await session.q(ev.message.text);
// 返信
request.post({
url: 'https://api.line.me/v2/bot/message/reply',
headers: {
'Authorization': `Bearer ${config.line_bot.channel_access_token}`
},
json: {
replyToken: ev.replyToken,
messages: [{
type: 'text',
text: res
}]
}
}, (err, res, body) => {
if (err) {
console.error(err);
return;
}
});
bot.react(ev);
});
app.post('/hooks/line', (req, res, next) => {