import 'package:flutter/material.dart';
import 'package:flutter_svg/svg.dart';

/// Flutter code sample for [SegmentedButton].

void main() {
  runApp(const SegmentedButtonApp());
}

class SegmentedButtonApp extends StatelessWidget {
  const SegmentedButtonApp({super.key});

  @override
  Widget build(BuildContext context) {
    return MaterialApp(
      theme: ThemeData(useMaterial3: true),
      home: const Scaffold(
        body: Center(
          child: Column(
            mainAxisAlignment: MainAxisAlignment.center,
            children: <Widget>[
              Spacer(),
              MultipleChoice(),
              Spacer(),
            ],
          ),
        ),
      ),
    );
  }
}

enum Sizes {
  extraSmall,
  small,
}

class SvgPath {
  static const String heat = "assets/svg/circle-heat-svgrepo-com.svg";
  static const String cloudArrow =
      "assets/svg/cloud-arrow-down-svgrepo-com.svg";
}

class MultipleChoice extends StatefulWidget {
  const MultipleChoice({super.key});

  @override
  State<MultipleChoice> createState() => _MultipleChoiceState();
}

class _MultipleChoiceState extends State<MultipleChoice> {
  Set<Sizes> selection = {};
  var segments = <ButtonSegment<Sizes>>[
    ButtonSegment<Sizes>(
        value: Sizes.extraSmall, icon: getSvgFromPath(SvgPath.cloudArrow)),
    ButtonSegment<Sizes>(
        value: Sizes.small, icon: getSvgFromPath(SvgPath.heat)),
  ];

  @override
  Widget build(BuildContext context) {
    return SegmentedButton<Sizes>(
      style: SegmentedButton.styleFrom(splashFactory: NoSplash.splashFactory),
      segments: segments,
      selected: selection,
      emptySelectionAllowed: true,
      showSelectedIcon: false,
      onSelectionChanged: (Set<Sizes> newSelection) {
        final selectedSegments = segments.map((e) {
          if (newSelection.contains(e.value)) {
            return ButtonSegment<Sizes>(
              value: e.value,
              icon: ColorFiltered(
                colorFilter:
                    const ColorFilter.mode(Colors.red, BlendMode.srcIn),
                child: e.icon,
              ),
            );
          } else {
            return ButtonSegment<Sizes>(
              value: e.value,
              icon: ColorFiltered(
                colorFilter:
                    const ColorFilter.mode(Colors.black, BlendMode.srcIn),
                child: e.icon,
              ),
            );
          }
        }).toList();
        setState(() {
          selection = newSelection;
          segments = selectedSegments;
        });
      },
      multiSelectionEnabled: true,
    );
  }
}

Widget getSvgFromPath(String path, [Color color = Colors.black]) {
  return SvgPicture.asset(
    path,
    colorFilter: ColorFilter.mode(
      color,
      BlendMode.srcIn,
    ),
  );
}