aiden-shared-calculations-unified 1.0.5 → 1.0.7

package/calculations/socialPosts/social_event_correlation.js

@@ -0,0 +1,122 @@
+/**
+ * @fileoverview Correlates social post volume spikes with topics from Gemini analysis.
+ * This calculation ONLY uses the social post insights context.
+ */
+
+class SocialEventCorrelation {
+    constructor() {
+        this.todayTickerVolume = {};
+        this.yesterdayTickerVolume = {};
+        this.todayTopicCounts = {}; // { "TICKER": { "topic": count } }
+        this.processed = false;
+    }
+
+    /**
+     * Helper to process a set of posts and populate volume/topic counts.
+     * @param {object} socialPostInsights - Map of { [postId]: postData }
+     * @param {object} volumeTarget - The object to store volume counts.
+     * @param {object} [topicTarget] - Optional. The object to store topic counts.
+     */
+    _processPosts(socialPostInsights, volumeTarget, topicTarget = null) {
+        if (!socialPostInsights) return;
+        const posts = Object.values(socialPostInsights);
+
+        for (const post of posts) {
+            if (!post.tickers || !Array.isArray(post.tickers)) continue;
+
+            for (const ticker of post.tickers) {
+                // 1. Aggregate Volume
+                volumeTarget[ticker] = (volumeTarget[ticker] || 0) + 1;
+
+                // 2. Aggregate Topics (if target provided)
+                if (topicTarget) {
+                    if (!topicTarget[ticker]) {
+                        topicTarget[ticker] = {};
+                    }
+                    // New structure: post.sentiment is { overallSentiment: "...", topics: [...] }
+                    const topics = post.sentiment?.topics;
+                    if (topics && Array.isArray(topics)) {
+                        for (const topic of topics) {
+                            // Normalize topic to lowercase for consistent counting
+                            const normalizedTopic = topic.toLowerCase();
+                            topicTarget[ticker][normalizedTopic] = (topicTarget[ticker][normalizedTopic] || 0) + 1;
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    /**
+     * @param {null} todayPortfolio - Not used.
+     * @param {null} yesterdayPortfolio - Not used.
+     * @param {null} userId - Not used.
+     * @param {object} context - Shared context (e.g., mappings).
+     * @param {object} todayInsights - Daily instrument insights.
+     * @param {object} yesterdayInsights - Yesterday's instrument insights.
+     * @param {object} todaySocialPostInsights - Map of { [postId]: postData } for today.
+     * @param {object} yesterdaySocialPostInsights - Map of { [postId]: postData } for yesterday.
+     */
+    async process(todayPortfolio, yesterdayPortfolio, userId, context, todayInsights, yesterdayInsights, todaySocialPostInsights, yesterdaySocialPostInsights) {
+        
+        if (this.processed) {
+            return; // Run only once
+        }
+        this.processed = true;
+
+        // 1. Process yesterday's posts for baseline volume
+        this._processPosts(yesterdaySocialPostInsights, this.yesterdayTickerVolume);
+
+        // 2. Process today's posts for volume AND topic correlation
+        this._processPosts(todaySocialPostInsights, this.todayTickerVolume, this.todayTopicCounts);
+    }
+
+    async getResult() {
+        if (!this.processed) return {};
+
+        const results = {};
+        const allTickers = new Set([
+            ...Object.keys(this.todayTickerVolume),
+            ...Object.keys(this.yesterdayTickerVolume)
+        ]);
+
+        for (const ticker of allTickers) {
+            const todayVolume = this.todayTickerVolume[ticker] || 0;
+            // Use 1 as baseline if yesterday was 0 to avoid -100% change, but 0 for spike calc
+            const yesterdayVolumeForCalc = this.yesterdayTickerVolume[ticker] || 0;
+            
+            let volumeSpikePercent = 0;
+            if (yesterdayVolumeForCalc > 0) {
+                 volumeSpikePercent = ((todayVolume - yesterdayVolumeForCalc) / yesterdayVolumeForCalc) * 100;
+            } else if (todayVolume > 0) {
+                volumeSpikePercent = todayVolume * 100.0; // e.g., 0 to 5 posts is a 500% increase
+            }
+
+            // Only report tickers with activity today
+            if (todayVolume === 0) continue;
+
+            // Correlate topics
+            const topics = this.todayTopicCounts[ticker] || {};
+            const correlatedEvents = Object.entries(topics)
+                .map(([topic, postCount]) => ({ topic, postCount }))
+                .sort((a, b) => b.postCount - a.postCount); // Sort by count descending
+
+            results[ticker] = {
+                volume: todayVolume,
+                volumeSpikePercent: volumeSpikePercent,
+                correlatedEvents: correlatedEvents
+            };
+        }
+
+        return results;
+    }
+
+    reset() {
+        this.todayTickerVolume = {};
+        this.yesterdayTickerVolume = {};
+        this.todayTopicCounts = {};
+        this.processed = false;
+    }
+}
+
+module.exports = SocialEventCorrelation;

package/calculations/socialPosts/social_sentiment_aggregation.js

@@ -1,6 +1,7 @@
 /**
  * @fileoverview Aggregates sentiment data from social posts.
  * This calculation ONLY uses the social post insights context, not portfolio data.
+ * UPDATED: To read sentiment from the `sentiment.overallSentiment` property.
  */
 
 class SocialSentimentAggregation {
@@ -36,7 +37,9 @@ class SocialSentimentAggregation {
         this.totalPosts = posts.length;
 
         for (const post of posts) {
-            const sentiment = post.sentiment || 'Neutral';
+            // --- FIX: Read from the nested overallSentiment property ---
+            const sentiment = post.sentiment?.overallSentiment || 'Neutral';
+            // --- END FIX ---
 
             // 1. Aggregate global sentiment counts
             if (this.sentimentCounts.hasOwnProperty(sentiment)) {
@@ -109,4 +112,4 @@ class SocialSentimentAggregation {
     }
 }
 
-module.exports = SocialSentimentAggregation;
+module.exports = SocialSentimentAggregation;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aiden-shared-calculations-unified",
-  "version": "1.0.5",
+  "version": "1.0.7",
   "description": "Shared calculation modules for the BullTrackers Computation System.",
   "main": "index.js",
   "files": [