flutter_flutter/dev/tools/gen_defaults/lib/fab_template.dart

// Copyright 2014 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import 'template.dart';

class FABTemplate extends TokenTemplate {
  const FABTemplate(
    super.blockName,
    super.fileName,
    super.tokens, {
    super.colorSchemePrefix = '_colors.',
    super.textThemePrefix = '_textTheme.',
  });

  @override
  String generate() =>
      '''
class _${blockName}DefaultsM3 extends FloatingActionButtonThemeData {
  _${blockName}DefaultsM3(this.context, this.type, this.hasChild)
    : super(
        elevation: ${elevation("md.comp.fab.primary.container")},
        focusElevation: ${elevation("md.comp.fab.primary.focus.container")},
        hoverElevation: ${elevation("md.comp.fab.primary.hover.container")},
        highlightElevation: ${elevation("md.comp.fab.primary.pressed.container")},
        enableFeedback: true,
        sizeConstraints: const BoxConstraints.tightFor(
          width: ${getToken("md.comp.fab.primary.container.width")},
          height: ${getToken("md.comp.fab.primary.container.height")},
        ),
        smallSizeConstraints: const BoxConstraints.tightFor(
          width: ${getToken("md.comp.fab.primary.small.container.width")},
          height: ${getToken("md.comp.fab.primary.small.container.height")},
        ),
        largeSizeConstraints: const BoxConstraints.tightFor(
          width: ${getToken("md.comp.fab.primary.large.container.width")},
          height: ${getToken("md.comp.fab.primary.large.container.height")},
        ),
        extendedSizeConstraints: const BoxConstraints.tightFor(
          height: ${getToken("md.comp.extended-fab.primary.container.height")},
        ),
        extendedIconLabelSpacing: 8.0,
      );

  final BuildContext context;
  final _FloatingActionButtonType type;
  final bool hasChild;
  late final ColorScheme _colors = Theme.of(context).colorScheme;
  late final TextTheme _textTheme = Theme.of(context).textTheme;

  bool get _isExtended => type == _FloatingActionButtonType.extended;

  @override Color? get foregroundColor => ${componentColor("md.comp.fab.primary.icon")};
  @override Color? get backgroundColor => ${componentColor("md.comp.fab.primary.container")};
  @override Color? get splashColor => ${componentColor("md.comp.fab.primary.pressed.state-layer")};
  @override Color? get focusColor => ${componentColor("md.comp.fab.primary.focus.state-layer")};
  @override Color? get hoverColor => ${componentColor("md.comp.fab.primary.hover.state-layer")};

  @override
  ShapeBorder? get shape => switch (type) {
    _FloatingActionButtonType.regular  => ${shape("md.comp.fab.primary.container")},
    _FloatingActionButtonType.small    => ${shape("md.comp.fab.primary.small.container")},
    _FloatingActionButtonType.large    => ${shape("md.comp.fab.primary.large.container")},
    _FloatingActionButtonType.extended => ${shape("md.comp.extended-fab.primary.container")},
  };

  @override
  double? get iconSize => switch (type) {
    _FloatingActionButtonType.regular  => ${getToken("md.comp.fab.primary.icon.size")},
    _FloatingActionButtonType.small    => ${getToken("md.comp.fab.primary.small.icon.size")},
    _FloatingActionButtonType.large    => ${getToken("md.comp.fab.primary.large.icon.size")},
    _FloatingActionButtonType.extended => ${getToken("md.comp.extended-fab.primary.icon.size")},
  };

  @override EdgeInsetsGeometry? get extendedPadding => EdgeInsetsDirectional.only(start: hasChild && _isExtended ? 16.0 : 20.0, end: 20.0);
  @override TextStyle? get extendedTextStyle => ${textStyle("md.comp.extended-fab.primary.label-text")};
}
''';
}