From 526a3583294eb5109f0e2c01456ed1863db97b87 Mon Sep 17 00:00:00 2001
From: Pawan Kumar <pawan.stardust@gmail.com>
Date: Wed, 10 May 2023 17:34:03 +0530
Subject: [PATCH] chore: Add the ability to use icons in WdsButton (#23014)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Adds the ability to pass icon in button component and adds button group
to the storybook

## Description

Fixes #21924
Fixes #21926

Media
> A video or a GIF is preferred. when using Loom, don’t embed because it
looks like it’s a GIF. instead, just link to the video

## Type of change
- Bug fix

## How Has This Been Tested?
- Manual


### Test Plan
> Add Testsmith test cases links that relate to this PR

### Issues raised during DP testing
> Link issues raised during DP testing for better visiblity and tracking
(copy link from comments dropped on this PR)


## Checklist:
### Dev activity
- [ ] My code follows the style guidelines of this project
- [ ] I have performed a self-review of my own code
- [ ] I have commented my code, particularly in hard-to-understand areas
- [ ] I have made corresponding changes to the documentation
- [ ] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] New and existing unit tests pass locally with my changes
- [ ] PR is being merged under a feature flag


### QA activity:
- [ ] Test plan has been approved by relevant developers
- [ ] Test plan has been peer reviewed by QA
- [ ] Cypress test cases have been added and approved by either SDET or
manual QA
- [ ] Organized project review call with relevant stakeholders after
Round 1/2 of QA
- [ ] Added Test Plan Approved label after reveiwing all Cypress test
---
 .../src/components/Button/Button.stories.mdx  |  23 +++
 .../headless/src/components/Button/Button.tsx |   9 +-
 .../headless/src/components/Icon/Icon.tsx     |  32 +++
 .../headless/src/components/Icon/index.tsx    |   1 +
 .../design-system/headless/src/index.ts       |   1 +
 .../design-system/widgets/jest.config.js      |   5 +
 .../design-system/widgets/package.json        |   3 +-
 .../src/components/Button/Button.stories.mdx  |  53 +++++
 .../src/components/Button/Button.test.tsx     |  66 ++++++
 .../widgets/src/components/Button/Button.tsx  |  21 +-
 .../src/components/Button/index.styled.tsx    |  11 +-
 .../ButtonGroup/ButtonGroup.stories.mdx       | 192 ++++++++++++++++++
 .../components/ButtonGroup/ButtonGroup.tsx    |  22 +-
 .../components/ButtonGroup/index.styled.tsx   |  80 +++++---
 .../packages/storybook/.storybook/main.js     |   2 +
 15 files changed, 477 insertions(+), 44 deletions(-)
 create mode 100644 app/client/packages/design-system/headless/src/components/Button/Button.stories.mdx
 create mode 100644 app/client/packages/design-system/headless/src/components/Icon/Icon.tsx
 create mode 100644 app/client/packages/design-system/headless/src/components/Icon/index.tsx
 create mode 100644 app/client/packages/design-system/widgets/jest.config.js
 create mode 100644 app/client/packages/design-system/widgets/src/components/Button/Button.test.tsx
 create mode 100644 app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.stories.mdx

diff --git a/app/client/packages/design-system/headless/src/components/Button/Button.stories.mdx b/app/client/packages/design-system/headless/src/components/Button/Button.stories.mdx
new file mode 100644
index 0000000000..90d9263ad7
--- /dev/null
+++ b/app/client/packages/design-system/headless/src/components/Button/Button.stories.mdx
@@ -0,0 +1,23 @@
+import { Canvas, Meta, Story, ArgsTable } from "@storybook/addon-docs";
+
+import { Button } from "./";
+
+<Meta
+  title="Design-system/headless/Button"
+  component={Button}
+  args={{
+    children: "Button",
+  }}
+/>
+
+export const Template = (args) => <Button {...args} />;
+
+# Button
+
+A button is a clickable element that is used to trigger an action.
+
+<Canvas>
+  <Story name="Button">{Template.bind({})}</Story>
+</Canvas>
+
+<ArgsTable story="Button" of={Button} />
diff --git a/app/client/packages/design-system/headless/src/components/Button/Button.tsx b/app/client/packages/design-system/headless/src/components/Button/Button.tsx
index f86105f786..aefdad25b2 100644
--- a/app/client/packages/design-system/headless/src/components/Button/Button.tsx
+++ b/app/client/packages/design-system/headless/src/components/Button/Button.tsx
@@ -6,9 +6,14 @@ import { useFocusRing } from "@react-aria/focus";
 import { useHover } from "@react-aria/interactions";
 import { useFocusableRef } from "@react-spectrum/utils";
 import type { FocusableRef } from "@react-types/shared";
-import type { ButtonProps as SpectrumButtonProps } from "@react-types/button";
+import type {
+  AriaButtonProps as SpectrumAriaBaseButtonProps,
+  ButtonProps as SpectrumButtonProps,
+} from "@react-types/button";
 
-export interface ButtonProps extends SpectrumButtonProps {
+export interface ButtonProps
+  extends SpectrumButtonProps,
+    SpectrumAriaBaseButtonProps {
   className?: string;
 }
 
diff --git a/app/client/packages/design-system/headless/src/components/Icon/Icon.tsx b/app/client/packages/design-system/headless/src/components/Icon/Icon.tsx
new file mode 100644
index 0000000000..dd78944ebd
--- /dev/null
+++ b/app/client/packages/design-system/headless/src/components/Icon/Icon.tsx
@@ -0,0 +1,32 @@
+import React from "react";
+import type { AriaLabelingProps, DOMProps } from "@react-types/shared";
+import type { ReactElement } from "react";
+import { filterDOMProps } from "@react-aria/utils";
+
+export interface IconProps extends DOMProps, AriaLabelingProps {
+  "aria-label"?: string;
+  children: ReactElement;
+  "aria-hidden"?: boolean | "false" | "true";
+  role?: string;
+}
+
+export function Icon(props: IconProps) {
+  const {
+    "aria-hidden": ariaHiddenProp,
+    "aria-label": ariaLabel,
+    children,
+    role = "img",
+    ...otherProps
+  } = props;
+
+  const ariaHidden = !ariaHiddenProp ? undefined : ariaHiddenProp;
+
+  return React.cloneElement(children, {
+    ...filterDOMProps(otherProps),
+    focusable: "false",
+    "aria-label": ariaLabel,
+    "aria-hidden": ariaLabel ? ariaHidden || undefined : true,
+    role,
+    "data-icon": "",
+  });
+}
diff --git a/app/client/packages/design-system/headless/src/components/Icon/index.tsx b/app/client/packages/design-system/headless/src/components/Icon/index.tsx
new file mode 100644
index 0000000000..65d727be32
--- /dev/null
+++ b/app/client/packages/design-system/headless/src/components/Icon/index.tsx
@@ -0,0 +1 @@
+export { Icon } from "./Icon";
diff --git a/app/client/packages/design-system/headless/src/index.ts b/app/client/packages/design-system/headless/src/index.ts
index dcb257a9bc..9213dac78b 100644
--- a/app/client/packages/design-system/headless/src/index.ts
+++ b/app/client/packages/design-system/headless/src/index.ts
@@ -2,3 +2,4 @@
 export * from "./components/Button";
 export * from "./components/Checkbox";
 export * from "./components/Field";
+export * from "./components/Icon";
diff --git a/app/client/packages/design-system/widgets/jest.config.js b/app/client/packages/design-system/widgets/jest.config.js
new file mode 100644
index 0000000000..c24244a098
--- /dev/null
+++ b/app/client/packages/design-system/widgets/jest.config.js
@@ -0,0 +1,5 @@
+module.exports = {
+  preset: "ts-jest",
+  roots: ["<rootDir>/src"],
+  testEnvironment: "jsdom",
+};
diff --git a/app/client/packages/design-system/widgets/package.json b/app/client/packages/design-system/widgets/package.json
index a6b28d4f96..b476864730 100644
--- a/app/client/packages/design-system/widgets/package.json
+++ b/app/client/packages/design-system/widgets/package.json
@@ -6,7 +6,8 @@
   "license": "MIT",
   "scripts": {
     "lint:ci": "eslint --cache .",
-    "prettier:ci": "prettier --check ."
+    "prettier:ci": "prettier --check .",
+    "test:unit": "jest"
   },
   "dependencies": {
     "@capsizecss/core": "^3.1.0",
diff --git a/app/client/packages/design-system/widgets/src/components/Button/Button.stories.mdx b/app/client/packages/design-system/widgets/src/components/Button/Button.stories.mdx
index 9b324a42cd..16d3442a0e 100644
--- a/app/client/packages/design-system/widgets/src/components/Button/Button.stories.mdx
+++ b/app/client/packages/design-system/widgets/src/components/Button/Button.stories.mdx
@@ -1,5 +1,8 @@
 import { Canvas, Meta, Story, ArgsTable } from "@storybook/addon-docs";
+
 import { Button } from "./";
+import { Icon } from "@design-system/headless";
+import EmotionHappyLineIcon from "remixicon-react/EmotionHappyLineIcon";
 
 <Meta
   title="Design-system/widgets/Button"
@@ -78,3 +81,53 @@ There are 3 variants of the button component.
     {Template.bind({})}
   </Story>
 </Canvas>
+
+# With Icon
+
+<Canvas>
+  <Story
+    name="With Icon"
+    args={{
+      children: "With Icon",
+      icon: (
+        <Icon>
+          <EmotionHappyLineIcon />
+        </Icon>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+</Canvas>
+
+# Icon Position
+
+<Canvas>
+  <Story
+    name="Icon Position - Start"
+    args={{
+      children: "Icon Start",
+      icon: (
+        <Icon>
+          <EmotionHappyLineIcon />
+        </Icon>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+  <Story
+    name="Icon Position - End"
+    args={{
+      children: "Icon End",
+      icon: (
+        <Icon>
+          <EmotionHappyLineIcon />
+        </Icon>
+      ),
+      iconPosition: "end",
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+</Canvas>
diff --git a/app/client/packages/design-system/widgets/src/components/Button/Button.test.tsx b/app/client/packages/design-system/widgets/src/components/Button/Button.test.tsx
new file mode 100644
index 0000000000..558aa676aa
--- /dev/null
+++ b/app/client/packages/design-system/widgets/src/components/Button/Button.test.tsx
@@ -0,0 +1,66 @@
+import React from "react";
+import "@testing-library/jest-dom";
+import { Icon } from "@design-system/headless";
+import { render, screen } from "@testing-library/react";
+import EmotionHappyLineIcon from "remixicon-react/EmotionHappyLineIcon";
+
+import { Button } from "./";
+
+describe("@design-system/widgets/Button", () => {
+  it("renders children when passed", () => {
+    render(<Button>Click me</Button>);
+    expect(screen.getByRole("button")).toHaveTextContent("Click me");
+  });
+
+  it("passes type to button component", () => {
+    render(<Button type="submit" />);
+    expect(screen.getByRole("button")).toHaveAttribute("type", "submit");
+  });
+
+  it("sets variant based on prop", () => {
+    render(<Button variant="primary" />);
+    expect(screen.getByRole("button")).toHaveAttribute(
+      "data-variant",
+      "primary",
+    );
+  });
+
+  it("sets disabled attribute based on prop", () => {
+    render(<Button isDisabled />);
+    expect(screen.getByRole("button")).toBeDisabled();
+    expect(screen.getByRole("button")).toHaveAttribute("data-disabled");
+  });
+
+  it("sets data-loading attribute and icon based on loading prop", () => {
+    render(<Button isLoading />);
+    expect(screen.getByRole("button")).toHaveAttribute("data-loading");
+
+    const icon = screen.getByRole("button").querySelector("[data-icon]");
+    expect(icon).toBeInTheDocument();
+  });
+
+  it("renders icon when passed", () => {
+    const { container } = render(
+      <Button
+        icon={
+          <Icon>
+            <EmotionHappyLineIcon />
+          </Icon>
+        }
+      />,
+    );
+
+    const icon = container.querySelector("button [data-icon]") as HTMLElement;
+    expect(icon).toBeInTheDocument();
+  });
+
+  it("sets icon position attribute based on the prop ", () => {
+    const { container } = render(<Button iconPosition="end" />);
+
+    const button = container.querySelector("button") as HTMLElement;
+    expect(button).toHaveAttribute("data-icon-position", "end");
+
+    const styles = window.getComputedStyle(button);
+    expect(styles.flexDirection).toBe("row-reverse");
+  });
+});
diff --git a/app/client/packages/design-system/widgets/src/components/Button/Button.tsx b/app/client/packages/design-system/widgets/src/components/Button/Button.tsx
index a92fef88ec..b8c0a0e13e 100644
--- a/app/client/packages/design-system/widgets/src/components/Button/Button.tsx
+++ b/app/client/packages/design-system/widgets/src/components/Button/Button.tsx
@@ -1,4 +1,5 @@
 import React, { forwardRef } from "react";
+import { Icon as HeadlessIcon } from "@design-system/headless";
 import type {
   ButtonProps as HeadlessButtonProps,
   ButtonRef as HeadlessButtonRef,
@@ -20,6 +21,7 @@ export interface ButtonProps extends Omit<HeadlessButtonProps, "className"> {
   fontFamily?: fontFamilyTypes;
   isFitContainer?: boolean;
   isFocused?: boolean;
+  icon?: React.ReactNode;
   iconPosition?: "start" | "end";
 }
 
@@ -28,6 +30,8 @@ export const Button = forwardRef(
     const {
       children,
       fontFamily,
+      icon,
+      iconPosition = "start",
       isFitContainer = false,
       isLoading,
       // eslint-disable-next-line -- TODO add onKeyUp when the bug is fixedhttps://github.com/adobe/react-spectrum/issues/4350
@@ -38,19 +42,28 @@ export const Button = forwardRef(
 
     const renderChildren = () => {
       if (isLoading) {
-        return <Spinner />;
+        return (
+          <HeadlessIcon>
+            <Spinner />
+          </HeadlessIcon>
+        );
       }
 
       return (
-        <Text fontFamily={fontFamily} lineClamp={1}>
-          {children}
-        </Text>
+        <>
+          {icon}
+          <Text fontFamily={fontFamily} lineClamp={1}>
+            {children}
+          </Text>
+        </>
       );
     };
 
     return (
       <StyledButton
+        data-button=""
         data-fit-container={isFitContainer ? "" : undefined}
+        data-icon-position={iconPosition === "start" ? undefined : "end"}
         data-loading={isLoading ? "" : undefined}
         data-variant={variant}
         ref={ref}
diff --git a/app/client/packages/design-system/widgets/src/components/Button/index.styled.tsx b/app/client/packages/design-system/widgets/src/components/Button/index.styled.tsx
index a2e0fa9759..15ce147d81 100644
--- a/app/client/packages/design-system/widgets/src/components/Button/index.styled.tsx
+++ b/app/client/packages/design-system/widgets/src/components/Button/index.styled.tsx
@@ -8,7 +8,7 @@ export const StyledButton = styled(HeadlessButton)<ButtonProps>`
   align-items: center;
   cursor: pointer;
   outline: 0;
-  gap: var(--spacing-4);
+  gap: var(--spacing-1);
   padding: var(--spacing-2) var(--spacing-4);
   height: calc(var(--sizing-root-unit) * 8);
   border-radius: var(--border-radius-1);
@@ -76,4 +76,13 @@ export const StyledButton = styled(HeadlessButton)<ButtonProps>`
     pointer-events: none;
     opacity: var(--opacity-disabled);
   }
+
+  & [data-icon] {
+    height: calc(var(--sizing-root-unit) * 5);
+    width: calc(var(--sizing-root-unit) * 5);
+  }
+
+  &[data-icon-position="end"] {
+    flex-direction: row-reverse;
+  }
 `;
diff --git a/app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.stories.mdx b/app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.stories.mdx
new file mode 100644
index 0000000000..1d5c44132b
--- /dev/null
+++ b/app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.stories.mdx
@@ -0,0 +1,192 @@
+import { Canvas, Meta, Story, ArgsTable } from "@storybook/addon-docs";
+
+import { Button } from "../Button";
+import { ButtonGroup } from "./";
+
+<Meta
+  title="Design-system/widgets/Button Group"
+  component={ButtonGroup}
+  args={{
+    children: (
+      <>
+        <Button>Option 1</Button>
+        <Button>Option 2</Button>
+        <Button>Option 3</Button>
+      </>
+    ),
+  }}
+/>
+
+export const Template = (args) => <ButtonGroup {...args} />;
+
+## Button Group
+
+A button group is a group of buttons that are visually connected together.
+
+<Canvas>
+  <Story name="Button Group">{Template.bind({})}</Story>
+</Canvas>
+
+<ArgsTable story="Button Group" of={ButtonGroup} />
+
+# Variants
+
+<Canvas>
+  <Story
+    name="Button Group - Primary"
+    args={{
+      children: (
+        <>
+          <Button variant="primary">Option 1</Button>
+          <Button variant="primary">Option 2</Button>
+          <Button variant="primary">Option 3</Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+  <Story
+    name="Button Group -  Secondary"
+    args={{
+      children: (
+        <>
+          <Button variant="secondary">Option 1</Button>
+          <Button variant="secondary">Option 2</Button>
+          <Button variant="secondary">Option 3</Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+  <Story
+    name="Button Group - Tertiary"
+    args={{
+      children: (
+        <>
+          <Button variant="tertiary">Option 1</Button>
+          <Button variant="tertiary">Option 2</Button>
+          <Button variant="tertiary">Option 3</Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+</Canvas>
+
+# Orientation
+
+<Canvas>
+  <Story
+    name="Button Group - Vertical Primary"
+    args={{
+      orientation: "vertical",
+      children: (
+        <>
+          <Button variant="primary">Option 2</Button>
+          <Button variant="primary">Option 2</Button>
+          <Button variant="primary">Option 3</Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+  <Story
+    name="Button Group - Vertical Secondary"
+    args={{
+      orientation: "vertical",
+      children: (
+        <>
+          <Button variant="secondary">Option 2</Button>
+          <Button variant="secondary">Option 2</Button>
+          <Button variant="secondary">Option 3</Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+  <Story
+    name="Button Group - Vertical Tertiary"
+    args={{
+      orientation: "vertical",
+      children: (
+        <>
+          <Button variant="tertiary">Option 2</Button>
+          <Button variant="tertiary">Option 2</Button>
+          <Button variant="tertiary">Option 3</Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+</Canvas>
+
+# Disabled
+
+<Canvas>
+  <Story
+    name="Button Group Primary Disabled"
+    args={{
+      children: (
+        <>
+          <Button variant="primary" isDisabled>
+            Option 2
+          </Button>
+          <Button variant="primary" isDisabled>
+            Option 2
+          </Button>
+          <Button variant="primary" isDisabled>
+            Option 3
+          </Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+  <Story
+    name="Button Group Secondary Disabled"
+    args={{
+      children: (
+        <>
+          <Button variant="secondary" isDisabled>
+            Option 2
+          </Button>
+          <Button variant="secondary" isDisabled>
+            Option 2
+          </Button>
+          <Button variant="secondary" isDisabled>
+            Option 3
+          </Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+  <Story
+    name="Button Group Tertiary Disabled"
+    args={{
+      children: (
+        <>
+          <Button variant="tertiary" isDisabled>
+            Option 2
+          </Button>
+          <Button variant="tertiary" isDisabled>
+            Option 2
+          </Button>
+          <Button variant="tertiary" isDisabled>
+            Option 3
+          </Button>
+        </>
+      ),
+    }}
+  >
+    {Template.bind({})}
+  </Story>
+</Canvas>
diff --git a/app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.tsx b/app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.tsx
index 8d865a99d4..675bc50093 100644
--- a/app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.tsx
+++ b/app/client/packages/design-system/widgets/src/components/ButtonGroup/ButtonGroup.tsx
@@ -3,11 +3,12 @@ import React, { forwardRef } from "react";
 import { StyledContainer } from "./index.styled";
 
 // types
-export enum Orientation {
-  VERTICAL = "vertical",
-  HORIZONTAL = "horizontal",
-}
+export const ORIENTATION = {
+  VERTICAL: "vertical",
+  HORIZONTAL: "horizontal",
+} as const;
 
+type Orientation = (typeof ORIENTATION)[keyof typeof ORIENTATION];
 export interface ButtonGroupProps
   extends React.ComponentPropsWithoutRef<"div"> {
   children?: React.ReactNode;
@@ -17,8 +18,17 @@ export interface ButtonGroupProps
 // component
 export const ButtonGroup = forwardRef<HTMLDivElement, ButtonGroupProps>(
   (props, ref) => {
-    const { orientation = Orientation.HORIZONTAL, ...others } = props;
-    return <StyledContainer orientation={orientation} ref={ref} {...others} />;
+    const { orientation = ORIENTATION.HORIZONTAL, ...others } = props;
+
+    return (
+      <StyledContainer
+        data-orientation={
+          orientation === ORIENTATION.VERTICAL ? "vertical" : undefined
+        }
+        ref={ref}
+        {...others}
+      />
+    );
   },
 );
 
diff --git a/app/client/packages/design-system/widgets/src/components/ButtonGroup/index.styled.tsx b/app/client/packages/design-system/widgets/src/components/ButtonGroup/index.styled.tsx
index 2ff533fe63..b5fe780214 100644
--- a/app/client/packages/design-system/widgets/src/components/ButtonGroup/index.styled.tsx
+++ b/app/client/packages/design-system/widgets/src/components/ButtonGroup/index.styled.tsx
@@ -3,13 +3,16 @@ import styled from "styled-components";
 import type { ButtonGroupProps } from "./ButtonGroup";
 
 export const StyledContainer = styled.div<ButtonGroupProps>`
-  --border-width: 1px;
-
   display: flex;
   height: 100%;
   width: 100%;
-  flex-direction: ${({ orientation }) =>
-    orientation === "vertical" ? "column" : "row"};
+  flex-direction: row;
+  align-items: center;
+  justify-content: center;
+
+  &[data-orientation="vertical"] {
+    flex-direction: column;
+  }
 
   & [data-button] {
     // increasing z index to make sure the focused button is on top of the others
@@ -17,48 +20,65 @@ export const StyledContainer = styled.div<ButtonGroupProps>`
       z-index: 1;
     }
 
-    &:is([data-variant="filled"]):not([data-disabled]) {
-      border-color: var(--wds-vs-color-border-onaccent);
-    }
-
-    &:is([data-variant="light"]):not([data-disabled]) {
-      border-color: var(--wds-vs-color-border-onaccent-light);
-    }
-
     &:first-child {
       border-bottom-right-radius: 0;
-      ${({ orientation }) =>
-        orientation === "vertical"
-          ? "border-bottom-left-radius: 0; border-bottom-width: calc(var(--border-width) / 2);"
-          : "border-top-right-radius: 0; border-right-width: calc(var(--border-width) / 2);"}
     }
 
     &:last-of-type {
       border-top-left-radius: 0;
-      ${({ orientation }) =>
-        orientation === "vertical"
-          ? "border-top-right-radius: 0; border-top-width: calc(var(--border-width) / 2);"
-          : "border-bottom-left-radius: 0; border-left-width: calc(var(--border-width) / 2);"}
     }
 
     &:not(:first-child):not(:last-of-type) {
       border-radius: 0;
+    }
+  }
 
-      ${({ orientation }) =>
-        orientation === "vertical"
-          ? "border-top-width: calc(var(--border-width) / 2); border-bottom-width: calc(var(--border-width) / 2);"
-          : "border-left-width: calc(var(--border-width) / 2); border-right-width: calc(var(--border-width) / 2);"}
+  &:not([data-orientation="vertical"]) [data-button] {
+    &:first-child {
+      border-top-right-radius: 0;
+      border-right-width: calc(var(--border-width-1) / 2);
+    }
+
+    &:last-of-type {
+      border-bottom-left-radius: 0;
+      border-left-width: calc(var(--border-width-1) / 2);
+    }
+
+    &:not(:first-child):not(:last-of-type) {
+      border-left-width: calc(var(--border-width-1) / 2);
+      border-right-width: calc(var(--border-width-1) / 2);
     }
 
     & + [data-button] {
-      ${({ orientation }) =>
-        orientation === "vertical"
-          ? "margin-top: calc(var(--border-width) * -1);"
-          : "margin-left: calc(var(--border-width) * -1);"}
+      margin-left: calc(var(--border-width-1) * -1);
 
       @media (min-resolution: 192dpi) {
-        ${({ orientation }) =>
-          orientation === "vertical" ? "margin-top: 0px;" : "margin-left: 0px;"}
+        margin-left: 0px;
+      }
+    }
+  }
+
+  &[data-orientation="vertical"] [data-button] {
+    &:first-child {
+      border-bottom-left-radius: 0;
+      border-bottom-width: calc(var(--border-width-1) / 2);
+    }
+
+    &:last-of-type {
+      border-top-right-radius: 0;
+      border-top-width: calc(var(--border-width-1) / 2);
+    }
+
+    &:not(:first-child):not(:last-of-type) {
+      border-top-width: calc(var(--border-width-1) / 2);
+      border-bottom-width: calc(var(--border-width-1) / 2);
+    }
+
+    & + [data-button] {
+      margin-top: calc(var(--border-width-1) * -1);
+
+      @media (min-resolution: 192dpi) {
+        margin-top: 0px;
       }
     }
   }
diff --git a/app/client/packages/storybook/.storybook/main.js b/app/client/packages/storybook/.storybook/main.js
index d8b71a5e68..d168963faa 100644
--- a/app/client/packages/storybook/.storybook/main.js
+++ b/app/client/packages/storybook/.storybook/main.js
@@ -23,7 +23,9 @@ async function webpackConfig(config) {
 module.exports = {
   stories: [
     "../../design-system/widgets/src/**/*.stories.mdx",
+    "../../design-system/headless/src/**/*.stories.mdx",
     "../../design-system/widgets/src/**/*.stories.@(js|jsx|ts|tsx)",
+    "../../design-system/headless/src/**/*.stories.@(js|jsx|ts|tsx)",
   ],
   addons: [
     "@storybook/addon-links",