ai/src/friend.ts

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;
		}
	}

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

	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];
	}

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

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

		this.save();
	}

	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++;
		this.doc.lastLoveIncrementedAt = today;
		this.doc.todayLoveIncrements = (this.doc.todayLoveIncrements || 0) + 1;
		this.save();
	}

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

	public save = () => {
		this.ai.friends.update(this.doc);
	}
}