@nocobase/plugin-data-source-manager 1.5.0-beta.1 → 1.6.0-alpha.1

package/dist/server/plugin.js

@@ -58,6 +58,81 @@ const canRefreshStatus = ["loaded", "loading-failed", "reloading-failed"];
 class PluginDataSourceManagerServer extends import_server.Plugin {
   dataSourceErrors = {};
   dataSourceStatus = {};
+  async handleSyncMessage(message) {
+    const { type } = message;
+    if (type === "syncRole") {
+      const { roleName, dataSourceKey } = message;
+      const dataSource = this.app.dataSourceManager.dataSources.get(dataSourceKey);
+      const dataSourceRole = await this.app.db.getRepository("dataSourcesRoles").findOne({
+        filter: {
+          dataSourceKey,
+          roleName
+        }
+      });
+      await dataSourceRole.writeToAcl({
+        acl: dataSource.acl
+      });
+    }
+    if (type === "syncRoleResource") {
+      const { roleName, dataSourceKey, resourceName } = message;
+      const dataSource = this.app.dataSourceManager.dataSources.get(dataSourceKey);
+      const dataSourceRoleResource = await this.app.db.getRepository("dataSourcesRolesResources").findOne({
+        filter: {
+          dataSourceKey,
+          roleName,
+          name: resourceName
+        }
+      });
+      await dataSourceRoleResource.writeToACL({
+        acl: dataSource.acl
+      });
+    }
+    if (type === "loadDataSource") {
+      const { dataSourceKey } = message;
+      const dataSourceModel = await this.app.db.getRepository("dataSources").findOne({
+        filter: {
+          key: dataSourceKey
+        }
+      });
+      if (!dataSourceModel) {
+        return;
+      }
+      await dataSourceModel.loadIntoApplication({
+        app: this.app
+      });
+    }
+    if (type === "loadDataSourceField") {
+      const { key } = message;
+      const fieldModel = await this.app.db.getRepository("dataSourcesFields").findOne({
+        filter: {
+          key
+        }
+      });
+      fieldModel.load({
+        app: this.app
+      });
+    }
+    if (type === "removeDataSourceCollection") {
+      const { dataSourceKey, collectionName } = message;
+      const dataSource = this.app.dataSourceManager.dataSources.get(dataSourceKey);
+      dataSource.collectionManager.removeCollection(collectionName);
+    }
+    if (type === "removeDataSourceField") {
+      const { key } = message;
+      const fieldModel = await this.app.db.getRepository("dataSourcesFields").findOne({
+        filter: {
+          key
+        }
+      });
+      fieldModel.unload({
+        app: this.app
+      });
+    }
+    if (type === "removeDataSource") {
+      const { dataSourceKey } = message;
+      this.app.dataSourceManager.dataSources.delete(dataSourceKey);
+    }
+  }
   dataSourceLoadingProgress = {};
   async beforeLoad() {
     this.app.db.registerModels({
@@ -114,6 +189,15 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
         model.loadIntoApplication({
           app: this.app
         });
+        this.sendSyncMessage(
+          {
+            type: "loadDataSource",
+            dataSourceKey: model.get("key")
+          },
+          {
+            transaction: options.transaction
+          }
+        );
       }
     });
     this.app.db.on("dataSources.afterCreate", async (model, options) => {
@@ -241,6 +325,7 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
         }
       }
     });
+    const self = this;
     this.app.actions({
       async ["dataSources:listEnabled"](ctx, next) {
         const dataSources = await ctx.db.getRepository("dataSources").find({
@@ -280,6 +365,10 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
           dataSourceModel.loadIntoApplication({
             app: ctx.app
           });
+          ctx.app.syncMessageManager.publish(self.name, {
+            type: "loadDataSource",
+            dataSourceKey: dataSourceModel.get("key")
+          });
         }
         ctx.body = {
           status: plugin.dataSourceStatus[filterByTk]
@@ -307,21 +396,49 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
         model.set("dataSourceKey", collection.get("dataSourceKey"));
       }
     });
-    this.app.db.on("dataSourcesCollections.afterDestroy", async (model) => {
+    this.app.db.on("dataSourcesCollections.afterDestroy", async (model, options) => {
       const dataSource = this.app.dataSourceManager.dataSources.get(model.get("dataSourceKey"));
       if (dataSource) {
         dataSource.collectionManager.removeCollection(model.get("name"));
       }
+      this.sendSyncMessage(
+        {
+          type: "removeDataSourceCollection",
+          dataSourceKey: model.get("dataSourceKey"),
+          collectionName: model.get("name")
+        },
+        {
+          transaction: options.transaction
+        }
+      );
     });
-    this.app.db.on("dataSourcesFields.afterSaveWithAssociations", async (model) => {
+    this.app.db.on("dataSourcesFields.afterSaveWithAssociations", async (model, options) => {
       model.load({
         app: this.app
       });
+      this.sendSyncMessage(
+        {
+          type: "loadDataSourceField",
+          key: model.get("key")
+        },
+        {
+          transaction: options.transaction
+        }
+      );
     });
-    this.app.db.on("dataSourcesFields.afterDestroy", async (model) => {
+    this.app.db.on("dataSourcesFields.afterDestroy", async (model, options) => {
       model.unload({
         app: this.app
       });
+      this.sendSyncMessage(
+        {
+          type: "removeDataSourceField",
+          key: model.get("key")
+        },
+        {
+          transaction: options.transaction
+        }
+      );
     });
     this.app.db.on(
       "dataSourcesCollections.afterSaveWithAssociations",
@@ -332,8 +449,17 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
         });
       }
     );
-    this.app.db.on("dataSources.afterDestroy", async (model) => {
+    this.app.db.on("dataSources.afterDestroy", async (model, options) => {
       this.app.dataSourceManager.dataSources.delete(model.get("key"));
+      this.sendSyncMessage(
+        {
+          type: "removeDataSource",
+          dataSourceKey: model.get("key")
+        },
+        {
+          transaction: options.transaction
+        }
+      );
     });
     this.app.on("afterStart", async (app2) => {
       const dataSourcesRecords = await this.app.db.getRepository("dataSources").find({
@@ -359,6 +485,17 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
           acl: dataSource.acl,
           transaction
         });
+        this.sendSyncMessage(
+          {
+            type: "syncRoleResource",
+            roleName: model.get("roleName"),
+            dataSourceKey: model.get("dataSourceKey"),
+            resourceName: model.get("name")
+          },
+          {
+            transaction
+          }
+        );
       }
     );
     this.app.db.on(
@@ -373,6 +510,17 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
           acl: dataSource.acl,
           transaction
         });
+        this.sendSyncMessage(
+          {
+            type: "syncRoleResource",
+            roleName: resource.get("roleName"),
+            dataSourceKey: resource.get("dataSourceKey"),
+            resourceName: resource.get("name")
+          },
+          {
+            transaction
+          }
+        );
       }
     );
     this.app.db.on("dataSourcesRolesResources.afterDestroy", async (model, options) => {
@@ -382,6 +530,17 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
       if (role) {
         role.revokeResource(model.get("name"));
       }
+      this.sendSyncMessage(
+        {
+          type: "syncRoleResource",
+          roleName,
+          dataSourceKey: model.get("dataSourceKey"),
+          resourceName: model.get("name")
+        },
+        {
+          transaction: options.transaction
+        }
+      );
     });
     this.app.db.on("dataSourcesRoles.afterSave", async (model, options) => {
       const { transaction } = options;
@@ -400,6 +559,16 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
         hooks: false,
         transaction
       });
+      this.sendSyncMessage(
+        {
+          type: "syncRole",
+          roleName: model.get("roleName"),
+          dataSourceKey: model.get("dataSourceKey")
+        },
+        {
+          transaction
+        }
+      );
     });
     this.app.on("acl:writeResources", async ({ roleName, transaction }) => {
       const dataSource = this.app.dataSourceManager.dataSources.get("main");
@@ -415,7 +584,6 @@ class PluginDataSourceManagerServer extends import_server.Plugin {
         transaction
       });
     });
-    const self = this;
     this.app.resourceManager.use(async function appendDataToRolesCheck(ctx, next) {
       const action = ctx.action;
       await next();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nocobase/plugin-data-source-manager",
-  "version": "1.5.0-beta.1",
+  "version": "1.6.0-alpha.1",
   "main": "dist/server/index.js",
   "displayName": "Data source manager",
   "displayName.zh-CN": "数据源管理",
@@ -17,5 +17,5 @@
   "keywords": [
     "Data model tools"
   ],
-  "gitHead": "10c6f1f3d90e91f3aabfa80449c7ef062e90f6af"
+  "gitHead": "a4c91015e34ec0c9a427451b5fbdfb5fedc4f3d7"
 }