import 'package:flutter/material.dart';

class TabCard extends StatefulWidget {
  final List<Widget> titles;
  final List<Widget> bodies;
  final MainAxisAlignment titleAlign;
  final Alignment bodyAlign;
  final FlexFit titleFit;
  final EdgeInsetsGeometry? titlePadding;
  final BoxDecoration titleDecoration;
  final BoxDecoration titleActiveDecoration;
  final Axis direction;

  const TabCard({
    Key? key,
    required this.titles,
    required this.bodies,
    this.direction = Axis.vertical,
    this.titleAlign = MainAxisAlignment.start,
    this.titlePadding,
    this.titleDecoration =
        const BoxDecoration(color: Color.fromRGBO(0, 0, 0, .1)),
    this.titleActiveDecoration = const BoxDecoration(color: Colors.white),
    this.titleFit = FlexFit.loose,
    this.bodyAlign = Alignment.topLeft,
  }) : super(key: key);

  @override
  State<TabCard> createState() => TabCardState();
}

class TabCardState extends State<TabCard> {
  int index = 0;
  late List<Widget> titles;

  late ValueNotifier<int> onTabChange;

  @override
  void initState() {
    super.initState();
    onTabChange = ValueNotifier<int>(index);

    titles = widget.titles.map<Widget>((e) {
      int curIndex = widget.titles.indexOf(e);
      return Flexible(
        flex: widget.titleFit == FlexFit.tight ? 0 : 1,
        fit: widget.titleFit,
        child: TabCardTitleItem(
          myIndex: curIndex,
          child: e,
        ),
      );
    }).toList();
  }

  void updateIndex(int i) {
    setState(() {
      index = i;
    });
    onTabChange.value = i;
  }

  @override
  Widget build(BuildContext context) {
    return Flex(
      direction: widget.direction,
      children: [
        Container(
          decoration: widget.titleDecoration,
          child: Flex(
            direction: widget.direction == Axis.horizontal
                ? Axis.vertical
                : Axis.horizontal,
            mainAxisSize: MainAxisSize.max,
            mainAxisAlignment: widget.titleAlign,
            children: titles,
          ),
        ),
        Expanded(
          child: IndexedStack(
            index: index,
            alignment: widget.bodyAlign,
            sizing: StackFit.expand,
            children: widget.bodies,
          ),
        ),
      ],
    );
  }
}

class TabCardTitleItem extends StatefulWidget {
  final int myIndex;
  final Widget child;

  const TabCardTitleItem({Key? key, required this.myIndex, required this.child})
      : super(key: key);

  @override
  State<TabCardTitleItem> createState() => TabCardTitleItemState();
}

class TabCardTitleItemState extends State<TabCardTitleItem> {
  bool isActive = false;
  TabCardState? tabCard;

  @override
  void initState() {
    super.initState();
    tabCard = context.findRootAncestorStateOfType<TabCardState>();
    if (tabCard != null) {
      if (widget.myIndex == tabCard!.index) {
        isActive = true;
      }
      tabCard?.onTabChange.addListener(indexListener);
    }
  }

  void indexListener() {
    setState(() {
      isActive = tabCard!.index == widget.myIndex;
    });
  }

  @override
  Widget build(BuildContext context) {
    return GestureDetector(
      onTap: () {
        tabCard?.updateIndex(widget.myIndex);
      },
      child: AnimatedContainer(
        duration: const Duration(milliseconds: 300),
        curve: Curves.easeOutQuint,
        padding: tabCard!.widget.titlePadding,
        decoration: isActive
            ? tabCard!.widget.titleActiveDecoration
            : tabCard!.widget.titleDecoration,
        child: Center(child: widget.child),
      ),
    );
  }

  @override
  void dispose() {
    tabCard?.onTabChange.removeListener(indexListener);
    super.dispose();
  }
}