import mongoose from "mongoose";
import { ObjectID } from "mongodb";

const Schema = mongoose.Schema;

ObjectID.prototype.valueOf = function() {
  return this.toString();
};

const CompanySchema = new Schema(
  {
    name: {
      type: String,
      unique: true,
      required: true
    },
    logo: {
      type: String,
      unique: true,
      required: true
    },
    level: {
      type: Number,
      default: 0
    },
    freights: [
      {
        type: Schema.Types.ObjectId,
        ref: "Freight"
      }
    ]
  },
  { timestamps: { updatedAt: "updated_at", createdAt: "created_at" } }
);

export default mongoose.model("Company", CompanySchema);