@axium/storage 0.14.0 → 0.14.2

package/dist/server/config.js

@@ -1,8 +1,14 @@
 import { getConfig, Severity } from '@axium/core';
 import { addEvent } from '@axium/server/audit';
+import { statfsSync } from 'node:fs';
 import * as z from 'zod';
 import '../polyfills.js';
+import { getTotalUse } from './db.js';
 export const defaultCASMime = [/video\/.*/, /audio\/.*/];
+function getSystemAvailable() {
+    const { bavail, bsize } = statfsSync(getConfig('@axium/storage').data, { bigint: true });
+    return (bavail * bsize) / 1000000n;
+}
 addEvent({
     source: '@axium/storage',
     name: 'storage_type_mismatch',
@@ -24,11 +30,26 @@ let _getLimits = null;
 export function useLimits(handler) {
     _getLimits = handler;
 }
+let _cachedUnlimited, _cachedTime = 0;
+/**
+ * Used when there is no user limit, that way users have an idea of how close they are to filling up system storage.
+ * @returns The maximum size that we can use in MB
+ */
+async function _unlimitedLimit() {
+    if (_cachedUnlimited && Date.now() - _cachedTime < 300_000)
+        return _cachedUnlimited;
+    _cachedUnlimited = getSystemAvailable() + (await getTotalUse()) / 1000000n;
+    _cachedTime = Date.now();
+    return _cachedUnlimited;
+}
 export async function getLimits(userId) {
+    let limits;
     try {
-        return await _getLimits(userId);
+        limits = await _getLimits(userId);
     }
     catch {
-        return getConfig('@axium/storage').limits;
+        limits = structuredClone(getConfig('@axium/storage').limits);
     }
+    limits.user_size ||= Number(await _unlimitedLimit());
+    return limits;
 }

package/dist/server/db.d.ts

@@ -42,3 +42,4 @@ export declare function getRecursive(this: {
 }>;
 export declare function getRecursiveIds(...ids: string[]): AsyncGenerator<string>;
 export declare function deleteRecursive(deleteSelf: boolean, ...itemId: string[]): Promise<void>;
+export declare function getTotalUse(): Promise<bigint>;

package/dist/server/db.js

@@ -64,3 +64,10 @@ export async function deleteRecursive(deleteSelf, ...itemId) {
     for (const id of toDelete)
         unlinkSync(join(getConfig('@axium/storage').data, id));
 }
+export async function getTotalUse() {
+    const { size } = await database
+        .selectFrom('storage')
+        .select(eb => eb.fn.sum('size').as('size'))
+        .executeTakeFirstOrThrow();
+    return BigInt(size);
+}

package/dist/server/hooks.js

@@ -5,15 +5,13 @@ import { count, database } from '@axium/server/database';
 import { mkdirSync } from 'node:fs';
 import '../common.js';
 import './index.js';
+import { getTotalUse } from './db.js';
 export function load() {
     mkdirSync(getConfig('@axium/storage').data, { recursive: true });
 }
 export async function statusText() {
     const { storage: items } = await count('storage');
-    const { size } = await database
-        .selectFrom('storage')
-        .select(eb => eb.fn.sum('size').as('size'))
-        .executeTakeFirstOrThrow();
+    const size = await getTotalUse();
     return `${items} items totaling ${formatBytes(Number(size))}`;
 }
 export async function clean(opt) {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@axium/storage",
-	"version": "0.14.0",
+	"version": "0.14.2",
 	"author": "James Prevett <axium@jamespre.dev>",
 	"description": "User file storage for Axium",
 	"funding": {