pch_xyz/ai
1
0
Fork 0
mirror of https://github.com/syuilo/ai.git synced 2024-09-19 19:31:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor

Browse source
This commit is contained in:
syuilo 2018-08-27 19:04:09 +09:00
parent 668ea62724
commit 87f6729c98
4 changed files with 23 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/ai.ts
View file

@ -123,17 +123,6 @@ export default class 藍 {
this.modules.push(module);
}
/**
* 指定したユーザーの「呼んでもらいたい名前」を取得します
*/
public getName = (userId: string) => {
const friend = this.friends.findOne({
userId: userId
});
return friend != null ? friend.name : null;
}
private onMessage = (msg: any) => {
switch (msg.type) {
// メンションされたとき

12
src/message-like.ts
View file

@ -26,10 +26,22 @@ export default class MessageLike {
return this.messageOrNote.replyId;
}
public friend: ReturnType<藍['friends']['findOne']>;
constructor(ai: 藍, messageOrNote: any, isMessage: boolean) {
this.ai = ai;
this.messageOrNote = messageOrNote;
this.isMessage = isMessage;
this.friend = this.ai.friends.findOne({
userId: this.userId
});
if (this.friend == null) {
this.friend = this.ai.friends.insertOne({
userId: this.userId
});
}
}
public reply = async (text: string, cw?: string) => {

39
src/modules/core/index.ts
View file

@ -16,16 +16,6 @@ export default class CoreModule implements IModule {
if (!msg.text) return false;
if (msg.text.includes('って呼んで') && !msg.text.startsWith('って呼んで')) {
let friend = this.ai.friends.findOne({
userId: msg.userId
});
if (friend == null) {
friend = this.ai.friends.insertOne({
userId: msg.userId
});
}
const name = msg.text.match(/^(.+?)って呼んで/)[1];
if (name.length > 10) {
@ -41,13 +31,12 @@ export default class CoreModule implements IModule {
name.endsWith('様');
if (withSan) {
friend.name = name;
this.ai.friends.update(friend);
msg.friend.name = name;
this.ai.friends.update(msg.friend);
msg.reply(serifs.core.setNameOk.replace('{name}', name));
} else {
msg.reply(serifs.core.san).then(reply => {
this.ai.subscribeReply(this, msg.userId, msg.isMessage, msg.isMessage ? msg.userId : reply.id, {
friend: friend,
name: name
});
});
@ -55,24 +44,16 @@ export default class CoreModule implements IModule {
return true;
} else if (msg.text.includes('おはよう')) {
const friend = this.ai.friends.findOne({
userId: msg.userId
});
if (friend && friend.name) {
msg.reply(serifs.core.goodMorningWithName.replace('{name}', friend.name));
if (msg.friend.name) {
msg.reply(serifs.core.goodMorningWithName.replace('{name}', msg.friend.name));
} else {
msg.reply(serifs.core.goodMorning);
}
return true;
} else if (msg.text.includes('おやすみ')) {
const friend = this.ai.friends.findOne({
userId: msg.userId
});
if (friend && friend.name) {
msg.reply(serifs.core.goodNightWithName.replace('{name}', friend.name));
if (msg.friend.name) {
msg.reply(serifs.core.goodNightWithName.replace('{name}', msg.friend.name));
} else {
msg.reply(serifs.core.goodNight);
}
@ -87,16 +68,16 @@ export default class CoreModule implements IModule {
if (msg.text == null) return;
const done = () => {
this.ai.friends.update(data.friend);
msg.reply(serifs.core.setNameOk.replace('{name}', data.friend.name));
this.ai.friends.update(msg.friend);
msg.reply(serifs.core.setNameOk.replace('{name}', msg.friend.name));
this.ai.unsubscribeReply(this, msg.userId);
};
if (msg.text.includes('はい')) {
data.friend.name = data.name + 'さん';
msg.friend.name = data.name + 'さん';
done();
} else if (msg.text.includes('いいえ')) {
data.friend.name = data.name;
msg.friend.name = data.name;
done();
} else {
msg.reply(serifs.core.yesOrNo).then(reply => {

2
src/modules/timer/index.ts
View file

@ -35,7 +35,7 @@ export default class TimerModule implements IModule {
const str = `${hours ? hoursQuery[0] : ''}${minutes ? minutesQuery[0] : ''}${seconds ? secondsQuery[0] : ''}`;
setTimeout(() => {
const name = this.ai.getName(msg.userId);
const name = msg.friend.name;
this.ai.sendMessage(msg.userId, {
text: name ? serifs.timer.notifyWithName.replace('{time}', str).replace('{name}', name) : serifs.timer.notify.replace('{time}', str)
});