Fix transpiler TypeScript generation and add Series support to math.sum

oakscriptjs/src/math/index.ts

@@ -10,6 +10,7 @@
  */
 
 import { float, int, simple_int, series_float } from '../types';
+import { Series } from '../runtime/series';
 
 /**
  * Returns the absolute value of a number.
@@ -192,14 +193,15 @@ export function avg(...values: float[]): float {
 /**
  * Returns the sum of values over a sliding window.
  *
- * @param source - Series of values
+ * @param source - Series of values or array
  * @param length - Window length for summation
- * @returns Series with rolling sum values
+ * @returns Series or array with rolling sum values
  *
  * @remarks
  * - Calculates sum of last `length` values at each point
  * - Returns NaN for first `length-1` values (insufficient data)
  * - Window slides forward one value at a time
+ * - When passed a Series, returns a Series; when passed an array, returns an array
  *
  * @example
  * ```typescript
@@ -209,21 +211,28 @@ export function avg(...values: float[]): float {
  *
  * @see {@link https://www.tradingview.com/pine-script-reference/v6/#fun_math.sum | PineScript math.sum}
  */
-export function sum(source: series_float, length: simple_int): series_float {
+export function sum(source: series_float | Series, length: simple_int): series_float | Series {
+  // Handle Series objects - extract to array first
+  const sourceArray = source instanceof Series ? source.toArray() : source;
   const result: series_float = [];
 
-  for (let i = 0; i < source.length; i++) {
+  for (let i = 0; i < sourceArray.length; i++) {
     if (i < length - 1) {
       result.push(NaN);
     } else {
       let total = 0;
       for (let j = 0; j < length; j++) {
-        total += source[i - j]!;
+        total += sourceArray[i - j]!;
       }
       result.push(total);
     }
   }
 
+  // If input was a Series, return a Series
+  if (source instanceof Series) {
+    return Series.fromArray(source.bars, result);
+  }
+
   return result;
 }
 

oakscriptjs/src/runtime/series.ts

@@ -41,6 +41,14 @@ export class Series {
     this.extractor = extractor;
   }
 
+  /**
+   * Get the underlying bar data
+   * @returns Bar array
+   */
+  get bars(): Bar[] {
+    return this.data;
+  }
+
   // ============================================
   // Static Factory Methods
   // ============================================