@chdlabs/my-shield 0.2.0 → 0.3.1

package/src/storage/MemoryAdapter.js

@@ -0,0 +1,56 @@
+/**
+ * In-Memory Storage Adapter
+ *
+ * Default ephemeral adapter for non-clustered, single-process deployments
+ * with zero filesystem or network overhead.
+ *
+ * @author ChdLabs
+ * @license MIT
+ */
+import { BaseStorageAdapter } from './BaseAdapter.js';
+
+export class MemoryAdapter extends BaseStorageAdapter {
+    constructor() {
+        super();
+        this.records = new Map();
+    }
+
+    async init(shield) {
+        await super.init(shield);
+    }
+
+    async loadBlockedIPs() {
+        const now = Date.now();
+        const active = [];
+        for (const [ip, item] of this.records.entries()) {
+            if (item.expiresAt && item.expiresAt <= now) {
+                this.records.delete(ip);
+            } else {
+                active.push(item);
+            }
+        }
+        return active;
+    }
+
+    async saveBlockedIP(ip, reason, metadata = {}, expiresAt = null) {
+        if (!ip) return;
+        this.records.set(ip, {
+            ip,
+            reason,
+            metadata,
+            expiresAt,
+            savedAt: Date.now()
+        });
+    }
+
+    async removeBlockedIP(ip) {
+        if (!ip) return;
+        this.records.delete(ip);
+    }
+
+    async close() {
+        this.records.clear();
+    }
+}
+
+export default MemoryAdapter;

package/src/storage/index.js

@@ -0,0 +1,53 @@
+/**
+ * Storage Subsystem Exports & Factory
+ *
+ * @author ChdLabs
+ * @license MIT
+ */
+import { BaseStorageAdapter } from './BaseAdapter.js';
+import { MemoryAdapter } from './MemoryAdapter.js';
+import { FileStorageAdapter } from './FileStorageAdapter.js';
+import { CallbackStorageAdapter } from './CallbackStorageAdapter.js';
+
+export {
+    BaseStorageAdapter,
+    MemoryAdapter,
+    FileStorageAdapter,
+    CallbackStorageAdapter
+};
+
+/**
+ * Instantiate appropriate storage adapter based on options
+ * @param {string|Object|BaseStorageAdapter} [storageOption]
+ * @param {Object} [logger]
+ * @returns {BaseStorageAdapter}
+ */
+export function createStorageAdapter(storageOption, logger = console) {
+    if (!storageOption || storageOption === 'memory') {
+        return new MemoryAdapter();
+    }
+
+    if (storageOption instanceof BaseStorageAdapter) {
+        return storageOption;
+    }
+
+    if (typeof storageOption === 'object') {
+        // Callback adapter: has onLoad, onBlock, or onUnblock
+        if (storageOption.onLoad || storageOption.onBlock || storageOption.onUnblock) {
+            return new CallbackStorageAdapter({ ...storageOption, logger });
+        }
+
+        // File adapter: has type === 'file' or directory or filename
+        if (storageOption.type === 'file' || storageOption.directory || storageOption.filename) {
+            return new FileStorageAdapter({ ...storageOption, logger });
+        }
+    }
+
+    if (storageOption === 'file') {
+        return new FileStorageAdapter({ logger });
+    }
+
+    return new MemoryAdapter();
+}
+
+export default createStorageAdapter;