infisical/backend/src/models/machineMembershipOrg.ts

import { Schema, Types, model } from "mongoose";
import { ADMIN, CUSTOM, MEMBER, NO_ACCESS} from "../variables";

export interface IMachineMembershipOrg {
  _id: Types.ObjectId;
  machineIdentity: Types.ObjectId;
  organization: Types.ObjectId;
  role: "admin" | "member" | "no-access" | "custom";
  customRole: Types.ObjectId;
}

const machineMembershipOrgSchema = new Schema<IMachineMembershipOrg>(
  {
    machineIdentity: {
      type: Schema.Types.ObjectId,
      ref: "MachineIdentity"
    },
    organization: {
      type: Schema.Types.ObjectId,
      ref: "Organization"
    },
    role: {
      type: String,
      enum: [ADMIN, MEMBER, NO_ACCESS, CUSTOM],
      required: true
    },
    customRole: {
      type: Schema.Types.ObjectId,
      ref: "Role"
    }
  },
  {
    timestamps: true
  }
);

export const MachineMembershipOrg = model<IMachineMembershipOrg>("MachineMembershipOrg", machineMembershipOrgSchema);