import mongoose from 'mongoose';
import { generateId } from '../../utils.js';
const { Schema } = mongoose;

// Define the main filament SKU schema - color and cost live at SKU level
const filamentSkuSchema = new Schema(
  {
    _reference: { type: String, default: () => generateId()() },
    barcode: { type: String, required: false },
    filament: { type: Schema.Types.ObjectId, ref: 'filament', required: true },
    name: { type: String, required: true },
    description: { type: String, required: false },
    color: { type: String, required: true },
    cost: { type: Number, required: false },
    overrideCost: { type: Boolean, default: false },
    costTaxRate: { type: Schema.Types.ObjectId, ref: 'taxRate', required: false },
    costWithTax: { type: Number, required: false },
  },
  { timestamps: true }
);

filamentSkuSchema.index({ name: 'text', barcode: 'text', description: 'text', color: 'text' });

// Add virtual id getter
filamentSkuSchema.virtual('id').get(function () {
  return this._id;
});

// Configure JSON serialization to include virtuals
filamentSkuSchema.set('toJSON', { virtuals: true });

filamentSkuSchema.statics.recalculate = async function (filamentSku, user) {
  const orderItemModel = mongoose.model('orderItem');
  const skuId = filamentSku._id;
  const draftOrderItems = await orderItemModel
    .find({
      'state.type': 'draft',
      itemType: 'filament',
      sku: skuId,
    })
    .populate('order')
    .lean();

  for (const orderItem of draftOrderItems) {
    await orderItemModel.recalculate(orderItem, user);
  }
};

// Create and export the model
export const filamentSkuModel = mongoose.model('filamentSku', filamentSkuSchema);