import * as d3 from 'd3'
import {
  Observable,
  Subject,
  combineLatest,
  takeUntil,
  map,
  distinctUntilChanged,
  of,
  iif,
  switchMap,
  shareReplay
} from 'rxjs'
import type {
  ChartParams,
  ComputedDatumMultiValue,
  ComputedDataMultiValue,
  ComputedXYDatumMultiValue,
  ContainerSize,
  ContainerPositionScaled,
  DataFormatterMultiValue,
  Layout
} from '../../lib/core-types'
import type { BaseRacingLabelsParams } from '../../lib/plugins-basic-types'
import type { BasePluginFn } from './types'
import { getColor, getDatumColor, getClassName, getUniID } from '../utils/orbchartsUtils'
import { getD3TransitionEase } from '../utils/d3Utils'
import { multiValueContainerSelectionsObservable } from '../multiValue/multiValueObservables'

interface BaseRacingAxisContext {
  selection: d3.Selection<any, unknown, any, unknown>
  computedData$: Observable<ComputedDataMultiValue>
  visibleComputedRankingData$: Observable<ComputedDatumMultiValue[][]>
  rankingScaleList$: Observable<Array<d3.ScalePoint<string>>>
  xScale$: Observable<(n: number) => number>
  fullParams$: Observable<BaseRacingLabelsParams>
  fullDataFormatter$: Observable<DataFormatterMultiValue>
  fullChartParams$: Observable<ChartParams>
  transitionEase$: Observable<string>
  // layout$: Observable<Layout>
  containerPosition$: Observable<ContainerPositionScaled[]>
  containerSize$: Observable<ContainerSize>
  isCategorySeprate$: Observable<boolean>
  // xyValueIndex$: Observable<[number, number]>
}

type ClipPathDatum = {
  id: string;
  // x: number;
  // y: number;
  width: number;
  height: number;
}

// const pluginName = 'RacingAxis'

const yTickTextAnchor = 'end'
const yTickDominantBaseline = 'middle'
const yAxisLabelAnchor = 'end'
const yAxisLabelDominantBaseline = 'auto'
// const textClassName = getClassName(pluginName, 'yLabel')

function renderRacingAxisLabel ({ selection, textClassName, fullParams, containerSize, fullDataFormatter, fullChartParams }: {
  selection: d3.Selection<SVGGElement, any, any, any>,
  textClassName: string
  fullParams: BaseRacingLabelsParams
  // axisLabelAlign: TextAlign
  containerSize: ContainerSize
  fullDataFormatter: DataFormatterMultiValue,
  fullChartParams: ChartParams
  // textReverseTransform: string,
}) {
  const offsetX = fullParams.barLabel.padding - fullParams.axisLabel.offset[0]
  const offsetY = - fullParams.barLabel.padding - fullParams.axisLabel.offset[1]
  let labelX = - offsetX
  let labelY = - offsetY

  selection
    .attr('transform', d => `translate(0, ${containerSize.height})`)
    .selectAll<SVGTextElement, BaseRacingLabelsParams>(`text`)
    .data([fullParams])
    .join(
      enter => {
        return enter
          .append('text')
          .style('font-weight', 'bold')
      },
      update => update,
      exit => exit.remove()
    )
    .attr('text-anchor', yAxisLabelAnchor)
    .attr('dominant-baseline', yAxisLabelDominantBaseline)
    .attr('font-size', fullChartParams.styles.textSize)
    .style('fill', getColor(fullParams.axisLabel.colorType, fullChartParams))
    // .style('transform', textReverseTransform)
    // 偏移使用 x, y 而非 transform 才不會受到外層 scale 變形影響
    .attr('x', labelX)
    .attr('y', labelY)
    .text(d => fullDataFormatter.yAxis.label)
}

function renderRacingLabels ({ selection, fullParams, fullChartParams, transitionEase, rankingScale, valueScale, categoryData }: {
  selection: d3.Selection<SVGGElement, any, any, any>,
  fullParams: BaseRacingLabelsParams
  fullChartParams: ChartParams
  transitionEase: string
  rankingScale: d3.ScalePoint<string>
  valueScale: ((n: number) => number)
  categoryData: ComputedDatumMultiValue[]
  // textReverseTransformWithRotate: string,
  // xyValueIndex: [number, number]
}) {
  const labelData = fullParams.barLabel.position === 'none' ? [] : categoryData

  const labelSelection = selection
    .selectAll<SVGGElement, string>(`text`)
    .data(labelData, (d: ComputedDatumMultiValue) => d.id)
    // .join('g')
    // .classed(boxClassName, true)
    .join(
      enter => {
        return enter
          .append('text')
          .style('font-weight', 'bold')
          .style('pointer-events', 'none')
          .attr('x', d => valueScale(d.value[d.xValueIndex] ?? 0) - fullParams.barLabel.padding)
          .attr('y', d => rankingScale(d.label)!)
      },
      update => {
        return update
          .transition()
          .duration(fullChartParams.transitionDuration)
          // .ease(d3.easeLinear)
          .ease(getD3TransitionEase(transitionEase))
          // 偏移使用 x, y 而非 transform 才不會受到外層 scale 變形影響
          .attr('x', d => valueScale(d.value[d.xValueIndex] ?? 0) - fullParams.barLabel.padding)
          .attr('y', d => rankingScale(d.label)!)
      },
      exit => exit.remove()
    )
    .attr('text-anchor', yTickTextAnchor)
    .attr('dominant-baseline', yTickDominantBaseline)
    .attr('font-size', fullChartParams.styles.textSize)
    .style('fill', d => getDatumColor({
      colorType: fullParams.barLabel.colorType,
      fullChartParams: fullChartParams,
      datum: d
    }))
    // .style('transform', textReverseTransformWithRotate)
    .text(d => d.label)

  return labelSelection
}


function renderClipPath ({ defsSelection, clipPathData }: {
  defsSelection: d3.Selection<SVGDefsElement, any, any, any>
  clipPathData: ClipPathDatum[]
  // textReverseTransform: string
}) {
  const clipPath = defsSelection
    .selectAll<SVGClipPathElement, Layout>('clipPath')
    .data(clipPathData)
    .join(
      enter => {
        return enter
          .append('clipPath')
      },
      update => update,
      exit => exit.remove()
    )
    .attr('id', d => d.id)
    // .attr('transform', textReverseTransform)
    .each((d, i, g) => {
      const rect = d3.select(g[i])
        .selectAll<SVGRectElement, typeof d>('rect')
        .data([d])
        .join(
          enter => {
            return enter
              .append('rect')
          },
          update => update,
          exit => exit.remove()
        )
        .attr('x', _d => - _d.width)
        .attr('y', 0)
        .attr('width', _d => _d.width * 2)
        .attr('height', _d => _d.height)
    })
}

export const createBaseRacingLabels: BasePluginFn<BaseRacingAxisContext> = (pluginName: string, {
  selection,
  computedData$,
  visibleComputedRankingData$,
  rankingScaleList$,
  xScale$,
  fullParams$,
  fullDataFormatter$,
  fullChartParams$,
  transitionEase$,
  // layout$,
  containerPosition$,
  containerSize$,
  isCategorySeprate$,
  // xyValueIndex$
}) => {

  const destroy$ = new Subject()
  
  // const containerClassName = getClassName(pluginName, 'container')
  const boxClassName = getClassName(pluginName, 'box')
  const textClassName = getClassName(pluginName, 'text')
  const clipPathID = getUniID(pluginName, 'clipPath-box')

  const containerSelection$ = multiValueContainerSelectionsObservable({
    selection,
    pluginName,
    clipPathID: null,
    computedData$,
    containerPosition$,
    isCategorySeprate$,
  }).pipe(
    takeUntil(destroy$),
  )

  // const containerSelection$ = combineLatest({
  //   computedData: computedData$.pipe(
  //     distinctUntilChanged((a, b) => {
  //       // 只有當series的數量改變時，才重新計算
  //       return a.length === b.length
  //     }),
  //   ),
  //   isCategorySeprate: isCategorySeprate$
  // }).pipe(
  //   takeUntil(destroy$),
  //   switchMap(async (d) => d),
  //   map(data => {
  //     return data.isCategorySeprate
  //       // category分開的時候顯示各別axis
  //       ? data.computedData
  //       // category合併的時候只顯示第一個axis
  //       : [data.computedData[0]]
  //   }),
  //   map((computedData, i) => {
  //     return selection
  //       .selectAll<SVGGElement, ComputedDatumMultiValue[]>(`g.${containerClassName}`)
  //       .data(computedData, d => d[0] ? d[0].categoryIndex : i)
  //       .join('g')
  //       .classed(containerClassName, true)
  //   })
  // )

  // combineLatest({
  //   containerSelection: containerSelection$,
  //   containerPosition: containerPosition$
  // }).pipe(
  //   takeUntil(destroy$),
  //   switchMap(async d => d)
  // ).subscribe(data => {
  //   data.containerSelection
  //     .attr('transform', (d, i) => {
  //       const containerPosition = data.containerPosition[i] ?? data.containerPosition[0]
  //       const translate = containerPosition.translate
  //       const scale = containerPosition.scale
  //       // return `translate(${translate[0]}, ${translate[1]}) scale(${scale[0]}, ${scale[1]})`
  //       return `translate(${translate[0]}, ${translate[1]})`
  //     })
  //     // .attr('opacity', 0)
  //     // .transition()
  //     // .attr('opacity', 1)
  // })

  // const textReverseTransform$ = containerPosition$.pipe(
  //   takeUntil(destroy$),
  //   switchMap(async (d) => d),
  //   map(containerPosition => {
  //     // const axesRotateXYReverseValue = `rotateX(${data.gridAxesReverseTransform.rotateX}deg) rotateY(${data.gridAxesReverseTransform.rotateY}deg)`
  //     // const axesRotateReverseValue = `rotate(${data.gridAxesReverseTransform.rotate}deg)`
  //     const containerScaleReverseValue = `scale(${1 / containerPosition[0].scale[0]}, ${1 / containerPosition[0].scale[1]})`
  //     // 抵消最外層scale
  //     return `${containerScaleReverseValue}`
  //   }),
  //   distinctUntilChanged()
  // )

  // const textReverseTransformWithRotate$ = combineLatest({
  //   textReverseTransform: textReverseTransform$,
  //   fullParams: fullParams$,
  // }).pipe(
  //   takeUntil(destroy$),
  //   switchMap(async (d) => d),
  //   map(data => {
  //     // 必須按照順序（先抵消外層rotate，再抵消最外層scale，最後再做本身的rotate）
  //     return `${data.textReverseTransform} rotate(${data.fullParams.barLabel.rotate}deg)`
  //   })
  // )

  // const rankingLabelList$ = visibleComputedRankingData$.pipe(
  //   takeUntil(destroy$),
  //   map(data => {
  //     return data.map(categoryData => categoryData.map(d => d.label))
  //   })
  // )

  const valueScale$: Observable<((n: number) => number)> = fullParams$.pipe(
    map(fullParams => fullParams.barLabel.position),
    distinctUntilChanged(),
    switchMap(position => {
      return iif(
        () => position === 'inside',
        xScale$,
        of(() => 0)
      )
    })
  )

  containerSize$.subscribe(data => {
    const defsSelection = selection.selectAll<SVGDefsElement, any>('defs')
      .data([clipPathID])
      .join('defs')
    const clipPathData = [{
      id: clipPathID,
      width: data.width,
      height: data.height
    }]
    renderClipPath({
      defsSelection: defsSelection,
      clipPathData,
      // textReverseTransform: data.textReverseTransform
    })
  })

  combineLatest({
    containerSelection: containerSelection$,
    fullParams: fullParams$,
    // layout: layout$,
    containerSize: containerSize$,
    fullDataFormatter: fullDataFormatter$,
    fullChartParams: fullChartParams$,
    transitionEase: transitionEase$,
    visibleComputedRankingData: visibleComputedRankingData$,
    // rankingLabelList: rankingLabelList$,
    rankingScaleList: rankingScaleList$,
    valueScale: valueScale$,
    // textReverseTransform: textReverseTransform$,
    // textReverseTransformWithRotate: textReverseTransformWithRotate$,
    // xyValueIndex: xyValueIndex$
  }).pipe(
    takeUntil(destroy$),
    switchMap(async (d) => d),
  ).subscribe(data => {

    data.containerSelection.each((d, i, g) => {
      const _containerSelection = d3.select(g[i])
      // const rankingLabels = data.rankingLabelList[i]
      const rankingScale = data.rankingScaleList[i]
      if (!rankingScale) {
        return
      }
      
      // const containerClipPathID = `${clipPathID}-${i}`  
      const axisSelection = _containerSelection
        .selectAll<SVGGElement, any>(`g.${boxClassName}`)
        .data([i])
        .join('g')
        .attr('class', boxClassName)
        .attr('clip-path', `url(#${clipPathID})`)
      const axisLabelSelection = _containerSelection
        .selectAll<SVGGElement, BaseRacingLabelsParams>(`g.${textClassName}`)
        .data([data.fullParams])
        .join('g')
        .classed(textClassName, true)

      renderRacingLabels({
        selection: axisSelection,
        fullParams: data.fullParams,
        fullChartParams: data.fullChartParams,
        transitionEase: data.transitionEase,
        rankingScale: rankingScale,
        categoryData: data.visibleComputedRankingData[i],
        // textReverseTransformWithRotate: data.textReverseTransformWithRotate,
        valueScale: data.valueScale,
        // xyValueIndex: data.xyValueIndex,
      })
  
      renderRacingAxisLabel({
        selection: axisLabelSelection,
        textClassName,
        fullParams: data.fullParams,
        containerSize: data.containerSize,
        fullDataFormatter: data.fullDataFormatter,
        fullChartParams: data.fullChartParams,
        // textReverseTransform: data.textReverseTransform,
      })
    })

  })

  return () => {
    destroy$.next(undefined)
  }
}