// src/components/Tradingview/datafeed.ts
class DataFeedBase {
  configuration: DatafeedConfiguration

  constructor(props: Partial<ChartingLibraryWidgetOptions>) {
    this.configuration = {
      supports_time: true,
      supports_timescale_marks: true,
      supports_marks: true,
      // 支持的分辨率
      supported_resolutions: ['1', '5', '15', '30', '60', '240', '1D', '1W', '1M'],
      intraday_multipliers: ['1', '5', '15', '30', '60', '240', '1D', '1W', '1M']
    } as DatafeedConfiguration

    this.setActiveSymbolInfo = props.setActiveSymbolInfo
    this.removeActiveSymbol = props.removeActiveSymbol
    this.getDataFeedBarCallback = props.getDataFeedBarCallback
    this.isZh = props.locale === 'zh_TW'
  }

  // 图表初始化时调用，设置支持的配置
  onReady(callback) {
    setTimeout(() => {
      callback(this.configuration)
    }, 0)
  }

  // 解析品种信息
  async resolveSymbol(symbolName, onSymbolResolvedCallback, onResolveErrorCallback, extension) {
    const resolution = String(STORAGE_GET_TRADINGVIEW_RESOLUTION() || '')
    const ENV = getEnv()
    const urlPrefix = ENV.isApp ? getInjectParams().baseUrl : ''

    let symbolInfo
    if (!ENV.isApp) {
      // HTTP 请求获取品种信息
      const res = await request(`${urlPrefix}/api/trade-core/coreApi/symbols/symbol/detail?symbol=${symbolName}`)
      symbolInfo = res?.data || {}
    } else {
      // APP 内获取 RN 传递的数据
      symbolInfo = {
        ...(ENV?.injectParams?.symbolInfo || {}),
        ...(stores.global.symbolInfo || {})
      }
    }

    const currentSymbol = {
      ...symbolInfo,
      precision: symbolInfo?.symbolDecimal || 2,
      description: symbolInfo?.remark || '',
      exchange: '',
      session: '24x7',
      name: symbolInfo.symbol,
      dataSourceCode: symbolInfo.dataSourceCode
    }

    const commonSymbolInfo = {
      has_intraday: true,
      has_daily: true,
      has_weekly_and_monthly: true,
      intraday_multipliers: this.configuration.intraday_multipliers,
      supported_resolutions: this.configuration.supported_resolutions,
      data_status: 'streaming',
      format: 'price',
      minmov: 1,
      pricescale: Math.pow(10, currentSymbol.precision),
      ticker: currentSymbol?.name
    } as LibrarySymbolInfo

    const currentSymbolInfo = {
      ...commonSymbolInfo,
      ...currentSymbol,
      description: this.isZh ? currentSymbol.description : currentSymbol?.name,
      exchange: this.isZh ? currentSymbol?.exchange : '',
      session: '0000-0000|0000-0000:1234567;1',
      timezone: ['D', 'W', 'M', 'Y'].some((item) => resolution.endsWith(item))
        ? 'Etc/UTC'
        : 'Asia/Shanghai'
    } as LibrarySymbolInfo

    setTimeout(() => {
      onSymbolResolvedCallback(currentSymbolInfo)
    }, 0)
  }

  // 搜索品种
  searchSymbols(userInput, exchange, symbolType, onResultReadyCallback) {
    const keyword = userInput || ''
    const resultArr = symbolInfoArr
      .filter((item) => item.name.includes(keyword))
      .map((item) => ({
        symbol: item.name,
        name: item.name,
        full_name: `${item.name}`,
        description: this.isZh ? item.description : item.name,
        exchange: this.isZh ? item.exchange : '',
        type: item.type,
        ticker: item.name
      }))

    setTimeout(() => {
      onResultReadyCallback(resultArr)
    }, 0)
  }

  // 获取 K 线历史数据（核心方法）
  getBars(symbolInfo, resolution, periodParams, onHistoryCallback, onErrorCallback) {
    const { from, to, firstDataRequest, countBack } = periodParams
    this.setActiveSymbolInfo({ symbolInfo, resolution })
    this.getDataFeedBarCallback({
      symbolInfo,
      resolution,
      from,
      to,
      countBack,
      onHistoryCallback,
      onErrorCallback,
      firstDataRequest
    })
  }

  // 订阅实时数据更新
  subscribeBars(symbolInfo, resolution, onRealtimeCallback, subscriberUID, onResetCacheNeededCallback) {
    this.setActiveSymbolInfo({
      symbolInfo,
      resolution,
      onRealtimeCallback,
      subscriberUID,
      onResetCacheNeededCallback
    })
    mitt.on('symbol_change', () => {
      onResetCacheNeededCallback()
    })
  }

  // 取消订阅
  unsubscribeBars(subscriberUID) {
    this.removeActiveSymbol(subscriberUID)
  }
}

export default DataFeedBase

// src/components/Tradingview/widgetOpts.tsx
import ma from './customIndicators/ma'

export default function getWidgetOpts(
  props,
  containerRef: any,
  datafeedParams: any
): ChartingLibraryWidgetOptions {
  const ENV = getEnv()
  const theme = props.theme
  const bgColor = theme === 'dark' ? ThemeConst.black : ThemeConst.white
  const toolbar_bg = theme === 'dark' ? ThemeConst.black : '#fff'

  // 禁用的功能
  const disabled_features: ChartingLibraryFeatureset[] = [
    'header_compare',
    'symbol_search_hot_key',
    'study_templates',
    'header_saveload',
    'save_shortcut',
    'header_undo_redo',
    'symbol_info',
    'timeframes_toolbar',
    'scales_date_format',
    'header_fullscreen_button',
    'display_market_status'
  ]

  // 移动端额外禁用
  if (props.isMobile) {
    disabled_features.push(
      'header_symbol_search',
      'context_menus',
      'show_chart_property_page',
      'header_screenshot',
      'adaptive_logo',
      'left_toolbar'
    )
  }

  const widgetOptions: ChartingLibraryWidgetOptions = {
    fullscreen: true,
    autosize: true,
    timezone: 'exchange',
    library_path: `${ENV.isApp ? '.' : ''}/static/charting_library/`,
    datafeed: new DataFeedBase(datafeedParams),
    symbol: props.symbol,
    client_id: 'tradingview.com',
    user_id: 'public_user_id',
    locale: props.locale as LanguageCode,
    interval: isPC() ? '15' : '1',
    theme,
    toolbar_bg,
    container: containerRef,
    symbol_search_request_delay: 1000,
    auto_save_delay: 5,
    study_count_limit: 5,
    allow_symbol_change: true,
    overrides: {
      'paneProperties.background': `${bgColor}`
    },
    disabled_features,
    enabled_features: [
      'hide_resolution_in_legend',
      'display_legend_on_all_charts'
    ],
    custom_css_url: ENV.isApp ? `./styles/index.css` : `/static/styles/index.css`,
    favorites: {
      intervals: ['1', '5', '15', '30', '60']
    },
    custom_indicators_getter: function (PineJS) {
      return Promise.resolve([ma(PineJS)])
    },
    loading_screen: {
      backgroundColor: 'transparent',
      foregroundColor: 'transparent'
    }
  }

  return widgetOptions
}

// src/stores/ws.ts
class WsStore {
  tvWidget = null
  @observable lastbar = {}
  @observable activeSymbolInfo = {}

  // HTTP 获取历史 K 线数据
  getHttpHistoryBars = async (symbolInfo, resolution, from, to, countBack, firstDataRequest) => {
    const klineType = {
      1: '1min',
      5: '5min',
      15: '15min',
      30: '30min',
      60: '60min',
      240: '4hour',
      '1D': '1day',
      '1W': '1week',
      '1M': '1mon'
    }[resolution] || '1min'

    const res = await request.get(`${url}/api/trade-market/marketApi/kline/symbol/klineList`, {
      params: {
        symbol: symbolInfo.symbol,
        first: firstDataRequest,
        current: 1,
        size: document.documentElement.clientWidth >= 1200 ? 500 : 200,
        klineType,
        klineTime: to * 1000
      }
    })

    const list = res?.data || []
    return list.map((item) => {
      const [klineTime, open, high, low, close] = (item || '').split(',')
      return {
        open: Number(open),
        close: Number(close),
        high: Number(high),
        low: Number(low),
        time: resolution.includes('M')
          ? Number(klineTime) + 8 * 60 * 60 * 1000
          : Number(klineTime)
      }
    }).reverse()
  }

  // 更新最后一条 K 线
  updateBar = (socketData, currentSymbol) => {
    const precision = currentSymbol.precision
    const lastBar = this.lastbar
    const resolution = currentSymbol.resolution
    const serverTime = socketData?.priceData?.id / 1000
    const bid = socketData?.priceData?.buy

    let rounded = serverTime
    if (!isNaN(resolution) || resolution.includes('D')) {
      const coeff = (resolution.includes('D') ? 1440 : Number(resolution)) * 60
      rounded = Math.floor(serverTime / coeff) * coeff
    }

    const lastBarSec = lastBar?.time / 1000

    if (rounded > lastBarSec) {
      // 新建 K 线
      return {
        time: rounded * 1000,
        open: Number(bid),
        high: Number(bid),
        low: Number(bid),
        close: Number(bid)
      }
    } else {
      // 更新当前 K 线
      return {
        time: lastBar.time,
        open: lastBar.open,
        high: Math.max(lastBar.high, Number(bid)),
        low: Math.min(lastBar.low, Number(bid)),
        close: Number(bid)
      }
    }
  }

  // 处理 WebSocket 消息
  @action
  message(res) {
    if (res?.header?.msgId === 'symbol') {
      const quoteBody = this.parseQuoteBodyData(res?.body)
      if (quoteBody?.symbol === this.activeSymbolInfo?.symbolInfo?.name) {
        const newLastBar = this.updateBar(quoteBody, {
          resolution: this.activeSymbolInfo.resolution,
          precision: this.activeSymbolInfo.symbolInfo.precision,
          symbolInfo: this.activeSymbolInfo.symbolInfo
        })
        if (newLastBar) {
          this.activeSymbolInfo.onRealtimeCallback?.(newLastBar)
          this.lastbar = newLastBar
        }
      }
    }
  }

  // Datafeed 回调
  @action
  getDataFeedBarCallback = (obj = {}) => {
    const { symbolInfo, resolution, firstDataRequest, from, to, countBack, onHistoryCallback } = obj
    this.getHttpHistoryBars(symbolInfo, resolution, from, to, countBack, firstDataRequest)
      .then((bars) => {
        if (bars?.length) {
          onHistoryCallback(bars, { noData: false })
          this.lastbar = bars.at(-1)
        } else {
          onHistoryCallback(bars, { noData: true })
        }
      })
  }
}

export default wsStore

// src/components/Tradingview/customIndicators/ma.ts
const customerMovingAverage = (PineJS: PineJS) => {
  const indicators: CustomIndicator = {
    name: 'Customer Moving Average',
    metainfo: {
      _metainfoVersion: 51,
      id: 'Customer Moving Average@tv-basicstudies-1',
      name: 'Customer Moving Average',
      description: 'Customer Moving Average',
      shortDescription: 'MA',
      is_price_study: true,
      isCustomIndicator: true,
      format: { type: 'price' },
      defaults: {
        styles: {
          plot_0: { linestyle: 0, linewidth: 1, plottype: 0, trackPrice: false, transparency: 35, visible: true, color: '#FF0000' },
          plot_1: { linestyle: 0, linewidth: 1, plottype: 0, trackPrice: false, transparency: 35, visible: true, color: '#00FF00' },
          plot_2: { linestyle: 0, linewidth: 1, plottype: 0, trackPrice: false, transparency: 35, visible: true, color: '#00FFFF' }
        },
        inputs: { in_0: 5, in_1: 10, in_2: 30 },
        precision: 4
      },
      plots: [
        { id: 'plot_0', type: 'line' },
        { id: 'plot_1', type: 'line' },
        { id: 'plot_2', type: 'line' }
      ],
      inputs: [
        { id: 'in_0', name: 'Length', defval: 9, type: 'integer', min: 1, max: 1e4 },
        { id: 'in_1', name: 'Length1', defval: 10, type: 'integer', min: 1, max: 1e4 },
        { id: 'in_2', name: 'Length2', defval: 30, type: 'integer', min: 1, max: 1e4 }
      ]
    },
    constructor: function (this: LibraryPineStudy<IPineStudyResult>) {
      this.main = function (context, inputCallback) {
        const close = PineJS.Std.close(context)
        const len1 = inputCallback(0)
        const len2 = inputCallback(1)
        const len3 = inputCallback(2)

        const value1 = PineJS.Std.sma(close, len1, context)
        const value2 = PineJS.Std.sma(close, len2, context)
        const value3 = PineJS.Std.sma(close, len3, context)

        return [
          { value: value1, offset: 0 },
          { value: value2, offset: 0 },
          { value: value3, offset: 0 }
        ]
      }
    }
  }
  return indicators
}

export default customerMovingAverage

// src/components/Tradingview/theme.ts
export const getTradingviewThemeCssVar = (theme: ThemeName) => {
  const primary = ThemeConst.primary
  const textPrimary = ThemeConst.textPrimary
  const isDark = theme === 'dark'

  return {
    '--tv-color-toolbar-button-text': '#7B7E80',
    '--tv-color-toolbar-button-text-active': textPrimary,
    '--tv-color-toolbar-button-text-active-hover': textPrimary,
    '--tv-color-toolbar-toggle-button-background-active': primary,
    '--tv-color-toolbar-toggle-button-background-active-hover': primary,
    '--tv-color-popup-element-text-active': '#131722',
    '--tv-color-popup-element-background-active': '#f0f3fa',
    ...(isDark ? { '--tv-color-pane-background': ThemeConst.black } : {})
  }
}