import mongooseUniqueValidator from "mongoose-unique-validator"; import {AppMaster} from "@onebro/appmaster"; import { DeepPartial, IModelCreator, IStatus, IQuery, mapSelectedData, getSchemaRefs, ObjectId} from "@onebro/oba-common"; import { SwiftTradersMPAlertConfig, SwiftTradersMPAlertModel, SwiftTradersMPAlert, SwiftTradersMPAlertJson, SwiftTradersMPAlertStatuses as statuses, SwiftTradersMPAlertStatus} from "../types"; import {swiftTradersMPAlertSchema as alert} from "../schemas"; const refs = getSchemaRefs(alert); export const swiftTradersMPAlertModels: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?:SwiftTradersMPAlert){ 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 SwiftTradersMPAlert.findById(o): await SwiftTradersMPAlert.findOne(SwiftTradersMPAlert.translateAliases(o)); if(!p) throw m.e.doesNotExist("alert"); return p;}; alert.statics._create = async function(o:SwiftTradersMPAlertConfig){return new SwiftTradersMPAlert({ ...o, author:new ObjectId(o.author), });}; alert.statics._update = async function(o:SwiftTradersMPAlert,p:DeepPartial){ await o.set(p).save(); return await SwiftTradersMPAlert._populate(o);}; alert.statics._remove = async function(o:SwiftTradersMPAlert){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:SwiftTradersMPAlert[] = 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 SwiftTradersMPAlert.aggregate(pipeline) .exec() .then((R:Partial<{[k in keyof (SwiftTradersMPAlert & SwiftTradersMPAlertJson)]:any}>[]) => { return {alerts:R.map(o => { if(o.status) o.status = { name:statuses[o.status.a0 as SwiftTradersMPAlertStatus], 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 SwiftTradersMPAlert; const json:SwiftTradersMPAlertJson = { id,created,updated,status,statusStr, type,author:author.preview(), body,img,link,symbol,sendAt, published:new Date(created), subscribers}; return json;}; const SwiftTradersMPAlert = m.db.model("onebrother","SwiftTradersMPAlert",alert); await SwiftTradersMPAlert.init(); return {SwiftTradersMPAlert}; };