import 'package:flutter/material.dart';

enum ExpandableState {
  expanded,
  collapsed,
}

class ExpandableController {
  VoidCallback? toggle;
  ExpandableState state = ExpandableState.collapsed;
}

class Expandable extends StatefulWidget {
  const Expandable({
    Key? key,
    required this.header,
    required this.body,
    this.animationController,
    this.animation,
    this.animationDurationMultiplier = 1.0,
    this.onExpandChanged,
    this.controller,
  }) : super(key: key);

  final Widget header;
  final Widget body;
  final AnimationController? animationController;
  final Animation<double>? animation;
  final double animationDurationMultiplier;
  final void Function(ExpandableState)? onExpandChanged;
  final ExpandableController? controller;

  @override
  State<Expandable> createState() => _ExpandableState();
}

class _ExpandableState extends State<Expandable> with TickerProviderStateMixin {
  late final AnimationController animationController;
  late final Animation<double> animation;
  late final Duration duration;
  late final ExpandableController? controller;

  ExpandableState _toggleState = ExpandableState.collapsed;

  Future<void> toggle() async {
    if (animation.isDismissed) {
      await animationController.forward();
      _toggleState = ExpandableState.collapsed;
      widget.onExpandChanged?.call(_toggleState);
    } else if (animation.isCompleted) {
      await animationController.reverse();
      _toggleState = ExpandableState.expanded;
      widget.onExpandChanged?.call(_toggleState);
    }
    controller?.state = _toggleState;
  }

  @override
  void initState() {
    controller = widget.controller;
    controller?.toggle = toggle;

    duration = Duration(
      milliseconds: (500 * widget.animationDurationMultiplier).toInt(),
    );
    animationController = widget.animationController ??
        AnimationController(
          vsync: this,
          duration: duration,
        );
    animation = widget.animation ??
        Tween<double>(begin: 0.0, end: 1.0).animate(
          CurvedAnimation(
            curve: Curves.easeInOut,
            parent: animationController,
          ),
        );
    super.initState();
  }

  @override
  void dispose() {
    animationController.dispose();
    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    return Column(
      mainAxisSize: MainAxisSize.min,
      children: [
        GestureDetector(
          onTap: toggle,
          child: Container(
            color: Colors.transparent,
            child: widget.header,
          ),
        ),
        SizeTransition(
          sizeFactor: animation,
          axisAlignment: 1.0,
          child: widget.body,
        ),
      ],
    );
  }
}