@363045841yyt/klinechart-core 0.7.12 → 0.8.1-alpha.1

package/src/engine/renderers/Indicator/vwap.ts

@@ -7,6 +7,8 @@ import { resolveStateKey } from '../../indicators/indicatorMetadata'
 import { createSparseVisibleStateComposer } from '../../indicators/visibleStateComposers'
 import type { IndicatorScheduler, VWAPSchedulerConfig } from '../../indicators/scheduler'
 import { calcVWAPData } from '../../indicators/calculators'
+import type { TitleInfo } from '../../indicators/indicatorMetadata'
+import type { KLineData } from '../../../types/price'
 
 const VWAP_COLOR = '#ec4899'
 
@@ -109,22 +111,34 @@ export function createVWAPRendererPlugin(options: { paneId?: string } = {}): Ren
     }
 }
 
+export function getVWAPTitleInfo(
+  _data: KLineData[],
+  index: number | null,
+  _params: Record<string, number | boolean | string>,
+  host: PluginHost,
+  paneId: string,
+): TitleInfo | null {
+  if (index === null) return null
+  const state = host.getSharedState<VWAPRenderState>(createVWAPStateKey(paneId))
+  const value = state?.series[index]
+  if (value === undefined) return null
+
+  return {
+    name: 'VWAP',
+    params: [],
+    values: [{ label: 'VWAP', value, color: VWAP_COLOR }],
+  }
+}
+
 @Indicator({
     name: 'vwap',
     displayName: 'VWAP',
     category: 'volume',
-    stateKey: createVWAPStateKey,
     defaultPaneId: 'sub_VWAP',
     visibleState: { compose: createSparseVisibleStateComposer('vwap', EMPTY_VWAP_STATE) },
     scale: { indicatorKey: 'vwap', label: 'VWAP', decimals: 2 },
-    updateConfig: (scheduler, params, paneId) => {
-        (scheduler as IndicatorScheduler).updateIndicatorConfig('vwap', params, paneId)
-    },
-    applyResult: (host, state, paneId) => {
-        host.setSharedState(createVWAPStateKey(paneId), state as any, 'indicator_scheduler')
-    },
+    getTitleInfo: getVWAPTitleInfo,
     runtime: {
-        configKey: 'vwap',
         defaultConfig: { sessionResetGapMs: 0, showVWAP: true },
         computeKey: 'calcVWAPData',
         compute: (data, c) => calcVWAPData(data, c.sessionResetGapMs),

package/src/engine/renderers/Indicator/wma.ts

@@ -1,9 +1,10 @@
 import type { RendererPluginWithHost, RenderContext, PluginHost } from '../../../plugin'
+import type { KLineData } from '../../../types/price'
 import { RENDERER_PRIORITY } from '../../../plugin'
 import type { WMARenderState } from '../../indicators/wmaState'
 import { createWMAStateKey, EMPTY_WMA_STATE } from '../../indicators/wmaState'
 import { Indicator } from '../../indicators/indicatorDefinitionRegistry'
-import { resolveStateKey } from '../../indicators/indicatorMetadata'
+import { resolveStateKey, type TitleInfo, type TitleValueItem, type GetTitleInfoFn } from '../../indicators/indicatorMetadata'
 import { createSparseVisibleStateComposer } from '../../indicators/visibleStateComposers'
 import type { IndicatorScheduler, WMASchedulerConfig } from '../../indicators/scheduler'
 import { calcWMAData } from '../../indicators/calculators'
@@ -121,23 +122,40 @@ export function createWMARendererPlugin(options: WMARendererOptions = {}): Rende
     }
 }
 
+export const getWMATitleInfo: GetTitleInfoFn = (
+    _data: KLineData[],
+    index: number | null,
+    _params: Record<string, number | boolean | string>,
+    pluginHost: PluginHost,
+    paneId: string,
+): TitleInfo | null => {
+    if (index === null) return null
+
+    const stateKey = createWMAStateKey(paneId)
+    const state = pluginHost?.getSharedState<WMARenderState>(stateKey)
+    if (!state || state.visibleMin > state.visibleMax) return null
+
+    const value = state.series[index]
+    if (value === undefined) return null
+
+    return {
+        name: 'WMA',
+        params: [state.params.period],
+        values: [{ label: 'WMA', value, color: '#10b981' }],
+    }
+}
+
 @Indicator({
     name: 'wma',
     displayName: 'WMA',
+    getTitleInfo: getWMATitleInfo,
     category: 'main',
-    stateKey: createWMAStateKey,
     defaultPaneId: 'main',
     allowMainPane: true,
     mainPane: { rendererName: 'wma_main', toActiveConfig: (params, active) => ({ ...params, showWMA: active }) },
     visibleState: { compose: createSparseVisibleStateComposer('wma', EMPTY_WMA_STATE) },
     scale: { indicatorKey: 'wma', label: 'WMA', decimals: 2 },
-    updateConfig: (scheduler, params, paneId) => {
-        (scheduler as IndicatorScheduler).updateIndicatorConfig('wma', params, paneId)
-    },
-    applyResult: (host, state, paneId) => {
-        host.setSharedState(createWMAStateKey(paneId), state as any, 'indicator_scheduler')
-    },
-    runtime: { configKey:'wma', defaultConfig:{period:10,showWMA:true}, computeKey:'calcWMAData', compute:(data,c)=>calcWMAData(data,c.period) },
+    runtime: { defaultConfig:{period:10,showWMA:true}, computeKey:'calcWMAData', compute:(data,c)=>calcWMAData(data,c.period) },
 })
 class WMADefinition {
     static rendererFactory = createWMARendererPlugin

package/src/engine/renderers/Indicator/wmsr.ts

@@ -10,6 +10,7 @@ import { resolveStateKey } from '../../indicators/indicatorMetadata'
 import type { IndicatorScheduler, WMSRSchedulerConfig } from '../../indicators/scheduler'
 import { createWmsrScaleRendererPlugin } from './scale/wmsr_scale'
 import { calcWMSRData } from '../../indicators/calculators'
+import type { KLineData } from '../../../types/price'
 
 type LinePoint = { x: number; y: number }
 
@@ -293,14 +294,15 @@ function drawWMSRLineWithCanvas2D(
  * 获取 WMSR 标题信息（供 paneTitle 使用）
  */
 export function getWMSRTitleInfo(
-    index: number,
-    period: number,
+    _data: KLineData[],
+    index: number | null,
+    params: Record<string, number | boolean | string>,
     pluginHost: PluginHost,
-    paneId: string = 'sub_WMSR',
-    theme: 'light' | 'dark' = 'light',
-    isAsiaMarket?: boolean
+    paneId: string,
 ): { name: string; params: number[]; values: Array<{ label: string; value: number; color: string }> } | null {
-    const colors = resolveThemeColors(theme, isAsiaMarket)
+    if (index === null) return null
+    const period = (params.period as number) ?? 14
+    const colors = resolveThemeColors('light')
     const state = pluginHost.getSharedState<WMSRRenderState>(createWMSRStateKey(paneId))
     if (!state) return null
 
@@ -320,18 +322,11 @@ export function getWMSRTitleInfo(
     name: 'wmsr',
     displayName: 'WMSR',
     category: 'oscillator',
-    stateKey: createWMSRStateKey,
     defaultPaneId: 'sub_WMSR',
     visibleState: { compose: createFixedRangeSparseVisibleStateComposer('wmsr', EMPTY_WMSR_STATE) },
     scaleRendererFactory: createWmsrScaleRendererPlugin,
-    updateConfig: (scheduler, params, paneId) => {
-    (scheduler as IndicatorScheduler).updateIndicatorConfig('wmsr', params, paneId)
-  },
-    applyResult: (host, state, paneId) => {
-        host.setSharedState(createWMSRStateKey(paneId), state as any, 'indicator_scheduler')
-    },
+    getTitleInfo: getWMSRTitleInfo,
     runtime: {
-        configKey: 'wmsr',
         defaultConfig: { period: 14, showWMSR: true },
         computeKey: 'calcWMSRData',
         compute: (data, c) => calcWMSRData(data, c.period),

package/src/engine/renderers/Indicator/zones.ts

@@ -5,7 +5,7 @@ import type { ZonesRenderState } from '../../indicators/zonesState'
 import { createZonesStateKey, EMPTY_ZONES_STATE } from '../../indicators/zonesState'
 import { Indicator } from '../../indicators/indicatorDefinitionRegistry'
 import { createFixedUnitVisibleStateComposer } from '../../indicators/visibleStateComposers'
-import { resolveStateKey } from '../../indicators/indicatorMetadata'
+import { resolveStateKey, type TitleInfo, type TitleValueItem, type GetTitleInfoFn } from '../../indicators/indicatorMetadata'
 import type { IndicatorScheduler, ZonesSchedulerConfig } from '../../indicators/scheduler'
 import { calcZonesData } from '../../indicators/calculators'
 
@@ -90,23 +90,42 @@ export function createZonesRendererPlugin(options: { paneId?: string } = {}): Re
     }
 }
 
+export const getZonesTitleInfo: GetTitleInfoFn = (_data, index, _params, host, paneId) => {
+    if (index === null) return null
+
+    const stateKey = createZonesStateKey(paneId)
+    const state = host?.getSharedState<ZonesRenderState>(stateKey)
+    if (!state) return null
+
+    const activeZones = state.series.filter(
+        z => z.startIndex <= index && (z.endIndex === undefined || z.endIndex >= index)
+    )
+    if (activeZones.length === 0) return null
+
+    const values: TitleValueItem[] = activeZones.slice(0, 5).map(z => ({
+        label: z.kind,
+        value: z.high,
+        color: z.kind.includes('Bull') ? '#22c55e' : '#ef4444',
+    }))
+
+    return {
+        name: 'Zones',
+        params: [activeZones.length],
+        values,
+    }
+}
+
 @Indicator({
     name: 'zones',
     displayName: 'Zones',
+    getTitleInfo: getZonesTitleInfo,
     category: 'main',
-    stateKey: createZonesStateKey,
     defaultPaneId: 'main',
     allowMainPane: true,
     mainPane: { rendererName: 'zones_main', toActiveConfig: (params, active) => ({ ...params, showFVG: active, showOB: active, showFilledZones: active }) },
     scale: { indicatorKey: 'zones', label: 'Zones', decimals: 2 },
     visibleState: { compose: createFixedUnitVisibleStateComposer('zones', EMPTY_ZONES_STATE) },
-    updateConfig: (scheduler, params, paneId) => {
-        (scheduler as IndicatorScheduler).updateIndicatorConfig('zones', params, paneId)
-    },
-    applyResult: (host, state, paneId) => {
-        host.setSharedState(createZonesStateKey(paneId), state as any, 'indicator_scheduler')
-    },
-    runtime: { configKey:'zones', defaultConfig:{showFVG:true,showOB:true,showFilledZones:true,obLookback:20}, computeKey:'calcZonesData', compute:(data,c)=>calcZonesData(data,c.obLookback,5,2,'close') },
+    runtime: { defaultConfig:{showFVG:true,showOB:true,showFilledZones:true,obLookback:20}, computeKey:'calcZonesData', compute:(data,c)=>calcZonesData(data,c.obLookback,5,2,'close') },
 })
 class ZonesDefinition {
     static rendererFactory = createZonesRendererPlugin

package/src/engine/renderers/__tests__/mainIndicatorLegend.renderer.test.ts

@@ -8,11 +8,13 @@ import { ENE_STATE_KEY } from '@/core/indicators/eneState'
 import type { PluginHost, RenderContext, RendererPluginWithHost } from '@/plugin'
 import type { KLineData } from '@/types/price'
 import type { Pane } from '@/core/layout/pane'
+import type { GetTitleInfoFn, TitleInfo, TitleValueItem } from '@/engine/indicators/indicatorMetadata'
+import type { IndicatorScheduler } from '@/engine/indicators/indicatorScheduler'
 
 // Type helper for tests - we know these methods exist on the implementation
 interface TestableLegendRenderer extends RendererPluginWithHost {
   draw: (context: RenderContext) => void
-  getConfig: () => { yPaddingPx: number; indicators: Record<string, { enabled: boolean; params: Record<string, unknown> }> }
+  getConfig: () => { yPaddingPx: number }
   setConfig: (config: Record<string, unknown>) => void
 }
 
@@ -33,10 +35,73 @@ function createMockCanvasContext(): CanvasRenderingContext2D {
   } as unknown as CanvasRenderingContext2D
 }
 
+/**
+ * Create a mock scheduler that returns indicator metadata with getTitleInfo.
+ * Keys are case-insensitive (matching real IndicatorScheduler behavior).
+ */
+function createMockScheduler(
+  metadataMap: Record<string, { getTitleInfo: GetTitleInfoFn }>,
+  activeMainIndicators?: string[]
+): IndicatorScheduler {
+  const activeSet = new Set((activeMainIndicators ?? ['ma']).map((i: string) => i.toLowerCase()))
+  return {
+    getIndicatorMetadata: vi.fn((id: string) => metadataMap[id.toLowerCase()] ?? null),
+    getMainIndicators: vi.fn(() => Object.entries(metadataMap).map(([id, meta]) => ({
+      name: id,
+      getTitleInfo: meta.getTitleInfo,
+      category: 'main',
+    }))),
+    isMainIndicatorActive: vi.fn((id: string) => activeSet.has(id.toLowerCase())),
+    getMainIndicatorParams: vi.fn(() => ({})),
+  } as unknown as IndicatorScheduler
+}
+
+/**
+ * Create a mock getTitleInfo for MA that reads from shared state
+ */
+function createMAGetTitleInfo(): GetTitleInfoFn {
+  return (data, index, params, host, paneId): TitleInfo => {
+    const state = host.getSharedState<MARenderState>(MA_STATE_KEY)
+    if (!state) return null
+
+    const ci = index ?? (state.series[5]?.length ?? 100) - 1
+    const values: TitleValueItem[] = []
+    for (const period of state.enabledPeriods) {
+      const v = state.series[period]?.[ci]
+      if (v !== undefined) {
+        values.push({
+          label: `MA${period}`,
+          value: v,
+          color: '#888888',
+        })
+      }
+    }
+    return { name: 'MA', values: values.length > 0 ? values : undefined }
+  }
+}
+
+/**
+ * Create simple mock getTitleInfo for BOLL/EXPMA/ENE
+ */
+function createSimpleGetTitleInfo(name: string): GetTitleInfoFn {
+  return (): TitleInfo => ({
+    name,
+    params: [20, 2],
+    values: [
+      { label: 'MID', value: 100.00, color: '#FF0000' },
+      { label: 'UP', value: 120.00, color: '#00FF00' },
+      { label: 'DN', value: 80.00, color: '#0000FF' },
+    ],
+  })
+}
+
 /**
  * 创建 mock PluginHost
  */
-function createMockPluginHost(state?: MARenderState): PluginHost {
+function createMockPluginHost(
+  state?: MARenderState,
+  scheduler?: IndicatorScheduler
+): PluginHost {
   return {
     setSharedState: vi.fn(),
     getSharedState: vi.fn(<T>(key: string): T | undefined => {
@@ -59,6 +124,10 @@ function createMockPluginHost(state?: MARenderState): PluginHost {
     off: vi.fn(),
     once: vi.fn(),
     emit: vi.fn(),
+    getService: vi.fn((name: string) => {
+      if (name === 'indicatorScheduler') return scheduler
+      return undefined
+    }),
   } as unknown as PluginHost
 }
 
@@ -149,14 +218,9 @@ describe('createMainIndicatorLegendRendererPlugin', () => {
     expect(typeof plugin.onInstall).toBe('function')
   })
 
-  it('should declare all indicator namespace keys', () => {
+  it('should not declare any namespace keys (individual renderers own their state)', () => {
     const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
-    expect(plugin.getDeclaredNamespaces()).toEqual([
-      MA_STATE_KEY,
-      BOLL_STATE_KEY,
-      EXPMA_STATE_KEY,
-      ENE_STATE_KEY,
-    ])
+    expect(plugin.getDeclaredNamespaces()).toEqual([])
   })
 })
 
@@ -168,17 +232,13 @@ describe('MainIndicatorLegend draw', () => {
     ctx = createMockCanvasContext()
   })
 
-  it('should not draw MA when MA is disabled', () => {
+  it('should not draw MA when MA is not active', () => {
     const state = createTestMARenderState()
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, [])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
-    // Disable MA
-    plugin.setConfig({
-      indicators: { MA: { enabled: false, params: {} } },
-    })
-
     const context = createMockRenderContext(ctx)
     plugin.draw(context)
 
@@ -190,7 +250,8 @@ describe('MainIndicatorLegend draw', () => {
 
   it('should draw MA values from StateStore', () => {
     const state = createTestMARenderState()
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
@@ -215,7 +276,8 @@ describe('MainIndicatorLegend draw', () => {
       },
       enabledPeriods: [5],
     })
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
@@ -227,7 +289,7 @@ describe('MainIndicatorLegend draw', () => {
 
     // Should show value 150 at index 50 (100 + 50)
     const maValueCalls = fillTextCalls.filter(call =>
-      String(call[0]).includes('150.00')
+      String(call[0]).includes('150.000')
     )
     expect(maValueCalls.length).toBeGreaterThan(0)
   })
@@ -239,7 +301,8 @@ describe('MainIndicatorLegend draw', () => {
       },
       enabledPeriods: [5],
     })
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
@@ -261,7 +324,7 @@ describe('MainIndicatorLegend draw', () => {
 
     // Should show last value (109) at index 9
     const maValueCalls = fillTextCalls.filter(call =>
-      String(call[0]).includes('109.00')
+      String(call[0]).includes('109.000')
     )
     expect(maValueCalls.length).toBeGreaterThan(0)
   })
@@ -283,27 +346,29 @@ describe('MainIndicatorLegend draw', () => {
       visibleMax: -Infinity,
       enabledPeriods: [],
     })
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
     const context = createMockRenderContext(ctx)
     plugin.draw(context)
 
-    // Should not draw any MA values
+    // Should not draw any MA period values (name 'MA' may still appear but no period texts)
     const fillTextCalls = vi.mocked(ctx.fillText).mock.calls
     const ma5Calls = fillTextCalls.filter(call => String(call[0]).includes('MA5'))
     expect(ma5Calls).toHaveLength(0)
   })
 
-  it('should display values with 2 decimal places', () => {
+  it('should display values with 3 decimal places', () => {
     const state = createTestMARenderState({
       series: {
         5: Array.from({ length: 100 }, () => 123.4567),
       },
       enabledPeriods: [5],
     })
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
@@ -314,29 +379,31 @@ describe('MainIndicatorLegend draw', () => {
 
     // Should show formatted value
     const formattedValueCalls = fillTextCalls.filter(call =>
-      String(call[0]).includes('123.46')
+      String(call[0]).includes('123.457')
     )
     expect(formattedValueCalls.length).toBeGreaterThan(0)
   })
 
   it('should use correct colors for each MA period', () => {
     const state = createTestMARenderState()
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
     const context = createMockRenderContext(ctx)
     plugin.draw(context)
 
-    const fillStyleSetter = vi.mocked(ctx).fillStyle as unknown as ReturnType<typeof vi.fn>
-
-    // Should have set fillStyle for each period's color
-    expect(fillStyleSetter).not.toBeUndefined()
+    // Should have drawn MA period values with proper colors
+    const fillTextCalls = vi.mocked(ctx.fillText).mock.calls
+    const ma5Calls = fillTextCalls.filter(call => String(call[0]).includes('MA5'))
+    expect(ma5Calls.length).toBeGreaterThan(0)
   })
 
   it('should save and restore context', () => {
     const state = createTestMARenderState()
-    const mockHost = createMockPluginHost(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
+    const mockHost = createMockPluginHost(state, scheduler)
     plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 }) as TestableLegendRenderer
     plugin.onInstall(mockHost)
 
@@ -357,10 +424,15 @@ describe('MainIndicatorLegend MA data source', () => {
       enabledPeriods: [5],
     })
     const mockGetSharedState = vi.fn().mockReturnValue(state)
+    const scheduler = createMockScheduler({ ma: { getTitleInfo: createMAGetTitleInfo() } }, ['ma'])
     const mockHost = {
       setSharedState: vi.fn(),
       getSharedState: mockGetSharedState,
       clearByOwner: vi.fn(),
+      getService: vi.fn((name: string) => {
+        if (name === 'indicatorScheduler') return scheduler
+        return undefined
+      }),
     } as unknown as PluginHost
 
     const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
@@ -373,7 +445,7 @@ describe('MainIndicatorLegend MA data source', () => {
     })
     plugin.draw(context)
 
-    // Verify it read from StateStore
+    // Verify it read from StateStore (via getTitleInfo calling getSharedState)
     expect(mockGetSharedState).toHaveBeenCalledWith(MA_STATE_KEY)
 
     const fillTextCalls = vi.mocked(ctx.fillText).mock.calls
@@ -387,14 +459,12 @@ describe('MainIndicatorLegend MA data source', () => {
 })
 
 describe('MainIndicatorLegend config management', () => {
-  it('getConfig should return current config', () => {
+  it('getConfig should return current yPaddingPx', () => {
     const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
 
     const config = plugin.getConfig()
 
     expect(config.yPaddingPx).toBe(20)
-    expect(config.indicators.MA.enabled).toBe(true)
-    expect(config.indicators.BOLL.enabled).toBe(false)
   })
 
   it('setConfig should update yPaddingPx', () => {
@@ -405,37 +475,18 @@ describe('MainIndicatorLegend config management', () => {
     const config = plugin.getConfig()
     expect(config.yPaddingPx).toBe(30)
   })
-
-  it('setConfig should merge indicator config', () => {
-    const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
-
-    plugin.setConfig({
-      indicators: {
-        MA: { enabled: false, params: {} },
-        BOLL: { enabled: true, params: { period: 26 } },
-      },
-    })
-
-    const config = plugin.getConfig()
-    expect(config.indicators.MA.enabled).toBe(false)
-    expect(config.indicators.BOLL.enabled).toBe(true)
-    expect(config.indicators.BOLL.params.period).toBe(26)
-  })
 })
 
 describe('MainIndicatorLegend with other indicators', () => {
-  it('should draw BOLL when enabled', () => {
-    const mockHost = createMockPluginHost(undefined)
+  it('should draw BOLL when active', () => {
+    const scheduler = createMockScheduler(
+      { boll: { getTitleInfo: createSimpleGetTitleInfo('BOLL') } },
+      ['boll']
+    )
+    const mockHost = createMockPluginHost(undefined, scheduler)
     const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
     plugin.onInstall(mockHost)
 
-    plugin.setConfig({
-      indicators: {
-        MA: { enabled: false, params: {} },
-        BOLL: { enabled: true, params: { period: 20, multiplier: 2 } },
-      },
-    })
-
     const ctx = createMockCanvasContext()
     const context = createMockRenderContext(ctx)
     plugin.draw(context)
@@ -449,18 +500,15 @@ describe('MainIndicatorLegend with other indicators', () => {
     expect(bollLabelCalls.length).toBeGreaterThan(0)
   })
 
-  it('should draw EXPMA when enabled', () => {
-    const mockHost = createMockPluginHost(undefined)
+  it('should draw EXPMA when active', () => {
+    const scheduler = createMockScheduler(
+      { expma: { getTitleInfo: createSimpleGetTitleInfo('EXPMA') } },
+      ['expma']
+    )
+    const mockHost = createMockPluginHost(undefined, scheduler)
     const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
     plugin.onInstall(mockHost)
 
-    plugin.setConfig({
-      indicators: {
-        MA: { enabled: false, params: {} },
-        EXPMA: { enabled: true, params: { fastPeriod: 12, slowPeriod: 50 } },
-      },
-    })
-
     const ctx = createMockCanvasContext()
     const context = createMockRenderContext(ctx)
     plugin.draw(context)
@@ -474,18 +522,15 @@ describe('MainIndicatorLegend with other indicators', () => {
     expect(expmaLabelCalls.length).toBeGreaterThan(0)
   })
 
-  it('should draw ENE when enabled', () => {
-    const mockHost = createMockPluginHost(undefined)
+  it('should draw ENE when active', () => {
+    const scheduler = createMockScheduler(
+      { ene: { getTitleInfo: createSimpleGetTitleInfo('ENE') } },
+      ['ene']
+    )
+    const mockHost = createMockPluginHost(undefined, scheduler)
     const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
     plugin.onInstall(mockHost)
 
-    plugin.setConfig({
-      indicators: {
-        MA: { enabled: false, params: {} },
-        ENE: { enabled: true, params: { period: 10, deviation: 11 } },
-      },
-    })
-
     const ctx = createMockCanvasContext()
     const context = createMockRenderContext(ctx)
     plugin.draw(context)
@@ -498,4 +543,22 @@ describe('MainIndicatorLegend with other indicators', () => {
     )
     expect(eneLabelCalls.length).toBeGreaterThan(0)
   })
+
+  it('should draw any registered main indicator when active (WMA example)', () => {
+    const scheduler = createMockScheduler(
+      { wma: { getTitleInfo: createSimpleGetTitleInfo('WMA') } },
+      ['wma']
+    )
+    const mockHost = createMockPluginHost(undefined, scheduler)
+    const plugin = createMainIndicatorLegendRendererPlugin({ yPaddingPx: 20 })
+    plugin.onInstall(mockHost)
+
+    const ctx = createMockCanvasContext()
+    const context = createMockRenderContext(ctx)
+    plugin.draw(context)
+
+    const fillTextCalls = vi.mocked(ctx.fillText).mock.calls
+    const wmaLabelCalls = fillTextCalls.filter(call => String(call[0]).includes('WMA'))
+    expect(wmaLabelCalls.length).toBeGreaterThan(0)
+  })
 })