@did-space/core 0.1.68 → 0.1.70

package/dist/space/index.d.ts

@@ -24,6 +24,11 @@ export declare class Space implements SpaceProtocol {
     isListable(options: PermissionOptions): Promise<boolean>;
     isReadable(options: PermissionOptions): Promise<boolean>;
     isWritable(options: PermissionOptions): Promise<boolean>;
+    getSpaceStatus(): Promise<{
+        totalCapacity: number;
+        usedCapacity: number;
+        isFull: boolean;
+    }>;
     write(options: WriteOptions): Promise<void>;
     delete(options: DeleteOptions): Promise<void>;
     read(options: ReadOptions): Promise<Stream>;

package/dist/space/index.js

@@ -8,8 +8,12 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Space = void 0;
+const xbytes_1 = __importDefault(require("xbytes"));
 class Space {
     constructor(driver) {
         this.driver = driver;
@@ -99,8 +103,48 @@ class Space {
             return this.driver.isWritable(options);
         });
     }
+    getSpaceStatus() {
+        var _a, _b;
+        return __awaiter(this, void 0, void 0, function* () {
+            // FIXME: 临时性解决方案,需要新增拓容功能,需要去除白名单功能
+            const [totalCapacity, usedCapacity] = yield Promise.all([this.driver.get('size'), this.driver.getSpaceSize()]);
+            const whiteList = ((_b = (_a = process.env) === null || _a === void 0 ? void 0 : _a.SPACE_WHITE_LIST) !== null && _b !== void 0 ? _b : '')
+                .split(',')
+                .map((s) => s.trim())
+                .filter(Boolean);
+            // 跳过白名单
+            if (whiteList.length && whiteList.includes(this.driver.options.root)) {
+                return {
+                    isFull: false,
+                    totalCapacity,
+                    usedCapacity,
+                };
+            }
+            // 存储空间不足
+            if (usedCapacity >= totalCapacity) {
+                return {
+                    isFull: true,
+                    totalCapacity,
+                    usedCapacity,
+                };
+            }
+            return {
+                isFull: false,
+                totalCapacity,
+                usedCapacity,
+            };
+        });
+    }
     write(options) {
         return __awaiter(this, void 0, void 0, function* () {
+            // FIXME: 插入数据和编辑数据应该分离?
+            if (yield this.exists(options)) {
+                return this.driver.write(options);
+            }
+            const spaceStatus = yield this.getSpaceStatus();
+            if (spaceStatus.isFull) {
+                throw new Error(`Insufficient free space, total capacity is ${(0, xbytes_1.default)(spaceStatus.totalCapacity)}, used capacity is ${(0, xbytes_1.default)(spaceStatus.usedCapacity)}`);
+            }
             return this.driver.write(options);
         });
     }
@@ -136,6 +180,14 @@ class Space {
     }
     writeAsOwner(key, data) {
         return __awaiter(this, void 0, void 0, function* () {
+            // FIXME: 插入数据和编辑数据应该分离
+            if (yield this.existsAsOwner(key)) {
+                return this.driver.writeAsOwner(key, data);
+            }
+            const spaceStatus = yield this.getSpaceStatus();
+            if (spaceStatus.isFull) {
+                throw new Error(`Insufficient free space, total capacity is ${(0, xbytes_1.default)(spaceStatus.totalCapacity)}, used capacity is ${(0, xbytes_1.default)(spaceStatus.usedCapacity)}`);
+            }
             return this.driver.writeAsOwner(key, data);
         });
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@did-space/core",
-  "version": "0.1.68",
+  "version": "0.1.70",
   "publishConfig": {
     "access": "public"
   },
@@ -61,5 +61,5 @@
     "ts-jest": "^28.0.6",
     "typescript": "4.4.4"
   },
-  "gitHead": "df55bed8c071fbc8f72a6566914c2e698bc4f5e5"
+  "gitHead": "1c67a8376a2ab06e06c771d34149be4fefe1e7c7"
 }