import BaseModel from './BaseModel';
import { index, getModelForClass, prop, DocumentType } from '@typegoose/typegoose';

@index({ serverId: 1, cityId: 1, createdAt: 1   })
export default class GuildActivityCity extends BaseModel {

    @prop({ required: true })
    serverId: number;       // 分服
    
    @prop({ required: true })
    cityId: number;       // 城池id

    @prop({ required: true, type: String, default: [] })
    declareGuilds: string[];       // 宣战的公会的code

    @prop({ required: true, default: 0 })
    declareCount: number;       // 宣战数量

    @prop({ required: true, default: "" })
    guardGuildCode: string;       // 占领的军团的code

    @prop({ required: true, default: "" })
    guardGuildName: string;       // 占领的军团名

    @prop({ required: true })
    guardTime: Date;       // 占领日期

    @prop({ required: true })
    status: number;       // 0-过去的 1-现在的 2-未来的

    // 每天宣战一次
    public static async getAllCities(serverId: number) {
        let rec: GuildActivityCityType[] = await GuildActivityCityModel.find({ serverId, status: 1 }).lean();
        
        return rec;
    }

    public static async findMinDeclareCountCity(serverId: number, cityIds: number[]) {
        let rec: GuildActivityCityType[] = await GuildActivityCityModel.find({ serverId, cityId: { $in: cityIds }, status: 1 }).sort({ declareCount: 1 }).lean();
        return rec;
    }

    public static async getCity(serverId: number, cityId: number) {
        let rec: GuildActivityCityType = await GuildActivityCityModel.findOne({ serverId, cityId, status: 1}).lean();
        
        return rec;
    }


    // 宣言
    public static async declare(serverId: number, cityId: number, guildCode: string) {
        let rec: GuildActivityCityType = await GuildActivityCityModel.findOneAndUpdate(
            { serverId, cityId, status: 1 }, 
            { $push: { declareGuilds: guildCode }, $inc: {declareCount: 1 } }, 
            {new: true, upsert: true}).lean();

        return rec;
    }

    // 占领
    public static async guard(serverId: number, cityId: number, guildCode: string, guildName: string) {
        let rec: GuildActivityCityType = await GuildActivityCityModel.findOneAndUpdate(
            { serverId, cityId, status: 1 }, 
            { $set: { guardGuildCode: guildCode, guardGuildName: guildName, guardTime: new Date()} }, 
            { new: true }).lean();
        return rec;
    }

    // 活动结束后需要清空占领情况
    public static async clearGuardAndStartNext(serverId?: number) {
        if(serverId) {
            await GuildActivityCityModel.updateMany({ serverId, status: 1 }, { $set: { status: 0 } });
            await GuildActivityCityModel.updateMany({ serverId, status: 2 }, { $set: { status: 1 } });
        } else {
            await GuildActivityCityModel.updateMany({ status: 1 }, { $set: { status: 0 } });
            await GuildActivityCityModel.updateMany({ status: 2 }, { $set: { status: 1 } });
        }
    }
}

export const GuildActivityCityModel = getModelForClass(GuildActivityCity);

export interface GuildActivityCityType extends Pick<DocumentType<GuildActivityCity>, keyof GuildActivityCity> { };
export type GuildActivityCityUpdateParam = Partial<GuildActivityCityType>; // 将所有字段变成可选项