forked from FoundKeyGang/FoundKey
✌️
This commit is contained in:
parent
d1f5d62251
commit
96b6ef4d9b
2 changed files with 122 additions and 55 deletions
|
@ -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;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -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) => {
|
||||||
|
|
Loading…
Reference in a new issue