ai/src/friend.ts

import autobind from 'autobind-decorator';
import 藍 from './ai';
import IModule from './module';
import getDate from './utils/get-date';
import { User } from './misskey/user';

export type FriendDoc = {
	userId: string;
	user: User;
	name?: string;
	love?: number;
	lastLoveIncrementedAt?: string;
	todayLoveIncrements?: number;
	perModulesData?: any;
};

export default class Friend {
	private ai: 藍;

	public get userId() {
		return this.doc.userId;
	}

	public get name() {
		return this.doc.name;
	}

	public get love() {
		return this.doc.love || 0;
	}

	public doc: FriendDoc;

	constructor(ai: 藍, opts: { user?: User, doc?: FriendDoc }) {
		this.ai = ai;

		if (opts.user) {
			this.doc = this.ai.friends.findOne({
				userId: opts.user.id
			});

			if (this.doc == null) {
				this.doc = this.ai.friends.insertOne({
					userId: opts.user.id,
					user: opts.user
				});
			} else {
				this.doc.user = opts.user;
				this.save();
			}
		} else {
			this.doc = opts.doc;
		}
	}

	@autobind
	public updateUser(user: User) {
		this.doc.user = user;
		this.save();
	}

	@autobind
	public getPerModulesData(module: IModule) {
		if (this.doc.perModulesData == null) {
			this.doc.perModulesData = {};
			this.doc.perModulesData[module.name] = {};
			this.save();
		} else if (this.doc.perModulesData[module.name] == null) {
			this.doc.perModulesData[module.name] = {};
			this.save();
		}

		return this.doc.perModulesData[module.name];
	}

	@autobind
	public setPerModulesData(module: IModule, data: any) {
		if (this.doc.perModulesData == null) {
			this.doc.perModulesData = {};
		}

		this.doc.perModulesData[module.name] = data;

		this.save();
	}

	@autobind
	public incLove() {
		const today = getDate();

		if (this.doc.lastLoveIncrementedAt != today) {
			this.doc.todayLoveIncrements = 0;
		}

		// 1日に上げられる親愛度は最大3
		if (this.doc.lastLoveIncrementedAt == today && this.doc.todayLoveIncrements >= 3) return;

		if (this.doc.love == null) this.doc.love = 0;
		this.doc.love++;

		// 最大 100
		if (this.doc.love > 100) this.doc.love = 100;

		this.doc.lastLoveIncrementedAt = today;
		this.doc.todayLoveIncrements = (this.doc.todayLoveIncrements || 0) + 1;
		this.save();
	}

	@autobind
	public decLove() {
		if (this.doc.love == null) this.doc.love = 0;
		this.doc.love--;

		// 最低 -30
		if (this.doc.love < -30) this.doc.love = -30;

		// 親愛度マイナスなら名前を忘れる
		if (this.doc.love < 0) {
			this.doc.name = null;
		}

		this.save();
	}

	@autobind
	public updateName(name: string) {
		this.doc.name = name;
		this.save();
	}

	@autobind
	public save() {
		this.ai.friends.update(this.doc);
	}
}