const mongoose = require('mongoose') const ObjectID = mongoose.Schema.Types.ObjectId const fileSchema = new mongoose.Schema( { fileName: { type: String, required: true, }, owner: { type: ObjectID, ref: 'User', }, fileEndpoint: String, bucketName: { type: String, required: true, }, }, { toJSON: { virtuals: true }, toObject: { virtuals: true }, }, { timestamps: true } ) fileSchema.virtual('publicUrl').get(function () { return this.fileEndpoint + '/' + this.bucketName + '/' + this.fileName }) fileSchema.virtual('privateUrl').get(function () { return process.env.API_URL + '/api/files/get-image/' + this.bucketName + '/' + this.fileName }) const File = mongoose.model('file', fileSchema) module.exports = File