import { LotRec, DividendRec } from './../domain/dbGeneral';
import BaseModel from './BaseModel';
import { index, getModelForClass, prop, DocumentType, modelOptions } from '@typegoose/typegoose';
import { genCode } from '../pubUtils/util';

/**
  * 分红记录表
**/
@modelOptions({ schemaOptions: { id: false } })
@index({ code: 1 })
@index({ guildCode: 1 })
export default class Dividend extends BaseModel {
    @prop({ required: true, default: 0 })
    type: number; // 0：初始值，1：演武；2：蛮夷入侵；3：诸侯混战；4：粮草先行
    @prop({ required: true, default: '' })
    guildCode: string; // 军团编号
    @prop({ required: true, default: '' })
    sourceCode: string; // 来源的唯一标识，如活动编号
    @prop({ required: true, default: '' })
    code: string; // 分红记录唯一标识
    @prop({ required: true, type: LotRec, default: [] })
    lots: LotRec[];
    @prop({ required: true, default: 0 })
    totalPrice: number; // 分红总金额
    @prop({ required: true, type: DividendRec, default: [] })
    dividends: DividendRec[];

    public static async createDividend(data: DividendParam) {
        const code = genCode(8);
        const docData = new DividendModel();
        const result: DividendType = await DividendModel.findOneAndUpdate({ code }, { ...docData.toJSON(), ...data, code }, { upsert: true, new: true }).select('-_id -__v').lean();
        return result;
    }

    public static async findDividend(code: string) {
        const result = await DividendModel.findOne({ code }).select('-_id -__v').lean();
        return result;
    }
}

export const DividendModel = getModelForClass(Dividend);

export interface DividendType extends Pick<DocumentType<Dividend>, keyof Dividend>{}
export type DividendParam = Partial<DividendType>;