@abtnode/core 1.16.52-beta-20251003-083412-fdfc4e36 → 1.16.52-beta-20251008-091027-c46c73e3

package/lib/states/audit-log.js

@@ -10,6 +10,7 @@ const isArray = require('lodash/isArray');
 const isString = require('lodash/isString');
 const mapValues = require('lodash/mapValues');
 const map = require('lodash/map');
+const omit = require('lodash/omit');
 const { joinURL } = require('ufo');
 const { Op } = require('sequelize');
 const { getDisplayName } = require('@blocklet/meta/lib/util');
@@ -273,13 +274,15 @@ const getLogContent = async (action, args, context, result, info, node) => {
     expandSite(args.id, info, node),
     expandUser(
       args.teamDid,
-      args.userDid || get(args, 'user.did') || args.ownerDid || args.did,
+      args.userDid || get(args, 'user.did') || args.ownerDid || args.did || context.user.did,
       args.passportId,
       info,
       node
     ),
   ]);
 
+  const prefix = process.env.NODE_ENV === 'production' ? info.routing.adminPath : '';
+
   switch (action) {
     // blocklets
     case 'installBlocklet':
@@ -684,7 +687,6 @@ const getLogContent = async (action, args, context, result, info, node) => {
     case 'followUser':
     case 'unfollowUser':
       const followerUser = await node.getUser({ teamDid: args.teamDid, user: { did: args.followerDid } });
-      const prefix = process.env.NODE_ENV === 'production' ? info.routing.adminPath : '';
       return `[${followerUser.fullName}](${joinURL(prefix, '/team/members')}) ${action === 'followUser' ? 'followed' : 'unfollowed'} user ${user}`;
 
     // connect to aigne
@@ -693,6 +695,49 @@ const getLogContent = async (action, args, context, result, info, node) => {
     case 'disconnectToAigne':
       return `Disconnect to Aigne(${args.url})`;
 
+    // org 相关
+    case 'createOrg':
+      return `${user} created an org(${args.name})`;
+    case 'deleteOrg':
+      return `${user} deleted the org(${args.id})`;
+    case 'updateOrg':
+      return `${user} updated the org(${args.org.name}): \n${Object.keys(omit(args.org, ['id']))
+        .map((x) => `- ${x}: ${args.org[x]}`)
+        .join('\n')}`;
+    case 'removeOrgMember': {
+      const [member, org, operator] = await Promise.all([
+        node.getUser({ teamDid: args.teamDid, user: { did: args.userDid } }),
+        node.getOrg({ teamDid: args.teamDid, id: args.orgId }, context),
+        node.getUser({ teamDid: args.teamDid, user: { did: context.user.did } }),
+      ]);
+      return `[${operator.fullName}](${joinURL(prefix, '/team/members')}) removed a member([${member.fullName}](${joinURL(prefix, '/team/members')})) from the org(${org.name})`;
+    }
+    case 'inviteMembersToOrg': {
+      const [operator, joinedOrg] = await Promise.all([
+        node.getUser({ teamDid: args.teamDid, user: { did: context.user.did } }),
+        node.getOrg({ teamDid: args.teamDid, id: args.orgId }, context),
+      ]);
+      if (args.inviteType === 'internal') {
+        const dids = args.userDids || [];
+        const { users } = await node.getUsers({ teamDid: args.teamDid, dids: dids.slice(0, 5) });
+        const invitedMembers = users.length < dids.length ? [...users, '...'] : users;
+        return `[${operator.fullName}](${joinURL(prefix, '/team/members')}) invited ${dids.length} ${dids.length > 1 ? 'members' : 'member'} to join the org(${joinedOrg.name}): \n${invitedMembers.map((x) => (x.fullName ? `- [${x.fullName}](${joinURL(prefix, '/team/members')})` : `- ${x}`)).join('\n')}`;
+      }
+      return `[${operator.fullName}](${joinURL(prefix, '/team/members')}) invited members to join the org(${joinedOrg.name})`;
+    }
+    case 'addOrgResource': {
+      const org = await node.getOrg({ teamDid: args.teamDid, id: args.orgId }, context);
+      const resourceIds = (args.resourceIds || []).slice(0, 5);
+      const displayResourceIds = resourceIds.length < args.resourceIds.length ? [...resourceIds, '...'] : resourceIds;
+      return `${user} added ${resourceIds.length} ${resourceIds.length > 1 ? 'resources' : 'resource'} to the org(${org.name}): \n${displayResourceIds.map((x) => `- ${x}`).join('\n')}`;
+    }
+    case 'migrateOrgResource': {
+      const org = await node.getOrg({ teamDid: args.teamDid, id: args.to }, context);
+      const resourceIds = (args.resourceIds || []).slice(0, 5);
+      const displayResourceIds = resourceIds.length < args.resourceIds.length ? [...resourceIds, '...'] : resourceIds;
+      return `${user} migrated ${resourceIds.length} ${resourceIds.length > 1 ? 'resources' : 'resource'} to the org(${org.name}): \n${displayResourceIds.map((x) => `- ${x}`).join('\n')}`;
+    }
+
     default:
       return action;
   }
@@ -797,6 +842,13 @@ const getLogCategory = (action) => {
     case 'updateUserInfo':
     case 'followUser':
     case 'unfollowUser':
+    case 'createOrg':
+    case 'updateOrg':
+    case 'deleteOrg':
+    case 'removeOrgMember':
+    case 'inviteMembersToOrg':
+    case 'addOrgResource':
+    case 'migrateOrgResource':
       return 'team';
 
     // accessKeys

package/lib/states/blocklet.js

@@ -198,7 +198,10 @@ class BlockletState extends BaseState {
 
   async getBlockletStatus(did) {
     const doc = await this.getBlocklet(did);
-    return doc ? doc.status : null;
+    if (!doc) {
+      return null;
+    }
+    return doc.status;
   }
 
   async hasBlocklet(did) {
@@ -469,7 +472,7 @@ class BlockletState extends BaseState {
   /**
    * refresh ports for blocklet if occupied during starting workflow
    */
-  async refreshBlockletPorts(did, componentDids = []) {
+  async refreshBlockletPorts(did, componentDids = [], isGreen = false) {
     const blocklet = await this.getBlocklet(did);
     if (!blocklet) {
       throw new CustomError(404, `Blocklet does not exist on refresh: ${did}`);
@@ -479,7 +482,11 @@ class BlockletState extends BaseState {
 
     await forEachComponentV2(blocklet, async (component) => {
       if (!shouldSkipComponent(component.meta.did, componentDids)) {
-        component.ports = await refreshPorts(component.ports, {
+        let oldPorts = component[isGreen ? 'greenPorts' : 'ports'];
+        if (!oldPorts || Object.keys(oldPorts).length === 0) {
+          oldPorts = component.ports;
+        }
+        component[isGreen ? 'greenPorts' : 'ports'] = await refreshPorts(oldPorts, {
           blackList: [...occupiedExternalPorts.keys(), ...occupiedInternalPorts.keys()],
         });
       }
@@ -556,7 +563,11 @@ class BlockletState extends BaseState {
    *   componentDids?: Array<Did>
    * }}
    */
-  async setBlockletStatus(did, status, { componentDids, operator = 'daemon' } = {}) {
+  async setBlockletStatus(
+    did,
+    status,
+    { componentDids, operator = 'daemon', isGreen = false, isGreenAndBlue = false } = {}
+  ) {
     logger.info('setBlockletStatus', { did, status, componentDids, operator });
     if (typeof status === 'undefined') {
       throw new Error('Unsupported blocklet status');
@@ -589,7 +600,11 @@ class BlockletState extends BaseState {
           return;
         }
 
-        component.status = status;
+        component[isGreen ? 'greenStatus' : 'status'] = status;
+        if (isGreenAndBlue) {
+          component.greenStatus = status;
+          component.status = status;
+        }
         if (status === BlockletStatus.running) {
           component.startedAt = new Date();
           component.stoppedAt = null;
@@ -602,7 +617,8 @@ class BlockletState extends BaseState {
         component.inProgressStart = Date.now();
       });
 
-      const updateData = pick(doc, ['status', 'children']);
+      const isSetStatus = status === BlockletStatus.downloading || status === BlockletStatus.waiting;
+      const updateData = isSetStatus ? pick(doc, ['status', 'children']) : pick(doc, ['children']);
       updateData.operator = operator;
 
       const res = await this.updateBlocklet(did, updateData);

package/lib/states/index.js

@@ -18,6 +18,7 @@ const BackupState = require('./backup');
 const TrafficInsightState = require('./traffic-insight');
 const RuntimeInsightState = require('./runtime-insight');
 const BlacklistState = require('./blacklist');
+const OrgState = require('./org');
 
 const { getDbFilePath } = require('../util');
 
@@ -46,6 +47,7 @@ const init = (dataDirs, config = {}) => {
   const trafficInsight = new TrafficInsightState(models.TrafficInsight, config);
   const runtimeInsight = new RuntimeInsightState(models.RuntimeInsight, { ...config, maxPageSize: 8640 });
   const blacklistState = new BlacklistState(models.Blacklist, config);
+  const orgState = new OrgState(models.Org, config, models);
 
   return {
     node: nodeState,
@@ -64,6 +66,7 @@ const init = (dataDirs, config = {}) => {
     trafficInsight,
     runtimeInsight,
     blacklist: blacklistState,
+    org: orgState,
   };
 };