/*
 * Copyright 2023 Hongen Wang All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import 'dart:io';
import 'dart:math';

import 'package:flutter/material.dart';
import 'package:proxypin/network/util/logger.dart';
import 'package:proxypin/ui/component/json/theme.dart';
import 'package:proxypin/ui/component/search/search_controller.dart';

import '../../../utils/platform.dart';

class JsonText extends StatefulWidget {
  final ColorTheme colorTheme;
  final dynamic json;
  final String indent;
  final ScrollController? scrollController;
  final SearchTextController? searchController;

  const JsonText({
    super.key,
    required this.json,
    this.indent = '  ',
    required this.colorTheme,
    this.scrollController,
    this.searchController,
  });

  @override
  State<JsonText> createState() => _JsonTextState();
}

class _JsonTextState extends State<JsonText> {
  ScrollController? trackingScrollController;
  SearchTextController? searchController;

  @override
  void initState() {
    super.initState();
    searchController = widget.searchController;
  }

  @override
  void dispose() {
    trackingScrollController?.dispose();
    trackingScrollController = null;
    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    if (searchController == null) {
      return jsonTextWidget(context);
    }
    return AnimatedBuilder(
      animation: searchController!,
      builder: (context, child) {
        return jsonTextWidget(context);
      },
    );
  }

  Widget jsonTextWidget(BuildContext context) {
    var jsonParser = JsonParser(widget.json, widget.colorTheme, widget.indent, searchController);
    var textList = jsonParser.getJsonTree();
    WidgetsBinding.instance.addPostFrameCallback((_) {
      searchController?.updateMatchCount(jsonParser.searchMatchTotal);
      // 自动滚动到当前高亮项
      scrollToMatch(jsonParser);
    });
    if (textList.length < 1500) {
      return SelectableText.rich(TextSpan(children: textList), showCursor: true);
    } else {
      return SizedBox(
          width: double.infinity,
          height: MediaQuery.of(context).size.height - 160,
          child: SingleChildScrollView(
              physics: Platforms.isDesktop() ? null : const BouncingScrollPhysics(),
              controller: Platforms.isDesktop() ? null : trackingScroll(),
              child: SelectableText.rich(TextSpan(children: textList), showCursor: true)));
    }
  }

  void scrollToMatch(JsonParser jsonParser) {
    if (searchController != null && jsonParser.matchKeys.isNotEmpty) {
      final currentIndex = searchController!.currentMatchIndex.value;
      if (currentIndex >= 0 && currentIndex < jsonParser.matchKeys.length) {
        final key = jsonParser.matchKeys[currentIndex];
        final context = key.currentContext;
        if (context != null) {
          Scrollable.ensureVisible(
            context,
            duration: const Duration(milliseconds: 300),
            alignment: 0.5, // 高亮项在视图中的位置
          );
        }
      }
    }
  }

  ///滚动条
  ScrollController trackingScroll() {
    if (trackingScrollController != null) {
      return trackingScrollController!;
    }

    var trackingScroll = TrackingScrollController();
    ScrollController? scrollController = widget.scrollController;

    double offset = 0;
    trackingScroll.addListener(() {
      if (trackingScroll.offset < -10 || (trackingScroll.offset < 30 && trackingScroll.offset < offset)) {
        if (scrollController != null && scrollController.offset >= 0) {
          scrollController.jumpTo(scrollController.offset - max((offset - trackingScroll.offset), 15));
        }
      }
      offset = trackingScroll.offset;
    });

    if (Platform.isIOS && scrollController != null) {
      scrollController.addListener(() {
        if (scrollController.offset >= scrollController.position.maxScrollExtent) {
          scrollController.jumpTo(scrollController.position.maxScrollExtent);
          trackingScroll
              .jumpTo(trackingScroll.offset + (scrollController.offset - scrollController.position.maxScrollExtent));
        }
      });
    }

    trackingScrollController = trackingScroll;
    return trackingScroll;
  }
}

class JsonParser {
  final dynamic json;
  final ColorTheme colorTheme;
  final String indent;
  final SearchTextController? searchController;
  int searchMatchTotal = 0;
  final List<GlobalKey> matchKeys = [];

  JsonParser(this.json, this.colorTheme, this.indent, this.searchController);

  int getLength() {
    if (json is Map) {
      return json.length;
    } else if (json is List) {
      return json.length;
    } else {
      return json == null ? 0 : json.toString().length;
    }
  }

  List<TextSpan> getJsonTree() {
    matchKeys.clear(); // 每次渲染前清空
    List<TextSpan> textList = [];
    if (json is Map) {
      textList.add(const TextSpan(text: '{ \n'));
      textList.addAll(getMapText(json, prefix: indent));
    } else if (json is List) {
      textList.add(const TextSpan(text: '[ \n'));
      textList.addAll(getArrayText(json));
    } else {
      textList.add(TextSpan(text: json == null ? '' : json.toString()));
      textList.add(TextSpan(text: '\n'));
    }
    return textList;
  }

  /// 获取Map json
  List<TextSpan> getMapText(Map<String, dynamic> map,
      {String openPrefix = '', String prefix = '', String suffix = ''}) {
    var result = <TextSpan>[];

    var entries = map.entries;
    for (int i = 0; i < entries.length; i++) {
      var entry = entries.elementAt(i);
      String postfix = '${i == entries.length - 1 ? '' : ','} ';

      var textSpan = TextSpan(text: prefix, children: [
        ..._highlightMatches('"${entry.key}"', textColor: colorTheme.propertyKey),
        const TextSpan(text: ': '),
        getBasicValue(entry.value, postfix),
      ]);
      result.add(textSpan);
      result.add(TextSpan(text: '\n'));

      if (entry.value is Map<String, dynamic>) {
        result.addAll(getMapText(entry.value, openPrefix: prefix, prefix: '$prefix$indent', suffix: postfix));
      } else if (entry.value is List) {
        result.addAll(getArrayText(entry.value, openPrefix: prefix, prefix: '$prefix$indent', suffix: postfix));
      }
    }

    result.add(TextSpan(text: '$openPrefix}$suffix  \n'));
    return result;
  }

  /// 获取数组json
  List<TextSpan> getArrayText(List<dynamic> list, {String openPrefix = '', String prefix = '', String suffix = ''}) {
    var result = <TextSpan>[];
    result.add(TextSpan(text: '$openPrefix[ \n'));

    for (int i = 0; i < list.length; i++) {
      var value = list[i];
      String postfix = i == list.length - 1 ? '' : ',';

      result.add(getBasicValue(value, postfix, prefix: prefix));
      result.add(TextSpan(text: '\n'));

      if (value is Map<String, dynamic>) {
        result.addAll(getMapText(value, openPrefix: '$openPrefix ', prefix: '$prefix$indent', suffix: postfix));
      } else if (value is List) {
        result.addAll(getArrayText(value, openPrefix: '$openPrefix ', prefix: '$prefix$indent', suffix: postfix));
      }
    }

    result.add(TextSpan(text: '$openPrefix]$suffix \n'));
    return result;
  }

  /// 获取基本类型值 复杂类型会忽略
  TextSpan getBasicValue(dynamic value, String suffix, {String? prefix}) {
    if (value == null) {
      return TextSpan(
          text: prefix,
          children: [..._highlightMatches('null', textColor: colorTheme.keyword), TextSpan(text: suffix)]);
    }

    if (value is String) {
      return TextSpan(
          text: prefix,
          children: [..._highlightMatches('"$value"', textColor: colorTheme.string), TextSpan(text: suffix)]);
    }

    if (value is num) {
      return TextSpan(
          text: prefix,
          children: [..._highlightMatches(value.toString(), textColor: colorTheme.number), TextSpan(text: suffix)]);
    }

    if (value is bool) {
      return TextSpan(
          text: prefix,
          children: [..._highlightMatches(value.toString(), textColor: colorTheme.keyword), TextSpan(text: suffix)]);
    }

    if (value is List) {
      return TextSpan(children: _highlightMatches("${prefix ?? ''}["));
    }

    return TextSpan(children: _highlightMatches("${prefix ?? ''}{"));
  }

  List<InlineSpan> _highlightMatches(String text, {Color? textColor}) {
    if (searchController == null || searchController?.shouldSearch() == false) {
      return [TextSpan(text: text, style: TextStyle(color: textColor))];
    }

    final pattern = searchController!.value.pattern;
    final regex = searchController!.value.isRegExp
        ? RegExp(pattern, caseSensitive: searchController!.value.isCaseSensitive)
        : RegExp(RegExp.escape(pattern), caseSensitive: searchController!.value.isCaseSensitive);

    final spans = <InlineSpan>[];
    int start = 0;
    var allMatches = regex.allMatches(text).toList();
    final currentIndex = searchController!.currentMatchIndex.value;
    for (int i = 0; i < allMatches.length; i++) {
      final match = allMatches[i];
      if (match.start > start) {
        spans.add(TextSpan(text: text.substring(start, match.start), style: TextStyle(color: textColor)));
      }
      // 为每个高亮项分配一个 GlobalKey
      final key = GlobalKey();
      matchKeys.add(key);
      spans.add(WidgetSpan(
        child: Container(
          key: key,
          color: searchMatchTotal == currentIndex ? colorTheme.searchMatchCurrentColor : colorTheme.searchMatchColor,
          child: Text(
            text.substring(match.start, match.end),
            style: TextStyle(color: textColor),
          ),
        ),
      ));
      start = match.end;
      searchMatchTotal += 1; // 统计总匹配数
    }

    if (start < text.length) {
      spans.add(TextSpan(text: text.substring(start), style: TextStyle(color: textColor)));
    }
    return spans;
  }
}