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