aiden-shared-calculations-unified 1.0.84 → 1.0.87

package/calculations/core/short-position-per-stock.js

@@ -1,13 +1,10 @@
 /**
  * @fileoverview Calculation (Pass 1) for short positions per stock.
- * --- FIX ---
- * - Logic updated to check for *both* Speculator schema (pos.IsBuy === false)
- * and Normal schema (pos.Direction === 'Sell') to match schema.md.
+ * REFACTORED: Tracks exposure weight (percentage).
  */
-
 class ShortPositionPerStock {
     constructor() {
-        this.assets = new Map();
+        this.stockData = new Map();
         this.tickerMap = null;
     }
 
@@ -21,70 +18,65 @@ class ShortPositionPerStock {
         };
     }
 
-    static getDependencies() {
-        return [];
-    }
+    static getDependencies() { return []; }
 
     static getSchema() {
-        return {
+        const schema = {
             "type": "object",
-            "description": "Calculates the total count of short ('sell') positions for each asset.",
-            "patternProperties": {
-                "^.*$": { "type": "number" }
+            "properties": {
+                "short_count": { "type": "number" },
+                "total_short_exposure_weight": { "type": "number", "description": "Total allocation % for short positions." }
             },
-            "additionalProperties": { "type": "number" }
+            "required": ["short_count", "total_short_exposure_weight"]
         };
+        return { "type": "object", "patternProperties": { "^.*$": schema } };
     }
 
-    _initAsset(instrumentId) {
-        if (!this.assets.has(instrumentId)) {
-            this.assets.set(instrumentId, 0);
+    _initStock(instId) {
+        if (!this.stockData.has(instId)) {
+            this.stockData.set(instId, { count: 0, weight: 0 });
         }
     }
 
-    process(todayPortfolio, yesterdayPortfolio, userId, context, todayInsights, yesterdayInsights, fetchedDependencies) {
-        if (!this.tickerMap) {
-             this.tickerMap = context.instrumentToTicker;
-        }
+    process(context) {
+        const { extract } = context.math;
+        const { mappings, user } = context;
+        if (!this.tickerMap) this.tickerMap = mappings.instrumentToTicker;
 
-        const positions = todayPortfolio.PublicPositions || todayPortfolio.AggregatedPositions;
-        if (!positions || !Array.isArray(positions)) {
-            return;
-        }
+        const positions = extract.getPositions(user.portfolio.today, user.type);
 
         for (const pos of positions) {
-            // --- THIS IS THE FIX ---
-            // Check for Speculator (IsBuy) OR Normal (Direction)
-            const isShort = (pos.IsBuy === false) || (pos.Direction === 'Sell');
+            const direction = extract.getDirection(pos);
+            if (direction !== 'Sell') continue; // Only Shorts
 
-            if (isShort) {
-            // --- END FIX ---
-                const instrumentId = pos.InstrumentID;
-                if (!instrumentId) continue;
+            const instId = extract.getInstrumentId(pos);
+            if (!instId) continue;
 
-                this._initAsset(instrumentId);
-                this.assets.set(instrumentId, this.assets.get(instrumentId) + 1);
-            }
+            const weight = extract.getPositionWeight(pos, user.type);
+
+            this._initStock(instId);
+            const data = this.stockData.get(instId);
+            data.count++;
+            data.weight += weight;
         }
     }
 
     async getResult() {
-        if (!this.tickerMap) {
-            return {}; 
-        }
-
+        if (!this.tickerMap) return {};
         const result = {};
-        for (const [instrumentId, count] of this.assets.entries()) {
-            const ticker = this.tickerMap[instrumentId] || `id_${instrumentId}`;
-            result[ticker] = count;
+        for (const [instId, data] of this.stockData.entries()) {
+            const ticker = this.tickerMap[instId] || `id_${instId}`;
+            result[ticker] = {
+                short_count: data.count,
+                total_short_exposure_weight: data.weight
+            };
         }
         return result;
     }
 
     reset() {
-        this.assets.clear();
+        this.stockData.clear();
         this.tickerMap = null;
     }
 }
-
 module.exports = ShortPositionPerStock;

package/calculations/core/social-activity-aggregation.js

@@ -1,17 +1,9 @@
 /**
  * @fileoverview Calculation (Pass 2 - Meta) for social activity.
- * --- FIX ---
- * 1. Changed 'process' signature to 5-arg 'meta' standard.
- * 2. Changed data access to read 'dateStr.social' (Arg 1, the metaPayload).
- * 3. Changed 'socialDoc.posts' to 'socialDoc' (which IS the map of posts).
- * 4. Changed 'post.comments' to 'post.commentCount'.
- * 5. Changed 'post.likes' to 'post.likeCount'.
+ * REFACTORED: Uses context.social.today.
  */
 class SocialActivityAggregation {
-
-    constructor() {
-        this.result = {};
-    }
+    constructor() { this.result = {}; }
 
     static getMetadata() {
         return {
@@ -23,60 +15,45 @@ class SocialActivityAggregation {
         };
     }
 
-    static getDependencies() {
-        return [];
-    }
+    static getDependencies() { return []; }
 
     static getSchema() {
         return {
             "type": "object",
-            "description": "Aggregates the total social activity for the day.",
             "properties": {
                 "total_posts": { "type": "number" },
-                "total_comments": { "type": "number" },
-                "total_likes": { "type": "number" }
+                "unique_authors": { "type": "number" },
+                "total_likes": { "type": "number" },
+                "total_comments": { "type": "number" }
             },
-            "required": ["total_posts", "total_comments", "total_likes"]
+            "required": ["total_posts", "unique_authors", "total_likes", "total_comments"]
         };
     }
 
-    // --- THIS IS THE FIX ---
-    async process(dateStr, rootData, dependencies, config, fetchedDependencies) {
-
-        // Get social data from Arg 1 (metaPayload)
-        const socialDoc = dateStr.social; // This IS the map of posts
-
-        if (!socialDoc || typeof socialDoc !== 'object') {
-            this.result = { total_posts: 0, total_comments: 0, total_likes: 0 };
-            return;
-        }
-
-        const postsArray = Object.values(socialDoc);
-        let totalPosts = postsArray.length;
-        let totalComments = 0;
-        let totalLikes = 0;
-
-        for (const post of postsArray) {
-            // Use schema.md fields
-            totalComments += post.commentCount || 0;
-            totalLikes += post.likeCount || 0;
+    process(context) {
+        // Access social data from context
+        const socialPosts = context.social?.today || {};
+        
+        let posts = 0, likes = 0, comments = 0;
+        const authors = new Set();
+
+        for (const postId in socialPosts) {
+            const post = socialPosts[postId];
+            posts++;
+            likes += (post.likeCount || 0);
+            comments += (post.commentCount || 0);
+            if (post.postOwnerId) authors.add(post.postOwnerId);
         }
 
         this.result = {
-            total_posts: totalPosts,
-            total_comments: totalComments,
-            total_likes: totalLikes
+            total_posts: posts,
+            unique_authors: authors.size,
+            total_likes: likes,
+            total_comments: comments
         };
     }
-    // --- END FIX ---
-
-    async getResult(fetchedDependencies) {
-        return this.result;
-    }
 
-    reset() {
-        this.result = {};
-    }
+    async getResult() { return this.result; }
+    reset() { this.result = {}; }
 }
-
 module.exports = SocialActivityAggregation;

package/calculations/core/social-asset-posts-trend.js

@@ -1,125 +1,69 @@
 /**
- * @fileoverview Calculation (Pass 2 - Meta) for social posts trend.
- * --- FIX ---
- * 1. Changed 'process' signature to 5-arg 'meta' standard.
- * 2. Changed data access to read 'dateStr.social' (Arg 1).
- * 3. Changed 'socialDoc.posts' (Object) to 'socialDoc' (Object).
- * 4. Changed 'post.timestamp' to 'post.createdAt'.
- * 5. Changed 'post.instrumentIds' to 'post.tickers'.
- * 6. Removed 'instrumentToTicker' mapping as 'post.tickers' is already tickers.
+ * @fileoverview Calculation (Pass 2 - Meta) for asset mention trends.
+ * REFACTORED: Aggregates post counts per ticker.
  */
-
 class SocialAssetPostsTrend {
-
-    constructor() {
-        this.result = {};
-    }
+    constructor() { this.result = {}; }
 
     static getMetadata() {
         return {
             type: 'meta',
             rootDataDependencies: ['social'],
-            isHistorical: true,
+            isHistorical: false,
             userType: 'n/a',
             category: 'core_social'
         };
     }
 
-    static getDependencies() {
-        return [];
-    }
+    static getDependencies() { return []; }
 
     static getSchema() {
         const tickerSchema = {
             "type": "object",
             "properties": {
-                "posts_1d_pct": { "type": "number" },
-                "posts_7d_pct": { "type": "number" },
-                "trend_ratio": { "type": ["number", "null"] }
+                "post_count": { "type": "number" },
+                "sentiment_score": { "type": "number" }
             },
-            "required": ["posts_1d_pct", "posts_7d_pct", "trend_ratio"]
-        };
-
-        return {
-            "type": "object",
-            "description": "Calculates the 1-day vs 7-day percentage of total posts for each asset.",
-            "patternProperties": { "^.*$": tickerSchema },
-            "additionalProperties": tickerSchema
+            "required": ["post_count", "sentiment_score"]
         };
+        return { "type": "object", "patternProperties": { "^.*$": tickerSchema } };
     }
 
-    // --- THIS IS THE FIX ---
-    async process(dateStr, rootData, dependencies, config, fetchedDependencies) {
-
-        // Get social data from Arg 1 (metaPayload)
-        const socialDoc = dateStr.social; // This IS the map of posts
-        // Get the date from Arg 1
-        const todayDateStr = dateStr.date;
-
-        if (!socialDoc || typeof socialDoc !== 'object' || !todayDateStr) {
-            this.result = {};
-            return;
-        }
-
-        const today = new Date(todayDateStr + 'T00:00:00Z');
-        const sevenDaysAgo = new Date(today);
-        sevenDaysAgo.setUTCDate(today.getUTCDate() - 7);
-
-        const counts1d = new Map();
-        const counts7d = new Map();
-        let total1d = 0;
-        let total7d = 0;
-
-        for (const post of Object.values(socialDoc)) {
-            const postDate = new Date(post.createdAt); // Use 'createdAt'
-            const is1d = postDate.toISOString().slice(0, 10) === todayDateStr;
-            const is7d = postDate >= sevenDaysAgo;
-
-            if (!is7d) continue; 
-
-            // Use 'tickers' array directly from schema
-            for (const ticker of (post.tickers || [])) {
-                if (!ticker) continue;
-
-                if (is7d) {
-                    counts7d.set(ticker, (counts7d.get(ticker) || 0) + 1);
-                    total7d++;
-                }
-                if (is1d) {
-                    counts1d.set(ticker, (counts1d.get(ticker) || 0) + 1);
-                    total1d++;
-                }
+    process(context) {
+        const socialPosts = context.social?.today || {};
+        const assetStats = new Map(); // { ticker: { count: 0, sentimentSum: 0 } }
+
+        for (const postId in socialPosts) {
+            const post = socialPosts[postId];
+            if (!post.tickers || !Array.isArray(post.tickers)) continue;
+
+            // Convert sentiment to numeric: Bullish=1, Bearish=-1, Neutral=0
+            let sentVal = 0;
+            const sentiment = (post.sentiment?.overallSentiment || '').toLowerCase();
+            if (sentiment === 'bullish') sentVal = 1;
+            else if (sentiment === 'bearish') sentVal = -1;
+
+            for (const ticker of post.tickers) {
+                const t = ticker.toUpperCase();
+                if (!assetStats.has(t)) assetStats.set(t, { count: 0, sentimentSum: 0 });
+                
+                const stats = assetStats.get(t);
+                stats.count++;
+                stats.sentimentSum += sentVal;
             }
         }
 
-        const result = {};
-        const allTickers = new Set([...counts1d.keys(), ...counts7d.keys()]);
-
-        for (const ticker of allTickers) {
-            const c1 = counts1d.get(ticker) || 0;
-            const c7 = counts7d.get(ticker) || 0;
-
-            const pct1d = (total1d > 0) ? (c1 / total1d) * 100 : 0;
-            const pct7d = (total7d > 0) ? (c7 / total7d) * 100 : 0;
-
-            result[ticker] = {
-                posts_1d_pct: pct1d,
-                posts_7d_pct: pct7d,
-                trend_ratio: (pct7d > 0) ? (pct1d / pct7d) : null
+        const output = {};
+        for (const [ticker, stats] of assetStats.entries()) {
+            output[ticker] = {
+                post_count: stats.count,
+                sentiment_score: stats.count > 0 ? stats.sentimentSum / stats.count : 0
             };
         }
-
-        this.result = result;
-    }
-    // --- END FIX ---
-
-    async getResult(fetchedDependencies) {
-        return this.result;
+        this.result = output;
     }
 
-    reset() {
-        this.result = {};
-    }
+    async getResult() { return this.result; }
+    reset() { this.result = {}; }
 }
-
 module.exports = SocialAssetPostsTrend;

package/calculations/core/social-event-correlation.js

@@ -1,17 +1,9 @@
 /**
- * @fileoverview Calculation (Pass 2 - Meta) for social event correlation.
- * --- FIX ---
- * 1. Changed 'process' signature to 5-arg 'meta' standard.
- * 2. Changed data access to read 'dateStr.social' (Arg 1).
- * 3. Changed 'socialDoc.posts' to 'socialDoc' (Object).
- * 4. **Rewrote logic** to infer events from 'post.pollData' and
- * 'post.sentiment.topics' as 'post.event' does not exist in schema.
+ * @fileoverview Calculation (Pass 2 - Meta) for social event extraction.
+ * REFACTORED: Analyzes events from social posts.
  */
 class SocialEventCorrelation {
-
-    constructor() {
-        this.result = {};
-    }
+    constructor() { this.result = {}; }
 
     static getMetadata() {
         return {
@@ -23,103 +15,44 @@ class SocialEventCorrelation {
         };
     }
 
-    static getDependencies() {
-        return [];
-    }
+    static getDependencies() { return []; }
 
     static getSchema() {
-        const eventSchema = {
-            "type": "object",
-            "properties": {
-                "topic_distribution": { "type": "object" },
-                "sentiment_distribution": { "type": "object" },
-                "post_count": { "type": "number" }
-            }
-        };
-
         return {
             "type": "object",
-            "description": "Correlates social events (e.g., 'poll') with topic and sentiment distributions.",
-            "properties": {
-                "poll": eventSchema,
-                "earnings": eventSchema, // Inferred from topics
-                "launch": eventSchema // Inferred from topics
-            }
+            "description": "Map of events detected in social stream.",
+            "patternProperties": { "^.*$": { "type": "number", "description": "Count of mentions" } }
         };
     }
 
-    _initEvent(result, eventName) {
-        if (!result[eventName]) {
-            result[eventName] = {
-                topic_distribution: {},
-                sentiment_distribution: {},
-                post_count: 0
-            };
-        }
-    }
-
-    // --- THIS IS THE FIX ---
-    async process(dateStr, rootData, dependencies, config, fetchedDependencies) {
-        const result = {};
-
-        // Get social data from Arg 1 (metaPayload)
-        const socialDoc = dateStr.social; // This IS the map of posts
-
-        if (!socialDoc || typeof socialDoc !== 'object') {
-            this.result = {};
-            return;
-        }
-
-        for (const post of Object.values(socialDoc)) {
+    process(context) {
+        const socialPosts = context.social?.today || {};
+        const events = new Map();
 
-            // Infer events from schema.md fields
+        for (const postId in socialPosts) {
+            const post = socialPosts[postId];
+            // Extract 'topics' from the sentiment analysis result
             const topics = post.sentiment?.topics || [];
-            const sentiment = post.sentiment?.overallSentiment || 'Neutral';
-
-            const detectedEvents = new Set();
-
-            // Event 1: Check for Polls
-            if (post.pollData) {
-                detectedEvents.add('poll');
-            }
-
-            // Event 2: Check for Topics as Events
-            if (topics.includes('Earnings')) {
-                detectedEvents.add('earnings');
-            }
-            // Add any other topic-based events you want to track
-            // if (topics.includes('Launch')) {
-            //     detectedEvents.add('launch');
-            // }
-
-            if (detectedEvents.size === 0) continue;
-
-            for (const eventName of detectedEvents) {
-                this._initEvent(result, eventName);
-                const eventData = result[eventName];
-                eventData.post_count++;
-
-                // Aggregate topics
+            
+            if (Array.isArray(topics)) {
                 for (const topic of topics) {
-                    eventData.topic_distribution[topic] = (eventData.topic_distribution[topic] || 0) + 1;
+                    // Basic cleanup
+                    const cleanTopic = topic.trim().toUpperCase();
+                    if (cleanTopic.length > 2) {
+                        events.set(cleanTopic, (events.get(cleanTopic) || 0) + 1);
+                    }
                 }
-
-                // Aggregate sentiment
-                eventData.sentiment_distribution[sentiment] = (eventData.sentiment_distribution[sentiment] || 0) + 1;
             }
         }
+        // Filter for significant events (mentioned > 1 times)
+        const significant = [...events.entries()]
+            .filter(([_, count]) => count > 1)
+            .sort((a, b) => b[1] - a[1]);
 
-        this.result = result;
+        this.result = Object.fromEntries(significant);
     }
-    // --- END FIX ---
 
-    async getResult(fetchedDependencies) {
-        return this.result;
-    }
-
-    reset() {
-        this.result = {};
-    }
+    async getResult() { return this.result; }
+    reset() { this.result = {}; }
 }
-
 module.exports = SocialEventCorrelation;

package/calculations/core/social-sentiment-aggregation.js

@@ -1,16 +1,9 @@
 /**
- * @fileoverview Calculation (Pass 2 - Meta) for social sentiment.
- * --- FIX ---
- * 1. Changed 'process' signature to 5-arg 'meta' standard.
- * 2. Changed data access to read 'dateStr.social' (Arg 1).
- * 3. Changed 'socialDoc.posts' (Object) to 'socialDoc' (Object).
- * 4. Changed 'post.sentiment.label' to 'post.sentiment.overallSentiment'.
+ * @fileoverview Calculation (Pass 2 - Meta) for sentiment aggregation.
+ * REFACTORED: Uses context.social.today and sentiment analysis results.
  */
 class SocialSentimentAggregation {
-
-    constructor() {
-        this.result = {};
-    }
+    constructor() { this.result = {}; }
 
     static getMetadata() {
         return {
@@ -22,70 +15,53 @@ class SocialSentimentAggregation {
         };
     }
 
-    static getDependencies() {
-        return [];
-    }
+    static getDependencies() { return []; }
 
     static getSchema() {
         return {
             "type": "object",
-            "description": "Aggregates the overall social sentiment for the day.",
             "properties": {
                 "bullish": { "type": "number" },
                 "bearish": { "type": "number" },
                 "neutral": { "type": "number" },
                 "total_posts": { "type": "number" },
-                "sentiment_score": { "type": "number", "description": "Net score (bullish-bearish) / total" }
+                "sentiment_score": { "type": "number" }
             },
             "required": ["bullish", "bearish", "neutral", "total_posts", "sentiment_score"]
         };
     }
 
-    // --- THIS IS THE FIX ---
-    async process(dateStr, rootData, dependencies, config, fetchedDependencies) {
+    process(context) {
+        const socialPosts = context.social?.today || {};
         const counts = { bullish: 0, bearish: 0, neutral: 0 };
-        let totalPosts = 0;
-
-        // Get social data from Arg 1 (metaPayload)
-        const socialDoc = dateStr.social; // This IS the map of posts
-
-        if (!socialDoc || typeof socialDoc !== 'object') {
-            this.result = { bullish: 0, bearish: 0, neutral: 0, total_posts: 0, sentiment_score: 0 };
-            return;
-        }
+        let total = 0;
 
-        const postsArray = Object.values(socialDoc);
-        totalPosts = postsArray.length;
-
-        for (const post of postsArray) {
-            // Use 'overallSentiment' and convert to lowercase
+        for (const postId in socialPosts) {
+            const post = socialPosts[postId];
             const sentiment = (post.sentiment?.overallSentiment || 'neutral').toLowerCase();
+            
             if (counts.hasOwnProperty(sentiment)) {
                 counts[sentiment]++;
+                total++;
+            } else {
+                counts.neutral++; // Default fallback
+                total++;
             }
         }
 
-        const score = (totalPosts > 0) 
-            ? ((counts.bullish - counts.bearish) / totalPosts) 
-            : 0;
+        // Score: (Bullish - Bearish) / Total (-1 to 1 range)
+        const score = total > 0 ? (counts.bullish - counts.bearish) / total : 0;
 
         this.result = {
             bullish: counts.bullish,
             bearish: counts.bearish,
             neutral: counts.neutral,
-            total_posts: totalPosts,
+            total_posts: total,
             sentiment_score: score
         };
     }
-    // --- END FIX ---
-
-    async getResult(fetchedDependencies) {
-        return this.result;
-    }
 
-    reset() {
-        this.result = {};
-    }
+    async getResult() { return this.result; }
+    reset() { this.result = {}; }
 }
-
 module.exports = SocialSentimentAggregation;