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);