ai/src/ai.ts

399 lines
9.9 KiB
TypeScript
Raw Normal View History

2018-08-11 06:26:25 +00:00
// AI CORE
2019-05-10 02:55:07 +00:00
import * as fs from 'fs';
2019-01-14 15:14:22 +00:00
import autobind from 'autobind-decorator';
2018-08-26 21:16:56 +00:00
import * as loki from 'lokijs';
2018-08-11 06:26:25 +00:00
import * as request from 'request-promise-native';
2019-01-14 15:14:22 +00:00
import chalk from 'chalk';
2019-02-01 17:06:51 +00:00
import * as uuid from 'uuid/v4';
2019-01-15 09:47:22 +00:00
const delay = require('timeout-as-promise');
2018-08-11 06:26:25 +00:00
import config from './config';
2019-01-14 15:14:22 +00:00
import Module from './module';
2019-01-15 09:47:22 +00:00
import Message from './message';
2019-02-01 17:06:51 +00:00
import Friend, { FriendDoc } from './friend';
2018-08-28 21:30:48 +00:00
import { User } from './misskey/user';
2018-10-09 15:47:03 +00:00
import Stream from './stream';
2019-01-15 09:58:04 +00:00
import log from './utils/log';
2018-08-11 06:26:25 +00:00
2019-01-23 12:49:10 +00:00
type MentionHook = (msg: Message) => Promise<boolean | HandlerResult>;
type ContextHook = (msg: Message, data?: any) => Promise<void | HandlerResult>;
2019-02-01 17:06:51 +00:00
type TimeoutCallback = (data?: any) => void;
2019-01-14 15:14:22 +00:00
export type HandlerResult = {
reaction: string;
};
export type InstallerResult = {
2019-01-15 03:01:58 +00:00
mentionHook?: MentionHook;
contextHook?: ContextHook;
2019-02-01 17:06:51 +00:00
timeoutCallback?: TimeoutCallback;
2019-01-14 15:14:22 +00:00
};
2018-08-11 06:26:25 +00:00
/**
*
*/
export default class {
2018-08-28 21:30:48 +00:00
public account: User;
2018-10-09 15:47:03 +00:00
public connection: Stream;
2019-01-14 15:14:22 +00:00
public modules: Module[] = [];
2019-01-15 03:01:58 +00:00
private mentionHooks: MentionHook[] = [];
private contextHooks: { [moduleName: string]: ContextHook } = {};
2019-02-01 17:06:51 +00:00
private timeoutCallbacks: { [moduleName: string]: TimeoutCallback } = {};
2018-08-26 21:16:56 +00:00
public db: loki;
private contexts: loki.Collection<{
2019-01-15 09:47:22 +00:00
isDm: boolean;
2018-08-26 21:16:56 +00:00
noteId?: string;
userId?: string;
module: string;
key: string;
2018-08-26 21:59:18 +00:00
data?: any;
}>;
2019-02-01 17:06:51 +00:00
private timers: loki.Collection<{
id: string;
module: string;
insertedAt: number;
delay: number;
data?: any;
}>;
2018-08-27 11:22:59 +00:00
public friends: loki.Collection<FriendDoc>;
2019-05-10 02:55:07 +00:00
public moduleData: loki.Collection<any>;
2018-08-26 21:16:56 +00:00
2019-01-15 17:48:14 +00:00
/**
*
* @param account 使
* @param modules
*/
2019-01-15 17:10:42 +00:00
constructor(account: User, modules: Module[]) {
2018-08-11 06:26:25 +00:00
this.account = account;
2019-01-15 17:10:42 +00:00
this.modules = modules;
2018-08-11 06:26:25 +00:00
2019-01-15 03:29:11 +00:00
this.log('Lodaing the memory...');
2018-08-26 21:16:56 +00:00
this.db = new loki('memory.json', {
autoload: true,
autosave: true,
autosaveInterval: 1000,
2019-01-14 15:14:22 +00:00
autoloadCallback: err => {
if (err) {
2019-01-15 03:29:11 +00:00
this.log(chalk.red(`Failed to load the memory: ${err}`));
2019-01-14 15:14:22 +00:00
} else {
2019-01-15 03:29:11 +00:00
this.log(chalk.green('The memory loaded successfully'));
2019-01-15 17:10:42 +00:00
this.run();
2019-01-14 15:14:22 +00:00
}
}
2018-08-26 21:16:56 +00:00
});
}
2019-01-14 15:14:22 +00:00
@autobind
public log(msg: string) {
2019-01-15 03:29:11 +00:00
log(chalk`[{magenta AiOS}]: ${msg}`);
2019-01-14 15:14:22 +00:00
}
@autobind
2019-01-15 01:23:54 +00:00
private run() {
2018-08-26 21:16:56 +00:00
//#region Init DB
2019-01-24 00:34:03 +00:00
this.contexts = this.getCollection('contexts', {
2018-08-29 07:26:33 +00:00
indices: ['key']
});
2019-02-01 17:06:51 +00:00
this.timers = this.getCollection('timers', {
indices: ['module']
});
2019-01-24 00:34:03 +00:00
this.friends = this.getCollection('friends', {
2018-08-29 07:26:33 +00:00
indices: ['userId']
});
2019-05-10 02:55:07 +00:00
this.moduleData = this.getCollection('moduleData', {
indices: ['module']
});
2018-08-26 21:16:56 +00:00
//#endregion
2018-10-09 15:47:03 +00:00
// Init stream
this.connection = new Stream();
2018-08-11 06:26:25 +00:00
2018-10-09 15:47:03 +00:00
//#region Main stream
const mainStream = this.connection.useSharedConnection('main');
2018-08-11 06:26:25 +00:00
2018-10-09 15:47:03 +00:00
// メンションされたとき
2019-01-23 13:18:02 +00:00
mainStream.on('mention', async data => {
2018-10-09 15:47:03 +00:00
if (data.userId == this.account.id) return; // 自分は弾く
2019-01-14 15:14:22 +00:00
if (data.text && data.text.startsWith('@' + this.account.username)) {
2019-01-23 13:18:02 +00:00
// Misskeyのバグで投稿が非公開扱いになる
if (data.text == null) data = await this.api('notes/show', { noteId: data.id });
2019-01-15 09:47:22 +00:00
this.onReceiveMessage(new Message(this, data, false));
2018-10-09 15:47:03 +00:00
}
2018-08-13 21:14:47 +00:00
});
2018-10-09 15:47:03 +00:00
// 返信されたとき
2019-01-23 13:18:02 +00:00
mainStream.on('reply', async data => {
2018-10-09 15:47:03 +00:00
if (data.userId == this.account.id) return; // 自分は弾く
2019-01-23 13:18:02 +00:00
if (data.text && data.text.startsWith('@' + this.account.username)) return;
// Misskeyのバグで投稿が非公開扱いになる
if (data.text == null) data = await this.api('notes/show', { noteId: data.id });
2019-01-15 09:47:22 +00:00
this.onReceiveMessage(new Message(this, data, false));
2018-08-13 21:14:47 +00:00
});
2019-01-24 11:49:27 +00:00
// Renoteされたとき
mainStream.on('renote', async data => {
if (data.userId == this.account.id) return; // 自分は弾く
if (data.text == null && (data.files || []).length == 0) return;
// リアクションする
this.api('notes/reactions/create', {
noteId: data.id,
reaction: 'love'
});
});
2018-10-09 15:47:03 +00:00
// メッセージ
mainStream.on('messagingMessage', data => {
if (data.userId == this.account.id) return; // 自分は弾く
2019-01-15 09:47:22 +00:00
this.onReceiveMessage(new Message(this, data, true));
2018-08-13 21:14:47 +00:00
});
//#endregion
2018-08-11 06:26:25 +00:00
2018-10-09 15:47:03 +00:00
// Install modules
2019-01-14 15:14:22 +00:00
this.modules.forEach(m => {
this.log(`Installing ${chalk.cyan.italic(m.name)}\tmodule...`);
2019-01-15 17:10:42 +00:00
m.init(this);
2019-01-14 15:14:22 +00:00
const res = m.install();
if (res != null) {
2019-01-15 03:01:58 +00:00
if (res.mentionHook) this.mentionHooks.push(res.mentionHook);
if (res.contextHook) this.contextHooks[m.name] = res.contextHook;
2019-02-01 17:06:51 +00:00
if (res.timeoutCallback) this.timeoutCallbacks[m.name] = res.timeoutCallback;
2019-01-14 15:14:22 +00:00
}
});
2019-02-01 17:06:51 +00:00
// タイマー監視
this.crawleTimer();
setInterval(this.crawleTimer, 1000);
2019-01-14 15:14:22 +00:00
this.log(chalk.green.bold('Ai am now running!'));
2018-08-13 21:14:47 +00:00
}
2019-01-15 17:48:14 +00:00
/**
*
* ()
*/
2019-01-14 15:14:22 +00:00
@autobind
2019-01-15 09:47:22 +00:00
private async onReceiveMessage(msg: Message): Promise<void> {
2019-01-15 03:29:11 +00:00
this.log(chalk.gray(`<<< An message received: ${chalk.underline(msg.id)}`));
2018-08-11 06:26:25 +00:00
2019-01-15 09:52:20 +00:00
// Ignore message if the user is a bot
// To avoid infinity reply loop.
if (msg.user.isBot) {
return;
}
2019-01-15 09:47:22 +00:00
const isNoContext = !msg.isDm && msg.replyId == null;
2019-01-15 09:34:42 +00:00
// Look up the context
2019-01-15 09:47:22 +00:00
const context = isNoContext ? null : this.contexts.findOne(msg.isDm ? {
isDm: true,
userId: msg.userId
} : {
2019-01-15 09:47:22 +00:00
isDm: false,
noteId: msg.replyId
2018-08-11 06:26:25 +00:00
});
let reaction = 'love';
2019-01-15 17:48:14 +00:00
//#region
// コンテキストがあればコンテキストフック呼び出し
// なければそれぞれのモジュールについてフックが引っかかるまで呼び出し
if (context != null) {
2019-01-15 03:01:58 +00:00
const handler = this.contextHooks[context.module];
2019-01-23 12:49:10 +00:00
const res = await handler(msg, context.data);
if (res != null && typeof res === 'object') {
reaction = res.reaction;
}
} else {
2019-01-14 15:14:22 +00:00
let res: boolean | HandlerResult;
2019-01-23 12:49:10 +00:00
for (const handler of this.mentionHooks) {
res = await handler(msg);
if (res === true || typeof res === 'object') break;
}
if (res != null && typeof res === 'object') {
reaction = res.reaction;
}
}
2019-01-15 17:48:14 +00:00
//#endregion
2019-01-15 09:47:22 +00:00
await delay(1000);
if (msg.isDm) {
// 既読にする
this.api('messaging/messages/read', {
messageId: msg.id,
});
} else {
// リアクションする
if (reaction) {
this.api('notes/reactions/create', {
noteId: msg.id,
reaction: reaction
});
}
2019-01-15 09:47:22 +00:00
}
2018-08-11 06:26:25 +00:00
}
2019-02-01 17:06:51 +00:00
@autobind
private crawleTimer() {
const timers = this.timers.find();
for (const timer of timers) {
// タイマーが時間切れかどうか
if (Date.now() - (timer.insertedAt + timer.delay) >= 0) {
this.log(`Timer expired: ${timer.module} ${timer.id}`);
this.timers.remove(timer);
this.timeoutCallbacks[timer.module](timer.data);
}
}
}
2019-01-24 00:34:03 +00:00
/**
*
*/
@autobind
public getCollection(name: string, opts?: any): loki.Collection {
let collection: loki.Collection;
collection = this.db.getCollection(name);
if (collection == null) {
collection = this.db.addCollection(name, opts);
}
return collection;
}
2019-02-01 17:06:51 +00:00
@autobind
public lookupFriend(userId: User['id']): Friend {
const doc = this.friends.findOne({
userId: userId
});
if (doc == null) return null;
const friend = new Friend(this, { doc: doc });
return friend;
}
2019-05-10 02:55:07 +00:00
/**
*
*/
@autobind
2019-05-12 04:01:08 +00:00
public async upload(file: Buffer | fs.ReadStream, meta: any) {
2019-05-10 02:55:07 +00:00
const res = await request.post({
url: `${config.apiUrl}/drive/files/create`,
formData: {
i: config.i,
2019-05-12 04:01:08 +00:00
file: {
value: file,
options: meta
}
2019-05-10 02:55:07 +00:00
},
json: true
});
return res;
}
2019-01-15 17:48:14 +00:00
/**
* 稿
*/
2019-01-14 15:14:22 +00:00
@autobind
public async post(param: any) {
const res = await this.api('notes/create', param);
return res.createdNote;
2018-08-11 06:26:25 +00:00
}
2019-01-15 17:48:14 +00:00
/**
*
*/
2019-01-14 15:14:22 +00:00
@autobind
public sendMessage(userId: any, param: any) {
return this.api('messaging/messages/create', Object.assign({
2018-08-11 06:26:25 +00:00
userId: userId,
}, param));
}
2019-01-15 17:48:14 +00:00
/**
* APIを呼び出します
*/
2019-01-14 15:14:22 +00:00
@autobind
public api(endpoint: string, param?: any) {
2018-08-11 06:26:25 +00:00
return request.post(`${config.apiUrl}/${endpoint}`, {
json: Object.assign({
i: config.i
}, param)
});
};
2019-01-15 17:48:14 +00:00
/**
*
* @param module 待ち受けるモジュール名
* @param key
* @param isDm
* @param id ID稿ID
* @param data
*/
2019-01-14 15:14:22 +00:00
@autobind
2019-01-15 09:47:22 +00:00
public subscribeReply(module: Module, key: string, isDm: boolean, id: string, data?: any) {
this.contexts.insertOne(isDm ? {
isDm: true,
userId: id,
module: module.name,
key: key,
2018-08-26 21:59:18 +00:00
data: data
} : {
2019-01-15 09:47:22 +00:00
isDm: false,
noteId: id,
module: module.name,
key: key,
2018-08-26 21:59:18 +00:00
data: data
});
}
2019-01-15 17:48:14 +00:00
/**
*
* @param module 解除するモジュール名
* @param key
*/
2019-01-14 15:14:22 +00:00
@autobind
public unsubscribeReply(module: Module, key: string) {
2018-08-26 21:16:56 +00:00
this.contexts.findAndRemove({
key: key,
module: module.name
});
}
2019-02-01 17:06:51 +00:00
/**
*
*
* @param module モジュール名
* @param delay
* @param data
*/
@autobind
public setTimeoutWithPersistence(module: Module, delay: number, data?: any) {
const id = uuid();
this.timers.insertOne({
id: id,
module: module.name,
insertedAt: Date.now(),
delay: delay,
data: data
});
this.log(`Timer persisted: ${module.name} ${id} ${delay}ms`);
}
2018-08-11 06:26:25 +00:00
}