"use client";

import React, { useEffect, useMemo, useState } from "react";
import ReactDOM from "react-dom"; // 添加这行

import {
  Box,
  Button,
  Typography,
  Checkbox,
  FormControlLabel,
  IconButton,
  Card,
  CardContent,
  Chip,
  Tooltip,
  Collapse,
  Link,
} from "@mui/material";
import {
  Info as InfoIcon,
  LocationOn as LocationIcon,
} from "@mui/icons-material";
import { DatePicker } from "@mui/x-date-pickers/DatePicker";
import { LocalizationProvider } from "@mui/x-date-pickers/LocalizationProvider";
import { AdapterDayjs } from "@mui/x-date-pickers/AdapterDayjs";
import "dayjs/locale/zh-cn"; // 引入中文包
import dayjs, { Dayjs } from "dayjs";
import { api } from "@/lib/api";
import moment from "moment";
import { config, NETWORK_NAME } from "@config/config";
import { useNotification } from "@refinedev/core";

import { queryFeaturesByIds } from "@/utils/mapQueryService";
import { useData, useMap } from "@components/olmap/core/MapComponent";
import { GeoJSON } from "ol/format";
import VectorLayer from "ol/layer/Vector";
import VectorSource from "ol/source/Vector";
import { Stroke, Style, Icon } from "ol/style";
import Feature, { FeatureLike } from "ol/Feature";
import {
  along,
  lineString,
  length,
  toMercator,
  bbox,
  featureCollection,
} from "@turf/turf";
import { Point } from "ol/geom";
import { toLonLat } from "ol/proj";
import Timeline from "@components/olmap/core/Controls/Timeline";
import { SchemaItem, SchemeRecord } from "./types";

interface SchemeQueryProps {
  schemes?: SchemeRecord[];
  onSchemesChange?: (schemes: SchemeRecord[]) => void;
  onLocate?: (scheme: SchemeRecord) => void;
  network?: string;
}

const SCHEME_TYPE = "burst_analysis";

const SchemeQuery: React.FC<SchemeQueryProps> = ({
  schemes: externalSchemes,
  onSchemesChange,
  onLocate,
  network = NETWORK_NAME,
}) => {
  const [queryAll, setQueryAll] = useState<boolean>(true);
  const [queryDate, setQueryDate] = useState<Dayjs | null>(dayjs(new Date()));
  const [highlightLayer, setHighlightLayer] =
    useState<VectorLayer<VectorSource> | null>(null);
  const [highlightFeatures, setHighlightFeatures] = useState<Feature[]>([]);

  // 时间轴相关状态
  const [showTimeline, setShowTimeline] = useState(false);
  const [selectedDate, setSelectedDate] = useState<Date | undefined>(undefined);
  const [timeRange, setTimeRange] = useState<
    { start: Date; end: Date } | undefined
  >();
  const [internalSchemes, setInternalSchemes] = useState<SchemeRecord[]>([]);
  const [loading, setLoading] = useState<boolean>(false);
  const [expandedId, setExpandedId] = useState<number | null>(null);
  const [mapContainer, setMapContainer] = useState<HTMLElement | null>(null); // 地图容器元素

  const { open } = useNotification();

  const map = useMap();
  const data = useData();
  const { schemeName, setSchemeName } = data || {};

  // 使用外部提供的 schemes 或内部状态
  const schemes =
    externalSchemes !== undefined ? externalSchemes : internalSchemes;
  const setSchemes = onSchemesChange || setInternalSchemes;

  // 格式化日期为简短格式
  const formatTime = (timeStr: string) => {
    const time = moment(timeStr);
    return time.format("MM-DD");
  };

  const filteredSchemes = useMemo(() => {
    return schemes.filter((scheme) => scheme.type === SCHEME_TYPE);
  }, [schemes]);

  const handleQuery = async () => {
    if (!queryAll && !queryDate) return;

    setLoading(true);
    try {
      const response = await api.get(
        `${config.BACKEND_URL}/api/v1/getallschemes/?network=${network}`,
      );
      let filteredResults = response.data;

      if (!queryAll) {
        const formattedDate = queryDate!.format("YYYY-MM-DD");
        filteredResults = response.data.filter((item: SchemaItem) => {
          const itemDate = moment(item.create_time).format("YYYY-MM-DD");
          return itemDate === formattedDate;
        });
      }

      const nextSchemes = filteredResults.map((item: SchemaItem) => ({
        id: item.scheme_id,
        schemeName: item.scheme_name,
        type: item.scheme_type,
        user: item.username,
        create_time: item.create_time,
        startTime: item.scheme_start_time,
        schemeDetail: item.scheme_detail,
      }));
      setSchemes(nextSchemes);

      if (filteredResults.length === 0) {
        open?.({
          type: "error",
          message: "查询结果",
          description: queryAll
            ? "没有找到任何方案"
            : `${queryDate!.format("YYYY-MM-DD")} 没有找到相关方案`,
        });
      }
    } catch (error) {
      console.error("查询请求失败:", error);
      open?.({
        type: "error",
        message: "查询失败",
        description: "获取方案列表失败，请稍后重试",
      });
    } finally {
      setLoading(false);
    }
  };

  const handleLocatePipes = (pipeIds: string[]) => {
    if (pipeIds.length > 0) {
      queryFeaturesByIds(pipeIds, "geo_pipes_mat").then((features) => {
        if (features.length > 0) {
          // 设置高亮要素
          setHighlightFeatures(features);
          // 将 OpenLayers Feature 转换为 GeoJSON Feature
          const geojsonFormat = new GeoJSON();
          const geojsonFeatures = features.map((feature) =>
            geojsonFormat.writeFeatureObject(feature),
          );

          const extent = bbox(featureCollection(geojsonFeatures as any));

          if (extent) {
            map?.getView().fit(extent, { maxZoom: 18, duration: 1000 });
          }
        }
      });
    }
  };

  // 内部的方案查询函数
  const handleViewDetails = (id: number) => {
    const scheme = filteredSchemes.find((s) => s.id === id);
    if (!scheme) return;

    setShowTimeline(true);
    // 计算时间范围
    const schemeDate = scheme.startTime
      ? new Date(scheme.startTime)
      : undefined;
    if (scheme.startTime && scheme.schemeDetail?.modify_total_duration) {
      const start = new Date(scheme.startTime);
      const end = new Date(
        start.getTime() + scheme.schemeDetail.modify_total_duration * 1000,
      );
      setSelectedDate(schemeDate);
      setTimeRange({ start, end });
    }
    setSchemeName?.(scheme.schemeName);
    handleLocatePipes(scheme.schemeDetail?.burst_ID || []);
  };

  // 初始化管道图层和高亮图层
  useEffect(() => {
    if (!map) return;
    // 获取地图的目标容器
    const target = map.getTargetElement();
    if (target) {
      setMapContainer(target);
    }
    const burstPipeStyle = function (feature: FeatureLike) {
      const styles = [];
      // 线条样式（底层发光，主线条，内层高亮线）
      styles.push(
        new Style({
          stroke: new Stroke({
            color: "rgba(255, 0, 0, 0.3)",
            width: 12,
          }),
        }),
        new Style({
          stroke: new Stroke({
            color: "rgba(255, 0, 0, 1)",
            width: 6,
            lineDash: [15, 10],
          }),
        }),
        new Style({
          stroke: new Stroke({
            color: "rgba(255, 102, 102, 1)",
            width: 3,
            lineDash: [15, 10],
          }),
        }),
      );
      const geometry = feature.getGeometry();
      const lineCoords =
        geometry?.getType() === "LineString"
          ? (geometry as any).getCoordinates()
          : null;
      if (geometry) {
        const lineCoordsWGS84 = lineCoords.map((coord: []) => {
          const [lon, lat] = toLonLat(coord);
          return [lon, lat];
        });
        // 计算中点
        const lineStringFeature = lineString(lineCoordsWGS84);
        const lineLength = length(lineStringFeature);
        const midPoint = along(lineStringFeature, lineLength / 2).geometry
          .coordinates;
        // 在中点添加 icon 样式
        const midPointMercator = toMercator(midPoint);
        styles.push(
          new Style({
            geometry: new Point(midPointMercator),
            image: new Icon({
              src: "/icons/burst_pipe.svg",
              scale: 0.2,
              anchor: [0.5, 1],
            }),
          }),
        );
      }
      return styles;
    };
    // 创建高亮图层
    const highlightLayer = new VectorLayer({
      source: new VectorSource(),
      style: burstPipeStyle,
      maxZoom: 24,
      minZoom: 12,
      properties: {
        name: "爆管管段高亮",
        value: "burst_pipe_highlight",
      },
    });

    map.addLayer(highlightLayer);
    setHighlightLayer(highlightLayer);

    return () => {
      map.removeLayer(highlightLayer);
    };
  }, [map]);

  // 高亮要素的函数
  useEffect(() => {
    if (!highlightLayer) {
      return;
    }
    const source = highlightLayer.getSource();
    if (!source) {
      return;
    }
    // 清除之前的高亮
    source.clear();
    // 添加新的高亮要素
    highlightFeatures.forEach((feature) => {
      if (feature instanceof Feature) {
        source.addFeature(feature);
      }
    });
  }, [highlightFeatures, highlightLayer]);

  return (
    <>
      {/* 将时间轴渲染到地图容器中 */}
      {showTimeline &&
        mapContainer &&
        ReactDOM.createPortal(
          <Timeline
            schemeDate={selectedDate}
            timeRange={timeRange}
            disableDateSelection={!!timeRange}
            schemeName={schemeName}
            schemeType={SCHEME_TYPE}
          />,
          mapContainer, // 渲染到地图容器中，而不是 body
        )}
      <Box className="flex flex-col h-full">
        {/* 查询条件 - 单行布局 */}
        <Box className="mb-2 p-2 bg-gray-50 rounded">
          <Box className="flex items-center gap-2 justify-between">
            <Box className="flex items-center gap-2">
              <FormControlLabel
                control={
                  <Checkbox
                    checked={queryAll}
                    onChange={(e) => setQueryAll(e.target.checked)}
                    size="small"
                  />
                }
                label={<Typography variant="body2">查询全部</Typography>}
                className="m-0"
              />
              <LocalizationProvider
                dateAdapter={AdapterDayjs}
                adapterLocale="zh-cn"
              >
                <DatePicker
                  value={queryDate}
                  onChange={(value) =>
                    value && dayjs.isDayjs(value) && setQueryDate(value)
                  }
                  format="YYYY-MM-DD"
                  disabled={queryAll}
                  slotProps={{
                    textField: {
                      size: "small",
                      sx: { width: 200 },
                    },
                  }}
                />
              </LocalizationProvider>
            </Box>
            <Button
              variant="contained"
              onClick={handleQuery}
              disabled={loading}
              size="small"
              className="bg-blue-600 hover:bg-blue-700"
              sx={{ minWidth: 80 }}
            >
              {loading ? "查询中..." : "查询"}
            </Button>
          </Box>
        </Box>

        {/* 结果列表 */}
        <Box className="flex-1 overflow-auto">
          {filteredSchemes.length === 0 ? (
            <Box className="flex flex-col items-center justify-center h-full text-gray-400">
              <Box className="mb-4">
                <svg
                  width="80"
                  height="80"
                  viewBox="0 0 80 80"
                  fill="none"
                  className="opacity-40"
                >
                  <rect
                    x="10"
                    y="20"
                    width="60"
                    height="45"
                    rx="2"
                    stroke="currentColor"
                    strokeWidth="2"
                  />
                  <line
                    x1="10"
                    y1="30"
                    x2="70"
                    y2="30"
                    stroke="currentColor"
                    strokeWidth="2"
                  />
                </svg>
              </Box>
              <Typography variant="body2">总共 0 条</Typography>
              <Typography variant="body2" className="mt-1">
                No data
              </Typography>
            </Box>
          ) : (
            <Box className="space-y-2 p-2">
              <Typography variant="caption" className="text-gray-500 px-2">
                共 {filteredSchemes.length} 条记录
              </Typography>
              {filteredSchemes.map((scheme) => (
                <Card
                  key={scheme.id}
                  variant="outlined"
                  className="hover:shadow-md transition-shadow"
                >
                  <CardContent className="p-3 pb-2 last:pb-3">
                    {/* 主要信息行 */}
                    <Box className="flex items-start justify-between mb-2">
                      <Box className="flex-1 min-w-0">
                        <Box className="flex items-center gap-2 mb-1">
                          <Typography
                            variant="body2"
                            className="font-medium truncate"
                            title={scheme.schemeName}
                          >
                            {scheme.schemeName}
                          </Typography>
                          <Chip
                            label={
                              scheme.type === "burst_analysis"
                                ? "爆管模拟"
                                : scheme.type
                            }
                            size="small"
                            className="h-5"
                            color="primary"
                            variant="outlined"
                          />
                        </Box>
                        <Typography
                          variant="caption"
                          className="text-gray-500 block"
                        >
                          ID: {scheme.id} · 日期:{" "}
                          {formatTime(scheme.create_time)}
                        </Typography>
                      </Box>
                      {/* 操作按钮 */}
                      <Box className="flex gap-1 ml-2">
                        <Tooltip
                          title={
                            expandedId === scheme.id ? "收起详情" : "查看详情"
                          }
                        >
                          <IconButton
                            size="small"
                            onClick={() =>
                              setExpandedId(
                                expandedId === scheme.id ? null : scheme.id,
                              )
                            }
                            color="primary"
                            className="p-1"
                          >
                            <InfoIcon fontSize="small" />
                          </IconButton>
                        </Tooltip>
                        <Tooltip title="查看定位结果">
                          <IconButton
                            size="small"
                            onClick={() => onLocate?.(scheme)}
                            color="primary"
                            className="p-1"
                          >
                            <LocationIcon fontSize="small" />
                          </IconButton>
                        </Tooltip>
                      </Box>
                    </Box>

                    {/* 可折叠的详细信息 */}
                    <Collapse in={expandedId === scheme.id}>
                      <Box className="mt-2 pt-3 border-t border-gray-200">
                        {/* 信息网格布局 */}
                        <Box className="grid grid-cols-2 gap-x-4 gap-y-3 mb-3">
                          {/* 爆管详情列 */}
                          <Box className="space-y-2">
                            <Box className="space-y-1.5 pl-2">
                              <Box className="flex items-start gap-2">
                                <Typography
                                  variant="caption"
                                  className="text-gray-600 min-w-[70px] mt-0.5"
                                >
                                  管段ID:
                                </Typography>
                                <Box className="flex-1 flex flex-wrap gap-1">
                                  {scheme.schemeDetail?.burst_ID?.length ? (
                                    scheme.schemeDetail.burst_ID.map(
                                      (pipeId, index) => (
                                        <Link
                                          key={index}
                                          component="button"
                                          variant="caption"
                                          className="font-medium text-blue-600 hover:text-blue-800 underline cursor-pointer"
                                          onClick={(e) => {
                                            e.preventDefault();
                                            handleLocatePipes?.([pipeId]);
                                          }}
                                        >
                                          {pipeId}
                                        </Link>
                                      ),
                                    )
                                  ) : (
                                    <Typography
                                      variant="caption"
                                      className="font-medium text-gray-900"
                                    >
                                      N/A
                                    </Typography>
                                  )}
                                </Box>
                              </Box>
                              <Box className="flex items-center gap-2">
                                <Typography
                                  variant="caption"
                                  className="text-gray-600 min-w-[70px]"
                                >
                                  管径:
                                </Typography>
                                <Typography
                                  variant="caption"
                                  className="font-medium text-gray-900"
                                >
                                  560 mm
                                </Typography>
                              </Box>
                              <Box className="flex items-center gap-2">
                                <Typography
                                  variant="caption"
                                  className="text-gray-600 min-w-[70px]"
                                >
                                  爆管面积:
                                </Typography>
                                <Typography
                                  variant="caption"
                                  className="font-medium text-gray-900"
                                >
                                  {scheme.schemeDetail?.burst_size?.[0] ||
                                    "N/A"}{" "}
                                  cm²
                                </Typography>
                              </Box>
                              <Box className="flex items-center gap-2">
                                <Typography
                                  variant="caption"
                                  className="text-gray-600 min-w-[70px]"
                                >
                                  持续时间:
                                </Typography>
                                <Typography
                                  variant="caption"
                                  className="font-medium text-gray-900"
                                >
                                  {scheme.schemeDetail?.modify_total_duration ||
                                    "N/A"}{" "}
                                  秒
                                </Typography>
                              </Box>
                            </Box>
                          </Box>

                          {/* 方案信息列 */}
                          <Box className="space-y-2">
                            <Box className="space-y-1.5 pl-2">
                              <Box className="flex items-center gap-2">
                                <Typography
                                  variant="caption"
                                  className="text-gray-600 min-w-[70px]"
                                >
                                  用户:
                                </Typography>
                                <Typography
                                  variant="caption"
                                  className="font-medium text-gray-900"
                                >
                                  {scheme.user}
                                </Typography>
                              </Box>
                              <Box className="flex items-center gap-2">
                                <Typography
                                  variant="caption"
                                  className="text-gray-600 min-w-[70px]"
                                >
                                  创建时间:
                                </Typography>
                                <Typography
                                  variant="caption"
                                  className="font-medium text-gray-900"
                                >
                                  {moment(scheme.create_time).format(
                                    "YYYY-MM-DD HH:mm",
                                  )}
                                </Typography>
                              </Box>
                              <Box className="flex items-center gap-2">
                                <Typography
                                  variant="caption"
                                  className="text-gray-600 min-w-[70px]"
                                >
                                  开始时间:
                                </Typography>
                                <Typography
                                  variant="caption"
                                  className="font-medium text-gray-900"
                                >
                                  {moment(scheme.startTime).format(
                                    "YYYY-MM-DD HH:mm",
                                  )}
                                </Typography>
                              </Box>
                            </Box>
                          </Box>
                        </Box>

                        {/* 操作按钮区域 */}
                        <Box className="pt-2 border-t border-gray-100 flex gap-5">
                          {scheme.schemeDetail?.burst_ID?.length ? (
                            <Button
                              variant="outlined"
                              fullWidth
                              size="small"
                              className="border-blue-600 text-blue-600 hover:bg-blue-50"
                              onClick={() =>
                                handleLocatePipes?.(
                                  scheme.schemeDetail!.burst_ID,
                                )
                              }
                              sx={{
                                textTransform: "none",
                                fontWeight: 500,
                              }}
                            >
                              定位全部管道
                            </Button>
                          ) : null}
                          <Button
                            variant="contained"
                            fullWidth
                            size="small"
                            className="bg-blue-600 hover:bg-blue-700"
                            onClick={() => handleViewDetails(scheme.id)}
                            sx={{
                              textTransform: "none",
                              fontWeight: 500,
                            }}
                          >
                            查看分析结果
                          </Button>
                        </Box>
                      </Box>
                    </Collapse>
                  </CardContent>
                </Card>
              ))}
            </Box>
          )}
        </Box>
      </Box>
    </>
  );
};

export default SchemeQuery;
