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