import 'package:flutter/material.dart';
import 'package:gpt_markdown/custom_widgets/markdown_config.dart';

import 'package:flutter/foundation.dart';
import 'package:flutter_math_fork/flutter_math.dart';
import 'package:gpt_markdown/custom_widgets/custom_divider.dart';
import 'package:gpt_markdown/custom_widgets/custom_error_image.dart';
import 'package:gpt_markdown/custom_widgets/custom_rb_cb.dart';
import 'package:gpt_markdown/custom_widgets/selectable_adapter.dart';
import 'package:gpt_markdown/custom_widgets/unordered_ordered_list.dart';
import 'dart:math';

import 'custom_widgets/code_field.dart';
import 'custom_widgets/indent_widget.dart';
import 'custom_widgets/link_button.dart';

part 'theme.dart';
part 'markdown_component.dart';
part 'md_widget.dart';

/// This widget create a full markdown widget as a column view.
class GptMarkdown extends StatelessWidget {
  const GptMarkdown(
    this.data, {
    super.key,
    this.style,
    this.followLinkColor = false,
    this.textDirection = TextDirection.ltr,
    this.latexWorkaround,
    this.textAlign,
    this.imageBuilder,
    this.textScaler,
    this.onLinkTab,
    this.latexBuilder,
    this.codeBuilder,
    this.sourceTagBuilder,
    this.highlightBuilder,
    this.linkBuilder,
    this.maxLines,
    this.overflow,
  });

  /// The direction of the text.
  final TextDirection textDirection;

  /// The data to be displayed.
  final String data;

  /// The style of the text.
  final TextStyle? style;

  /// The alignment of the text.
  final TextAlign? textAlign;

  /// The text scaler.
  final TextScaler? textScaler;

  /// The callback function to handle link clicks.
  final void Function(String url, String title)? onLinkTab;

  /// The LaTeX workaround.
  final String Function(String tex)? latexWorkaround;
  final int? maxLines;

  /// The overflow.
  final TextOverflow? overflow;

  /// The LaTeX builder.
  final Widget Function(
    BuildContext context,
    String tex,
    TextStyle style,
    bool inline,
  )?
  latexBuilder;

  /// Whether to follow the link color.
  final bool followLinkColor;

  /// The code builder.
  final Widget Function(
    BuildContext context,
    String name,
    String code,
    bool closed,
  )?
  codeBuilder;

  /// The source tag builder.
  final Widget Function(BuildContext, String, TextStyle)? sourceTagBuilder;

  /// The highlight builder.
  final Widget Function(BuildContext context, String text, TextStyle style)?
  highlightBuilder;

  /// The link builder.
  final Widget Function(
    BuildContext context,
    String text,
    String url,
    TextStyle style,
  )?
  linkBuilder;

  /// The image builder.
  final Widget Function(BuildContext, String imageUrl)? imageBuilder;

  /// A method to remove extra lines inside block LaTeX.
  String _removeExtraLinesInsideBlockLatex(String text) {
    return text.replaceAllMapped(
      RegExp(r"\\\[(.*?)\\\]", multiLine: true, dotAll: true),
      (match) {
        String content = match[0] ?? "";
        return content.replaceAllMapped(RegExp(r"\n[\n\ ]+"), (match) => "\n");
      },
    );
  }

  @override
  Widget build(BuildContext context) {
    String tex = data.trim();
    tex = tex.replaceAllMapped(
      RegExp(r"(?<!\\)\$\$(.*?)(?<!\\)\$\$", dotAll: true),
      (match) => "\\[${match[1] ?? ""}\\]",
    );
    if (!tex.contains(r"\(")) {
      tex = tex.replaceAllMapped(
        RegExp(r"(?<!\\)\$(.*?)(?<!\\)\$"),
        (match) => "\\(${match[1] ?? ""}\\)",
      );
      tex = tex.splitMapJoin(
        RegExp(r"\[.*?\]|\(.*?\)"),
        onNonMatch: (p0) {
          return p0.replaceAll("\\\$", "\$");
        },
      );
    }
    tex = _removeExtraLinesInsideBlockLatex(tex);
    return ClipRRect(
      child: MdWidget(
        tex,
        config: GptMarkdownConfig(
          textDirection: textDirection,
          style: style,
          onLinkTab: onLinkTab,
          textAlign: textAlign,
          textScaler: textScaler,
          followLinkColor: followLinkColor,
          latexWorkaround: latexWorkaround,
          latexBuilder: latexBuilder,
          codeBuilder: codeBuilder,
          maxLines: maxLines,
          overflow: overflow,
          sourceTagBuilder: sourceTagBuilder,
          highlightBuilder: highlightBuilder,
          linkBuilder: linkBuilder,
          imageBuilder: imageBuilder,
        ),
      ),
    );
  }
}