@discomedia/utils 1.0.40 → 1.0.41

package/dist/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.0.40",
+  "version": "1.0.41",
   "author": "Disco Media",
   "description": "Utility functions used in Disco Media apps",
   "always-build-npm": true,

package/dist/test.js

@@ -13731,14 +13731,25 @@ class AlpacaMarketDataAPI extends EventEmitter {
             `Avg Volume: ${avgVolume.toLocaleString()}`);
     }
     /**
-     * Get all assets available for trade and data consumption from Alpaca
-     * @param params Optional query params: status (e.g. 'active'), asset_class (e.g. 'us_equity', 'crypto')
+     * Get assets available for trade and data consumption from Alpaca
+     * @param params Optional query params
+     * - status: 'active' | 'inactive' (default 'active')
+     * - asset_class: 'us_equity' | 'us_option' | 'crypto' (default 'us_equity')
+     * - shortable: if true (default), filters to assets with shortable=true and easy_to_borrow=true
      * @returns Array of AlpacaAsset objects
      * @see https://docs.alpaca.markets/reference/get-v2-assets-1
      */
     async getAssets(params) {
         // Endpoint: GET /v2/assets
-        return this.makeRequest('/assets', 'GET', params, 'api'); // use apiURL
+        const { status = 'active', asset_class = 'us_equity', shortable = true, } = {
+            status: params?.status ?? 'active',
+            asset_class: params?.asset_class ?? 'us_equity',
+            shortable: params?.shortable ?? true,
+        };
+        const assets = await this.makeRequest('/assets', 'GET', { status, asset_class }, 'api');
+        if (!shortable)
+            return assets;
+        return assets.filter((a) => a.shortable === true && a.easy_to_borrow === true);
     }
     /**
      * Get a single asset by symbol or asset_id
@@ -14382,88 +14393,84 @@ class AlpacaMarketDataAPI extends EventEmitter {
     }
 }
 // Export the singleton instance
-AlpacaMarketDataAPI.getInstance();
+const marketDataAPI = AlpacaMarketDataAPI.getInstance();
+
+var alpacaMarketDataApi = /*#__PURE__*/Object.freeze({
+    __proto__: null,
+    AlpacaMarketDataAPI: AlpacaMarketDataAPI,
+    marketDataAPI: marketDataAPI
+});
 
 // Test file for context functionality
-//testOpenRouter();
-//testGetMarketStatus();
-//testCryptoMarketData();
-//testGetPortfolioDailyHistory();
-async function testWebSocketConnectAndDisconnect() {
-    console.log('\n--- Testing WebSocket Connect and Disconnect ---');
+async function testGetAssetsShortableFilter() {
+    console.log('\n--- Testing getAssets params and shortable filter ---');
     const log = (message, options = { type: 'info' }) => {
         log$1(message, { ...options, source: 'Test' });
     };
-    // Ensure market data env vars are populated from TRADING_* if needed
-    if (!process.env.ALPACA_API_KEY && process.env.ALPACA_TRADING_API_KEY) {
-        process.env.ALPACA_API_KEY = process.env.ALPACA_TRADING_API_KEY;
-    }
-    if (!process.env.ALPACA_SECRET_KEY && process.env.ALPACA_TRADING_SECRET_KEY) {
-        process.env.ALPACA_SECRET_KEY = process.env.ALPACA_TRADING_SECRET_KEY;
-    }
-    if (!process.env.ALPACA_ACCOUNT_TYPE && process.env.ALPACA_TRADING_ACCOUNT_TYPE) {
-        process.env.ALPACA_ACCOUNT_TYPE = process.env.ALPACA_TRADING_ACCOUNT_TYPE;
-    }
-    const apiKey = process.env.ALPACA_API_KEY;
-    const secretKey = process.env.ALPACA_SECRET_KEY;
-    if (!apiKey || !secretKey) {
-        console.error('Missing Alpaca API credentials. Check .env ALPACA_TRADING_API_KEY/SECRET or ALPACA_API_KEY/SECRET.');
+    if (!process.env.ALPACA_API_KEY || !process.env.ALPACA_SECRET_KEY) {
+        console.log('Skipping getAssets test: Missing environment variables (ALPACA_API_KEY, ALPACA_SECRET_KEY)');
         return;
     }
-    const md = AlpacaMarketDataAPI.getInstance();
-    const symbols = ['SPY', 'TSLA', 'AAPL'];
-    // Track received minute bars per symbol
-    const barCounts = Object.fromEntries(symbols.map((s) => [s, 0]));
-    const firstSeenAt = {};
-    // Connect and subscribe to minute bars
-    log(`Connecting stock stream and subscribing to minute bars for ${symbols.join(', ')}`);
-    md.connectStockStream();
-    md.subscribe('stock', { bars: symbols });
-    const formatNY = (d) => d.toLocaleString('en-US', { timeZone: 'America/New_York' });
-    const onBar = (msg) => {
-        if (!symbols.includes(msg.S))
-            return;
-        barCounts[msg.S] += 1;
-        if (!firstSeenAt[msg.S]) {
-            firstSeenAt[msg.S] = formatNY(new Date(msg.t));
+    try {
+        const { marketDataAPI } = await Promise.resolve().then(function () { return alpacaMarketDataApi; });
+        // 1) Default call: should filter to shortable && easy_to_borrow and default to active us_equity
+        log('Calling getAssets() with defaults');
+        const filtered = await marketDataAPI.getAssets();
+        console.log(`Received ${filtered.length} filtered assets (default params)`);
+        const allFilteredValid = filtered.every((a) => a.shortable === true && a.easy_to_borrow === true);
+        if (!allFilteredValid) {
+            console.error('✗ Default getAssets() included assets that are not shortable and easy_to_borrow');
         }
-        log(`Bar ${msg.S} o=${msg.o.toFixed(2)} h=${msg.h.toFixed(2)} l=${msg.l.toFixed(2)} c=${msg.c.toFixed(2)} @ ${formatNY(new Date(msg.t))}`, { type: 'debug', symbol: msg.S });
-    };
-    md.on('stock-b', onBar);
-    // Wait until we have at least 2 bars per symbol, or timeout after 3 minutes
-    const minBarsPerSymbol = 2;
-    const timeoutMs = 3 * 60 * 1000;
-    const done = await new Promise((resolve) => {
-        const startedAt = Date.now();
-        const interval = setInterval(() => {
-            const hasAll = symbols.every((s) => barCounts[s] >= minBarsPerSymbol);
-            const elapsed = Date.now() - startedAt;
-            if (hasAll) {
-                clearInterval(interval);
-                resolve(true);
-            }
-            else if (elapsed >= timeoutMs) {
-                clearInterval(interval);
-                resolve(false);
-            }
-        }, 1000);
-    });
-    // Unsubscribe and disconnect
-    md.unsubscribe('stock', { bars: symbols });
-    md.disconnectStockStream();
-    md.removeListener('stock-b', onBar);
-    // Report results
-    const nowNY = formatNY(new Date());
-    console.log('\nStock bars summary:');
-    symbols.forEach((s) => {
-        console.log(`  ${s}: ${barCounts[s]} bars (first at ${firstSeenAt[s] ?? 'n/a'}; finished at ${nowNY})`);
-    });
-    if (done) {
-        console.log('✓ Received minimum bars for all symbols and disconnected.');
-    }
-    else {
-        console.warn('Completed with timeout before receiving minimum bars for all symbols.');
-    }
-}
-testWebSocketConnectAndDisconnect();
+        else {
+            console.log('✓ Default getAssets() correctly filtered shortable and easy_to_borrow assets');
+        }
+        // 2) Unfiltered call: shortable=false
+        log('Calling getAssets({ shortable: false })');
+        const unfiltered = await marketDataAPI.getAssets({ shortable: false });
+        console.log(`Received ${unfiltered.length} unfiltered assets`);
+        if (unfiltered.length < filtered.length) {
+            console.warn('Unfiltered list is smaller than filtered. This is unexpected but not fatal.');
+        }
+        else {
+            console.log('✓ Unfiltered list length is >= filtered list length');
+        }
+        // 3) Specific constraints: inactive crypto, not asserting non-empty
+        log("Calling getAssets({ status: 'inactive', asset_class: 'crypto', shortable: false })");
+        const inactiveCrypto = await marketDataAPI.getAssets({ status: 'inactive', asset_class: 'crypto', shortable: false });
+        console.log(`Received ${inactiveCrypto.length} inactive crypto assets`);
+        if (inactiveCrypto.length > 0) {
+            const valid = inactiveCrypto.every((a) => a.class === 'crypto' && a.status === 'inactive');
+            if (!valid) {
+                console.error('✗ Inactive crypto request returned assets with mismatched class or status');
+            }
+            else {
+                console.log('✓ Inactive crypto request returned only crypto with inactive status');
+            }
+        }
+        else {
+            console.log('No inactive crypto assets returned; skipping class/status assertions.');
+        }
+    }
+    catch (error) {
+        console.error('✗ getAssets test failed:', error);
+        if (error instanceof Error) {
+            console.error('Error message:', error.message);
+            console.error('Stack trace:', error.stack);
+        }
+    }
+}
+// testGetTradingDate();
+// testGetTradingStartAndEndDates();
+// testGetLastFullTradingDate();
+// testGetMarketOpenClose();
+// testGetNYTimeZone();
+// testGetNextMarketDay();
+// testCountTradingDays();
+// testGetPreviousMarketDay();
+// testOpenRouter();
+// testGetMarketStatus();
+// testCryptoMarketData();
+// testGetPortfolioDailyHistory();
+// testWebSocketConnectAndDisconnect();
+testGetAssetsShortableFilter();
 //# sourceMappingURL=test.js.map