@gabriel3615/ta_analysis/src/analysis/integration/plugins/chipPlugin.ts

stock ta analysis

import type {
  AnalyzerPlugin,
  AnalysisInputData,
  DirectionConversionResult,
  IntegrationContext,
} from '../IntegrationTypes.js';
import { TradeDirection } from '../../../types.js';

function clamp(value: number, min = 0, max = 100) {
  return Math.max(min, Math.min(max, value));
}

export function createChipPlugin(): AnalyzerPlugin {
  return {
    id: 'chip',
    category: 'main',
    extract(
      input: AnalysisInputData,
      _context: IntegrationContext
    ): DirectionConversionResult {
      const chip = input.analyses.chip;
      let direction: TradeDirection = TradeDirection.Neutral;
      const buy = chip.combinedBuySignalStrength;
      const sell = chip.combinedShortSignalStrength;
      const diff = buy - sell;
      let confidence = Math.max(buy, sell);
      if (diff > 15) direction = TradeDirection.Long;
      else if (diff < -15) direction = TradeDirection.Short;
      else confidence = confidence * 0.5;
      return { direction, confidence: clamp(confidence), source: 'chip' };
    },
    summarize(input: AnalysisInputData): string {
      const chip = input.analyses.chip;
      return `买入强度:${chip.combinedBuySignalStrength} 做空强度:${chip.combinedShortSignalStrength} 主周期:${chip.primaryTimeframe}`;
    },
  };
}