@363045841yyt/klinechart-core 0.7.3

package/src/semantic/validator.ts

@@ -0,0 +1,349 @@
+/**
+ * 语义化配置校验器
+ * 包含 JSON Schema 校验、安全校验、业务逻辑校验
+ */
+
+import type { SemanticChartConfig, DataConfig, ValidationResult, SecurityResult, MarkerStyle } from './types'
+
+// ============ 常量定义 ============
+
+/** 禁止的属性键（防止原型污染） */
+const FORBIDDEN_KEYS = ['__proto__', 'constructor', 'prototype']
+
+/** 颜色值正则（严格校验） */
+const COLOR_PATTERN =
+  /^(#[0-9a-fA-F]{3,8}|rgb\(\s*\d{1,3}\s*,\s*\d{1,3}\s*,\s*\d{1,3}\s*\)|rgba\(\s*\d{1,3}\s*,\s*\d{1,3}\s*,\s*\d{1,3}\s*,\s*[\d.]+\s*\))$/
+
+/** 股票代码正则规则
+ * 注意：北交所规则持续扩充，以交易所官方公告为准
+ * 规则版本日期：2025-01
+ */
+const SYMBOL_PATTERNS = {
+  SH: /^(600|601|603|605|688)\d{3}$/, // 上交所
+  SZ: /^(000|001|002|003|300|301)\d{3}$/, // 深交所
+  BJ: /^(83|87|43|82)\d{4}$/, // 北交所
+}
+
+/** 安全限制 */
+interface SecurityLimits {
+  maxJsonSize: number
+  maxIndicators: number
+  maxCustomMarkers: number
+}
+
+/** 根据 period 计算最大日期范围（天数） */
+function getMaxDateRangeDays(period: DataConfig['period']): number {
+  const LIMITS: Record<string, number> = {
+    '5min': 30,
+    '15min': 60,
+    '30min': 90,
+    '60min': 180,
+    daily: 365 * 3,
+    weekly: 365 * 5,
+    monthly: 365 * 10,
+  }
+  return LIMITS[period] ?? 365
+}
+
+// ============ 单例 Canvas（颜色校验用） ============
+
+const _colorCanvas = typeof document !== 'undefined' ? document.createElement('canvas') : null
+const _colorCtx = _colorCanvas?.getContext('2d') ?? null
+
+// ============ 校验器类 ============
+
+export class SemanticConfigValidator {
+  private limits: SecurityLimits
+  private _ajv: Promise<any> | null = null
+
+  constructor() {
+    this.limits = {
+      maxJsonSize: 64 * 1024, // 64KB
+      maxIndicators: 10,
+      maxCustomMarkers: 100,
+    }
+  }
+
+  private async getAjv(): Promise<any> {
+    if (!this._ajv) {
+      this._ajv = (async () => {
+        const Ajv = (await import('ajv')).default
+        const ajv = new Ajv({ allErrors: true, allowUnionTypes: true })
+        ajv.addFormat('date', {
+          type: 'string',
+          validate: (data: string) => /^\d{4}-\d{2}-\d{2}$/.test(data),
+        })
+        const schemaModule = await import('./schema.json')
+        ajv.addSchema(schemaModule.default || schemaModule)
+        return ajv
+      })()
+    }
+    return this._ajv
+  }
+
+  /**
+   * 入口校验（在 JSON.parse 之前调用）
+   * 检查原始字符串大小
+   */
+  validateRawInput(raw: string): ValidationResult {
+    const byteLength =
+      typeof TextEncoder !== 'undefined'
+        ? new TextEncoder().encode(raw).byteLength
+        : raw.length * 3 // SSR 降级：保守估计
+
+    if (byteLength > this.limits.maxJsonSize) {
+      return { valid: false, errors: ['JSON payload too large (max 64KB)'] }
+    }
+    return { valid: true }
+  }
+
+  /**
+   * JSON Schema 校验
+   */
+  async validate(config: unknown): Promise<ValidationResult> {
+    // 1. 类型检查
+    if (!config || typeof config !== 'object') {
+      return { valid: false, errors: ['Config must be an object'] }
+    }
+
+    // 2. 原型污染检查
+    const protoCheck = this.checkPrototypePollution(config)
+    if (!protoCheck.valid) {
+      return protoCheck
+    }
+
+    // 3. JSON Schema 校验
+    try {
+      const ajv = await this.getAjv()
+      const valid = ajv.validate('https://kmap.dev/schemas/semantic-chart-config/1.0.0', config)
+      if (!valid) {
+        const errors = ajv.errors?.map((e: { instancePath: string; message?: string }) => `${e.instancePath} ${e.message}`) || ['Schema validation failed']
+        return { valid: false, errors }
+      }
+    } catch {
+      return { valid: false, errors: ['Schema validation unavailable (ajv not loaded)'] }
+    }
+
+    return { valid: true }
+  }
+
+  /**
+   * 安全校验（纯同步）
+   */
+  securityCheck(config: SemanticChartConfig): SecurityResult {
+    const violations: string[] = []
+
+    // 1. 检查日期范围限制
+    const dateCheck = this.checkDateRange(config.data)
+    if (!dateCheck.valid) {
+      violations.push(...(dateCheck.errors || []))
+    }
+
+    // 2. 检查股票代码格式
+    const symbolCheck = this.checkSymbol(config.data)
+    if (!symbolCheck.valid) {
+      violations.push(...(symbolCheck.errors || []))
+    }
+
+    // 3. 检查指标数量限制
+    if (config.indicators) {
+      const mainCount = config.indicators.main?.length || 0
+      const subCount = config.indicators.sub?.length || 0
+      if (mainCount + subCount > this.limits.maxIndicators) {
+        violations.push(`Too many indicators (max ${this.limits.maxIndicators})`)
+      }
+    }
+
+    // 4. 检查标记数量限制
+    const markerCount = config.markers?.customMarkers?.length || 0
+    if (markerCount > this.limits.maxCustomMarkers) {
+      violations.push(`Too many custom markers (max ${this.limits.maxCustomMarkers})`)
+    }
+
+    // 5. 检查颜色值格式
+    if (config.markers?.customMarkers) {
+      for (const marker of config.markers.customMarkers) {
+        if (marker.style) {
+          const colorCheck = this.checkMarkerStyle(marker.style, marker.id)
+          violations.push(...colorCheck)
+        }
+        // 检查日期格式
+        const dateCheck = this.checkMarkerDate(marker.date, marker.id)
+        violations.push(...dateCheck)
+      }
+    }
+
+    return {
+      passed: violations.length === 0,
+      violations: violations.length > 0 ? violations : undefined,
+    }
+  }
+
+  // ============ 私有方法 ============
+
+  private checkPrototypePollution(obj: unknown, path = ''): ValidationResult {
+    if (!obj || typeof obj !== 'object') {
+      return { valid: true }
+    }
+
+    const errors: string[] = []
+    for (const key of Object.keys(obj as Record<string, unknown>)) {
+      if (FORBIDDEN_KEYS.includes(key)) {
+        errors.push(`Forbidden key "${key}" found at ${path}`)
+      }
+    }
+
+    return errors.length > 0 ? { valid: false, errors } : { valid: true }
+  }
+
+  private checkDateRange(data: DataConfig): ValidationResult {
+    const errors: string[] = []
+
+    const startDate = new Date(data.startDate)
+    const endDate = new Date(data.endDate)
+
+    if (isNaN(startDate.getTime())) {
+      errors.push(`Invalid startDate: ${data.startDate}`)
+    }
+    if (isNaN(endDate.getTime())) {
+      errors.push(`Invalid endDate: ${data.endDate}`)
+    }
+
+    if (errors.length > 0) {
+      return { valid: false, errors }
+    }
+
+    const maxDays = getMaxDateRangeDays(data.period)
+    const diffDays = Math.ceil((endDate.getTime() - startDate.getTime()) / (1000 * 60 * 60 * 24))
+
+    if (diffDays > maxDays) {
+      errors.push(`Date range exceeds maximum for period "${data.period}" (max ${maxDays} days, got ${diffDays})`)
+    }
+
+    if (diffDays < 0) {
+      errors.push('endDate must be after startDate')
+    }
+
+    return errors.length > 0 ? { valid: false, errors } : { valid: true }
+  }
+
+  private checkSymbol(data: DataConfig): ValidationResult {
+    const { symbol, exchange } = data
+
+    if (exchange) {
+      const pattern = SYMBOL_PATTERNS[exchange]
+      if (!pattern.test(symbol)) {
+        return { valid: false, errors: [`Invalid symbol "${symbol}" for exchange "${exchange}"`] }
+      }
+    } else {
+      // 自动识别
+      const valid = Object.values(SYMBOL_PATTERNS).some((p) => p.test(symbol))
+      if (!valid) {
+        return { valid: false, errors: [`Invalid symbol format: "${symbol}"`] }
+      }
+    }
+
+    return { valid: true }
+  }
+
+  private checkMarkerStyle(style: MarkerStyle, markerId: string): string[] {
+    const errors: string[] = []
+    const colorFields = ['fillColor', 'strokeColor', 'textColor'] as const
+
+    for (const field of colorFields) {
+      const value = style[field]
+      if (typeof value === 'string' && !COLOR_PATTERN.test(value)) {
+        errors.push(`Invalid ${field} in marker "${markerId}": ${value}`)
+      }
+    }
+
+    // 检查颜色值安全性（浏览器环境）
+    if (_colorCtx) {
+      for (const field of colorFields) {
+        const value = style[field]
+        if (typeof value === 'string') {
+          _colorCtx.fillStyle = '#000000'
+          _colorCtx.fillStyle = value
+          // 浏览器会过滤非法值，如果返回的不是合法颜色说明有问题
+          if (_colorCtx.fillStyle === '#000000' && value !== '#000000' && !value.startsWith('#000')) {
+            errors.push(`Potentially unsafe ${field} in marker "${markerId}": ${value}`)
+          }
+        }
+      }
+    }
+
+    return errors
+  }
+
+  /** 日期格式正则：YYYY-MM-DD 或 YYYY-MM-DD HH:mm */
+  private static readonly DATE_PATTERN = /^\d{4}-\d{2}-\d{2}( \d{2}:\d{2})?$/
+
+  private checkMarkerDate(date: string, markerId: string): string[] {
+    const errors: string[] = []
+
+    if (!SemanticConfigValidator.DATE_PATTERN.test(date)) {
+      errors.push(`Invalid date format in marker "${markerId}": ${date} (expected YYYY-MM-DD or YYYY-MM-DD HH:mm)`)
+      return errors
+    }
+
+    // 验证日期是否有效
+    const hasTime = date.includes(' ')
+    let parsed: Date
+
+    if (hasTime) {
+      parsed = new Date(date)
+    } else {
+      // 对于纯日期，解析为 UTC
+      const [year, month, day] = date.split('-').map(Number)
+      parsed = new Date(Date.UTC(year!, month! - 1, day!))
+    }
+
+    if (isNaN(parsed.getTime())) {
+      errors.push(`Invalid date value in marker "${markerId}": ${date}`)
+    }
+
+    return errors
+  }
+}
+
+// ============ 工具函数导出 ============
+
+/**
+ * 净化参数对象（防止原型污染）
+ */
+export function sanitizeParams(params: Record<string, unknown>): Record<string, unknown> {
+  const safe: Record<string, unknown> = Object.create(null)
+  for (const [key, value] of Object.entries(params)) {
+    if (!FORBIDDEN_KEYS.includes(key)) {
+      safe[key] = value
+    }
+  }
+  return safe
+}
+
+/**
+ * 净化颜色值（浏览器环境）
+ */
+export function sanitizeColor(input: string): string | null {
+  if (!_colorCtx) return null
+  _colorCtx.fillStyle = '#000000'
+  _colorCtx.fillStyle = input
+  return _colorCtx.fillStyle
+}
+
+/**
+ * 校验颜色值格式
+ */
+export function validateColor(color: string): boolean {
+  return COLOR_PATTERN.test(color)
+}
+
+/**
+ * 校验股票代码
+ */
+export function validateSymbol(symbol: string, exchange?: 'SH' | 'SZ' | 'BJ'): boolean {
+  if (exchange) {
+    return SYMBOL_PATTERNS[exchange].test(symbol)
+  }
+  return Object.values(SYMBOL_PATTERNS).some((p) => p.test(symbol))
+}

package/src/types/kLine.ts

@@ -0,0 +1,13 @@
+import type { KLineData } from './price'
+
+export type kLineTrend = 'up' | 'down' | 'flat'
+
+export function getKLineTrend(KLineData: KLineData): kLineTrend {
+  if (KLineData.open > KLineData.close) {
+    return 'down'
+  } else if (KLineData.open < KLineData.close) {
+    return 'up'
+  } else {
+    return 'flat'
+  }
+}

package/src/types/price.ts

@@ -0,0 +1,56 @@
+export interface KLineData {
+  /* 时间戳（毫秒） */
+  timestamp: number
+  /* 开盘价 */
+  open: number
+  /* 最高价 */
+  high: number
+  /* 最低价 */
+  low: number
+  /* 收盘价 */
+  close: number
+  /** 股票代码（东财等数据源会提供） */
+  stockCode?: string
+  /** 成交量 */
+  volume?: number
+  /** 成交额 */
+  turnover?: number
+  /** 振幅 */
+  amplitude?: number
+  /** 涨跌幅 */
+  changePercent?: number
+  /** 涨跌额 */
+  changeAmount?: number
+  /** 换手率 */
+  turnoverRate?: number
+}
+
+export interface KLineDailyDongCaiResponse extends KLineData {
+  stockCode: string
+  volume: number
+  turnover: number
+  amplitude: number
+  changePercent: number
+  changeAmount: number
+  turnoverRate: number
+}
+
+
+export function toKLineData(arr: KLineDailyDongCaiResponse[]): KLineData[] {
+  return arr
+    .map((e) => ({
+      timestamp: e.timestamp,
+      open: e.open,
+      high: e.high,
+      low: e.low,
+      close: e.close,
+      stockCode: e.stockCode,
+      volume: e.volume,
+      turnover: e.turnover,
+      amplitude: e.amplitude,
+      changePercent: e.changePercent,
+      changeAmount: e.changeAmount,
+      turnoverRate: e.turnoverRate,
+    }))
+    .sort((a, b) => a.timestamp - b.timestamp)
+}

package/src/types/volumePrice.ts

@@ -0,0 +1,33 @@
+/**
+ * 量价关系类型枚举
+ */
+export enum VolumePriceRelation {
+  /** 量价齐升（放量上涨）- 价格上涨且成交量显著放大 */
+  RISE_WITH_VOLUME = 'rise_with_volume',
+  /** 量价背离（缩量上涨）- 价格上涨但成交量萎缩 */
+  RISE_WITHOUT_VOLUME = 'rise_without_volume',
+  /** 量增价跌 - 价格下跌且成交量放大 */
+  FALL_WITH_VOLUME = 'fall_with_volume',
+  /** 量缩价跌 - 价格下跌且成交量萎缩 */
+  FALL_WITHOUT_VOLUME = 'fall_without_volume',
+  /** 中性状态 - 其他情况 */
+  OTHERS = 'others'
+}
+
+/** 量价关系计算配置 */
+export interface VolumePriceConfig {
+  /** 成交量放大阈值（相对于均值的倍数） */
+  volumeAmplifyThreshold: number
+  /** 成交量萎缩阈值（相对于均值的倍数） */
+  volumeShrinkThreshold: number
+  /** 计算均值的周期 */
+  avgPeriod: number
+}
+
+/** 默认量价关系计算配置 */
+export const DEFAULT_VOLUME_PRICE_CONFIG: VolumePriceConfig = {
+  volumeAmplifyThreshold: 1.5,
+  volumeShrinkThreshold: 0.8,
+  avgPeriod: 20,
+}
+

package/src/utils/dateFormat.ts

@@ -0,0 +1,208 @@
+/**
+ * 日期格式化工具函数集合
+ * 统一管理项目中所有日期相关的格式化逻辑
+ */
+
+// ========== 模块级复用的 Intl.DateTimeFormat 实例 ==========
+// Intl.DateTimeFormat 构造极其昂贵（~36ms），必须复用
+const YMD_FORMATTER = new Intl.DateTimeFormat('zh-CN', {
+    timeZone: 'Asia/Shanghai',
+    year: 'numeric',
+    month: '2-digit',
+    day: '2-digit',
+})
+
+// ========== 缓存配置 ==========
+const YMD_CACHE_SIZE = 1024
+const ymdCache = new Map<number, string>()
+
+/**
+ * 将时间戳格式化为 YYYYMMDD 格式（纯数字，无分隔符）
+ * @param timestamp - 时间戳（毫秒）
+ * @returns 格式化后的日期字符串，例如 "20250114"
+ *
+ * @example
+ * formatDateToYYYYMMDDNoDash(1736793600000) // "20250114"
+ */
+export function formatDateToYYYYMMDDNoDash(timestamp: number): string {
+    const d = new Date(timestamp)
+    const year = d.getFullYear()
+    const month = String(d.getMonth() + 1).padStart(2, '0')
+    const day = String(d.getDate()).padStart(2, '0')
+    return `${year}${month}${day}`
+}
+
+/**
+ * 获取当前日期的 YYYYMMDD 格式（纯数字，无分隔符）
+ * @returns 当前日期的格式化字符串，例如 "20250114"
+ *
+ * @example
+ * getCurrentDateYYYYMMDD() // "20250114"（根据实际日期）
+ */
+export function getCurrentDateYYYYMMDD(): string {
+    const d = new Date()
+    const year = d.getFullYear()
+    const month = String(d.getMonth() + 1).padStart(2, '0')
+    const day = String(d.getDate()).padStart(2, '0')
+    return `${year}${month}${day}`
+}
+
+/**
+ * 将时间戳格式化为 YYYY-MM-DD 格式（上海时区）
+ * @param timestamp - 时间戳（毫秒）
+ * @returns 格式化后的日期字符串，例如 "2025-01-14"
+ *
+ * @example
+ * formatDateToYYYYMMDD(1736793600000) // "2025-01-14"
+ */
+export function formatDateToYYYYMMDD(timestamp: number): string {
+    // 缓存命中检查
+    const cached = ymdCache.get(timestamp)
+    if (cached !== undefined) return cached
+
+    // 使用复用的 formatter，避免每次构造开销
+    const parts = YMD_FORMATTER.formatToParts(new Date(timestamp))
+
+    // 用 for 循环替代 .reduce，避免临时对象分配
+    let y = '', m = '', d = ''
+    for (let i = 0; i < parts.length; i++) {
+        const p = parts[i]
+        if (p.type === 'year') y = p.value
+        else if (p.type === 'month') m = p.value
+        else if (p.type === 'day') d = p.value
+    }
+
+    const result = `${y}-${m}-${d}`
+
+    // 写入缓存，防膨胀
+    if (ymdCache.size >= YMD_CACHE_SIZE) ymdCache.clear()
+    ymdCache.set(timestamp, result)
+
+    return result
+}
+
+// ========== formatMonthOrYear 缓存 ==========
+const MONTH_YEAR_CACHE_SIZE = 512
+const monthYearCache = new Map<number, { text: string; isYear: boolean }>()
+
+/**
+ * 格式化月份或年份用于显示
+ * 当年为 1 月时显示年份，其他月份显示"X月"格式
+ * @param timestamp - 时间戳（毫秒）
+ * @returns 包含文本和是否为年份的标志
+ *
+ * @example
+ * formatMonthOrYear(1704067200000) // { text: "2024", isYear: true }  (2024年1月)
+ * formatMonthOrYear(1706745600000) // { text: "2月", isYear: false } (2024年2月)
+ */
+export function formatMonthOrYear(timestamp: number): { text: string; isYear: boolean } {
+    // 缓存命中检查
+    const cached = monthYearCache.get(timestamp)
+    if (cached !== undefined) return cached
+
+    const d = new Date(timestamp)
+    const year = d.getFullYear()
+    const month = d.getMonth() + 1
+    // 当年 1 月：直接标注年份；其它月份：标注"X月"
+    const result = month === 1 ? { text: String(year), isYear: true } : { text: `${month}月`, isYear: false }
+
+    // 写入缓存，防膨胀
+    if (monthYearCache.size >= MONTH_YEAR_CACHE_SIZE) monthYearCache.clear()
+    monthYearCache.set(timestamp, result)
+
+    return result
+}
+
+/**
+ * 生成月份键值用于比较（YYYY-M 格式）
+ * 注意：月份未补零，用于快速比较月份是否相同
+ * @param timestamp - 时间戳（毫秒）
+ * @returns 月份键值，例如 "2025-1"
+ *
+ * @example
+ * monthKey(1736793600000) // "2025-1"
+ */
+/**
+ * 生成月份键值用于比较
+ * 返回数字 year * 12 + month，比字符串比较更快且无分配
+ * 使用 new Date 保证本地时区正确（与显示一致）
+ *
+ * @example
+ * monthKey(1736793600000) // 24301 (2025*12 + 0)
+ */
+export function monthKey(timestamp: number): number {
+    const d = new Date(timestamp)
+    return d.getFullYear() * 12 + d.getMonth()
+}
+
+// ========== 便捷别名 ==========
+
+/**
+ * formatDateToYYYYMMDD 的别名，保持与历史代码的兼容性
+ * timestamp 是"上海时区当天 00:00:00"映射到 UTC 的值；显示时强制按上海时区格式化
+ * @param ts - 时间戳（毫秒）
+ * @returns 格式化后的日期字符串，例如 "2025-01-14"
+ */
+export const formatShanghaiDate = formatDateToYYYYMMDD
+
+/**
+ * formatDateToYYYYMMDD 的别名，用于十字线日期标签显示
+ * 按上海时区格式化，避免不同时区出现日期偏移
+ * @param ts - 时间戳（毫秒）
+ * @returns 格式化后的日期字符串，例如 "2025-01-14"
+ */
+export const formatYMDShanghai = formatDateToYYYYMMDD
+
+/**
+ * 查找每个月份第一个K线的索引
+ * @param data - K线数据数组（按时间升序排列）
+ * @returns 月边界索引数组，例如 [0, 30, 60] 表示第0、30、60个K线分别是每月的第一个交易日
+ * 
+ * @example
+ * // 假设数据：[1/2, 1/3, 2/1, 2/2, 3/1, 3/2]
+ * findMonthBoundaries(data) // [0, 2, 4]
+ * // 解释：第0个K线是1月第一个，第2个K线是2月第一个，第4个K线是3月第一个
+ */
+// findMonthBoundaries 缓存
+let _cacheDataRef: Array<{ timestamp: number } | undefined> | null = null
+let _cacheLen = 0
+let _cacheFirstTs = 0
+let _cacheLastTs = 0
+let _cacheResult: number[] = []
+
+/**
+ * 查找每个月份第一个K线的索引
+ * 结果按数据引用缓存，同一份数据多次调用直接返回缓存
+ */
+export function findMonthBoundaries(data: Array<{ timestamp: number } | undefined>): number[] {
+    if (data.length === 0) return []
+
+    // 缓存命中：同一引用 + 同长度 + 首尾时间戳不变
+    if (_cacheDataRef === data && _cacheLen === data.length) {
+        const firstTs = data[0]?.timestamp
+        const lastTs = data[data.length - 1]?.timestamp
+        if (firstTs === _cacheFirstTs && lastTs === _cacheLastTs) {
+            return _cacheResult
+        }
+    }
+
+    const boundaries: number[] = [0]
+    let lastKey = monthKey(data[0]!.timestamp)
+
+    for (let i = 1; i < data.length; i++) {
+        const cur = data[i]
+        if (!cur) continue
+        const curKey = monthKey(cur.timestamp)
+        if (curKey !== lastKey) {
+            boundaries.push(i)
+            lastKey = curKey
+        }
+    }
+
+    _cacheDataRef = data
+    _cacheLen = data.length
+    _cacheFirstTs = data[0]?.timestamp ?? 0
+    _cacheLastTs = data[data.length - 1]?.timestamp ?? 0
+    _cacheResult = boundaries
+    return boundaries
+}