relnet/src/models/relationship.model.ts

import mongoose, { Document, Schema } from 'mongoose';


export const RELATIONSHIP_TYPES = [
  'acquaintance', 'friend', 'partner', 'family', 'secondDegree', 'colleague', 'teacher', 'exPartner', 'custom',
];

export interface IRelationship extends Document {
  _id: string;
  source: mongoose.Types.ObjectId;
  target: mongoose.Types.ObjectId;
  type: string;
  customType?: string;
  network: mongoose.Types.ObjectId;
}

const RelationshipSchema = new Schema(
  {
    source: {
      type: Schema.Types.ObjectId,
      ref: 'Person',
      required: true,
    },
    target: {
      type: Schema.Types.ObjectId,
      ref: 'Person',
      required: true,
    },
    type: {
      type: String,
      required: [true, 'Relationship type is required'],
      enum: RELATIONSHIP_TYPES,
    },
    customType: {
      type: String,
      trim: true,
    },
    network: {
      type: Schema.Types.ObjectId,
      ref: 'Network',
      required: true,
    },
  },
  { timestamps: true },
);

// Create compound index to ensure unique relationships in a network
RelationshipSchema.index({ source: 1, target: 1, network: 1 }, { unique: true });

export default mongoose.model<IRelationship>('Relationship', RelationshipSchema);