2022-03-04 23:17:28 +08:00
|
|
|
import Database from '@nocobase/database';
|
2022-01-19 10:02:52 +08:00
|
|
|
import { uid } from '@nocobase/utils';
|
|
|
|
import { Model } from 'sequelize';
|
|
|
|
|
2022-03-04 23:17:28 +08:00
|
|
|
const setTargetKey = (db: Database, model: Model) => {
|
2022-04-09 14:10:00 +08:00
|
|
|
if (model.get('targetKey')) {
|
|
|
|
return;
|
|
|
|
}
|
2022-03-04 23:17:28 +08:00
|
|
|
const target = model.get('target') as any;
|
|
|
|
if (db.hasCollection(target)) {
|
|
|
|
const targetModel = db.getCollection(target).model;
|
|
|
|
model.set('targetKey', targetModel.primaryKeyAttribute || 'id');
|
2022-04-09 14:10:00 +08:00
|
|
|
} else {
|
|
|
|
model.set('targetKey', 'id');
|
2022-03-04 23:17:28 +08:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
const setSourceKey = (db: Database, model: Model) => {
|
2022-04-09 14:10:00 +08:00
|
|
|
if (model.get('sourceKey')) {
|
|
|
|
return;
|
|
|
|
}
|
2022-03-04 23:17:28 +08:00
|
|
|
const source = model.get('collectionName') as any;
|
|
|
|
if (db.hasCollection(source)) {
|
|
|
|
const sourceModel = db.getCollection(source).model;
|
|
|
|
model.set('sourceKey', sourceModel.primaryKeyAttribute || 'id');
|
2022-04-09 14:10:00 +08:00
|
|
|
} else {
|
|
|
|
model.set('sourceKey', 'id');
|
2022-03-04 23:17:28 +08:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2022-02-14 21:57:13 +08:00
|
|
|
export const beforeInitOptions = {
|
2022-03-04 23:17:28 +08:00
|
|
|
belongsTo(model: Model, { database }) {
|
2022-01-19 10:02:52 +08:00
|
|
|
const defaults = {
|
2022-04-09 14:10:00 +08:00
|
|
|
// targetKey: 'id',
|
2022-01-19 10:02:52 +08:00
|
|
|
foreignKey: `f_${uid()}`,
|
|
|
|
};
|
|
|
|
for (const key in defaults) {
|
|
|
|
if (model.get(key)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
model.set(key, defaults[key]);
|
|
|
|
}
|
2022-03-04 23:17:28 +08:00
|
|
|
setTargetKey(database, model);
|
2022-01-19 10:02:52 +08:00
|
|
|
},
|
2022-03-04 23:17:28 +08:00
|
|
|
belongsToMany(model: Model, { database }) {
|
2022-01-19 10:02:52 +08:00
|
|
|
const defaults = {
|
2022-04-09 14:10:00 +08:00
|
|
|
// targetKey: 'id',
|
|
|
|
// sourceKey: 'id',
|
2022-01-19 10:02:52 +08:00
|
|
|
through: `t_${uid()}`,
|
|
|
|
foreignKey: `f_${uid()}`,
|
|
|
|
otherKey: `f_${uid()}`,
|
|
|
|
};
|
|
|
|
for (const key in defaults) {
|
|
|
|
if (model.get(key)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
model.set(key, defaults[key]);
|
|
|
|
}
|
2022-03-04 23:17:28 +08:00
|
|
|
setTargetKey(database, model);
|
|
|
|
setSourceKey(database, model);
|
2022-01-19 10:02:52 +08:00
|
|
|
},
|
2022-03-04 23:17:28 +08:00
|
|
|
hasMany(model: Model, { database }) {
|
2022-01-19 10:02:52 +08:00
|
|
|
const defaults = {
|
2022-04-09 14:10:00 +08:00
|
|
|
// targetKey: 'id',
|
|
|
|
// sourceKey: 'id',
|
2022-01-19 10:02:52 +08:00
|
|
|
foreignKey: `f_${uid()}`,
|
|
|
|
target: `t_${uid()}`,
|
|
|
|
};
|
|
|
|
for (const key in defaults) {
|
|
|
|
if (model.get(key)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
model.set(key, defaults[key]);
|
|
|
|
}
|
2022-03-04 23:17:28 +08:00
|
|
|
setTargetKey(database, model);
|
|
|
|
setSourceKey(database, model);
|
2022-01-19 10:02:52 +08:00
|
|
|
},
|
2022-03-04 23:17:28 +08:00
|
|
|
hasOne(model: Model, { database }) {
|
2022-01-19 10:02:52 +08:00
|
|
|
const defaults = {
|
2022-04-09 14:10:00 +08:00
|
|
|
// sourceKey: 'id',
|
2022-01-19 10:02:52 +08:00
|
|
|
foreignKey: `f_${uid()}`,
|
|
|
|
};
|
|
|
|
for (const key in defaults) {
|
|
|
|
if (model.get(key)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
model.set(key, defaults[key]);
|
|
|
|
}
|
2022-03-04 23:17:28 +08:00
|
|
|
setSourceKey(database, model);
|
2022-01-19 10:02:52 +08:00
|
|
|
},
|
|
|
|
};
|