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

View file

@ -10,20 +10,83 @@ import prominence = require('prominence');
const redis = prominence(_redis); 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) => { module.exports = async (app: express.Application) => {
if (config.line_bot == null) return; if (config.line_bot == null) return;
const handler = new EventEmitter(); const handler = new EventEmitter();
handler.on('message', async (ev) => { handler.on('message', async (ev) => {
// テキスト以外(スタンプなど)は無視
if (ev.message.type !== 'text') return;
const sourceId = ev.source.userId; const sourceId = ev.source.userId;
const sessionId = `line-bot-sessions:${sourceId}`; const sessionId = `line-bot-sessions:${sourceId}`;
const _session = await redis.get(sessionId); const _session = await redis.get(sessionId);
let session: BotCore; let bot: LineBot;
if (_session == null) { if (_session == null) {
const user = await User.findOne({ 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, { User.update(user._id, {
$set: { $set: {
line: { line: {
@ -44,7 +107,7 @@ module.exports = async (app: express.Application) => {
}); });
}); });
session.on('signout', user => { bot.on('signout', user => {
User.update(user._id, { User.update(user._id, {
$set: { $set: {
line: { 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 { } else {
session = BotCore.import(JSON.parse(_session)); bot = LineBot.import(JSON.parse(_session));
} }
session.on('updated', () => { bot.on('updated', () => {
redis.set(sessionId, JSON.stringify(session.export())); redis.set(sessionId, JSON.stringify(bot.export()));
}); });
const res = await session.q(ev.message.text); bot.react(ev);
// 返信
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;
}
});
}); });
app.post('/hooks/line', (req, res, next) => { app.post('/hooks/line', (req, res, next) => {