@dignetwork/dig-sdk 0.0.1-alpha.18 → 0.0.1-alpha.180

package/dist/blockchain/Wallet.js

@@ -38,9 +38,10 @@ const chia_root_resolver_1 = require("chia-root-resolver");
 const chia_config_loader_1 = require("chia-config-loader");
 const encryption_1 = require("../utils/encryption");
 const buffer_1 = require("buffer");
-const datalayer_driver_1 = require("datalayer-driver");
+const datalayer_driver_1 = require("@dignetwork/datalayer-driver");
 const config_1 = require("../utils/config");
 const FileCache_1 = require("../utils/FileCache");
+const config_2 = require("../utils/config");
 const KEYRING_FILE = "keyring.json";
 const CACHE_DURATION = 5 * 60 * 1000; // 5 minutes
 exports.DEFAULT_FEE_COIN_COST = 64000000;
@@ -174,31 +175,62 @@ class Wallet {
         const signature = (0, datalayer_driver_1.signMessage)(buffer_1.Buffer.from(message, "utf-8"), privateSyntheticKey);
         return signature.toString("hex");
     }
-    async verifyKeyOwnershipSignature(nonce, signature, publicKey) {
+    static async verifyKeyOwnershipSignature(nonce, signature, publicKey) {
         const message = `Signing this message to prove ownership of key.\n\nNonce: ${nonce}`;
         return (0, datalayer_driver_1.verifySignedMessage)(buffer_1.Buffer.from(signature, "hex"), buffer_1.Buffer.from(publicKey, "hex"), buffer_1.Buffer.from(message, "utf-8"));
     }
     async selectUnspentCoins(peer, coinAmount, feeBigInt, omitCoins = []) {
-        const cache = new FileCache_1.FileCache("reserved_coins");
-        const cachedReservedCoins = cache.getCachedKeys();
-        const now = Date.now();
-        const omitCoinIds = omitCoins.map((coin) => (0, datalayer_driver_1.getCoinId)(coin).toString("hex"));
-        cachedReservedCoins.forEach((coinId) => {
-            const reservation = cache.get(coinId);
-            if (reservation && reservation.expiry > now) {
-                omitCoinIds.push(coinId);
+        const cache = new FileCache_1.FileCache("reserved_coins", config_2.USER_DIR_PATH);
+        const ownerPuzzleHash = await this.getOwnerPuzzleHash();
+        // Define a function to attempt selecting unspent coins
+        const trySelectCoins = async () => {
+            const now = Date.now();
+            const omitCoinIds = omitCoins.map((coin) => (0, datalayer_driver_1.getCoinId)(coin).toString("hex"));
+            // Update omitCoinIds with currently valid reserved coins
+            const cachedReservedCoins = cache.getCachedKeys();
+            cachedReservedCoins.forEach((coinId) => {
+                const reservation = cache.get(coinId);
+                if (reservation && reservation.expiry > now) {
+                    if (!omitCoinIds.includes(coinId)) {
+                        omitCoinIds.push(coinId);
+                    }
+                }
+                else {
+                    cache.delete(coinId);
+                }
+            });
+            const coinsResp = await peer.getAllUnspentCoins(ownerPuzzleHash, config_1.MIN_HEIGHT, buffer_1.Buffer.from(config_1.MIN_HEIGHT_HEADER_HASH, "hex"));
+            const unspentCoins = coinsResp.coins.filter((coin) => !omitCoinIds.includes((0, datalayer_driver_1.getCoinId)(coin).toString("hex")));
+            const selectedCoins = (0, datalayer_driver_1.selectCoins)(unspentCoins, feeBigInt + coinAmount);
+            return selectedCoins;
+        };
+        let selectedCoins = [];
+        let retry = true;
+        while (retry) {
+            selectedCoins = await trySelectCoins();
+            if (selectedCoins.length > 0) {
+                // Coins have been successfully selected
+                retry = false;
             }
             else {
-                cache.delete(coinId);
+                const now = Date.now();
+                // Check if there are any valid cached reserved coins left
+                const cachedReservedCoins = cache.getCachedKeys().filter((coinId) => {
+                    const reservation = cache.get(coinId);
+                    return reservation && reservation.expiry > now;
+                });
+                if (cachedReservedCoins.length > 0) {
+                    // Wait 10 seconds and try again
+                    console.log("No unspent coins available. Waiting 10 seconds...");
+                    await new Promise((resolve) => setTimeout(resolve, 10000));
+                }
+                else {
+                    // No unspent coins and no reserved coins
+                    throw new Error("No unspent coins available.");
+                }
             }
-        });
-        const ownerPuzzleHash = await this.getOwnerPuzzleHash();
-        const coinsResp = await peer.getAllUnspentCoins(ownerPuzzleHash, config_1.MIN_HEIGHT, buffer_1.Buffer.from(config_1.MIN_HEIGHT_HEADER_HASH, "hex"));
-        const unspentCoins = coinsResp.coins.filter((coin) => !omitCoinIds.includes((0, datalayer_driver_1.getCoinId)(coin).toString("hex")));
-        const selectedCoins = (0, datalayer_driver_1.selectCoins)(unspentCoins, feeBigInt + coinAmount);
-        if (selectedCoins.length === 0) {
-            throw new Error("No unspent coins available.");
         }
+        // Reserve the selected coins
         selectedCoins.forEach((coin) => {
             const coinId = (0, datalayer_driver_1.getCoinId)(coin).toString("hex");
             cache.set(coinId, { coinId, expiry: Date.now() + CACHE_DURATION });

package/dist/blockchain/coins.d.ts

@@ -1,6 +1,5 @@
-import { Peer, Coin, CoinSpend } from "datalayer-driver";
+import { Peer, CoinSpend } from "@dignetwork/datalayer-driver";
 export declare const DEFAULT_FEE_COIN_COST = 64000000;
 export declare const calculateFeeForCoinSpends: (peer: Peer, coinSpends: CoinSpend[] | null) => Promise<bigint>;
-export declare const selectUnspentCoins: (peer: Peer, coinAmount: bigint, feeBigInt: bigint, omitCoins?: Coin[], walletName?: string) => Promise<Coin[]>;
 export declare const isCoinSpendable: (peer: Peer, coinId: Buffer) => Promise<boolean>;
 //# sourceMappingURL=coins.d.ts.map

package/dist/blockchain/coins.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"coins.d.ts","sourceRoot":"","sources":["../../src/blockchain/coins.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,IAAI,EACJ,IAAI,EAEJ,SAAS,EAEV,MAAM,kBAAkB,CAAC;AAK1B,eAAO,MAAM,qBAAqB,WAAa,CAAC;AAWhD,eAAO,MAAM,yBAAyB,SAC9B,IAAI,cACE,SAAS,EAAE,GAAG,IAAI,KAC7B,OAAO,CAAC,MAAM,CA4BhB,CAAC;AAEF,eAAO,MAAM,kBAAkB,SACvB,IAAI,cACE,MAAM,aACP,MAAM,cACN,IAAI,EAAE,eACL,MAAM,KACjB,OAAO,CAAC,IAAI,EAAE,CAoEhB,CAAC;AAGF,eAAO,MAAM,eAAe,SACpB,IAAI,UACF,MAAM,KACb,OAAO,CAAC,OAAO,CAWjB,CAAC"}
+{"version":3,"file":"coins.d.ts","sourceRoot":"","sources":["../../src/blockchain/coins.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,IAAI,EAGJ,SAAS,EAEV,MAAM,8BAA8B,CAAC;AAMtC,eAAO,MAAM,qBAAqB,WAAa,CAAC;AAWhD,eAAO,MAAM,yBAAyB,SAC9B,IAAI,cACE,SAAS,EAAE,GAAG,IAAI,KAC7B,OAAO,CAAC,MAAM,CA4BhB,CAAC;AAEF,eAAO,MAAM,eAAe,SACpB,IAAI,UACF,MAAM,KACb,OAAO,CAAC,OAAO,CAWjB,CAAC"}

package/dist/blockchain/coins.js

@@ -1,10 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isCoinSpendable = exports.selectUnspentCoins = exports.calculateFeeForCoinSpends = exports.DEFAULT_FEE_COIN_COST = void 0;
-const datalayer_driver_1 = require("datalayer-driver");
-const Wallet_1 = require("./Wallet");
+exports.isCoinSpendable = exports.calculateFeeForCoinSpends = exports.DEFAULT_FEE_COIN_COST = void 0;
 const config_1 = require("../utils/config");
-const FileCache_1 = require("../utils/FileCache");
 exports.DEFAULT_FEE_COIN_COST = 64000000;
 // Set cache expiration time (5 minutes)
 const CACHE_DURATION = 5 * 60 * 1000; // 5 minutes in milliseconds
@@ -38,57 +35,6 @@ const calculateFeeForCoinSpends = async (peer, coinSpends) => {
     */
 };
 exports.calculateFeeForCoinSpends = calculateFeeForCoinSpends;
-const selectUnspentCoins = async (peer, coinAmount, feeBigInt, omitCoins = [], walletName = "default") => {
-    // Initialize the cache for reserved coins
-    const cache = new FileCache_1.FileCache("reserved_coins");
-    // Get all cached reserved coins
-    const cachedReservedCoins = cache.getCachedKeys();
-    // Filter expired reserved coins and omit valid reservations
-    const now = Date.now();
-    const omitCoinIds = omitCoins.map((coin) => (0, datalayer_driver_1.getCoinId)(coin).toString("hex"));
-    const validReservedCoins = cachedReservedCoins.filter((coinId) => {
-        const reservation = cache.get(coinId);
-        if (reservation && reservation.expiry > now) {
-            // Valid reservation, add to omit list
-            omitCoinIds.push(coinId);
-            return true;
-        }
-        else {
-            // Reservation expired, remove it
-            cache.delete(coinId);
-            return false;
-        }
-    });
-    // Fetch all unspent coins from the peer
-    const wallet = await Wallet_1.Wallet.load(walletName);
-    const ownerPuzzleHash = await wallet.getOwnerPuzzleHash();
-    const coinsResp = await peer.getAllUnspentCoins(ownerPuzzleHash, config_1.MIN_HEIGHT, Buffer.from(config_1.MIN_HEIGHT_HEADER_HASH, "hex"));
-    if (process.env.DIG_DEBUG === "1") {
-        console.log("Unspent Coins:", coinsResp); // Debugging
-    }
-    const unspentCoins = coinsResp.coins;
-    // Filter out the omitted coins
-    const filteredUnspentCoins = unspentCoins.filter((coin) => !omitCoinIds.includes((0, datalayer_driver_1.getCoinId)(coin).toString("hex")));
-    if (process.env.DIG_DEBUG === "1") {
-        console.log("Unspent Coins after filtering:", filteredUnspentCoins); // Debugging
-    }
-    // Select coins needed for the transaction
-    const selectedCoins = (0, datalayer_driver_1.selectCoins)(filteredUnspentCoins, feeBigInt + coinAmount);
-    if (process.env.DIG_DEBUG === "1") {
-        console.log("Selected Coins:", selectedCoins); // Debugging
-    }
-    // If no coins are selected, throw an error
-    if (selectedCoins.length === 0) {
-        throw new Error("No unspent coins available.");
-    }
-    // Cache the selected coins as reserved for the next 5 minutes
-    selectedCoins.forEach((coin) => {
-        const coinId = (0, datalayer_driver_1.getCoinId)(coin).toString("hex");
-        cache.set(coinId, { coinId, expiry: Date.now() + CACHE_DURATION });
-    });
-    return selectedCoins;
-};
-exports.selectUnspentCoins = selectUnspentCoins;
 const isCoinSpendable = async (peer, coinId) => {
     try {
         const spent = await peer.isCoinSpent(coinId, config_1.MIN_HEIGHT, Buffer.from(config_1.MIN_HEIGHT_HEADER_HASH, "hex"));

package/dist/blockchain/index.d.ts

@@ -5,4 +5,5 @@ export * from './FullNodePeer';
 export * from './DataStoreSerializer';
 export * from './ServerCoin';
 export * from "./Wallet";
+export * from './StoreMonitorRegistry';
 //# sourceMappingURL=index.d.ts.map

package/dist/blockchain/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/blockchain/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AACnC,OAAO,KAAK,KAAK,MAAM,SAAS,CAAC;AACjC,cAAc,aAAa,CAAC;AAC5B,cAAa,gBAAgB,CAAC;AAC9B,cAAc,uBAAuB,CAAC;AACtC,cAAc,cAAc,CAAC;AAC7B,cAAc,UAAU,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/blockchain/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AACnC,OAAO,KAAK,KAAK,MAAM,SAAS,CAAC;AACjC,cAAc,aAAa,CAAC;AAC5B,cAAa,gBAAgB,CAAC;AAC9B,cAAc,uBAAuB,CAAC;AACtC,cAAc,cAAc,CAAC;AAC7B,cAAc,UAAU,CAAC;AACzB,cAAc,wBAAwB,CAAC"}

package/dist/blockchain/index.js

@@ -34,3 +34,4 @@ __exportStar(require("./FullNodePeer"), exports);
 __exportStar(require("./DataStoreSerializer"), exports);
 __exportStar(require("./ServerCoin"), exports);
 __exportStar(require("./Wallet"), exports);
+__exportStar(require("./StoreMonitorRegistry"), exports);

package/dist/types.d.ts

@@ -23,6 +23,7 @@ export interface DigConfig {
 export interface RootHistoryItem {
     root_hash: string;
     timestamp: Number | undefined;
+    synced: boolean;
 }
 export interface DatFile {
     root: string;
@@ -47,6 +48,7 @@ export interface ServerCoinData {
     coin: CoinData;
     createdAt: string;
     epoch: number;
+    rootHash: string;
 }
 export interface IncentiveProgramData {
     storeId: string;

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,MAAM;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,qBAAqB;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,SAAS;IACxB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;CAC/B;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,KAAK,EAAE;QACL,CAAC,GAAG,EAAE,MAAM,GAAG;YACb,IAAI,EAAE,MAAM,CAAC;YACb,MAAM,EAAE,MAAM,CAAC;SAChB,CAAC;KACH,CAAC;CACH;AAED,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,QAAQ;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,QAAQ,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,MAAM,EAAE,OAAO,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,cAAc,CAAC;CAC9B"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,MAAM;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,qBAAqB;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,SAAS;IACxB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,KAAK,EAAE;QACL,CAAC,GAAG,EAAE,MAAM,GAAG;YACb,IAAI,EAAE,MAAM,CAAC;YACb,MAAM,EAAE,MAAM,CAAC;SAChB,CAAC;KACH,CAAC;CACH;AAED,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,QAAQ;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,QAAQ,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,MAAM,EAAE,OAAO,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,cAAc,CAAC;CAC9B"}

package/dist/utils/ContentScanner.d.ts

@@ -0,0 +1,63 @@
+/**
+ * @class IllegalContentScanner
+ * Scans files for illegal content using ClamAV running in a Docker container and deletes files if flagged.
+ */
+export declare class IllegalContentScanner {
+    private clamAVContainerName;
+    private virusTotalApiKey?;
+    private openAiApiKey?;
+    /**
+     * @constructor
+     * @param clamAVContainerName The name of the Docker container running ClamAV (default: 'clamav').
+     */
+    constructor(clamAVContainerName?: string);
+    /**
+     * Scans a file for illegal content using ClamAV, VirusTotal, and OpenAI Moderation API.
+     * If a service is missing its API key, the scan for that service is skipped.
+     * @param filePath The path of the file to scan.
+     * @returns {Promise<void>} A promise that resolves after the scan and potential deletion.
+     */
+    scanFile(filePath: string): Promise<void>;
+    /**
+     * Scans a file using ClamAV running in Docker.
+     * @param filePath The path of the file to scan.
+     * @returns {Promise<string>} The result of the ClamAV scan.
+     */
+    private scanWithClamAV;
+    /**
+     * Submits a file to VirusTotal for scanning.
+     * @param filePath The path of the file to scan.
+     * @returns {Promise<any>} The result from VirusTotal.
+     */
+    private scanWithVirusTotal;
+    /**
+     * Scans file content (text) using OpenAI's Moderation API.
+     * @param content The text content to scan.
+     * @returns {Promise<any>} The result from OpenAI's Moderation API.
+     */
+    private scanWithOpenAI;
+    /**
+     * Checks if the ClamAV scan result indicates a malicious file.
+     * @param result The ClamAV scan output.
+     * @returns {boolean} True if the file is flagged, false otherwise.
+     */
+    private isClamAVResultMalicious;
+    /**
+     * Checks if the VirusTotal result indicates a malicious file.
+     * @param result The VirusTotal scan result.
+     * @returns {boolean} True if the file is flagged, false otherwise.
+     */
+    private isVirusTotalResultMalicious;
+    /**
+     * Checks if the OpenAI Moderation result indicates harmful content.
+     * @param result The OpenAI Moderation result.
+     * @returns {boolean} True if harmful content is detected, false otherwise.
+     */
+    private isOpenAiResultMalicious;
+    /**
+     * Deletes a file from the filesystem.
+     * @param filePath The file to delete.
+     */
+    private deleteFile;
+}
+//# sourceMappingURL=ContentScanner.d.ts.map

package/dist/utils/ContentScanner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContentScanner.d.ts","sourceRoot":"","sources":["../../src/utils/ContentScanner.ts"],"names":[],"mappings":"AAKA;;;GAGG;AACH,qBAAa,qBAAqB;IAC9B,OAAO,CAAC,mBAAmB,CAAS;IACpC,OAAO,CAAC,gBAAgB,CAAC,CAAS;IAClC,OAAO,CAAC,YAAY,CAAC,CAAS;IAE9B;;;OAGG;gBACS,mBAAmB,GAAE,MAAiB;IAMlD;;;;;OAKG;IACU,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqDtD;;;;OAIG;IACH,OAAO,CAAC,cAAc;IAYtB;;;;OAIG;YACW,kBAAkB;IAgBhC;;;;OAIG;YACW,cAAc;IAa5B;;;;OAIG;IACH,OAAO,CAAC,uBAAuB;IAI/B;;;;OAIG;IACH,OAAO,CAAC,2BAA2B;IAKnC;;;;OAIG;IACH,OAAO,CAAC,uBAAuB;IAI/B;;;OAGG;IACH,OAAO,CAAC,UAAU;CASrB"}

package/dist/utils/ContentScanner.js

@@ -0,0 +1,175 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IllegalContentScanner = void 0;
+const child_process_1 = require("child_process");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const axios_1 = __importDefault(require("axios"));
+/**
+ * @class IllegalContentScanner
+ * Scans files for illegal content using ClamAV running in a Docker container and deletes files if flagged.
+ */
+class IllegalContentScanner {
+    /**
+     * @constructor
+     * @param clamAVContainerName The name of the Docker container running ClamAV (default: 'clamav').
+     */
+    constructor(clamAVContainerName = "clamav") {
+        this.clamAVContainerName = clamAVContainerName;
+        this.virusTotalApiKey = process.env.VIRUSTOTAL_API_KEY || undefined;
+        this.openAiApiKey = process.env.OPENAI_API_KEY || undefined;
+    }
+    /**
+     * Scans a file for illegal content using ClamAV, VirusTotal, and OpenAI Moderation API.
+     * If a service is missing its API key, the scan for that service is skipped.
+     * @param filePath The path of the file to scan.
+     * @returns {Promise<void>} A promise that resolves after the scan and potential deletion.
+     */
+    async scanFile(filePath) {
+        try {
+            // 1. ClamAV Scan
+            console.log(`Starting ClamAV scan for file: ${filePath}`);
+            const clamAvResult = await this.scanWithClamAV(filePath);
+            if (this.isClamAVResultMalicious(clamAvResult)) {
+                console.log(`ClamAV flagged file: ${filePath}. Deleting...`);
+                this.deleteFile(filePath);
+                return;
+            }
+            else {
+                console.log(`ClamAV did not flag the file: ${filePath}.`);
+            }
+            // 2. VirusTotal Scan (if API key is provided)
+            if (this.virusTotalApiKey) {
+                console.log(`Submitting file to VirusTotal for scanning: ${filePath}`);
+                const virusTotalResult = await this.scanWithVirusTotal(filePath);
+                if (this.isVirusTotalResultMalicious(virusTotalResult)) {
+                    console.log(`VirusTotal flagged file: ${filePath}. Deleting...`);
+                    this.deleteFile(filePath);
+                    return;
+                }
+                else {
+                    console.log(`VirusTotal did not flag the file: ${filePath}.`);
+                }
+            }
+            else {
+                console.log(`VirusTotal API key not provided. Skipping VirusTotal scan.`);
+            }
+            // 3. OpenAI Moderation API (if API key is provided and file is text-based)
+            const fileExtension = path_1.default.extname(filePath).toLowerCase();
+            if (this.openAiApiKey && (fileExtension === ".txt" || fileExtension === ".json")) {
+                console.log(`Scanning file content with OpenAI Moderation API: ${filePath}`);
+                const fileContent = await fs_1.default.promises.readFile(filePath, "utf-8");
+                const openAiResult = await this.scanWithOpenAI(fileContent);
+                if (this.isOpenAiResultMalicious(openAiResult)) {
+                    console.log(`OpenAI flagged file: ${filePath}. Deleting...`);
+                    this.deleteFile(filePath);
+                    return;
+                }
+            }
+            else {
+                if (!this.openAiApiKey) {
+                    console.log(`OpenAI API key not provided. Skipping OpenAI scan.`);
+                }
+                else {
+                    console.log(`OpenAI scan skipped: File type not supported (${fileExtension}).`);
+                }
+            }
+            console.log(`File ${filePath} passed all multilayer scans.`);
+        }
+        catch (error) {
+            console.error(`Error scanning file: ${filePath}`, error);
+        }
+    }
+    /**
+     * Scans a file using ClamAV running in Docker.
+     * @param filePath The path of the file to scan.
+     * @returns {Promise<string>} The result of the ClamAV scan.
+     */
+    scanWithClamAV(filePath) {
+        return new Promise((resolve, reject) => {
+            (0, child_process_1.exec)(`docker exec ${this.clamAVContainerName} clamdscan ${filePath}`, (error, stdout, stderr) => {
+                if (error) {
+                    reject(`ClamAV scan error: ${stderr}`);
+                }
+                else {
+                    resolve(stdout);
+                }
+            });
+        });
+    }
+    /**
+     * Submits a file to VirusTotal for scanning.
+     * @param filePath The path of the file to scan.
+     * @returns {Promise<any>} The result from VirusTotal.
+     */
+    async scanWithVirusTotal(filePath) {
+        const fileContent = await fs_1.default.promises.readFile(filePath);
+        const formData = new FormData();
+        // @ts-ignore
+        formData.append("file", fileContent);
+        const response = await axios_1.default.post("https://www.virustotal.com/vtapi/v2/file/scan", formData, {
+            headers: {
+                "x-apikey": this.virusTotalApiKey,
+                // @ts-ignore
+                ...formData.getHeaders(),
+            },
+        });
+        return response.data;
+    }
+    /**
+     * Scans file content (text) using OpenAI's Moderation API.
+     * @param content The text content to scan.
+     * @returns {Promise<any>} The result from OpenAI's Moderation API.
+     */
+    async scanWithOpenAI(content) {
+        const response = await axios_1.default.post("https://api.openai.com/v1/moderations", { input: content }, {
+            headers: {
+                Authorization: `Bearer ${this.openAiApiKey}`,
+            },
+        });
+        return response.data;
+    }
+    /**
+     * Checks if the ClamAV scan result indicates a malicious file.
+     * @param result The ClamAV scan output.
+     * @returns {boolean} True if the file is flagged, false otherwise.
+     */
+    isClamAVResultMalicious(result) {
+        return result.includes("FOUND");
+    }
+    /**
+     * Checks if the VirusTotal result indicates a malicious file.
+     * @param result The VirusTotal scan result.
+     * @returns {boolean} True if the file is flagged, false otherwise.
+     */
+    isVirusTotalResultMalicious(result) {
+        // Check if VirusTotal flagged the file based on its multiple antivirus results
+        return result.positives > 0;
+    }
+    /**
+     * Checks if the OpenAI Moderation result indicates harmful content.
+     * @param result The OpenAI Moderation result.
+     * @returns {boolean} True if harmful content is detected, false otherwise.
+     */
+    isOpenAiResultMalicious(result) {
+        return result.flagged === true;
+    }
+    /**
+     * Deletes a file from the filesystem.
+     * @param filePath The file to delete.
+     */
+    deleteFile(filePath) {
+        fs_1.default.unlink(filePath, (err) => {
+            if (err) {
+                console.error(`Error deleting file: ${filePath}`, err);
+            }
+            else {
+                console.log(`File ${filePath} was deleted successfully.`);
+            }
+        });
+    }
+}
+exports.IllegalContentScanner = IllegalContentScanner;

package/dist/utils/DigCache.d.ts

@@ -0,0 +1,38 @@
+import NodeCache from "node-cache";
+export interface ICache {
+    set<T>(key: string, value: T, ttl?: number): Promise<boolean>;
+    get<T>(key: string): Promise<T | undefined>;
+    del(key: string): Promise<number>;
+    has(key: string): Promise<boolean>;
+    flushAll(): Promise<boolean>;
+    keys(pattern?: string): Promise<string[]>;
+    ttl(key: string, ttl?: number): Promise<boolean>;
+}
+declare class DigCache implements ICache {
+    private cache;
+    private useRedis;
+    /**
+     * Constructor with the same signature as NodeCache.
+     * @param options - NodeCache options
+     */
+    constructor(options?: NodeCache.Options);
+    /**
+     * Initializes the Redis connection if Redis is being used.
+     * Call this method before performing any cache operations.
+     */
+    initialize(): Promise<void>;
+    /**
+     * Gracefully disconnects the Redis client if Redis is being used.
+     * Call this method when your application is shutting down.
+     */
+    disconnect(): Promise<void>;
+    set<T>(key: string, value: T, ttl?: number): Promise<boolean>;
+    get<T>(key: string): Promise<T | undefined>;
+    del(key: string): Promise<number>;
+    has(key: string): Promise<boolean>;
+    flushAll(): Promise<boolean>;
+    keys(pattern?: string): Promise<string[]>;
+    ttl(key: string, ttl?: number): Promise<boolean>;
+}
+export { DigCache };
+//# sourceMappingURL=DigCache.d.ts.map

package/dist/utils/DigCache.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"DigCache.d.ts","sourceRoot":"","sources":["../../src/utils/DigCache.ts"],"names":[],"mappings":"AACA,OAAO,SAAS,MAAM,YAAY,CAAC;AAGnC,MAAM,WAAW,MAAM;IACrB,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9D,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC;IAC5C,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAClC,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACnC,QAAQ,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7B,IAAI,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1C,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CAElD;AAED,cAAM,QAAS,YAAW,MAAM;IAC9B,OAAO,CAAC,KAAK,CAA8B;IAC3C,OAAO,CAAC,QAAQ,CAAU;IAE1B;;;OAGG;gBACS,OAAO,CAAC,EAAE,SAAS,CAAC,OAAO;IAqBvC;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAYjC;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAW3B,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAkC7D,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;IAuB3C,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBjC,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAoBlC,QAAQ,IAAI,OAAO,CAAC,OAAO,CAAC;IAoB5B,IAAI,CAAC,OAAO,GAAE,MAAY,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAmB9C,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CA0BvD;AAED,OAAO,EAAE,QAAQ,EAAE,CAAC"}