bulltrackers-module 1.0.171 → 1.0.172

package/functions/computation-system/controllers/computation_controller.js

@@ -0,0 +1,215 @@
+/**
+ * @fileoverview Control Layer - Orchestrates Computation Execution
+ * UPDATE: Imported and exposed HistoryExtractor in the math context.
+ * UPDATE: Implemented strict User Type segregation. 'All' now defaults to 'Normal'.
+ */
+
+const { 
+    DataExtractor, 
+    HistoryExtractor, 
+    MathPrimitives, 
+    Aggregators, 
+    Validators, 
+    SCHEMAS, 
+    SignalPrimitives 
+} = require('../layers/math_primitives');
+
+const { 
+    loadDailyInsights, 
+    loadDailySocialPostInsights, 
+    getPortfolioPartRefs, 
+    getHistoryPartRefs, 
+    streamPortfolioData, 
+    streamHistoryData 
+} = require('../utils/data_loader');
+
+// ============================================================================
+// DATA LOADER WRAPPER
+// ============================================================================
+
+class DataLoader {
+    constructor(config, dependencies) {
+        this.config = config;
+        this.deps = dependencies;
+        this.cache = { mappings: null, insights: new Map(), social: new Map() };
+    }
+
+    async loadMappings() {
+        if (this.cache.mappings) return this.cache.mappings;
+        const { calculationUtils } = this.deps;
+        this.cache.mappings = await calculationUtils.loadInstrumentMappings();
+        return this.cache.mappings;
+    }
+
+    async loadInsights(dateStr) {
+        if (this.cache.insights.has(dateStr)) return this.cache.insights.get(dateStr);
+        const insights = await loadDailyInsights(this.config, this.deps, dateStr);
+        this.cache.insights.set(dateStr, insights);
+        return insights;
+    }
+
+    async loadSocial(dateStr) {
+        if (this.cache.social.has(dateStr)) return this.cache.social.get(dateStr);
+        const social = await loadDailySocialPostInsights(this.config, this.deps, dateStr);
+        this.cache.social.set(dateStr, social);
+        return social;
+    }
+}
+
+// ============================================================================
+// CONTEXT BUILDER
+// ============================================================================
+
+class ContextBuilder {
+    static buildPerUserContext(options) {
+        const {
+            todayPortfolio, yesterdayPortfolio, todayHistory, yesterdayHistory,
+            userId, userType, dateStr, metadata, mappings, insights, socialData,
+            computedDependencies, config, deps
+        } = options;
+
+        return {
+            // User Identity & Data
+            user: {
+                id: userId,
+                type: userType,
+                portfolio: { today: todayPortfolio, yesterday: yesterdayPortfolio },
+                history: { today: todayHistory, yesterday: yesterdayHistory }
+            },
+            // Global Time & Data
+            date: { today: dateStr },
+            insights: { today: insights?.today, yesterday: insights?.yesterday },
+            social: { today: socialData?.today, yesterday: socialData?.yesterday },
+            // Helpers
+            mappings: mappings || {},
+            math: {
+                extract: DataExtractor,
+                history: HistoryExtractor,
+                compute: MathPrimitives,
+                aggregate: Aggregators,
+                validate: Validators,
+                signals: SignalPrimitives,
+                schemas: SCHEMAS
+            },
+            computed: computedDependencies || {},
+            meta: metadata,
+            config,
+            deps
+        };
+    }
+
+    static buildMetaContext(options) {
+        const {
+            dateStr, metadata, mappings, insights, socialData,
+            computedDependencies, config, deps
+        } = options;
+
+        return {
+            date: { today: dateStr },
+            insights: { today: insights?.today, yesterday: insights?.yesterday },
+            social: { today: socialData?.today, yesterday: socialData?.yesterday },
+            mappings: mappings || {},
+            math: {
+                extract: DataExtractor,
+                history: HistoryExtractor,
+                compute: MathPrimitives,
+                aggregate: Aggregators,
+                validate: Validators,
+                signals: SignalPrimitives,
+                schemas: SCHEMAS
+            },
+            computed: computedDependencies || {},
+            meta: metadata,
+            config,
+            deps
+        };
+    }
+}
+
+// ============================================================================
+// EXECUTOR
+// ============================================================================
+
+class ComputationExecutor {
+    constructor(config, dependencies, dataLoader) {
+        this.config = config;
+        this.deps = dependencies;
+        this.loader = dataLoader;
+    }
+
+    async executePerUser(calcInstance, metadata, dateStr, portfolioData, historyData, computedDeps) {
+        const { logger } = this.deps;
+        
+        // --------------------------------------------------------------------
+        // 1. DETERMINE TARGET SCHEMA
+        // --------------------------------------------------------------------
+        // We strictly enforce separation here. 
+        // Unless 'speculator' is explicitly requested, we default to 'normal'.
+        // This effectively deprecates 'all' by treating it as 'normal' for safety.
+        const targetUserType = (metadata.userType === 'speculator')
+            ? SCHEMAS.USER_TYPES.SPECULATOR 
+            : SCHEMAS.USER_TYPES.NORMAL;
+        
+        const mappings = await this.loader.loadMappings();
+        const insights = metadata.rootDataDependencies?.includes('insights') 
+            ? { today: await this.loader.loadInsights(dateStr) } : null;
+
+        // Loop through user batch
+        for (const [userId, todayPortfolio] of Object.entries(portfolioData)) {
+            const yesterdayPortfolio = historyData ? historyData[userId] : null;
+            
+            // ----------------------------------------------------------------
+            // 2. IDENTIFY ACTUAL DATA TYPE
+            // ----------------------------------------------------------------
+            // We inspect the data structure to know what we are holding.
+            const actualUserType = todayPortfolio.PublicPositions 
+                ? SCHEMAS.USER_TYPES.SPECULATOR 
+                : SCHEMAS.USER_TYPES.NORMAL;
+
+            // ----------------------------------------------------------------
+            // 3. STRICT GATEKEEPING
+            // ----------------------------------------------------------------
+            // If the computation asked for 'normal' (or 'all'), but we have a 'speculator', SKIP.
+            // If the computation asked for 'speculator', but we have a 'normal', SKIP.
+            if (targetUserType !== actualUserType) continue;
+
+            const context = ContextBuilder.buildPerUserContext({
+                todayPortfolio, yesterdayPortfolio,
+                userId, userType: actualUserType, dateStr, metadata, mappings, insights,
+                computedDependencies: computedDeps, config: this.config, deps: this.deps
+            });
+
+            try {
+                await calcInstance.process(context);
+            } catch (e) {
+                logger.log('WARN', `Calc ${metadata.name} failed for user ${userId}: ${e.message}`);
+            }
+        }
+    }
+
+    async executeOncePerDay(calcInstance, metadata, dateStr, computedDeps) {
+        const mappings = await this.loader.loadMappings();
+        const insights = metadata.rootDataDependencies?.includes('insights') 
+            ? { today: await this.loader.loadInsights(dateStr) } : null;
+        const social = metadata.rootDataDependencies?.includes('social')
+            ? { today: await this.loader.loadSocial(dateStr) } : null;
+
+        const context = ContextBuilder.buildMetaContext({
+            dateStr, metadata, mappings, insights, socialData: social,
+            computedDependencies: computedDeps, config: this.config, deps: this.deps
+        });
+
+        return await calcInstance.process(context);
+    }
+}
+
+class ComputationController {
+    constructor(config, dependencies) {
+        this.config = config;
+        this.deps = dependencies;
+        this.loader = new DataLoader(config, dependencies);
+        this.executor = new ComputationExecutor(config, dependencies, this.loader);
+    }
+}
+
+module.exports = { ComputationController };