64 lines
2.0 KiB
JavaScript
64 lines
2.0 KiB
JavaScript
import mongoose from 'mongoose';
|
|
const { Schema } = mongoose;
|
|
|
|
// Define the moonraker connection schema
|
|
const moonrakerSchema = new Schema(
|
|
{
|
|
host: { type: String, required: true },
|
|
port: { type: Number, required: true },
|
|
protocol: { type: String, required: true },
|
|
apiKey: { type: String, default: null, required: false },
|
|
},
|
|
{ _id: false }
|
|
);
|
|
|
|
// Define the alert schema
|
|
const alertSchema = new Schema(
|
|
{
|
|
priority: { type: String, required: true }, // order to show
|
|
type: { type: String, required: true }, // selectFilament, error, info, message,
|
|
message: { type: String, required: false },
|
|
},
|
|
{ timestamps: true, _id: false }
|
|
);
|
|
|
|
// Define the main printer schema
|
|
const printerSchema = new Schema(
|
|
{
|
|
name: { type: String, required: true },
|
|
online: { type: Boolean, required: true, default: false },
|
|
state: {
|
|
type: { type: String, required: true, default: 'offline' },
|
|
percent: { type: Number, required: false },
|
|
},
|
|
connectedAt: { type: Date, default: null },
|
|
loadedFilament: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: 'filament',
|
|
default: null,
|
|
},
|
|
moonraker: { type: moonrakerSchema, required: true },
|
|
tags: [{ type: String }],
|
|
firmware: { type: String },
|
|
currentJob: { type: Schema.Types.ObjectId, ref: 'job' },
|
|
currentSubJob: { type: Schema.Types.ObjectId, ref: 'subJob' },
|
|
currentFilamentStock: { type: Schema.Types.ObjectId, ref: 'filamentStock' },
|
|
subJobs: [{ type: Schema.Types.ObjectId, ref: 'subJob' }],
|
|
vendor: { type: Schema.Types.ObjectId, ref: 'vendor', default: null },
|
|
host: { type: Schema.Types.ObjectId, ref: 'host', default: null },
|
|
alerts: [alertSchema],
|
|
},
|
|
{ timestamps: true }
|
|
);
|
|
|
|
// Add virtual id getter
|
|
printerSchema.virtual('id').get(function () {
|
|
return this._id.toHexString();
|
|
});
|
|
|
|
// Configure JSON serialization to include virtuals
|
|
printerSchema.set('toJSON', { virtuals: true });
|
|
|
|
// Create and export the model
|
|
export const printerModel = mongoose.model('printer', printerSchema);
|