80 lines
1.8 KiB
TypeScript
80 lines
1.8 KiB
TypeScript
import { TeamCollectionName } from '@fastgpt/global/support/user/team/constant';
|
|
import { OrgCollectionName } from '@fastgpt/global/support/user/team/org/constant';
|
|
import type { OrgSchemaType } from '@fastgpt/global/support/user/team/org/type';
|
|
import { connectionMongo, getMongoModel } from '../../../common/mongo';
|
|
import { OrgMemberCollectionName } from './orgMemberSchema';
|
|
import { getNanoid } from '@fastgpt/global/common/string/tools';
|
|
const { Schema } = connectionMongo;
|
|
|
|
function requiredStringPath(this: OrgSchemaType) {
|
|
return typeof this.path !== 'string';
|
|
}
|
|
|
|
export const OrgSchema = new Schema(
|
|
{
|
|
teamId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: TeamCollectionName,
|
|
required: true
|
|
},
|
|
pathId: {
|
|
// path id, only used for path
|
|
type: String,
|
|
required: true,
|
|
default: () => getNanoid()
|
|
},
|
|
path: {
|
|
type: String,
|
|
required: requiredStringPath // allow empty string, but not null
|
|
},
|
|
name: {
|
|
type: String,
|
|
required: true
|
|
},
|
|
avatar: String,
|
|
description: String,
|
|
updateTime: {
|
|
type: Date,
|
|
default: () => new Date()
|
|
}
|
|
},
|
|
{
|
|
// Auto update updateTime
|
|
timestamps: {
|
|
updatedAt: 'updateTime'
|
|
}
|
|
}
|
|
);
|
|
|
|
OrgSchema.virtual('members', {
|
|
ref: OrgMemberCollectionName,
|
|
localField: '_id',
|
|
foreignField: 'orgId'
|
|
});
|
|
// OrgSchema.virtual('permission', {
|
|
// ref: ResourcePermissionCollectionName,
|
|
// localField: '_id',
|
|
// foreignField: 'orgId',
|
|
// justOne: true
|
|
// });
|
|
|
|
try {
|
|
OrgSchema.index({
|
|
teamId: 1,
|
|
path: 1
|
|
});
|
|
OrgSchema.index(
|
|
{
|
|
teamId: 1,
|
|
pathId: 1
|
|
},
|
|
{
|
|
unique: true
|
|
}
|
|
);
|
|
} catch (error) {
|
|
console.log(error);
|
|
}
|
|
|
|
export const MongoOrgModel = getMongoModel<OrgSchemaType>(OrgCollectionName, OrgSchema);
|