102 lines
2.3 KiB
TypeScript
102 lines
2.3 KiB
TypeScript
import autobind from 'autobind-decorator';
|
|
import * as chalk from 'chalk';
|
|
const delay = require('timeout-as-promise');
|
|
|
|
import 藍 from './ai';
|
|
import Friend from './friend';
|
|
import { User } from './misskey/user';
|
|
import includes from './utils/includes';
|
|
import or from './utils/or';
|
|
import config from './config';
|
|
|
|
export default class Message {
|
|
private ai: 藍;
|
|
private messageOrNote: any;
|
|
public isDm: boolean;
|
|
|
|
public get id(): string {
|
|
return this.messageOrNote.id;
|
|
}
|
|
|
|
public get user(): User {
|
|
return this.messageOrNote.user;
|
|
}
|
|
|
|
public get userId(): string {
|
|
return this.messageOrNote.userId;
|
|
}
|
|
|
|
public get text(): string {
|
|
return this.messageOrNote.text;
|
|
}
|
|
|
|
public get extractedText(): string {
|
|
const host = new URL(config.host).host.replace(/\./g, '\\.');
|
|
return this.text
|
|
.replace(new RegExp(`^@${this.ai.account.username}@${host}\\s`, 'i'), '')
|
|
.replace(new RegExp(`^@${this.ai.account.username}\\s`, 'i'), '')
|
|
.trim();
|
|
}
|
|
|
|
public get replyId(): string {
|
|
return this.messageOrNote.replyId;
|
|
}
|
|
|
|
public friend: Friend;
|
|
|
|
constructor(ai: 藍, messageOrNote: any, isDm: boolean) {
|
|
this.ai = ai;
|
|
this.messageOrNote = messageOrNote;
|
|
this.isDm = isDm;
|
|
|
|
this.friend = new Friend(ai, { user: this.user });
|
|
|
|
// メッセージなどに付いているユーザー情報は省略されている場合があるので完全なユーザー情報を持ってくる
|
|
this.ai.api('users/show', {
|
|
userId: this.userId
|
|
}).then(user => {
|
|
this.friend.updateUser(user);
|
|
});
|
|
}
|
|
|
|
@autobind
|
|
public async reply(text: string | null, opts?: {
|
|
file?: any;
|
|
cw?: string;
|
|
renote?: string;
|
|
immediate?: boolean;
|
|
}) {
|
|
if (text == null) return;
|
|
|
|
this.ai.log(`>>> Sending reply to ${chalk.underline(this.id)}`);
|
|
|
|
if (!opts?.immediate) {
|
|
await delay(2000);
|
|
}
|
|
|
|
if (this.isDm) {
|
|
return await this.ai.sendMessage(this.messageOrNote.userId, {
|
|
text: text,
|
|
fileId: opts?.file?.id
|
|
});
|
|
} else {
|
|
return await this.ai.post({
|
|
replyId: this.messageOrNote.id,
|
|
text: text,
|
|
fileIds: opts?.file ? [opts?.file.id] : undefined,
|
|
cw: opts?.cw,
|
|
renoteId: opts?.renote
|
|
});
|
|
}
|
|
}
|
|
|
|
@autobind
|
|
public includes(words: string[]): boolean {
|
|
return includes(this.text, words);
|
|
}
|
|
|
|
@autobind
|
|
public or(words: (string | RegExp)[]): boolean {
|
|
return or(this.text, words);
|
|
}
|
|
}
|