ai/src/module.ts
2024-03-25 23:45:20 +09:00

81 lines
2 KiB
TypeScript

import { bindThis } from '@/decorators.js';
import , { InstallerResult } from '@/ai.js';
export default abstract class Module {
public abstract readonly name: string;
private maybeAi?: ;
private doc: any;
public init(ai: ) {
this.maybeAi = ai;
this.doc = this.ai.moduleData.findOne({
module: this.name
});
if (this.doc == null) {
this.doc = this.ai.moduleData.insertOne({
module: this.name,
data: {}
});
}
}
public abstract install(): InstallerResult;
protected get ai(): {
if (this.maybeAi == null) {
throw new TypeError('This module has not been initialized');
}
return this.maybeAi;
}
@bindThis
protected log(msg: string) {
this.ai.log(`[${this.name}]: ${msg}`);
}
/**
* コンテキストを生成し、ユーザーからの返信を待ち受けます
* @param key コンテキストを識別するためのキー
* @param id トークメッセージ上のコンテキストならばトーク相手のID、そうでないなら待ち受ける投稿のID
* @param data コンテキストに保存するオプションのデータ
*/
@bindThis
protected subscribeReply(key: string | null, id: string, data?: any) {
this.ai.subscribeReply(this, key, id, data);
}
/**
* 返信の待ち受けを解除します
* @param key コンテキストを識別するためのキー
*/
@bindThis
protected unsubscribeReply(key: string | null) {
this.ai.unsubscribeReply(this, key);
}
/**
* 指定したミリ秒経過後に、タイムアウトコールバックを呼び出します。
* このタイマーは記憶に永続化されるので、途中でプロセスを再起動しても有効です。
* @param delay ミリ秒
* @param data オプションのデータ
*/
@bindThis
public setTimeoutWithPersistence(delay: number, data?: any) {
this.ai.setTimeoutWithPersistence(this, delay, data);
}
@bindThis
protected getData() {
return this.doc.data;
}
@bindThis
protected setData(data: any) {
this.doc.data = data;
this.ai.moduleData.update(this.doc);
}
}