summaryrefslogtreecommitdiff
path: root/src/api/bot
diff options
context:
space:
mode:
authorsyuilo <syuilotan@yahoo.co.jp>2017-10-07 18:30:04 +0900
committersyuilo <syuilotan@yahoo.co.jp>2017-10-07 18:30:04 +0900
commit96b6ef4d9ba73dc04eb601ffedee52d6b6ab580a (patch)
tree0f5f53fb09c68c7988c9f33380880affdfb5b833 /src/api/bot
parentRefactor (diff)
downloadsharkey-96b6ef4d9ba73dc04eb601ffedee52d6b6ab580a.tar.gz
sharkey-96b6ef4d9ba73dc04eb601ffedee52d6b6ab580a.tar.bz2
sharkey-96b6ef4d9ba73dc04eb601ffedee52d6b6ab580a.zip
:v:
Diffstat (limited to 'src/api/bot')
-rw-r--r--src/api/bot/core.ts70
-rw-r--r--src/api/bot/interfaces/line.ts105
2 files changed, 121 insertions, 54 deletions
diff --git a/src/api/bot/core.ts b/src/api/bot/core.ts
index 4970f24469..4e168a6054 100644
--- a/src/api/bot/core.ts
+++ b/src/api/bot/core.ts
@@ -43,18 +43,26 @@ export default class BotCore extends EventEmitter {
};
}
+ protected _import(data) {
+ this.user = data.user ? initUser(data.user) : null;
+ this.setContext(data.context ? Context.import(this, data.context) : null);
+ }
+
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;
+ const bot = new BotCore();
+ bot._import(data);
+ return bot;
}
- public async q(query: string): Promise<string> {
+ 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;
}
}
diff --git a/src/api/bot/interfaces/line.ts b/src/api/bot/interfaces/line.ts
index 437f29cb3c..03dc2a85ba 100644
--- a/src/api/bot/interfaces/line.ts
+++ b/src/api/bot/interfaces/line.ts
@@ -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) => {