From 31770f51d9692658f2af41016b42276040f0ab41 Mon Sep 17 00:00:00 2001
From: Adil Hanney <adilh+github@orchid.anonaddy.com>
Date: Wed, 15 Feb 2023 04:37:11 +0000
Subject: [PATCH] Add `axis` parameter to `SelectHeaderStyleButton`

---
 lib/src/widgets/toolbar.dart                  |  1 +
 .../toolbar/select_header_style_button.dart   | 97 ++++++++++---------
 2 files changed, 54 insertions(+), 44 deletions(-)

diff --git a/lib/src/widgets/toolbar.dart b/lib/src/widgets/toolbar.dart
index 28bf610e..2bc5bb72 100644
--- a/lib/src/widgets/toolbar.dart
+++ b/lib/src/widgets/toolbar.dart
@@ -376,6 +376,7 @@ class QuillToolbar extends StatelessWidget implements PreferredSizeWidget {
         if (showHeaderStyle)
           SelectHeaderStyleButton(
             controller: controller,
+            axis: axis,
             iconSize: toolbarIconSize,
             iconTheme: iconTheme,
             afterButtonPressed: afterButtonPressed,
diff --git a/lib/src/widgets/toolbar/select_header_style_button.dart b/lib/src/widgets/toolbar/select_header_style_button.dart
index 72192398..5e1dee9f 100644
--- a/lib/src/widgets/toolbar/select_header_style_button.dart
+++ b/lib/src/widgets/toolbar/select_header_style_button.dart
@@ -10,6 +10,7 @@ import '../toolbar.dart';
 class SelectHeaderStyleButton extends StatefulWidget {
   const SelectHeaderStyleButton({
     required this.controller,
+    this.axis = Axis.horizontal,
     this.iconSize = kDefaultIconSize,
     this.iconTheme,
     this.attributes = const [
@@ -23,6 +24,7 @@ class SelectHeaderStyleButton extends StatefulWidget {
   }) : super(key: key);
 
   final QuillController controller;
+  final Axis axis;
   final double iconSize;
   final QuillIconTheme? iconTheme;
   final List<Attribute> attributes;
@@ -67,53 +69,60 @@ class _SelectHeaderStyleButtonState extends State<SelectHeaderStyleButton> {
       fontSize: widget.iconSize * 0.7,
     );
 
-    return Row(
-      mainAxisSize: MainAxisSize.min,
-      children: widget.attributes.map((attribute) {
-        final isSelected = _selectedAttribute == attribute;
-        return Padding(
-          // ignore: prefer_const_constructors
-          padding: EdgeInsets.symmetric(horizontal: !kIsWeb ? 1.0 : 5.0),
-          child: ConstrainedBox(
-            constraints: BoxConstraints.tightFor(
-              width: widget.iconSize * kIconButtonFactor,
-              height: widget.iconSize * kIconButtonFactor,
-            ),
-            child: RawMaterialButton(
-              hoverElevation: 0,
-              highlightElevation: 0,
-              elevation: 0,
-              visualDensity: VisualDensity.compact,
-              shape: RoundedRectangleBorder(
-                  borderRadius: BorderRadius.circular(
-                      widget.iconTheme?.borderRadius ?? 2)),
-              fillColor: isSelected
-                  ? (widget.iconTheme?.iconSelectedFillColor ??
-                      Theme.of(context).primaryColor)
-                  : (widget.iconTheme?.iconUnselectedFillColor ??
-                      theme.canvasColor),
-              onPressed: () {
-                final _attribute = _selectedAttribute == attribute
-                    ? Attribute.header
-                    : attribute;
-                widget.controller.formatSelection(_attribute);
-                widget.afterButtonPressed?.call();
-              },
-              child: Text(
-                _valueToText[attribute] ?? '',
-                style: style.copyWith(
-                  color: isSelected
-                      ? (widget.iconTheme?.iconSelectedColor ??
-                          theme.primaryIconTheme.color)
-                      : (widget.iconTheme?.iconUnselectedColor ??
-                          theme.iconTheme.color),
-                ),
+    final children =  widget.attributes.map((attribute) {
+      final isSelected = _selectedAttribute == attribute;
+      return Padding(
+        // ignore: prefer_const_constructors
+        padding: EdgeInsets.symmetric(horizontal: !kIsWeb ? 1.0 : 5.0),
+        child: ConstrainedBox(
+          constraints: BoxConstraints.tightFor(
+            width: widget.iconSize * kIconButtonFactor,
+            height: widget.iconSize * kIconButtonFactor,
+          ),
+          child: RawMaterialButton(
+            hoverElevation: 0,
+            highlightElevation: 0,
+            elevation: 0,
+            visualDensity: VisualDensity.compact,
+            shape: RoundedRectangleBorder(
+                borderRadius: BorderRadius.circular(
+                    widget.iconTheme?.borderRadius ?? 2)),
+            fillColor: isSelected
+                ? (widget.iconTheme?.iconSelectedFillColor ??
+                Theme.of(context).primaryColor)
+                : (widget.iconTheme?.iconUnselectedFillColor ??
+                theme.canvasColor),
+            onPressed: () {
+              final _attribute = _selectedAttribute == attribute
+                  ? Attribute.header
+                  : attribute;
+              widget.controller.formatSelection(_attribute);
+              widget.afterButtonPressed?.call();
+            },
+            child: Text(
+              _valueToText[attribute] ?? '',
+              style: style.copyWith(
+                color: isSelected
+                    ? (widget.iconTheme?.iconSelectedColor ??
+                    theme.primaryIconTheme.color)
+                    : (widget.iconTheme?.iconUnselectedColor ??
+                    theme.iconTheme.color),
               ),
             ),
           ),
-        );
-      }).toList(),
-    );
+        ),
+      );
+    }).toList();
+
+    return widget.axis == Axis.horizontal
+        ? Row(
+            mainAxisSize: MainAxisSize.min,
+            children: children,
+          )
+        : Column(
+            mainAxisSize: MainAxisSize.min,
+            children: children,
+          );
   }
 
   void _didChangeEditingValue() {