ai/src/ai.ts

243 lines
5.5 KiB
TypeScript
Raw Normal View History

2018-08-11 06:26:25 +00:00
// AI CORE
2018-08-26 21:16:56 +00:00
import * as loki from 'lokijs';
2018-08-11 06:26:25 +00:00
import * as WebSocket from 'ws';
import * as request from 'request-promise-native';
import config from './config';
import IModule from './module';
import MessageLike from './message-like';
2018-08-27 11:22:59 +00:00
import { FriendDoc } from './friend';
2018-08-28 21:30:48 +00:00
import { User } from './misskey/user';
2018-08-29 07:26:33 +00:00
import getCollection from './utils/get-collection';
2018-08-11 06:26:25 +00:00
const ReconnectingWebSocket = require('../node_modules/reconnecting-websocket/dist/reconnecting-websocket-cjs.js');
/**
*
*/
export default class {
2018-08-28 21:30:48 +00:00
public account: User;
2018-08-11 06:26:25 +00:00
/**
*
*/
private connection: any;
2018-08-13 21:14:47 +00:00
/**
*
*/
private localTimelineConnection: any;
2018-08-11 06:26:25 +00:00
private modules: IModule[] = [];
2018-08-26 21:16:56 +00:00
public db: loki;
private contexts: loki.Collection<{
isMessage: boolean;
noteId?: string;
userId?: string;
module: string;
key: string;
2018-08-26 21:59:18 +00:00
data?: any;
}>;
2018-08-27 11:22:59 +00:00
public friends: loki.Collection<FriendDoc>;
2018-08-26 21:16:56 +00:00
2018-08-28 21:30:48 +00:00
constructor(account: User, modules: IModule[]) {
2018-08-11 06:26:25 +00:00
this.account = account;
2018-08-28 07:02:20 +00:00
this.modules = modules;
2018-08-11 06:26:25 +00:00
2018-08-26 21:16:56 +00:00
this.db = new loki('memory.json', {
autoload: true,
autosave: true,
autosaveInterval: 1000,
autoloadCallback: this.init
});
}
private init = () => {
//#region Init DB
2018-08-29 07:26:33 +00:00
this.contexts = getCollection(this.db, 'contexts', {
indices: ['key']
});
this.friends = getCollection(this.db, 'friends', {
indices: ['userId']
});
2018-08-26 21:16:56 +00:00
//#endregion
// Install modules
this.modules.forEach(m => m.install(this));
2018-08-13 21:14:47 +00:00
//#region Home stream
2018-08-11 06:26:25 +00:00
this.connection = new ReconnectingWebSocket(`${config.wsUrl}/?i=${config.i}`, [], {
WebSocket: WebSocket
});
this.connection.addEventListener('open', () => {
console.log('home stream opened');
});
this.connection.addEventListener('close', () => {
console.log('home stream closed');
2018-08-25 05:18:16 +00:00
this.connection.reconnect();
2018-08-11 06:26:25 +00:00
});
this.connection.addEventListener('message', message => {
const msg = JSON.parse(message.data);
this.onMessage(msg);
});
2018-08-13 21:14:47 +00:00
//#endregion
2018-08-11 06:26:25 +00:00
2018-08-13 21:14:47 +00:00
//#region Local timeline stream
this.localTimelineConnection = new ReconnectingWebSocket(`${config.wsUrl}/local-timeline?i=${config.i}`, [], {
WebSocket: WebSocket
});
this.localTimelineConnection.addEventListener('open', () => {
console.log('local-timeline stream opened');
});
this.localTimelineConnection.addEventListener('close', () => {
console.log('local-timeline stream closed');
2018-08-27 06:40:41 +00:00
this.localTimelineConnection.reconnect();
2018-08-13 21:14:47 +00:00
});
this.localTimelineConnection.addEventListener('message', message => {
const msg = JSON.parse(message.data);
this.onLocalNote(msg.body);
});
//#endregion
2018-08-11 06:26:25 +00:00
}
private onMessage = (msg: any) => {
switch (msg.type) {
// メンションされたとき
case 'mention': {
if (msg.body.userId == this.account.id) return; // 自分は弾く
if (msg.body.text.startsWith('@' + this.account.username)) {
this.onMention(new MessageLike(this, msg.body, false));
}
break;
}
// 返信されたとき
case 'reply': {
if (msg.body.userId == this.account.id) return; // 自分は弾く
this.onMention(new MessageLike(this, msg.body, false));
break;
}
// メッセージ
case 'messaging_message': {
if (msg.body.userId == this.account.id) return; // 自分は弾く
this.onMention(new MessageLike(this, msg.body, true));
break;
}
default:
break;
}
}
2018-08-13 21:14:47 +00:00
private onLocalNote = (note: any) => {
this.modules.filter(m => m.hasOwnProperty('onLocalNote')).forEach(m => {
return m.onLocalNote(note);
});
}
2018-08-11 06:26:25 +00:00
private onMention = (msg: MessageLike) => {
console.log(`mention received: ${msg.id}`);
2018-08-26 21:16:56 +00:00
const context = !msg.isMessage && msg.replyId == null ? null : this.contexts.findOne(msg.isMessage ? {
isMessage: true,
userId: msg.userId
} : {
isMessage: false,
noteId: msg.replyId
2018-08-11 06:26:25 +00:00
});
let reaction = 'love';
if (context != null) {
const module = this.modules.find(m => m.name == context.module);
const res = module.onReplyThisModule(msg, context.data);
if (res != null && typeof res === 'object') {
reaction = res.reaction;
}
} else {
let res: ReturnType<IModule['onMention']>;
this.modules.filter(m => m.hasOwnProperty('onMention')).some(m => {
res = m.onMention(msg);
return res === true || typeof res === 'object';
});
if (res != null && typeof res === 'object') {
reaction = res.reaction;
}
}
setTimeout(() => {
if (msg.isMessage) {
// 既読にする
this.api('messaging/messages/read', {
messageId: msg.id,
});
} else {
// リアクションする
this.api('notes/reactions/create', {
noteId: msg.id,
reaction: reaction
});
}
}, 1000);
2018-08-11 06:26:25 +00:00
}
public post = async (param: any) => {
const res = await this.api('notes/create', param);
return res.createdNote;
2018-08-11 06:26:25 +00:00
}
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));
}
2018-08-13 08:54:56 +00:00
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)
});
};
2018-08-26 21:59:18 +00:00
public subscribeReply = (module: IModule, key: string, isMessage: boolean, id: string, data?: any) => {
2018-08-26 21:16:56 +00:00
this.contexts.insertOne(isMessage ? {
isMessage: true,
userId: id,
module: module.name,
key: key,
2018-08-26 21:59:18 +00:00
data: data
} : {
isMessage: false,
noteId: id,
module: module.name,
key: key,
2018-08-26 21:59:18 +00:00
data: data
});
}
public unsubscribeReply = (module: IModule, key: string) => {
2018-08-26 21:16:56 +00:00
this.contexts.findAndRemove({
key: key,
module: module.name
});
}
2018-08-11 06:26:25 +00:00
}