import mongooseUniqueValidator from "mongoose-unique-validator";
import {AppMaster} from "@onebro/appmaster";
import {
  DeepPartial,
  IModelCreator,
  IStatus,
  IQuery,
  mapSelectedData,
  getSchemaRefs,
  ObjectId} from "@onebro/oba-common";
import {
  SwiftTradersMessageConfig,
  SwiftTradersMessageModel,
  SwiftTradersMessage,
  SwiftTradersMessageJson,
  SwiftTradersMessageStatuses as statuses,
  SwiftTradersMessageStatus} from "../types";
import {swiftTradersMessageSchema as msg} from "../schemas";
const refs = getSchemaRefs(msg);

export const swiftTradersMessageModels:IModelCreator<AppMaster> = async m => {
  msg.plugin(mongooseUniqueValidator);
  msg.virtual("lastStatus").get(function(){return this.status?(this.status[this.status.length - 1] as any).json(statuses):null;});
  msg.virtual("lastStatus").set(function(o:IStatus<SwiftTradersMessageStatus>){this.status = [...this.status,o];});
  msg.virtual("statusStr").get(function(){return this.lastStatus?this.lastStatus.name + " @ " + this.lastStatus.time:"";});
  msg.statics._populate = async function(o?:SwiftTradersMessage){
    await o.populate(refs).execPopulate();
    return o;};
  msg.statics._find = async function(o:Partial<SwiftTradersMessage>|string){
    if(!o) return m.e.badinfo();
    const p = typeof o == "string"?
    await SwiftTradersMessage.findById(o):
    await SwiftTradersMessage.findOne(SwiftTradersMessage.translateAliases(o));
    if(!p) throw m.e.doesNotExist("message");
    return p;};
  msg.statics._create = async function(o:SwiftTradersMessageConfig){return new SwiftTradersMessage({
    ...o,
    author:new ObjectId(o.author),
    });};
  msg.statics._update = async function(o:SwiftTradersMessage,p:DeepPartial<SwiftTradersMessage>){
    await o.set(p).save();
    return await SwiftTradersMessage._populate(o);};
  msg.statics._remove = async function(o:SwiftTradersMessage){return await o.remove().then(() => ({removed:o._id}));};
  msg.statics._query = async function(o:IQuery){
    const {query,populate,limit,skip,sort,select} = o;
    const results:SwiftTradersMessage[] = await this.find(this.translateAliases(query))
      .populate(populate)
      //.where(where)
      .limit(limit)
      .skip(skip)
      .sort(sort)
      .exec();
    const msgs = mapSelectedData(select,results);
    return {msgs};};
  msg.statics._lookup = async function({path,query}:{path:string;query:any}){
    const pipeline:any[] = [];
    return SwiftTradersMessage.aggregate(pipeline)
    .exec()
    .then((R:Partial<{[k in keyof (SwiftTradersMessage & SwiftTradersMessageJson)]:any}>[]) => {
      return {msgs:R.map(o => {
        if(o.status) o.status = {
          name:statuses[o.status.a0 as SwiftTradersMessageStatus],
          time:o.status.a1,
          until:o.status.a2,
          info:o.status.a3};
        delete o._id;
        return o;})};
        });};
  msg.methods.json = function(){
    const {
      id,created,updated,lastStatus:status,statusStr,
      type,author,body,img,link,symbol,sendAt,subscribers} = this as SwiftTradersMessage;
    const json:SwiftTradersMessageJson = {
      id,created,updated,status,statusStr,
      type,author:author.preview(),
      body,img,link,symbol,sendAt,
      published:new Date(created),
      subscribers};
    return json;};
  const SwiftTradersMessage = m.db.model<SwiftTradersMessage,SwiftTradersMessageModel>("onebrother","SwiftTradersMessage",msg);
  await SwiftTradersMessage.init();
  return {SwiftTradersMessage};
};