bulltrackers-module 1.0.592 → 1.0.593

package/functions/old-generic-api/user-api/helpers/core/user_status_helpers.js

@@ -0,0 +1,195 @@
+/**
+ * @fileoverview User Status Helpers
+ * Functions to check user status (PI, signed-in, etc.)
+ */
+
+const { findLatestRankingsDate } = require('./data_lookup_helpers');
+
+/**
+ * Get Popular Investor master list from Firestore
+ * @param {Firestore} db - Firestore instance
+ * @param {object} collectionRegistry - Collection registry (optional)
+ * @param {object} logger - Logger instance (optional)
+ * @returns {Promise<object>} - Master list investors map { cid: { cid, username, firstSeenAt, lastSeenAt } }
+ */
+async function getPIMasterList(db, collectionRegistry = null, logger = null) {
+    try {
+        let masterListPath = 'system_state/popular_investor_master_list';
+        
+        if (collectionRegistry && collectionRegistry.getCollectionPath) {
+            try {
+                masterListPath = collectionRegistry.getCollectionPath('system', 'popularInvestorMasterList', {});
+            } catch (err) {
+                if (logger) {
+                    logger.log('WARN', `[getPIMasterList] Failed to get master list path from registry, using default: ${err.message}`);
+                }
+            }
+        }
+        
+        const masterListRef = db.doc(masterListPath);
+        const masterListDoc = await masterListRef.get();
+        
+        if (!masterListDoc.exists) {
+            return {};
+        }
+        
+        const masterListData = masterListDoc.data();
+        return masterListData.investors || {};
+    } catch (error) {
+        if (logger) {
+            logger.log('ERROR', `[getPIMasterList] Error loading master list: ${error.message}`);
+        }
+        return {};
+    }
+}
+
+/**
+ * Get PI username from CID using master list
+ * @param {Firestore} db - Firestore instance
+ * @param {string|number} piCid - Popular Investor CID
+ * @param {object} collectionRegistry - Collection registry (optional)
+ * @param {object} logger - Logger instance (optional)
+ * @returns {Promise<string|null>} - Username or null if not found
+ */
+async function getPIUsernameFromMasterList(db, piCid, collectionRegistry = null, logger = null) {
+    try {
+        const investors = await getPIMasterList(db, collectionRegistry, logger);
+        const cidStr = String(piCid);
+        const investor = investors[cidStr];
+        
+        if (investor && investor.username) {
+            return investor.username;
+        }
+        
+        return null;
+    } catch (error) {
+        if (logger) {
+            logger.log('ERROR', `[getPIUsernameFromMasterList] Error getting username for PI ${piCid}: ${error.message}`);
+        }
+        return null;
+    }
+}
+
+/**
+ * Get PI CID from username using master list
+ * @param {Firestore} db - Firestore instance
+ * @param {string} username - Popular Investor username
+ * @param {object} collectionRegistry - Collection registry (optional)
+ * @param {object} logger - Logger instance (optional)
+ * @returns {Promise<string|null>} - CID or null if not found
+ */
+async function getPICidFromMasterList(db, username, collectionRegistry = null, logger = null) {
+    try {
+        const investors = await getPIMasterList(db, collectionRegistry, logger);
+        const searchUsername = username.toLowerCase().trim();
+        
+        for (const [cid, investor] of Object.entries(investors)) {
+            if (investor.username && investor.username.toLowerCase().trim() === searchUsername) {
+                return cid;
+            }
+        }
+        
+        return null;
+    } catch (error) {
+        if (logger) {
+            logger.log('ERROR', `[getPICidFromMasterList] Error getting CID for username ${username}: ${error.message}`);
+        }
+        return null;
+    }
+}
+
+/**
+ * Check if a CID is a Popular Investor using master list
+ * @param {Firestore} db - Firestore instance
+ * @param {string|number} cid - User CID
+ * @param {object} collectionRegistry - Collection registry (optional)
+ * @param {object} logger - Logger instance (optional)
+ * @returns {Promise<boolean>} - True if PI, false otherwise
+ */
+async function isPopularInvestor(db, cid, collectionRegistry = null, logger = null) {
+    try {
+        const investors = await getPIMasterList(db, collectionRegistry, logger);
+        const cidStr = String(cid);
+        return investors.hasOwnProperty(cidStr);
+    } catch (error) {
+        if (logger) {
+            logger.log('ERROR', `[isPopularInvestor] Error checking if ${cid} is PI: ${error.message}`);
+        }
+        return false;
+    }
+}
+
+/**
+ * Check if a signed-in user is also a Popular Investor
+ * Returns ranking entry if found, null otherwise
+ * Checks dev overrides first for pretendToBePI flag
+ * @param {Firestore} db - Firestore instance
+ * @param {string|number} userCid - User CID
+ * @param {object} config - Configuration object
+ * @param {object} logger - Logger instance (optional)
+ * @returns {Promise<object|null>} - Ranking entry or null
+ */
+async function checkIfUserIsPI(db, userCid, config, logger = null) {
+    try {
+        // Check dev override first (for developer accounts)
+        const { getDevOverride } = require('../dev/dev_helpers');
+        const devOverride = await getDevOverride(db, userCid, config, logger);
+        
+        if (devOverride && devOverride.enabled && devOverride.pretendToBePI) {
+            // Generate fake ranking entry for dev testing
+            const fakeRankEntry = {
+                CustomerId: Number(userCid),
+                UserName: 'Dev Test PI',
+                AUMValue: 500000 + Math.floor(Math.random() * 1000000), // Random AUM between 500k-1.5M
+                Copiers: 150 + Math.floor(Math.random() * 200), // Random copiers between 150-350
+                RiskScore: 3 + Math.floor(Math.random() * 3), // Random risk score 3-5
+                Gain: 25 + Math.floor(Math.random() * 50), // Random gain 25-75%
+                WinRatio: 50 + Math.floor(Math.random() * 20), // Random win ratio 50-70%
+                Trades: 500 + Math.floor(Math.random() * 1000) // Random trades 500-1500
+            };
+            
+            if (logger && logger.log) {
+                logger.log('INFO', `[checkIfUserIsPI] DEV OVERRIDE: User ${userCid} pretending to be PI with fake ranking data`);
+            } else {
+                console.log(`[checkIfUserIsPI] DEV OVERRIDE: User ${userCid} pretending to be PI with fake ranking data`);
+            }
+            
+            return fakeRankEntry;
+        }
+        
+        // Otherwise, check master list (single source of truth)
+        const collectionRegistry = config.collectionRegistry || null;
+        const isPI = await isPopularInvestor(db, userCid, collectionRegistry, logger);
+        
+        if (!isPI) {
+            return null;
+        }
+        
+        // Get username from master list
+        const username = await getPIUsernameFromMasterList(db, userCid, collectionRegistry, logger);
+        
+        if (!username) {
+            return null;
+        }
+        
+        // Return a ranking-like entry for compatibility (minimal data)
+        // Note: Full ranking data (AUM, copiers, etc.) is not in master list
+        // If full ranking data is needed, it should be fetched from rankings collection separately
+        return {
+            CustomerId: Number(userCid),
+            UserName: username
+        };
+    } catch (error) {
+        console.error('[checkIfUserIsPI] Error checking if user is PI:', error);
+        return null;
+    }
+}
+
+module.exports = {
+    checkIfUserIsPI,
+    getPIMasterList,
+    getPIUsernameFromMasterList,
+    getPICidFromMasterList,
+    isPopularInvestor
+};
+