From 931a039f749a57eb5a5a01fdfd338dac1c38d86b Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Fri, 15 Nov 2024 14:35:50 -0700
Subject: [PATCH 01/11] feat(atomic-elements)!: Implemented ChipGroupField to
 support v3 Collections api

The ChipGroupField component implements the Collection API from
`@react-aria/collections`.

Reimplements `ChipGroup` as a simplified wrapper around ChipGroupField

Reimplements `Chip` as a leaf component using `createLeafComponent`.
Note that it is still usable stand-alone like it was before

BREAKING CHANGE: Chip no longer support `key` prop and instead uses
the `id` prop when used within a collection.
---
 .changeset/dull-wolves-look.md                |   5 +
 .../components/Chips/Chip/Chip.component.tsx  | 150 ++++++++------
 .../src/components/Chips/Chip/Chip.context.ts |   4 +
 .../src/components/Chips/Chip/Chip.spec.tsx   |  11 +-
 .../components/Chips/Chip/Chip.stories.tsx    |   5 +
 .../src/components/Chips/Chip/Chip.types.ts   |  29 +++
 .../Chip/__snapshots__/Chip.spec.tsx.snap     |   2 +-
 .../src/components/Chips/Chip/index.tsx       |   4 +-
 .../Chips/ChipGroup/ChipGroup.component.tsx   | 110 ++---------
 .../Chips/ChipGroup/ChipGroup.spec.tsx        |   8 +-
 .../Chips/ChipGroup/ChipGroup.stories.tsx     |  11 +-
 .../__snapshots__/ChipGroup.spec.tsx.snap     |   8 +-
 .../ChipGroupField.component.tsx              | 185 ++++++++++++++++++
 .../ChipGroupField/ChipGroupField.context.ts  |  15 ++
 .../ChipGroupField/ChipGroupField.spec.tsx    | 118 +++++++++++
 .../ChipGroupField/ChipGroupField.stories.tsx |  27 +++
 .../ChipGroupField/ChipGroupField.styles.ts}  |   3 +-
 .../ChipGroupField.spec.tsx.snap              |  69 +++++++
 .../components/Fields/ChipGroupField/index.ts |   3 +
 .../atomic-elements/src/components/index.ts   |   4 +
 20 files changed, 604 insertions(+), 167 deletions(-)
 create mode 100644 .changeset/dull-wolves-look.md
 create mode 100644 packages/atomic-elements/src/components/Chips/Chip/Chip.context.ts
 create mode 100644 packages/atomic-elements/src/components/Chips/Chip/Chip.types.ts
 create mode 100644 packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx
 create mode 100644 packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts
 create mode 100644 packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.spec.tsx
 create mode 100644 packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.stories.tsx
 rename packages/atomic-elements/src/components/{Chips/ChipGroup/ChipGroup.styles.ts => Fields/ChipGroupField/ChipGroupField.styles.ts} (78%)
 create mode 100644 packages/atomic-elements/src/components/Fields/ChipGroupField/__snapshots__/ChipGroupField.spec.tsx.snap
 create mode 100644 packages/atomic-elements/src/components/Fields/ChipGroupField/index.ts

diff --git a/.changeset/dull-wolves-look.md b/.changeset/dull-wolves-look.md
new file mode 100644
index 00000000..42abe38c
--- /dev/null
+++ b/.changeset/dull-wolves-look.md
@@ -0,0 +1,5 @@
+---
+"@atomicjolt/atomic-elements": major
+---
+
+Implemented ChipGroupField using the new Collections API & re-implemented ChipGroup & Chip on top of that base
diff --git a/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx b/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx
index df8ee24d..48bd29e0 100644
--- a/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx
+++ b/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx
@@ -1,98 +1,130 @@
-import React from "react";
-import classNames from "classnames";
-import type { ItemProps } from "react-stately";
-import { PressEvent, PressProps } from "@react-aria/interactions";
-import { AriaButtonProps } from "@react-aria/button";
-import { mergeProps } from "@react-aria/utils";
+import React, { useContext } from "react";
+import { filterDOMProps, mergeProps, useObjectRef } from "@react-aria/utils";
+import { useTag } from "@react-aria/tag";
+import { createLeafComponent } from "@react-aria/collections";
 
-import { copyStaticProperties } from "@utils/clone";
-import { useVariantClass } from "@hooks/variants";
-import { Item } from "@components/Collection";
 import { IconButton } from "@components/Buttons/IconButton";
 import { useConditionalPress } from "@hooks/useConditionalPress";
+import { useFocusRing } from "@hooks/useFocusRing";
+import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useRenderProps } from "@hooks";
+import { ChipGroupStateContext } from "@components/Fields/ChipGroupField/ChipGroupField.context";
+
+import { ChipArgs, ChipGroupChipProps, ChipInternalProps } from "./Chip.types";
 import { ChipContent, ChipWrapper } from "./Chip.styles";
-import { SuggestStrings, HasClassName } from "../../../types";
+import { ChipContext } from "./Chip.context";
 
-type ChipVariants = SuggestStrings<
-  "default" | "warning" | "success" | "danger"
->;
+export function ChipLeaf<T>(...args: ChipArgs<T>) {
+  const [props, ref] = useContextPropsV2(ChipContext, args[0], args[1]);
 
-export interface ChipProps<T> extends ItemProps<T>, PressProps, HasClassName {
-  children: React.ReactNode;
-  variant?: ChipVariants;
-  /** Handler that is called when the user
-   * clicks the remove button for the chip */
-  onRemove?: (e: PressEvent) => void;
-  isDisabled?: boolean;
-}
+  // We're being rendered standalone
+  if (args.length === 2) {
+    return (
+      <ChipInternal {...props} ref={ref} allowsRemoving={!!props.onRemove} />
+    );
+  }
 
-/** Chip component */
-export function Chip<T>(props: ChipProps<T>) {
-  return <ChipInternal {...props} allowsRemoving={!!props.onRemove} />;
+  // We're being rendered as part of a collection (i.e ChipGroup)
+  const item = args[2];
+  return <ChipGroupChip item={item} itemRef={ref} {...props} />;
 }
 
-copyStaticProperties(Item, Chip);
+/**
+ * Chip component. Can be used stand-alone, or within a parent
+ * `ChipGroup`
+ */
+export const Chip = createLeafComponent("item", ChipLeaf);
+
+function ChipGroupChip<T>(props: ChipGroupChipProps<T>) {
+  const { item } = props;
+  const ref = useObjectRef<HTMLDivElement>(props.itemRef);
+  const state = useContext(ChipGroupStateContext)!;
+  const {
+    rowProps,
+    gridCellProps,
+    removeButtonProps,
+    allowsRemoving,
+    isFocused,
+    isSelected,
+  } = useTag(props, state, ref);
+
+  const { focusProps, isFocusVisible } = useFocusRing({ within: true });
+
+  const isDisabled = state.disabledKeys.has(item.key);
 
-interface ChipInternalProps<T> extends ChipProps<T> {
-  wrapperProps?: React.DOMAttributes<HTMLDivElement>;
-  contentProps?: React.DOMAttributes<HTMLDivElement>;
-  removeButtonProps?: AriaButtonProps<"button">;
-  allowsRemoving?: boolean;
+  const renderProps = useRenderProps({
+    componentClassName: "aje-chip",
+    values: {
+      isSelected,
+      isFocusVisible,
+      isFocused,
+    },
+    ...props,
+  });
+
+  return (
+    <ChipWrapper
+      ref={ref}
+      {...mergeProps(rowProps, focusProps)}
+      {...renderProps}
+    >
+      <ChipContent {...gridCellProps}>
+        {renderProps.children}
+        {allowsRemoving && (
+          <IconButton
+            icon="close"
+            size="small"
+            variant="chip"
+            {...mergeProps(removeButtonProps, { isDisabled })}
+          />
+        )}
+      </ChipContent>
+    </ChipWrapper>
+  );
 }
 
-export const ChipInternal = React.forwardRef<
-  HTMLDivElement,
-  ChipInternalProps<any>
->(function ChipInternal<T>(
+export const ChipInternal = React.forwardRef(function ChipInternal<T>(
   props: ChipInternalProps<T>,
   ref: React.Ref<HTMLDivElement>
 ) {
   const {
     className,
-    variant = "default",
+    variant,
     onRemove,
     isDisabled,
     children,
-    wrapperProps = {},
-    contentProps = {},
-    removeButtonProps = {},
     allowsRemoving = false,
     ...rest
   } = props;
 
-  const variantClass = useVariantClass("aje-chip", variant);
   const { pressProps } = useConditionalPress(rest);
 
-  const allWrapperProps = [
-    wrapperProps,
-    { "aria-disabled": isDisabled || undefined },
-  ];
-
-  if (!isDisabled) {
-    allWrapperProps.push(pressProps);
-  }
+  const renderProps = useRenderProps({
+    componentClassName: "aje-chip",
+    values: {
+      isSelected: false,
+      isFocusVisible: false,
+      isFocused: false,
+    },
+    ...props,
+  });
 
   return (
     <ChipWrapper
-      className={classNames("aje-chip", variantClass, className)}
       ref={ref}
-      {...mergeProps(...allWrapperProps)}
+      {...mergeProps({ "aria-disabled": isDisabled || undefined }, pressProps)}
+      {...renderProps}
+      {...filterDOMProps(props as any)}
     >
-      <ChipContent {...contentProps}>
-        {children}
-
+      <ChipContent>
+        {renderProps.children}
         {allowsRemoving && (
           <IconButton
             icon="close"
             size="small"
             variant="chip"
-            {...mergeProps(
-              {
-                isDisabled,
-                onPress: onRemove,
-              },
-              removeButtonProps
-            )}
+            isDisabled={isDisabled}
+            onPress={onRemove}
           />
         )}
       </ChipContent>
diff --git a/packages/atomic-elements/src/components/Chips/Chip/Chip.context.ts b/packages/atomic-elements/src/components/Chips/Chip/Chip.context.ts
new file mode 100644
index 00000000..3c76fb8e
--- /dev/null
+++ b/packages/atomic-elements/src/components/Chips/Chip/Chip.context.ts
@@ -0,0 +1,4 @@
+import { createComponentContext } from "@utils/index";
+import { ChipProps } from "./Chip.types";
+
+export const ChipContext = createComponentContext<ChipProps<any>>();
diff --git a/packages/atomic-elements/src/components/Chips/Chip/Chip.spec.tsx b/packages/atomic-elements/src/components/Chips/Chip/Chip.spec.tsx
index f31188f5..ffd00b3d 100644
--- a/packages/atomic-elements/src/components/Chips/Chip/Chip.spec.tsx
+++ b/packages/atomic-elements/src/components/Chips/Chip/Chip.spec.tsx
@@ -1,5 +1,5 @@
-import { render } from "@testing-library/react";
-import { expect, describe, test } from "vitest";
+import { fireEvent, render, screen } from "@testing-library/react";
+import { expect, describe, test, vi } from "vitest";
 import { Chip } from ".";
 
 describe("Chip", () => {
@@ -7,4 +7,11 @@ describe("Chip", () => {
     const result = render(<Chip>Item</Chip>);
     expect(result.asFragment()).toMatchSnapshot();
   });
+
+  test("onRemove", () => {
+    const onRemove = vi.fn();
+    render(<Chip onRemove={onRemove}>Item</Chip>);
+    fireEvent.click(screen.getByRole("button"));
+    expect(onRemove).toHaveBeenCalledOnce();
+  });
 });
diff --git a/packages/atomic-elements/src/components/Chips/Chip/Chip.stories.tsx b/packages/atomic-elements/src/components/Chips/Chip/Chip.stories.tsx
index 1d377aa0..4569d5ac 100644
--- a/packages/atomic-elements/src/components/Chips/Chip/Chip.stories.tsx
+++ b/packages/atomic-elements/src/components/Chips/Chip/Chip.stories.tsx
@@ -23,6 +23,11 @@ export default {
         category: "Events",
       },
     },
+    onAction: {
+      table: {
+        disable: true,
+      },
+    },
   },
 } as Meta<typeof Chip>;
 
diff --git a/packages/atomic-elements/src/components/Chips/Chip/Chip.types.ts b/packages/atomic-elements/src/components/Chips/Chip/Chip.types.ts
new file mode 100644
index 00000000..305d39ae
--- /dev/null
+++ b/packages/atomic-elements/src/components/Chips/Chip/Chip.types.ts
@@ -0,0 +1,29 @@
+import { Node } from "react-stately";
+import { AriaLabelProps, DomProps, SuggestStrings } from "../../../types";
+import { PressEvent, PressProps } from "@react-aria/interactions";
+import { ItemProps } from "../../Collection";
+
+type ChipVariants = SuggestStrings<
+  "default" | "warning" | "success" | "danger"
+>;
+
+export interface ChipProps<T> extends ItemProps, PressProps {
+  variant?: ChipVariants;
+  /** Handler that is called when the user
+   * clicks the remove button for the chip */
+  onRemove?: (e: PressEvent) => void;
+  isDisabled?: boolean;
+}
+
+export type ChipArgs<T> =
+  | [ChipProps<T>, React.ForwardedRef<HTMLDivElement>]
+  | [ChipProps<T>, React.ForwardedRef<HTMLDivElement>, Node<T>];
+
+export interface ChipGroupChipProps<T> extends ChipProps<T> {
+  item: Node<T>;
+  itemRef: React.ForwardedRef<HTMLDivElement>;
+}
+
+export interface ChipInternalProps<T> extends ChipProps<T> {
+  allowsRemoving?: boolean;
+}
diff --git a/packages/atomic-elements/src/components/Chips/Chip/__snapshots__/Chip.spec.tsx.snap b/packages/atomic-elements/src/components/Chips/Chip/__snapshots__/Chip.spec.tsx.snap
index 8f5b6891..e3ec1788 100644
--- a/packages/atomic-elements/src/components/Chips/Chip/__snapshots__/Chip.spec.tsx.snap
+++ b/packages/atomic-elements/src/components/Chips/Chip/__snapshots__/Chip.spec.tsx.snap
@@ -3,7 +3,7 @@
 exports[`Chip > matches snapshot 1`] = `
 <DocumentFragment>
   <div
-    class="sc-keTIit ia-dvfO aje-chip aje-chip--default"
+    class="sc-keTIit ia-dvfO aje-chip"
   >
     <div
       class="sc-dstKZu gTCzyn"
diff --git a/packages/atomic-elements/src/components/Chips/Chip/index.tsx b/packages/atomic-elements/src/components/Chips/Chip/index.tsx
index c8e94d79..6d56711b 100644
--- a/packages/atomic-elements/src/components/Chips/Chip/index.tsx
+++ b/packages/atomic-elements/src/components/Chips/Chip/index.tsx
@@ -1,2 +1,2 @@
-export { Chip, ChipInternal } from "./Chip.component";
-export type { ChipProps } from "./Chip.component";
+export { Chip, ChipLeaf } from "./Chip.component";
+export type { ChipProps } from "./Chip.types";
diff --git a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.component.tsx b/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.component.tsx
index 6af3a3d8..63d50341 100644
--- a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.component.tsx
+++ b/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.component.tsx
@@ -1,18 +1,7 @@
-import { useRef } from "react";
-import type { ListState } from "react-stately";
-import classNames from "classnames";
-import {
-  AriaTagGroupProps,
-  AriaTagProps,
-  useTag,
-  useTagGroup,
-} from "@react-aria/tag";
-import { useListState } from "react-stately";
+import { AriaTagGroupProps } from "@react-aria/tag";
 import { AriaProps, FieldInputProps } from "../../../types";
 import { ErrorMessage, Label, Message } from "@components/Fields";
-import { ChipsWrapper, ChipGroupWrapper } from "./ChipGroup.styles";
-import { useFocusRing } from "../../../hooks/useFocusRing";
-import { ChipInternal } from "../Chip";
+import { ChipGroupField } from "@components/Fields/ChipGroupField";
 
 export interface ChipGroupProps<T extends object>
   extends AriaProps<AriaTagGroupProps<T>>,
@@ -22,93 +11,36 @@ export interface ChipGroupProps<T extends object>
   labelPlacement?: "above" | "inline";
 }
 
-/** Collection Component for displaying a group of chips.
- * Chips can be selected and removed by the user.
+/**
+ * A generic ChipGroup component that renders a group of chips with optional labels, messages, and error messages.
  */
 export function ChipGroup<T extends object>(props: ChipGroupProps<T>) {
   const {
     label,
     message,
     error,
-    isInvalid,
-    isDisabled,
-    isRequired,
-    className,
     labelPlacement = "above",
-    size = "full",
+    items,
+    children,
+    ...rest
   } = props;
 
-  const ref = useRef(null);
-
-  const state = useListState(props);
-  const { gridProps, labelProps, descriptionProps, errorMessageProps } =
-    useTagGroup(props, state, ref);
-
   return (
-    <ChipGroupWrapper
-      className={classNames(className, "aje-chip-group")}
-      isDisabled={isDisabled}
-      isInvalid={isInvalid}
-      isRequired={isRequired}
-      size={size}
-    >
+    <ChipGroupField {...rest}>
       {labelPlacement === "above" && label && (
-        <Label {...labelProps} $paddingBottom="0px">
-          {label}
-        </Label>
+        <Label $paddingBottom="0px">{label}</Label>
       )}
-      <ChipsWrapper {...gridProps} ref={ref}>
-        {labelPlacement === "inline" && label && (
-          <Label {...labelProps} $paddingBottom="0px">
-            {label}
-          </Label>
-        )}
-        {[...state.collection].map((item) => (
-          <ChipGroupChip key={item.key} item={item} state={state} />
-        ))}
-      </ChipsWrapper>
-      {message && <Message {...descriptionProps}>{message}</Message>}
-      {isInvalid && error && (
-        <ErrorMessage {...errorMessageProps}>{error}</ErrorMessage>
-      )}
-    </ChipGroupWrapper>
-  );
-}
-
-// NOTE: when Chip is rendered standalone, it renders the actual Chip component
-// When rendered within a ChipGroup, it actually renders the ChipInternal component
-
-interface ChipGroupChipProps<T> extends AriaTagProps<T> {
-  state: ListState<T>;
-}
-
-function ChipGroupChip<T>(props: ChipGroupChipProps<T>) {
-  const { item, state } = props;
-  const ref = useRef(null);
-  const { rowProps, gridCellProps, removeButtonProps, allowsRemoving } = useTag(
-    props,
-    state,
-    ref
-  );
-
-  const { focusProps } = useFocusRing({ within: true });
-
-  const isDisabled = state.disabledKeys.has(item.key);
-
-  return (
-    <ChipInternal
-      ref={ref}
-      wrapperProps={{ ...rowProps, ...focusProps }}
-      contentProps={gridCellProps}
-      removeButtonProps={{
-        ...removeButtonProps,
-        isDisabled,
-      }}
-      {...item.props}
-      allowsRemoving={allowsRemoving}
-      isDisabled={isDisabled}
-    >
-      {item.rendered}
-    </ChipInternal>
+      <ChipGroupField.List
+        items={items}
+        beforeChildren={
+          labelPlacement === "inline" &&
+          label && <Label $paddingBottom="0px">{label}</Label>
+        }
+      >
+        {children}
+      </ChipGroupField.List>
+      {message && <Message>{message}</Message>}
+      {error && <ErrorMessage>{error}</ErrorMessage>}
+    </ChipGroupField>
   );
 }
diff --git a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.spec.tsx b/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.spec.tsx
index cfcddbfb..6b168263 100644
--- a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.spec.tsx
+++ b/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.spec.tsx
@@ -24,13 +24,13 @@ describe("ChipGroup", () => {
 
   test("renders chips", () => {
     const chips = [
-      { key: "1", rendered: "Chip 1" },
-      { key: "2", rendered: "Chip 2" },
-      { key: "3", rendered: "Chip 3" },
+      { id: "1", rendered: "Chip 1" },
+      { id: "2", rendered: "Chip 2" },
+      { id: "3", rendered: "Chip 3" },
     ];
     render(
       <ChipGroup items={chips} label="Label">
-        {({ key, rendered }) => <Chip key={key}>{rendered}</Chip>}
+        {({ rendered }) => <Chip>{rendered}</Chip>}
       </ChipGroup>
     );
     chips.forEach((chip) => {
diff --git a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.stories.tsx b/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.stories.tsx
index 2e8e01ed..87d29309 100644
--- a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.stories.tsx
+++ b/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.stories.tsx
@@ -20,6 +20,9 @@ export default {
     },
     onRemove: {
       description: "Function to call when a chip is removed",
+      table: {
+        category: "Events",
+      },
     },
   },
 } as Meta<typeof ChipGroup>;
@@ -30,10 +33,10 @@ export const Primary: Story = {
   args: {
     label: "Chip Group",
     children: [
-      <Chip key="news">News</Chip>,
-      <Chip key="travel">Travel</Chip>,
-      <Chip key="gaming">Gaming</Chip>,
-      <Chip key="shopping">Shopping</Chip>,
+      <Chip id="news">News</Chip>,
+      <Chip id="travel">Travel</Chip>,
+      <Chip id="gaming">Gaming</Chip>,
+      <Chip id="shopping">Shopping</Chip>,
     ],
   },
 };
diff --git a/packages/atomic-elements/src/components/Chips/ChipGroup/__snapshots__/ChipGroup.spec.tsx.snap b/packages/atomic-elements/src/components/Chips/ChipGroup/__snapshots__/ChipGroup.spec.tsx.snap
index b460350b..27b205bf 100644
--- a/packages/atomic-elements/src/components/Chips/ChipGroup/__snapshots__/ChipGroup.spec.tsx.snap
+++ b/packages/atomic-elements/src/components/Chips/ChipGroup/__snapshots__/ChipGroup.spec.tsx.snap
@@ -3,7 +3,7 @@
 exports[`ChipGroup > matches snapshot 1`] = `
 <DocumentFragment>
   <div
-    class="sc-ovuCP euVikJ aje-chip-group is-full"
+    class="sc-ovuCP euVikJ aje-chip-group"
   >
     <label
       class="sc-blHHSb jKAkSr"
@@ -23,9 +23,9 @@ exports[`ChipGroup > matches snapshot 1`] = `
     >
       <div
         aria-label="Item"
-        class="sc-fFoeYl kvHja-d aje-chip aje-chip--default"
-        data-key="$.0"
-        id="react-aria-:r0:-$.0"
+        class="sc-fFoeYl kvHja-d aje-chip"
+        data-key="react-aria-1"
+        id="react-aria-:r0:-react-aria-1"
         role="row"
         tabindex="0"
       >
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx
new file mode 100644
index 00000000..1db70709
--- /dev/null
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx
@@ -0,0 +1,185 @@
+import React, {
+  DOMAttributes,
+  forwardRef,
+  RefAttributes,
+  useContext,
+} from "react";
+import { useTagGroup } from "@react-aria/tag";
+import { useListState } from "react-stately";
+import {
+  BaseCollection,
+  Collection,
+  CollectionBuilder,
+} from "@react-aria/collections";
+import {
+  CollectionBase,
+  FocusableElement,
+  Key,
+  MultipleSelection,
+  SelectionBehavior,
+} from "@react-types/shared";
+
+import {
+  FieldErrorContext,
+  FieldLabelContext,
+  FieldMessageContext,
+  FieldProps,
+} from "@components/Fields";
+import { useRenderProps } from "@hooks";
+import { useContextPropsV2 } from "@hooks/useContextProps";
+import { Provider } from "@components/Internal/Provider";
+import { useCollectionRenderer } from "@hooks/useCollectionRenderer";
+
+import { ChipsWrapper, ChipGroupWrapper } from "./ChipGroupField.styles";
+import {
+  ChipGroupFieldContext,
+  ChipGroupListContext,
+  ChipGroupStateContext,
+} from "./ChipGroupField.context";
+import {
+  AriaLabelProps,
+  DomProps,
+  ExtendedSize,
+  FieldStatusProps,
+  RenderStyleProps,
+} from "../../../types";
+import { ItemContext } from "@components/Collection";
+import { ChipLeaf } from "@components/Chips/Chip";
+
+export interface ChipGroupFieldProps
+  extends MultipleSelection,
+    RenderStyleProps<never>,
+    DomProps,
+    AriaLabelProps,
+    Omit<FieldStatusProps, "isReadOnly"> {
+  children: React.ReactNode;
+  /** Handler called when a user deletes a chip */
+  onRemove?: (removed: Set<Key>) => void;
+  /** How multiple selection should behave in the collection. */
+  selectionBehavior?: SelectionBehavior;
+  /** Size of the chip group */
+  size?: ExtendedSize;
+}
+
+interface ForwardedChipGroupField {
+  (props: ChipGroupFieldProps & RefAttributes<HTMLDivElement>): JSX.Element;
+  displayName: string;
+
+  /** The list of chips within a ChipGroupField */
+  List: typeof ChipGroupFieldList;
+}
+
+/** Building blocks for constructing a ChipGroup field. */
+export const ChipGroupField = forwardRef(function ChipGroupField<
+  T extends object,
+>(props: ChipGroupFieldProps, ref: React.ForwardedRef<HTMLDivElement>) {
+  [props, ref] = useContextPropsV2(ChipGroupFieldContext, props, ref);
+
+  return (
+    <CollectionBuilder content={props.children}>
+      {(collection: BaseCollection<T>) => (
+        <ChipGroupFieldInner
+          {...props}
+          collection={collection}
+          chipGroupRef={ref}
+        />
+      )}
+    </CollectionBuilder>
+  );
+}) as unknown as ForwardedChipGroupField;
+
+ChipGroupField.displayName = "ChipFieldGroup";
+
+interface ChipGroupFieldInnerProps<T extends object>
+  extends ChipGroupFieldProps {
+  collection: BaseCollection<T>;
+  chipGroupRef: React.RefObject<HTMLDivElement>;
+}
+
+function ChipGroupFieldInner<T extends object>(
+  props: ChipGroupFieldInnerProps<T>
+) {
+  const { isInvalid, isDisabled, isRequired, chipGroupRef } = props;
+
+  const state = useListState({
+    ...props,
+    children: undefined,
+  });
+
+  const { gridProps, labelProps, descriptionProps, errorMessageProps } =
+    useTagGroup({ ...props, label: true }, state, chipGroupRef);
+
+  const renderProps = useRenderProps({
+    componentClassName: "aje-chip-group",
+    ...props,
+    selectors: {
+      "data-disabled": isDisabled,
+      "data-invalid": isInvalid,
+      "data-required": isRequired,
+    },
+  });
+
+  return (
+    <ChipGroupWrapper {...renderProps}>
+      <Provider
+        values={[
+          [FieldLabelContext.Provider, labelProps],
+          [FieldMessageContext.Provider, descriptionProps],
+          [FieldErrorContext.Provider, { ...errorMessageProps, isInvalid }],
+          [ChipGroupStateContext.Provider, state],
+          [ChipGroupListContext.Provider, { ...gridProps, ref: chipGroupRef }],
+          [ItemContext.Provider, { render: ChipLeaf }],
+        ]}
+      >
+        {renderProps.children}
+      </Provider>
+    </ChipGroupWrapper>
+  );
+}
+
+export interface ChipGroupFieldListProps<T>
+  extends CollectionBase<T>,
+    Omit<DOMAttributes<FocusableElement>, "children"> {
+  beforeChildren?: React.ReactNode;
+}
+
+interface ForwardedChipGroupFieldList {
+  <T>(
+    props: ChipGroupFieldListProps<T> & RefAttributes<HTMLDivElement>
+  ): JSX.Element;
+  displayName: string;
+}
+
+const ChipGroupFieldList = forwardRef(function ChipGroupFieldList<
+  T extends object,
+>(props: ChipGroupFieldListProps<T>, ref: React.ForwardedRef<HTMLDivElement>) {
+  [props, ref] = useContextPropsV2(ChipGroupListContext, props, ref);
+  const state = useContext(ChipGroupStateContext);
+
+  // We're building the collection in the ChipGroupField component
+  if (!state) return <Collection {...props} />;
+
+  // We're rendering a built collection
+  return <ChipGroupFieldListInner {...props} ref={ref} />;
+}) as ForwardedChipGroupFieldList;
+
+ChipGroupFieldList.displayName = "ChipGroupField.List";
+ChipGroupField.List = ChipGroupFieldList;
+
+const ChipGroupFieldListInner = forwardRef(function ChipGroupFieldListInner(
+  props: ChipGroupFieldListProps<any>,
+  ref: React.ForwardedRef<HTMLDivElement>
+) {
+  const { items, children, beforeChildren, ...rest } = props;
+  const state = useContext(ChipGroupStateContext)!;
+  const { CollectionRenderer } = useCollectionRenderer();
+
+  return (
+    <ChipsWrapper {...rest} ref={ref}>
+      {beforeChildren}
+      <CollectionRenderer
+        collection={state.collection as BaseCollection<object>}
+      />
+    </ChipsWrapper>
+  );
+});
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts
new file mode 100644
index 00000000..43abcc62
--- /dev/null
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts
@@ -0,0 +1,15 @@
+import { ListState } from "react-stately";
+import { createComponentContext } from "@utils/index";
+import {
+  ChipGroupFieldListProps,
+  ChipGroupFieldProps,
+} from "./ChipGroupField.component";
+import { createContext } from "react";
+
+export const ChipGroupFieldContext =
+  createComponentContext<ChipGroupFieldProps<any>>();
+
+export const ChipGroupStateContext = createContext<ListState<any> | null>(null);
+
+export const ChipGroupListContext =
+  createComponentContext<ChipGroupFieldListProps<any>>();
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.spec.tsx b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.spec.tsx
new file mode 100644
index 00000000..1067f6e8
--- /dev/null
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.spec.tsx
@@ -0,0 +1,118 @@
+import { describe, test, expect, vi } from "vitest";
+import { fireEvent, render, screen } from "@testing-library/react";
+import { ChipGroupField } from ".";
+import { Chip } from "@components/Chips/Chip";
+import { Label } from "../Atoms/Label";
+
+describe("ChipGroupField", () => {
+  const chips = [
+    { id: "1", rendered: "Chip 1" },
+    { id: "2", rendered: "Chip 2" },
+    { id: "3", rendered: "Chip 3" },
+  ];
+
+  test("matches snapshot", () => {
+    const result = render(
+      <ChipGroupField>
+        <ChipGroupField.List>
+          <Chip>Item 1</Chip>
+          <Chip>Item 2</Chip>
+          <Chip>Item 3</Chip>
+        </ChipGroupField.List>
+      </ChipGroupField>
+    );
+    expect(result.asFragment()).toMatchSnapshot();
+  });
+
+  test("renders label", () => {
+    render(
+      <ChipGroupField>
+        <Label>My Chip Group</Label>
+        <ChipGroupField.List>
+          <Chip>Item 1</Chip>
+          <Chip>Item 2</Chip>
+          <Chip>Item 3</Chip>
+        </ChipGroupField.List>
+      </ChipGroupField>
+    );
+
+    expect(screen.getByText("My Chip Group")).not.toBeNull();
+  });
+
+  test("renders chips", () => {
+    render(
+      <ChipGroupField>
+        <ChipGroupField.List items={chips}>
+          {({ rendered }) => <Chip>{rendered}</Chip>}
+        </ChipGroupField.List>
+      </ChipGroupField>
+    );
+
+    chips.forEach((chip) => {
+      expect(screen.getByText(chip.rendered)).not.toBeNull();
+    });
+  });
+
+  test("onRemove", () => {
+    const onRemove = vi.fn();
+
+    render(
+      <ChipGroupField onRemove={onRemove}>
+        <ChipGroupField.List items={chips}>
+          {({ rendered }) => <Chip>{rendered}</Chip>}
+        </ChipGroupField.List>
+      </ChipGroupField>
+    );
+
+    const buttons = screen.getAllByRole("button");
+    expect(buttons.length).toEqual(3);
+    const button = buttons[0];
+    fireEvent.click(button);
+
+    expect(onRemove).toHaveBeenCalledOnce();
+    expect(onRemove).toHaveBeenCalledWith(new Set(["1"]));
+  });
+
+  test("selection", () => {
+    const onSelectionChange = vi.fn();
+
+    render(
+      <ChipGroupField
+        selectionMode="multiple"
+        onSelectionChange={onSelectionChange}
+      >
+        <ChipGroupField.List items={chips}>
+          {({ rendered }) => <Chip>{rendered}</Chip>}
+        </ChipGroupField.List>
+      </ChipGroupField>
+    );
+
+    const rows = screen.getAllByRole("row");
+
+    rows.forEach((row, index) => {
+      fireEvent.focus(row);
+      fireEvent.keyDown(row, { key: "Enter", code: "Enter", charCode: 13 });
+      expect(onSelectionChange).toHaveBeenCalledTimes(index + 1);
+    });
+  });
+
+  test("disabledKeys", () => {
+    render(
+      <ChipGroupField disabledKeys={["1", "2"]}>
+        <ChipGroupField.List items={chips}>
+          {({ rendered }) => <Chip>{rendered}</Chip>}
+        </ChipGroupField.List>
+      </ChipGroupField>
+    );
+
+    const rows = screen.getAllByRole("row");
+
+    rows.forEach((row, index) => {
+      if (index > 3) {
+        expect(row.dataset["aria-disabled"]).toEqual("true");
+      } else {
+        expect(row.dataset["aria-disabled"]).toBeUndefined();
+      }
+    });
+  });
+});
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.stories.tsx b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.stories.tsx
new file mode 100644
index 00000000..12344d23
--- /dev/null
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.stories.tsx
@@ -0,0 +1,27 @@
+import { Meta, StoryObj } from "@storybook/react";
+import { ChipGroupField } from ".";
+import { Label } from "../Atoms/Label";
+import { Chip } from "@components/Chips/Chip";
+
+export default {
+  title: "Fields/ChipGroupField",
+  component: ChipGroupField,
+  parameters: {
+    layout: "centered",
+  },
+} as Meta<typeof ChipGroupField>;
+
+type Story = StoryObj<typeof ChipGroupField>;
+
+export const Primary: Story = {
+  args: {
+    children: [
+      <Label key="label">Label</Label>,
+      <ChipGroupField.List key="list">
+        <Chip>Chip 1</Chip>
+        <Chip>Chip 2</Chip>
+        <Chip>Chip 3</Chip>
+      </ChipGroupField.List>,
+    ],
+  },
+};
diff --git a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.styles.ts b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.styles.ts
similarity index 78%
rename from packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.styles.ts
rename to packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.styles.ts
index ec5ce386..f653ff58 100644
--- a/packages/atomic-elements/src/components/Chips/ChipGroup/ChipGroup.styles.ts
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.styles.ts
@@ -1,8 +1,7 @@
 import styled from "styled-components";
 import mixins from "../../../styles/mixins";
-import { FieldWrapper } from "../../Internal/FieldWrapper";
 
-export const ChipGroupWrapper = styled(FieldWrapper)`
+export const ChipGroupWrapper = styled.div`
   ${mixins.FieldStatus}
   ${mixins.SizingX}
 
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/__snapshots__/ChipGroupField.spec.tsx.snap b/packages/atomic-elements/src/components/Fields/ChipGroupField/__snapshots__/ChipGroupField.spec.tsx.snap
new file mode 100644
index 00000000..e26d2426
--- /dev/null
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/__snapshots__/ChipGroupField.spec.tsx.snap
@@ -0,0 +1,69 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`ChipGroupField > matches snapshot 1`] = `
+<DocumentFragment>
+  <div
+    class="sc-ovuCP euVikJ aje-chip-group"
+  >
+    <div
+      aria-atomic="false"
+      aria-labelledby="react-aria-:r1:"
+      aria-live="off"
+      aria-relevant="additions"
+      class="sc-eOzmre foFCjW"
+      id="react-aria-:r0:"
+      role="grid"
+      tabindex="0"
+    >
+      <div
+        aria-label="Item 1"
+        class="sc-fFoeYl kvHja-d aje-chip"
+        data-key="react-aria-1"
+        id="react-aria-:r0:-react-aria-1"
+        role="row"
+        tabindex="0"
+      >
+        <div
+          aria-colindex="1"
+          class="sc-eauhAA jzqQfF"
+          role="gridcell"
+        >
+          Item 1
+        </div>
+      </div>
+      <div
+        aria-label="Item 2"
+        class="sc-fFoeYl kvHja-d aje-chip"
+        data-key="react-aria-2"
+        id="react-aria-:r0:-react-aria-2"
+        role="row"
+        tabindex="0"
+      >
+        <div
+          aria-colindex="1"
+          class="sc-eauhAA jzqQfF"
+          role="gridcell"
+        >
+          Item 2
+        </div>
+      </div>
+      <div
+        aria-label="Item 3"
+        class="sc-fFoeYl kvHja-d aje-chip"
+        data-key="react-aria-3"
+        id="react-aria-:r0:-react-aria-3"
+        role="row"
+        tabindex="0"
+      >
+        <div
+          aria-colindex="1"
+          class="sc-eauhAA jzqQfF"
+          role="gridcell"
+        >
+          Item 3
+        </div>
+      </div>
+    </div>
+  </div>
+</DocumentFragment>
+`;
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/index.ts b/packages/atomic-elements/src/components/Fields/ChipGroupField/index.ts
new file mode 100644
index 00000000..b022ea3c
--- /dev/null
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/index.ts
@@ -0,0 +1,3 @@
+export { ChipGroupField } from "./ChipGroupField.component";
+export * from "./ChipGroupField.context";
+export type { ChipGroupFieldProps } from "./ChipGroupField.component";
diff --git a/packages/atomic-elements/src/components/index.ts b/packages/atomic-elements/src/components/index.ts
index 41d17ae5..e6535597 100644
--- a/packages/atomic-elements/src/components/index.ts
+++ b/packages/atomic-elements/src/components/index.ts
@@ -93,6 +93,7 @@ export {
 export { SelectField } from "./Fields/SelectField";
 export { MultiSelectField } from "./Fields/MultiSelectField";
 export { ComboBoxField } from "./Fields/ComboBoxField";
+export { ChipGroupField } from "./Fields/ChipGroupField";
 
 export { FormatDate } from "./Format/FormatDate";
 export { FormatNumber } from "./Format/FormatNumber";
@@ -179,6 +180,9 @@ export type {
   NumberFieldProps,
   ComboInputProps,
 } from "./Fields";
+export type { SelectFieldProps } from "./Fields/SelectField";
+export type { ComboBoxFieldProps } from "./Fields/ComboBoxField";
+export type { ChipGroupFieldProps } from "./Fields/ChipGroupField";
 export type { TableProps } from "./Layout/Table/Table.types";
 export type { TabsProps } from "./Layout/Tabs";
 export type { CardProps } from "./Layout/Card";

From 5fd8d09e29494e4dda875d596cdf8dfdee9d3c9b Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 14:43:56 -0700
Subject: [PATCH 02/11] Refactoring Calendar implementation

---
 .../Buttons/Button/Button.component.tsx       |   4 +-
 .../Buttons/Button/Button.context.ts          |   6 +-
 .../Calendar/Calendar.component.tsx           | 197 +++++-------------
 .../DateAndTimes/Calendar/Calendar.context.ts |   6 +
 .../DateAndTimes/Calendar/Calendar.styles.ts  |  77 +------
 .../DateAndTimes/Calendar/Calendar.types.ts   |   7 +
 .../Calendar/components/CalendarCell.tsx      |  62 ++++++
 .../Calendar/components/CalendarGrid.tsx      |  51 +++++
 .../Calendar/components/CalendarTitle.tsx     |  39 ++++
 .../atomic-elements/src/components/index.ts   |   6 +-
 .../atomic-elements/src/styles/variables.ts   |   7 +-
 playground/src/tabs/Aria.tsx                  |  22 +-
 12 files changed, 253 insertions(+), 231 deletions(-)
 create mode 100644 packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts
 create mode 100644 packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts
 create mode 100644 packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx
 create mode 100644 packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarGrid.tsx
 create mode 100644 packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx

diff --git a/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx b/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
index ae731454..6e27e53e 100644
--- a/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
+++ b/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
@@ -17,6 +17,7 @@ import { useButtonLink } from "@hooks/useButtonLink";
 import { useContextPropsV2 } from "@hooks/useContextProps";
 import { StyledButton } from "./Button.styles";
 import { ButtonContext } from "./Button.context";
+import { SlotProps } from "@hooks/useSlottedContext";
 
 export type ButtonVariants = SuggestStrings<
   | "primary"
@@ -41,7 +42,8 @@ export interface ButtonProps
   extends AriaButtonOptions<"button">,
     LoadingProps,
     RenderBaseProps<ButtonRenderProps>,
-    HasVariant<ButtonVariants> {
+    HasVariant<ButtonVariants>,
+    SlotProps {
   as?: "button" | "a";
   size?: ExtendedSize;
 }
diff --git a/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts b/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts
index 29a63a1c..d26caffd 100644
--- a/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts
+++ b/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts
@@ -1,8 +1,4 @@
-import React from "react";
 import { ButtonProps } from ".";
 import { createComponentContext } from "@utils/index";
-import { HasIcon } from '../../../types';
 
-export const ButtonContext = createComponentContext<
-  ButtonProps & HasIcon
->();
+export const ButtonContext = createComponentContext<ButtonProps>();
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
index 7d86a849..55217554 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
@@ -1,42 +1,28 @@
-import { useRef } from "react";
-import classNames from "classnames";
 import {
   useCalendar,
-  AriaCalendarGridProps,
   AriaCalendarProps,
   DateValue,
-  useCalendarGrid,
-  useCalendarCell,
 } from "@react-aria/calendar";
 import { useLocale } from "@react-aria/i18n";
-import {
-  getWeeksInMonth,
-  CalendarDate,
-  createCalendar,
-  today,
-  getLocalTimeZone,
-} from "@internationalized/date";
-import { CalendarState, useCalendarState } from "react-stately";
+import { createCalendar } from "@internationalized/date";
+import { useCalendarState } from "react-stately";
 
-import { IconButton } from "../../../Buttons/IconButton";
-import { Button } from "../../../Buttons/Button";
-import { AriaProps, BaseProps, Size } from "../../../../types";
-import {
-  CalendarContent,
-  StyledCalendarCell,
-  CalendarFooter,
-  CalendarHeader,
-  CalendarWrapper,
-  CalendarCellButton,
-  CalendarWeekDay,
-  CalendarHeaderButtons,
-  CalendarHeaderTitle,
-} from "./Calendar.styles";
+import { ExtendedSize, RenderBaseProps } from "../../../../types";
+import { useRenderProps } from "@hooks";
+import { Provider } from "@components/Internal/Provider";
+import { ButtonContext } from "@components/Buttons/Button/Button.context";
+import { DEFAULT_SLOT } from "@hooks/useSlottedContext";
+import { ErrorMessageContext } from "@components/Fields/Atoms/ErrorMessage";
+import { CalendarWrapper } from "./Calendar.styles";
+import { CalendarStateContext, CalendarTitleContext } from "./Calendar.context";
+import { CalendarTitle } from "./components/CalendarTitle";
+import { CalendarGrid } from "./components/CalendarGrid";
+import { CalendarCell } from "./components/CalendarCell";
 
 export interface CalendarProps<T extends DateValue>
-  extends AriaProps<AriaCalendarProps<T>>,
-    BaseProps {
-  size?: Size;
+  extends AriaCalendarProps<T>,
+    RenderBaseProps<never> {
+  size?: ExtendedSize;
 }
 
 /**
@@ -61,130 +47,45 @@ export function Calendar<T extends DateValue>(props: CalendarProps<T>) {
     createCalendar,
   });
 
-  const { calendarProps, prevButtonProps, nextButtonProps, title } =
-    useCalendar(props, state);
-
-  const className = classNames("aje-calendar", `is-${size}`);
-
-  return (
-    <CalendarWrapper {...calendarProps} className={className}>
-      <CalendarHeader className="aje-calendar__header">
-        <CalendarHeaderTitle>{title}</CalendarHeaderTitle>
-        <CalendarHeaderButtons>
-          <IconButton
-            {...prevButtonProps}
-            icon="chevron_left"
-            variant="content"
-          />
-          <IconButton
-            {...nextButtonProps}
-            icon="chevron_right"
-            variant="content"
-          />
-        </CalendarHeaderButtons>
-      </CalendarHeader>
-      <CalendarGrid state={state} />
-      <CalendarFooter className="aje-calendar__footer">
-        {/* TODO: useCalendar doesn't appear to have the ability to set the value
-        to null, check if there's a way to do this? */}
-        {/* <Button
-          variant="calendar-footer"
-          onPress={() => {
-            state.setValue(null);
-          }}
-        >
-          Clear
-        </Button> */}
-        <Button
-          variant="calendar-footer"
-          onPress={() => {
-            const date = today(getLocalTimeZone());
-            state.setValue(date);
-            state.setFocusedDate(date);
-          }}
-        >
-          Today
-        </Button>
-      </CalendarFooter>
-    </CalendarWrapper>
-  );
-}
-
-interface CalendarGridProps extends AriaCalendarGridProps {
-  state: CalendarState;
-}
-
-function CalendarGrid(props: CalendarGridProps) {
-  const { state, ...rest } = props;
-  const { locale } = useLocale();
-  const { gridProps, headerProps, weekDays } = useCalendarGrid(rest, state);
-
-  // Get the number of weeks in the month so we can render the proper number of rows.
-  const weeksInMonth = getWeeksInMonth(state.visibleRange.start, locale);
-
-  return (
-    <CalendarContent {...gridProps} className="aje-calendar__content">
-      <thead {...headerProps}>
-        <tr>
-          {weekDays.map((day, index) => (
-            <CalendarWeekDay key={index}>{day}</CalendarWeekDay>
-          ))}
-        </tr>
-      </thead>
-      <tbody>
-        {[...new Array(weeksInMonth).keys()].map((weekIndex) => (
-          <tr key={weekIndex}>
-            {state
-              .getDatesInWeek(weekIndex)
-              .map((date, i) =>
-                date ? (
-                  <CalendarCell key={i} state={state} date={date} />
-                ) : (
-                  <td key={i} />
-                )
-              )}
-          </tr>
-        ))}
-      </tbody>
-    </CalendarContent>
-  );
-}
-
-interface CalendarCellProps {
-  state: CalendarState;
-  date: CalendarDate;
-}
-
-function CalendarCell(props: CalendarCellProps) {
-  const { state, date } = props;
-  const ref = useRef(null);
   const {
-    cellProps,
-    buttonProps,
-    isSelected,
-    isOutsideVisibleRange,
-    isDisabled,
-    isUnavailable,
-    formattedDate,
-  } = useCalendarCell({ date }, state, ref);
+    calendarProps,
+    prevButtonProps,
+    nextButtonProps,
+    title,
+    errorMessageProps,
+  } = useCalendar(props, state);
 
-  const cellClassName = classNames("aje-calendar__cell", {
-    "is-selected": isSelected,
-    "is-disabled": isDisabled,
-    "is-unavailable": isUnavailable,
+  const renderProps = useRenderProps({
+    componentClassName: "aje-calendar",
+    size,
+    ...props,
   });
 
   return (
-    <StyledCalendarCell {...cellProps} className={cellClassName}>
-      <CalendarCellButton
-        {...buttonProps}
-        ref={ref}
-        hidden={isOutsideVisibleRange}
+    <CalendarWrapper {...calendarProps} {...renderProps}>
+      <Provider
+        values={[
+          [CalendarStateContext.Provider, state],
+          [CalendarTitleContext.Provider, { title }],
+          [
+            ButtonContext.Provider,
+            {
+              slots: {
+                [DEFAULT_SLOT]: {},
+                "previous-month": prevButtonProps,
+                "next-month": nextButtonProps,
+              },
+            },
+          ],
+          [ErrorMessageContext.Provider, errorMessageProps],
+        ]}
       >
-        {formattedDate}
-      </CalendarCellButton>
-    </StyledCalendarCell>
+        {renderProps.children}
+      </Provider>
+    </CalendarWrapper>
   );
 }
 
-export default Calendar;
+Calendar.Title = CalendarTitle;
+Calendar.Grid = CalendarGrid;
+Calendar.Cell = CalendarCell;
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts
new file mode 100644
index 00000000..57410927
--- /dev/null
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts
@@ -0,0 +1,6 @@
+import { createContext } from 'react';
+import { CalendarState } from 'react-stately';
+
+export const CalendarStateContext = createContext<CalendarState | null>(null);
+
+export const CalendarTitleContext = createContext<{ title: string } | null>(null);
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts
index 56ff6ee2..e545fa89 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts
@@ -43,47 +43,16 @@ export const CalendarWrapper = styled.div`
   }
 `;
 
-export const CalendarHeader = styled.div`
-  display: grid;
-  grid-template-columns: 1.1fr 1fr;
-  margin: 0 15px;
-  margin-bottom: 5px;
-`;
-
 export const CalendarHeaderTitle = styled.h2`
   ${mixins.Bold}
   font-size: var(--font-size);
   color: var(--neutral600);
 `;
 
-export const CalendarHeaderButtons = styled.div`
-  display: flex;
-  align-items: center;
-  justify-content: space-between;
-`;
-
 export const CalendarContent = styled.table`
   height: var(--calendar-content-height);
   min-width: var(--calendar-content-height);
-`;
-
-export const CalendarFooter = styled.div`
-  display: flex;
-  align-items: center;
-  justify-content: flex-end;
-  margin-top: auto;
-
-  .aje-btn--calendar-footer {
-    border: none;
-    background-color: transparent;
-    color: var(--primary600);
-    padding: 5px;
-
-    &:hover {
-      cursor: pointer;
-      color: var(--primary800);
-    }
-  }
+  margin: auto;
 `;
 
 export const CalendarWeekDay = styled.th`
@@ -101,18 +70,18 @@ export const StyledCalendarCell = styled.td`
   color: var(--text-clr);
   height: var(--calendar-cell-size);
 
-  &.is-selected {
-    background-color: var(--time-selected-back-clr);
-    color: var(--time-selected-text-clr);
+  &[aria-selected="true"] {
+    background-color: var(--calendar-selected-bg-clr);
+    color: var(--calendar-selected-text-clr);
     border-radius: 3px;
 
     &:hover {
-      background-color: var(--time-selected-back-clr);
+      background-color: var(--calendar-selected-bg-clr);
     }
   }
 
   &[aria-invalid="true"] {
-    background-color: var(--error500);
+    background-color: var(--error-clr);
   }
 
   &:hover {
@@ -121,7 +90,7 @@ export const StyledCalendarCell = styled.td`
     background-color: var(--neutral200);
   }
 
-  &.is-disabled {
+  &[data-disabled] {
     color: var(--neutral400);
 
     &:hover {
@@ -129,32 +98,6 @@ export const StyledCalendarCell = styled.td`
       background-color: transparent;
     }
   }
-
-  /*
-
-  &.in-interval {
-    border-radius: 0px;
-    background-color: var(--neutral200);
-  }
-
-  &.in-potential-interval {
-    border-top: 1px dashed var(--neutral400);
-    border-bottom: 1px dashed var(--neutral400);
-  }
-
-  &.neighboring-month {
-    color: var(--neutral400);
-  }
-
-  &.interval-start {
-    background-color: var(--primary700);
-    border-radius: 5px 0px 0px 5px;
-  }
-
-  &.interval-end {
-    background-color: var(--primary700);
-    border-radius: 0px 5px 5px 0px;
-  } */
 `;
 
 export const CalendarCellButton = styled.div`
@@ -163,10 +106,4 @@ export const CalendarCellButton = styled.div`
   display: flex;
   justify-content: center;
   align-items: center;
-
-  &[hidden] {
-    visibility: hidden;
-    display: auto;
-    pointer-events: none;
-  }
 `;
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts
new file mode 100644
index 00000000..c230efc2
--- /dev/null
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts
@@ -0,0 +1,7 @@
+import { CalendarDate } from '@internationalized/date';
+
+export interface CalendarRenderProps {
+  focusedDate: CalendarDate;
+  isDisabled: boolean;
+  isFocused: boolean;
+}
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx
new file mode 100644
index 00000000..1f04b329
--- /dev/null
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx
@@ -0,0 +1,62 @@
+import { CalendarDate } from "@internationalized/date";
+import { useContext, useRef } from "react";
+import { CalendarStateContext } from "../Calendar.context";
+import { useCalendarCell } from "@react-aria/calendar";
+import { CalendarCellButton, StyledCalendarCell } from "../Calendar.styles";
+import { useRenderProps } from "@hooks";
+import { RenderBaseProps } from "../../../../../types";
+import { CalendarRenderProps } from "../Calendar.types";
+
+interface CalendarCellRenderProps {
+  focusedDate: CalendarDate;
+  isDisabled: boolean;
+  isSelected: boolean;
+  formattedDate: string;
+}
+
+interface CalendarCellProps extends RenderBaseProps<CalendarCellRenderProps> {
+  date: CalendarDate;
+  isDisabled?: boolean;
+}
+
+export function CalendarCell(props: CalendarCellProps) {
+  const state = useContext(CalendarStateContext)!;
+  const ref = useRef(null);
+
+  const {
+    cellProps,
+    buttonProps,
+    isSelected,
+    isOutsideVisibleRange,
+    isDisabled,
+    isUnavailable,
+    formattedDate,
+  } = useCalendarCell(props, state, ref);
+
+  const renderProps = useRenderProps({
+    componentClassName: "aje-calendar__cell",
+    ...props,
+    values: {
+      focusedDate: state.focusedDate,
+      formattedDate,
+      isDisabled,
+      isSelected,
+    },
+    selectors: {
+      "data-disabled": isDisabled,
+      "data-unavailable": isUnavailable,
+    },
+  });
+
+  return (
+    <StyledCalendarCell {...cellProps} {...renderProps}>
+      <CalendarCellButton
+        {...buttonProps}
+        ref={ref}
+        hidden={isOutsideVisibleRange}
+      >
+        {formattedDate}
+      </CalendarCellButton>
+    </StyledCalendarCell>
+  );
+}
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarGrid.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarGrid.tsx
new file mode 100644
index 00000000..e7ab03a7
--- /dev/null
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarGrid.tsx
@@ -0,0 +1,51 @@
+import React, { useContext } from "react";
+import { AriaCalendarGridProps, useCalendarGrid } from "@react-aria/calendar";
+import { CalendarStateContext } from "../Calendar.context";
+import { useLocale } from "@react-aria/i18n";
+import { CalendarDate, getWeeksInMonth } from "@internationalized/date";
+import { CalendarContent, CalendarWeekDay } from "../Calendar.styles";
+import { RenderStyleProps } from "../../../../../types";
+
+interface CalendarGridProps
+  extends AriaCalendarGridProps,
+    RenderStyleProps<never> {
+  children: (date: CalendarDate) => React.ReactNode;
+}
+
+export function CalendarGrid(props: CalendarGridProps) {
+  const state = useContext(CalendarStateContext)!;
+  const { locale } = useLocale();
+  const { gridProps, headerProps, weekDays } = useCalendarGrid(props, state);
+
+  // Get the number of weeks in the month so we can render the proper number of rows.
+  const weeksInMonth = getWeeksInMonth(state.visibleRange.start, locale);
+
+  return (
+    <CalendarContent {...gridProps} className="aje-calendar__content">
+      <thead {...headerProps}>
+        <tr>
+          {weekDays.map((day, index) => (
+            <CalendarWeekDay key={index}>{day}</CalendarWeekDay>
+          ))}
+        </tr>
+      </thead>
+      <tbody>
+        {[...new Array(weeksInMonth).keys()].map((weekIndex) => (
+          <tr key={weekIndex}>
+            {state
+              .getDatesInWeek(weekIndex)
+              .map((date, i) =>
+                date ? (
+                  <React.Fragment key={i}>
+                    {props.children(date)}
+                  </React.Fragment>
+                ) : (
+                  <td key={i} />
+                )
+              )}
+          </tr>
+        ))}
+      </tbody>
+    </CalendarContent>
+  );
+}
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx
new file mode 100644
index 00000000..a3e14aad
--- /dev/null
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx
@@ -0,0 +1,39 @@
+import { useContext } from "react";
+import { RenderBaseProps } from "../../../../../types";
+import { useRenderProps } from "@hooks";
+
+import {
+  CalendarStateContext,
+  CalendarTitleContext,
+} from "../Calendar.context";
+import { CalendarHeaderTitle } from "../Calendar.styles";
+import { CalendarRenderProps } from "../Calendar.types";
+
+interface CalendarTitleRenderProps extends CalendarRenderProps {
+  title: string;
+}
+
+interface CalendarTitleProps
+  extends RenderBaseProps<CalendarTitleRenderProps> {}
+
+export function CalendarTitle(props: CalendarTitleProps) {
+  const state = useContext(CalendarStateContext)!;
+  const { title } = useContext(CalendarTitleContext)!;
+
+  const renderProps = useRenderProps({
+    componentClassName: "aje-calendar__title",
+    values: {
+      focusedDate: state.focusedDate,
+      isDisabled: state.isDisabled,
+      isFocused: state.isFocused,
+      title: title,
+    },
+    ...props,
+  });
+
+  return (
+    <CalendarHeaderTitle {...renderProps}>
+      {props.children ? renderProps.children : title}
+    </CalendarHeaderTitle>
+  );
+}
diff --git a/packages/atomic-elements/src/components/index.ts b/packages/atomic-elements/src/components/index.ts
index 41d17ae5..03aaaab8 100644
--- a/packages/atomic-elements/src/components/index.ts
+++ b/packages/atomic-elements/src/components/index.ts
@@ -125,11 +125,7 @@ export { LiveMessage } from "./Accessibility/LiveMessage";
 export type { BannerProps, BannerVariants } from "./Banners/Banner";
 export type { ButtonProps } from "./Buttons/Button";
 export type { IconButtonProps } from "./Buttons/IconButton";
-export type {
-  ButtonGroupProps,
-  ButtonGroupChild,
-  ButtonGroupChildProps,
-} from "./Buttons/ButtonGroup";
+export type { ButtonGroupProps } from "./Buttons/ButtonGroup";
 export type { ComboBoxProps } from "./Dropdowns/Combobox";
 export type { CustomSelectProps } from "./Dropdowns/CustomSelect";
 export type { SelectProps, SelectValue } from "./Dropdowns/Select";
diff --git a/packages/atomic-elements/src/styles/variables.ts b/packages/atomic-elements/src/styles/variables.ts
index 3c051b5c..50075077 100644
--- a/packages/atomic-elements/src/styles/variables.ts
+++ b/packages/atomic-elements/src/styles/variables.ts
@@ -242,11 +242,16 @@ export const CssVariables = createGlobalStyle`
   --choose-label-padding-left: 32px;
   --choose-label-height: 24px;
 
-  /* # Calendar */
+  /* # TimePicker */
   --time-clr: var(--neutral200);
   --time-selected-back-clr: var(--primary600);
   --time-selected-text-clr: var(--neutral100);
 
+  /* # Calendar */
+  --calendar-clr: var(--neutral200);
+  --calendar-selected-bg-clr: var(--accent-clr);
+  --calendar-selected-text-clr: var(--neutral100);
+
   /* # Card */
   --card-bg-clr: var(--background-clr-alt);
   --card-text-clr: var(--text-clr);
diff --git a/playground/src/tabs/Aria.tsx b/playground/src/tabs/Aria.tsx
index 7d77cf09..3d5c27c2 100644
--- a/playground/src/tabs/Aria.tsx
+++ b/playground/src/tabs/Aria.tsx
@@ -1,5 +1,25 @@
 import React from "react";
+import {
+  Calendar,
+  Flex,
+  IconButton,
+} from "../../../packages/atomic-elements/src/index";
 
 export default function Aria() {
-  return <div>Hi there</div>;
+  return (
+    <div style={{ width: "320px", border: "1px solid black" }}>
+      <Calendar>
+        <Flex gap={20} alignItems="center" justifyContent="space-between">
+          <IconButton
+            slot="previous-month"
+            icon="chevron_left"
+            variant="ghost"
+          />
+          <Calendar.Title />
+          <IconButton slot="next-month" icon="chevron_right" variant="ghost" />
+        </Flex>
+        <Calendar.Grid>{(date) => <Calendar.Cell date={date} />}</Calendar.Grid>
+      </Calendar>
+    </div>
+  );
 }

From 368f8f0e2b1de0fe23081967c215e56d72a8e3e6 Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 14:57:07 -0700
Subject: [PATCH 03/11] Fixed TS builds

---
 .../Buttons/Button/Button.component.tsx       |  8 ++--
 .../Buttons/Button/Button.context.ts          |  4 +-
 .../IconButton/IconButton.component.tsx       |  2 +-
 .../ErrorMessage/ErrorMessage.context.ts      |  4 +-
 .../Fields/Atoms/Message/Message.context.ts   |  2 +-
 .../FormatTable/FormatTable.component.tsx     |  6 ++-
 .../Layout/Table/components/TableFooter.tsx   | 37 ---------------
 .../ConfirmationModal.component.tsx           |  3 +-
 .../ErrorModal/ErrorModal.component.tsx       |  2 +-
 .../Overlays/Modal/Modal.component.tsx        | 47 ++++++++++---------
 .../Overlays/Modal/Modal.context.ts           |  3 +-
 .../atomic-elements/src/components/index.ts   |  6 +--
 12 files changed, 42 insertions(+), 82 deletions(-)
 delete mode 100644 packages/atomic-elements/src/components/Layout/Table/components/TableFooter.tsx

diff --git a/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx b/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
index ae731454..74f6867c 100644
--- a/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
+++ b/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
@@ -5,7 +5,6 @@ import { mergeProps, useObjectRef } from "@react-aria/utils";
 import { SpinnerLoader } from "../../Loaders/SpinnerLoader";
 import {
   ExtendedSize,
-  HasIcon,
   HasVariant,
   LoadingProps,
   RenderBaseProps,
@@ -17,6 +16,7 @@ import { useButtonLink } from "@hooks/useButtonLink";
 import { useContextPropsV2 } from "@hooks/useContextProps";
 import { StyledButton } from "./Button.styles";
 import { ButtonContext } from "./Button.context";
+import { SlotProps } from "@hooks/useSlottedContext";
 
 export type ButtonVariants = SuggestStrings<
   | "primary"
@@ -41,7 +41,8 @@ export interface ButtonProps
   extends AriaButtonOptions<"button">,
     LoadingProps,
     RenderBaseProps<ButtonRenderProps>,
-    HasVariant<ButtonVariants> {
+    HasVariant<ButtonVariants>,
+    SlotProps {
   as?: "button" | "a";
   size?: ExtendedSize;
 }
@@ -54,8 +55,7 @@ export const Button = forwardRef<HTMLButtonElement, ButtonProps>(
   function Button(props, forwardedRef) {
     [props, forwardedRef] = useContextPropsV2(
       ButtonContext,
-      // Button doesn't have Icon props, but the context does
-      props as ButtonProps & HasIcon,
+      props,
       forwardedRef
     );
 
diff --git a/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts b/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts
index 29a63a1c..64d0ddf6 100644
--- a/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts
+++ b/packages/atomic-elements/src/components/Buttons/Button/Button.context.ts
@@ -1,8 +1,8 @@
 import React from "react";
 import { ButtonProps } from ".";
 import { createComponentContext } from "@utils/index";
-import { HasIcon } from '../../../types';
+import { CanHaveIcon } from '../../../types';
 
 export const ButtonContext = createComponentContext<
-  ButtonProps & HasIcon
+  ButtonProps & CanHaveIcon
 >();
diff --git a/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx b/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx
index 3973b9a9..51e94414 100644
--- a/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx
+++ b/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx
@@ -18,7 +18,7 @@ export interface IconButtonProps
 export const IconButton = forwardRef<HTMLButtonElement, IconButtonProps>(
   function IconButton(props, forwardedRef) {
     [props, forwardedRef] = useContextPropsV2(
-      ButtonContext,
+      ButtonContext as any,
       props,
       forwardedRef
     );
diff --git a/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.context.ts b/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.context.ts
index 3502f7bd..1d62605c 100644
--- a/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.context.ts
+++ b/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.context.ts
@@ -1,6 +1,4 @@
 import { createComponentContext } from "@utils/index";
 import { ErrorMessageProps } from "./ErrorMessage.component";
 
-export const ErrorMessageContext = createComponentContext<ErrorMessageProps>(
-  {} as any
-);
+export const ErrorMessageContext = createComponentContext<ErrorMessageProps>();
diff --git a/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.context.ts b/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.context.ts
index e5dfc1e0..cd21091b 100644
--- a/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.context.ts
+++ b/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.context.ts
@@ -1,4 +1,4 @@
 import { createComponentContext } from "@utils/index";
 import { MessageProps } from "./Message.component";
 
-export const MessageContext = createComponentContext<MessageProps>({});
+export const MessageContext = createComponentContext<MessageProps>();
diff --git a/packages/atomic-elements/src/components/Format/FormatTable/FormatTable.component.tsx b/packages/atomic-elements/src/components/Format/FormatTable/FormatTable.component.tsx
index 8988a68a..3d7c7f7e 100644
--- a/packages/atomic-elements/src/components/Format/FormatTable/FormatTable.component.tsx
+++ b/packages/atomic-elements/src/components/Format/FormatTable/FormatTable.component.tsx
@@ -20,8 +20,10 @@ export function FormatTable(props: FormatTableProps) {
       </Table.Header>
       <Table.Body items={value || []}>
         {(item) => (
-          <Table.Row>
-            {(colKey) => <Table.Cell key={colKey}>{item[colKey]}</Table.Cell>}
+          <Table.Row columns={columns}>
+            {(column) => (
+              <Table.Cell key={column.key}>{item[column.key]}</Table.Cell>
+            )}
           </Table.Row>
         )}
       </Table.Body>
diff --git a/packages/atomic-elements/src/components/Layout/Table/components/TableFooter.tsx b/packages/atomic-elements/src/components/Layout/Table/components/TableFooter.tsx
deleted file mode 100644
index cad5f0c1..00000000
--- a/packages/atomic-elements/src/components/Layout/Table/components/TableFooter.tsx
+++ /dev/null
@@ -1,37 +0,0 @@
-import { TableRowGroup } from "./TableRowGroup";
-import { StyledTableFooter } from "../Table.styles";
-import { TableState, TreeGridState } from "../Table.types";
-import { TableRow } from "./TableRow";
-import { TableCell } from "./TableCell";
-import { useRenderProps } from "@hooks";
-
-interface TableFooterProps<T> {
-  state: TableState<T> | TreeGridState<T>;
-}
-
-export function TableFooter<T>(props: TableFooterProps<T>) {
-  const { state } = props;
-
-  const { collection } = state;
-
-  const rows = [...collection.footer!.childNodes];
-
-  const renderProps = useRenderProps({
-    componentClassName: "aje-table__footer",
-    ...collection.body.props,
-  });
-
-  return (
-    <TableRowGroup type={StyledTableFooter} {...renderProps}>
-      {rows.map((row) => {
-        return (
-          <TableRow item={row} state={state} key={row.key}>
-            {[...collection.getChildren!(row.key)].map((cell) => {
-              return <TableCell key={cell.key} cell={cell} state={state} />;
-            })}
-          </TableRow>
-        );
-      })}
-    </TableRowGroup>
-  );
-}
diff --git a/packages/atomic-elements/src/components/Overlays/ConfirmationModal/ConfirmationModal.component.tsx b/packages/atomic-elements/src/components/Overlays/ConfirmationModal/ConfirmationModal.component.tsx
index 5057af4f..245c0f02 100644
--- a/packages/atomic-elements/src/components/Overlays/ConfirmationModal/ConfirmationModal.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/ConfirmationModal/ConfirmationModal.component.tsx
@@ -3,8 +3,7 @@ import { Button } from "../../Buttons/Button";
 import { Modal, BaseModalProps } from "../Modal";
 import { OverlayTriggerStateContext } from "../OverlayTrigger/context";
 
-export interface ConfirmationModalProps
-  extends Omit<BaseModalProps, "children"> {
+export interface ConfirmationModalProps extends BaseModalProps {
   children: React.ReactNode;
   /** Must include a title. Titles are always in Title case. */
   title: string;
diff --git a/packages/atomic-elements/src/components/Overlays/ErrorModal/ErrorModal.component.tsx b/packages/atomic-elements/src/components/Overlays/ErrorModal/ErrorModal.component.tsx
index da6fdfe4..2af48a21 100644
--- a/packages/atomic-elements/src/components/Overlays/ErrorModal/ErrorModal.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/ErrorModal/ErrorModal.component.tsx
@@ -3,7 +3,7 @@ import { Button } from "../../Buttons/Button";
 import { MaterialIcon } from "../../Icons/MaterialIcon";
 import { Modal, BaseModalProps } from "../Modal";
 
-export interface ErrorModalProps extends Omit<BaseModalProps, "children"> {
+export interface ErrorModalProps extends BaseModalProps {
   children: React.ReactNode;
   /** Must include a title. Titles are always in Title case. */
   title: string;
diff --git a/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx b/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx
index fa83f3bd..c275ef28 100644
--- a/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx
@@ -9,8 +9,7 @@ import {
   OverlayTriggerState,
   useOverlayTriggerState,
 } from "react-stately";
-import { useVariantClass } from "../../../hooks/variants";
-import { HasClassName } from "../../../types";
+import { RenderStyleProps } from "../../../types";
 import {
   ModalBackground,
   ModalWrapper,
@@ -20,52 +19,46 @@ import {
   ModalTitle,
 } from "./Modal.styles";
 import classNames from "classnames";
-import { useContextProps } from "../../../hooks/useContextProps";
+import { useContextPropsV2 } from "../../../hooks/useContextProps";
 import { ModalContext } from "./Modal.context";
 import { OverlayTriggerStateContext } from "../OverlayTrigger/context";
+import { useRenderProps } from "@hooks";
+import { filterDOMProps } from "@react-aria/utils";
 
 type CloseModal = () => void;
-export interface BaseModalProps extends HasClassName, OverlayTriggerProps {
-  id?: string;
-  children?: React.ReactNode | ((close: CloseModal) => React.ReactNode);
 
+export interface BaseModalProps
+  extends RenderStyleProps<never>,
+    OverlayTriggerProps {
+  id?: string;
   /** Centers the modal within the viewport */
   centered?: boolean;
 }
 
 export interface ModalProps extends BaseModalProps {
   variant?: string;
+  children?: React.ReactNode | ((close: CloseModal) => React.ReactNode);
 }
 
 /**
  * Modal Component to render content overlayed on top of the page content.
  */
 export function Modal(props: ModalProps) {
-  const {
-    children,
-    centered = false,
-    variant = "default",
-    className,
-    ...rest
-  } = useContextProps(ModalContext, props);
+  let ref = useRef(null);
+  [props, ref] = useContextPropsV2(ModalContext, props, ref);
+
+  const { children, centered = false, variant = "default", ...rest } = props;
 
   const contextState = useContext(OverlayTriggerStateContext);
   const localState = useOverlayTriggerState(props);
   const state = contextState ?? localState;
 
-  const variantClassName = useVariantClass("aje-modal", variant);
-
   if (!state.isOpen) {
     return null;
   }
 
   return (
-    <ModalOverlay
-      {...rest}
-      state={state}
-      className={classNames(className, variantClassName)}
-      centered={centered}
-    >
+    <ModalOverlay state={state} centered={centered} variant={variant} {...rest}>
       {children}
     </ModalOverlay>
   );
@@ -73,13 +66,20 @@ export function Modal(props: ModalProps) {
 
 interface ModalOverlayProps extends BaseModalProps, AriaModalOverlayProps {
   state: OverlayTriggerState;
+  children: React.ReactNode | ((close: CloseModal) => React.ReactNode);
+  variant?: string;
 }
 
 function ModalOverlay(props: ModalOverlayProps) {
-  const { children, state, centered, className } = props;
+  const { children, state, centered } = props;
   const ref = useRef(null);
   const { modalProps, underlayProps } = useModalOverlay(props, state, ref);
 
+  const renderProps = useRenderProps({
+    componentClassName: "aje-modal",
+    ...props,
+  });
+
   return (
     <Overlay>
       <ModalBackground
@@ -87,7 +87,8 @@ function ModalOverlay(props: ModalOverlayProps) {
         className={classNames({ "is-centered": centered })}
       >
         <ModalWrapper
-          className={classNames("aje-modal", className)}
+          {...renderProps}
+          {...filterDOMProps(props)}
           onClick={(e: React.MouseEvent<HTMLDivElement>) => e.stopPropagation()}
           ref={ref}
           id={props.id}
diff --git a/packages/atomic-elements/src/components/Overlays/Modal/Modal.context.ts b/packages/atomic-elements/src/components/Overlays/Modal/Modal.context.ts
index 0a74de63..b750c048 100644
--- a/packages/atomic-elements/src/components/Overlays/Modal/Modal.context.ts
+++ b/packages/atomic-elements/src/components/Overlays/Modal/Modal.context.ts
@@ -1,4 +1,5 @@
 import React from "react";
 import { ModalProps } from ".";
+import { createComponentContext } from '@utils/index';
 
-export const ModalContext = React.createContext<ModalProps | null>(null);
+export const ModalContext = createComponentContext<ModalProps>();
diff --git a/packages/atomic-elements/src/components/index.ts b/packages/atomic-elements/src/components/index.ts
index 41d17ae5..03aaaab8 100644
--- a/packages/atomic-elements/src/components/index.ts
+++ b/packages/atomic-elements/src/components/index.ts
@@ -125,11 +125,7 @@ export { LiveMessage } from "./Accessibility/LiveMessage";
 export type { BannerProps, BannerVariants } from "./Banners/Banner";
 export type { ButtonProps } from "./Buttons/Button";
 export type { IconButtonProps } from "./Buttons/IconButton";
-export type {
-  ButtonGroupProps,
-  ButtonGroupChild,
-  ButtonGroupChildProps,
-} from "./Buttons/ButtonGroup";
+export type { ButtonGroupProps } from "./Buttons/ButtonGroup";
 export type { ComboBoxProps } from "./Dropdowns/Combobox";
 export type { CustomSelectProps } from "./Dropdowns/CustomSelect";
 export type { SelectProps, SelectValue } from "./Dropdowns/Select";

From 490cd8d25ee179bf27552cd6bf6a386890b6e993 Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:01:11 -0700
Subject: [PATCH 04/11] Misc updates

---
 .../Calendar/Calendar.component.tsx           |  40 +++++-
 .../DateAndTimes/Calendar/Calendar.context.ts |  13 +-
 .../Calendar/Calendar.stories.tsx             | 126 ++++++++++++++++--
 .../DateAndTimes/Calendar/Calendar.styles.ts  |   6 +
 .../TimeInput/TimeInput.component.tsx         |   1 +
 .../TimeInput/TimeInput.stories.tsx           |   2 +-
 6 files changed, 165 insertions(+), 23 deletions(-)

diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
index 55217554..5d743877 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
@@ -1,3 +1,4 @@
+import { useContext, useRef } from "react";
 import {
   useCalendar,
   AriaCalendarProps,
@@ -7,22 +8,30 @@ import { useLocale } from "@react-aria/i18n";
 import { createCalendar } from "@internationalized/date";
 import { useCalendarState } from "react-stately";
 
-import { ExtendedSize, RenderBaseProps } from "../../../../types";
+import { AriaProps, ExtendedSize, RenderBaseProps } from "../../../../types";
 import { useRenderProps } from "@hooks";
 import { Provider } from "@components/Internal/Provider";
 import { ButtonContext } from "@components/Buttons/Button/Button.context";
 import { DEFAULT_SLOT } from "@hooks/useSlottedContext";
+import { useContextPropsV2 } from "@hooks/useContextProps";
 import { ErrorMessageContext } from "@components/Fields/Atoms/ErrorMessage";
-import { CalendarWrapper } from "./Calendar.styles";
-import { CalendarStateContext, CalendarTitleContext } from "./Calendar.context";
+
 import { CalendarTitle } from "./components/CalendarTitle";
 import { CalendarGrid } from "./components/CalendarGrid";
 import { CalendarCell } from "./components/CalendarCell";
+import { CalendarWrapper } from "./Calendar.styles";
+import {
+  CalendarContext,
+  CalendarStateContext,
+  CalendarTitleContext,
+} from "./Calendar.context";
 
 export interface CalendarProps<T extends DateValue>
-  extends AriaCalendarProps<T>,
+  extends AriaProps<AriaCalendarProps<T>>,
     RenderBaseProps<never> {
   size?: ExtendedSize;
+  isDisabled?: boolean;
+  isInvalid?: boolean;
 }
 
 /**
@@ -36,9 +45,20 @@ export interface CalendarProps<T extends DateValue>
  * import { parseDate } from "@internationalized/date";
  *
  * const [date, setDate] = useState(parseDate("2021-01-01"));
- * <Calendar value={date} onChange={setDate} />
+ *
+ * <Calendar value={date} onChange={setDate}>
+ *   <Calendar.Title />
+ *   <Calendar.Grid>
+ *     {(date) => <Calendar.Cell date={date} />}
+ *   </Calendar.Grid>
+ * </Calendar>
  */
 export function Calendar<T extends DateValue>(props: CalendarProps<T>) {
+  let ref = useRef(null);
+  // TS is getting confused because react-aria uses some conditional types that
+  // it doesn't understand how to resolve when using useContextPropsV2
+  [props, ref] = useContextPropsV2(CalendarContext as any, props, ref);
+
   const { size = "medium" } = props;
   const { locale } = useLocale();
   const state = useCalendarState({
@@ -62,7 +82,7 @@ export function Calendar<T extends DateValue>(props: CalendarProps<T>) {
   });
 
   return (
-    <CalendarWrapper {...calendarProps} {...renderProps}>
+    <CalendarWrapper {...calendarProps} {...renderProps} ref={ref}>
       <Provider
         values={[
           [CalendarStateContext.Provider, state],
@@ -77,7 +97,10 @@ export function Calendar<T extends DateValue>(props: CalendarProps<T>) {
               },
             },
           ],
-          [ErrorMessageContext.Provider, errorMessageProps],
+          [
+            ErrorMessageContext.Provider,
+            { isInvalid: props.isInvalid, ...errorMessageProps },
+          ],
         ]}
       >
         {renderProps.children}
@@ -86,6 +109,9 @@ export function Calendar<T extends DateValue>(props: CalendarProps<T>) {
   );
 }
 
+/** The current Month & Year formatted & localized */
 Calendar.Title = CalendarTitle;
+/** The grid of days in the calendar */
 Calendar.Grid = CalendarGrid;
+/** A cell in the calendar grid */
 Calendar.Cell = CalendarCell;
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts
index 57410927..75633317 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.context.ts
@@ -1,6 +1,13 @@
-import { createContext } from 'react';
-import { CalendarState } from 'react-stately';
+import { createContext } from "react";
+import { CalendarState } from "react-stately";
+
+import { createComponentContext } from "@utils/index";
+import { CalendarProps } from "./Calendar.component";
+
+export const CalendarContext = createComponentContext<CalendarProps<any>>();
 
 export const CalendarStateContext = createContext<CalendarState | null>(null);
 
-export const CalendarTitleContext = createContext<{ title: string } | null>(null);
+export const CalendarTitleContext = createContext<{ title: string } | null>(
+  null
+);
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.stories.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.stories.tsx
index 49287ef0..084870af 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.stories.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.stories.tsx
@@ -1,6 +1,10 @@
 import { Meta, StoryObj } from "@storybook/react";
 import { today, getLocalTimeZone } from "@internationalized/date";
 import { Calendar } from ".";
+import { IconButton } from "@components/Buttons/IconButton";
+import { Flex } from "@components/Layout/Flex/Flex";
+import { FieldStateControls, RenderPropsArgTypes } from "@sb/helpers";
+import { ErrorMessage } from "@components/Fields";
 
 const meta: Meta<typeof Calendar> = {
   title: "Inputs/Date & Time/Calendar",
@@ -9,25 +13,32 @@ const meta: Meta<typeof Calendar> = {
     layout: "centered",
   },
   argTypes: {
+    ...RenderPropsArgTypes,
+    children: {
+      control: false,
+      table: {
+        disable: true,
+      },
+    },
     value: {
       control: false,
       description:
-        "The current value of the input. Accepts a object from the `@internationalized/date` library.",
+        "The current value of the calendar. Accepts a object from the `@internationalized/date` library.",
     },
     defaultValue: {
       control: false,
       description:
-        "Set the initial value of an uncontrolled input. Accepts a object from the `@internationalized/date` library.",
+        "Set the initial value of an uncontrolled calendar. Accepts a object from the `@internationalized/date` library.",
     },
     defaultFocusedValue: {
       control: false,
       description:
-        "Set the initial focused value of an uncontrolled input. Accepts a object from the `@internationalized/date` library.",
+        "Set the initial focused value of an uncontrolled calendar. Accepts a object from the `@internationalized/date` library.",
     },
     focusedValue: {
       control: false,
       description:
-        "The current focused value of the input. Accepts a object from the `@internationalized/date` library.",
+        "The current focused value of the calendary. Accepts a object from the `@internationalized/date` library.",
     },
     isDateUnavailable: {
       control: false,
@@ -44,9 +55,27 @@ const meta: Meta<typeof Calendar> = {
       description:
         "The minimum date that can be selected. Accepts a object from the `@internationalized/date` library.",
     },
+    isDisabled: {
+      control: "boolean",
+      description: "Whether the calendar is disabled.",
+    },
+    isInvalid: {
+      control: "boolean",
+      description: "Whether the calendar is in an invalid state.",
+    },
     onFocusChange: {
       control: false,
       description: "Callback for when the focus date changes.",
+      table: {
+        category: "Events",
+      },
+    },
+    onChange: {
+      control: false,
+      description: "Callback for when the selected date changes.",
+      table: {
+        category: "Events",
+      },
     },
   },
 };
@@ -55,24 +84,53 @@ export default meta;
 
 type Story = StoryObj<typeof Calendar>;
 
-export const Primary: Story = {
-  args: {
-    size: "medium",
-  },
-};
-
 const date = today(getLocalTimeZone());
 const minValue = date.set({ month: date.month - 1 });
 const maxValue = date.set({ month: date.month + 1 });
 
-export const WithDefaultValue: Story = {
+export const Primary: Story = {
   args: {
-    ...Primary.args,
+    size: "medium",
     defaultValue: date,
+    children: [
+      <Flex
+        key="header"
+        gap={20}
+        alignItems="center"
+        justifyContent="space-between"
+      >
+        <IconButton slot="previous-month" icon="chevron_left" variant="ghost" />
+        <Calendar.Title />
+        <IconButton slot="next-month" icon="chevron_right" variant="ghost" />
+      </Flex>,
+      <Calendar.Grid key="grid">
+        {(date) => <Calendar.Cell date={date} />}
+      </Calendar.Grid>,
+    ],
+  },
+  parameters: {
+    docs: {
+      source: {
+        language: "jsx",
+        code: `
+<Calendar>
+  <Flex gap={20} alignItems="center" justifyContent="space-between">
+    <IconButton slot="previous-month" icon="chevron_left" variant="ghost" />
+    <Calendar.Title />
+    <IconButton slot="next-month" icon="chevron_right" variant="ghost" />
+  </Flex>
+  <Calendar.Grid>
+    {(date) => <Calendar.Cell date={date} />}
+  </Calendar.Grid>
+</Calendar>
+        `,
+      },
+    },
   },
 };
 
 export const WithMaxAndMinValues: Story = {
+  ...Primary,
   args: {
     ...Primary.args,
     defaultValue: date,
@@ -80,3 +138,47 @@ export const WithMaxAndMinValues: Story = {
     maxValue,
   },
 };
+
+export const WithErrorMessage: Story = {
+  ...Primary,
+  args: {
+    ...Primary.args,
+    isInvalid: true,
+    children: [
+      <Flex
+        key="header"
+        gap={20}
+        alignItems="center"
+        justifyContent="space-between"
+      >
+        <IconButton slot="previous-month" icon="chevron_left" variant="ghost" />
+        <Calendar.Title />
+        <IconButton slot="next-month" icon="chevron_right" variant="ghost" />
+      </Flex>,
+      <Calendar.Grid key="grid">
+        {(date) => <Calendar.Cell date={date} />}
+      </Calendar.Grid>,
+      <ErrorMessage key="error">The Selected Date is Invalid</ErrorMessage>,
+    ],
+  },
+  parameters: {
+    docs: {
+      source: {
+        language: "jsx",
+        code: `
+<Calendar isInvalid>
+  <Flex gap={20} alignItems="center" justifyContent="space-between">
+    <IconButton slot="previous-month" icon="chevron_left" variant="ghost" />
+    <Calendar.Title />
+    <IconButton slot="next-month" icon="chevron_right" variant="ghost" />
+  </Flex>
+  <Calendar.Grid>
+    {(date) => <Calendar.Cell date={date} />}
+  </Calendar.Grid>
+  <ErrorMessage key="error">The Selected Date is Invalid</ErrorMessage>,
+</Calendar>
+        `,
+      },
+    },
+  },
+};
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts
index e545fa89..78230f45 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.styles.ts
@@ -82,6 +82,12 @@ export const StyledCalendarCell = styled.td`
 
   &[aria-invalid="true"] {
     background-color: var(--error-clr);
+
+    &:hover {
+      cursor: pointer;
+      border-radius: 3px;
+      background-color: var(--error-clr);
+    }
   }
 
   &:hover {
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.component.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.component.tsx
index ffb7bdc5..899809d2 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.component.tsx
@@ -56,6 +56,7 @@ export function TimeInput<T extends TimeValue>(props: TimeInputProps<T>) {
         {...fieldProps}
         ref={ref}
         className={"aje-input__date-segments"}
+        padding={["left", "right"]}
       >
         {state.segments.map((segment, i) => (
           <DateSegment key={i} segment={segment} state={state} />
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.stories.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.stories.tsx
index ed5d1d81..6284156d 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.stories.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/TimeInput/TimeInput.stories.tsx
@@ -52,7 +52,7 @@ const date = now(getLocalTimeZone());
 export const Primary: Story = {
   args: {
     defaultValue: date,
-    label: "Date input",
+    label: "Time input",
     size: "medium",
     hideTimeZone: true,
     shouldForceLeadingZeros: false,

From 1b165245b9f74497fb5a4457e2eb25e5ceee2ddd Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:14:02 -0700
Subject: [PATCH 05/11] Update DatePicker to use new Calendar API

---
 .../Calendar/Calendar.component.tsx           |  2 +-
 .../DatePicker/DatePicker.component.tsx       | 58 +++++++++++--------
 .../DatePicker/DatePicker.styles.ts           |  3 +-
 .../Overlays/Popover/Popover.styles.ts        |  2 +-
 4 files changed, 36 insertions(+), 29 deletions(-)

diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
index 5d743877..4c83ec70 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
@@ -1,4 +1,4 @@
-import { useContext, useRef } from "react";
+import { useRef } from "react";
 import {
   useCalendar,
   AriaCalendarProps,
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.component.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.component.tsx
index 002b4343..625778ca 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.component.tsx
@@ -5,21 +5,21 @@ import {
   DateValue,
 } from "@react-aria/datepicker";
 import { useDatePickerState } from "react-stately";
-import { Size, AriaProps, FieldInputProps } from "../../../../types";
+import { AriaProps, FieldInputProps } from "../../../../types";
 import { DatePickerComboInput, DatePickerWrapper } from "./DatePicker.styles";
-import classNames from "classnames";
 import { DateInput } from "../DateInput";
 import { IconButton } from "../../../Buttons/IconButton";
 import { Popover } from "../../../Overlays/Popover";
-import { Dialog } from "../../../Overlays/Dialog";
 import { Calendar } from "../Calendar";
 import { ErrorMessage, Label, Message } from "../../../Fields";
 import { OverlayTriggerStateContext } from "@components/Overlays/OverlayTrigger/context";
+import { useRenderProps } from "@hooks";
+import { Flex } from "@components/Layout/Flex/Flex";
+import { filterDOMProps } from "@react-aria/utils";
 
-export type DatePickerProps<T extends DateValue> = AriaProps<
-  AriaDatePickerProps<T>
-> &
-  FieldInputProps;
+export interface DatePickerProps<T extends DateValue>
+  extends AriaProps<AriaDatePickerProps<T>>,
+    FieldInputProps {}
 
 /** DatePicker is a combination of a `<DateInput/>` component and a dropdown `<Calendar />` component. */
 export function DatePicker<T extends DateValue>(props: DatePickerProps<T>) {
@@ -30,7 +30,6 @@ export function DatePicker<T extends DateValue>(props: DatePickerProps<T>) {
     labelProps,
     fieldProps,
     buttonProps,
-    dialogProps,
     calendarProps,
     errorMessageProps,
     descriptionProps,
@@ -39,28 +38,21 @@ export function DatePicker<T extends DateValue>(props: DatePickerProps<T>) {
   const {
     size = "medium",
     error,
-    className,
-    isDisabled,
     isRequired,
-    isReadOnly,
     isInvalid,
     label,
     message,
   } = props;
 
-  const calendarSize: Size = ["auto", "full"].includes(size)
-    ? "medium"
-    : (size as Size);
+  const calendarSize = ["auto", "full"].includes(size) ? "medium" : size;
+
+  const renderProps = useRenderProps({
+    componentClassName: "aje-date-picker",
+    ...props,
+  });
 
   return (
-    <DatePickerWrapper
-      className={classNames("aje-date-picker", className)}
-      size={size}
-      isInvalid={isInvalid}
-      isDisabled={isDisabled}
-      isRequired={isRequired}
-      isReadOnly={isReadOnly}
-    >
+    <DatePickerWrapper {...filterDOMProps(props)} {...renderProps}>
       {label && <Label {...labelProps}>{label}</Label>}
       {message && <Message {...descriptionProps}>{message}</Message>}
 
@@ -68,6 +60,7 @@ export function DatePicker<T extends DateValue>(props: DatePickerProps<T>) {
         {...groupProps}
         ref={ref}
         className={"aje-input__date-segments"}
+        padding={["left", "right"]}
       >
         <DateInput {...fieldProps} size="full" isRequired={isRequired} />
         <IconButton icon="today" variant="content" {...buttonProps} />
@@ -79,9 +72,24 @@ export function DatePicker<T extends DateValue>(props: DatePickerProps<T>) {
 
       <OverlayTriggerStateContext.Provider value={state}>
         <Popover triggerRef={ref} placement="bottom start" variant="datepicker">
-          <Dialog {...dialogProps}>
-            <Calendar {...calendarProps} size={calendarSize} />
-          </Dialog>
+          <Calendar {...calendarProps} size={calendarSize}>
+            <Flex justifyContent="space-between" alignItems="center">
+              <IconButton
+                icon="chevron_left"
+                variant="ghost"
+                slot="previous-month"
+              />
+              <Calendar.Title />
+              <IconButton
+                icon="chevron_right"
+                variant="ghost"
+                slot="next-month"
+              />
+            </Flex>
+            <Calendar.Grid>
+              {(date) => <Calendar.Cell date={date} />}
+            </Calendar.Grid>
+          </Calendar>
         </Popover>
       </OverlayTriggerStateContext.Provider>
     </DatePickerWrapper>
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.styles.ts b/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.styles.ts
index 4cd495b0..1461f145 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.styles.ts
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/DatePicker/DatePicker.styles.ts
@@ -1,10 +1,9 @@
 import styled from "styled-components";
 import mixins from "../../../../styles/mixins";
-import { FieldWrapper } from "../../../Internal/FieldWrapper";
 import { ComboInput } from "../../../Fields";
 import { StyledComboInput } from "../../../Fields/ComboInput";
 
-export const DatePickerWrapper = styled(FieldWrapper)`
+export const DatePickerWrapper = styled.div`
   ${mixins.FieldStatus}
   ${mixins.SizingX}
 
diff --git a/packages/atomic-elements/src/components/Overlays/Popover/Popover.styles.ts b/packages/atomic-elements/src/components/Overlays/Popover/Popover.styles.ts
index 482f3d3c..ef0e9d1e 100644
--- a/packages/atomic-elements/src/components/Overlays/Popover/Popover.styles.ts
+++ b/packages/atomic-elements/src/components/Overlays/Popover/Popover.styles.ts
@@ -32,7 +32,7 @@ export const PopoverContent = styled.div<{
   }
 
   &.aje-popover--datepicker {
-    --popover-min-height: 330px;
+    --popover-min-height: auto;
     --popover-box-shadow: 0 2px 4px 0 rgba(0, 0, 0, 0.1);
     --popover-radius: 5px;
     --popover-border: 1px solid var(--neutral300);

From b4d1e5d2dfd7ecf0fd5c8e1c802e233dbb4186db Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:21:07 -0700
Subject: [PATCH 06/11] Remove old useContextProps in favor of V2

---
 .../components/Dropdowns/Menu/MenuTrigger.tsx |  2 +-
 .../ComboBoxField/ComboBoxField.component.tsx |  4 ++--
 .../MultiSelectField.component.tsx            |  2 +-
 .../SelectField/SelectField.component.tsx     |  2 +-
 .../OverlayTrigger.component.tsx              |  2 +-
 .../Overlays/Popover/Popover.component.tsx    | 12 ++++++-----
 .../Overlays/Popover/Popover.context.ts       |  4 ++++
 .../components/Overlays/Popover/context.ts    |  4 ----
 .../Overlays/ToolTip/ToolTip.component.tsx    | 21 ++++++-------------
 .../Overlays/ToolTip/ToolTip.context.ts       |  3 ++-
 10 files changed, 25 insertions(+), 31 deletions(-)
 create mode 100644 packages/atomic-elements/src/components/Overlays/Popover/Popover.context.ts
 delete mode 100644 packages/atomic-elements/src/components/Overlays/Popover/context.ts

diff --git a/packages/atomic-elements/src/components/Dropdowns/Menu/MenuTrigger.tsx b/packages/atomic-elements/src/components/Dropdowns/Menu/MenuTrigger.tsx
index d80df8c0..294fa6f1 100644
--- a/packages/atomic-elements/src/components/Dropdowns/Menu/MenuTrigger.tsx
+++ b/packages/atomic-elements/src/components/Dropdowns/Menu/MenuTrigger.tsx
@@ -7,7 +7,7 @@ import {
 } from "react-stately";
 import { HasChildren } from "../../../types";
 import { Provider } from "../../Internal/Provider";
-import { PopoverContext } from "../../Overlays/Popover/context";
+import { PopoverContext } from "../../Overlays/Popover/Popover.context";
 import { MenuContext } from "./Menu.context";
 import { OverlayTriggerStateContext } from "../../Overlays/OverlayTrigger/context";
 
diff --git a/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx b/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx
index 1970362f..d7dbed87 100644
--- a/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx
@@ -9,7 +9,7 @@ import { Provider } from "@components/Internal/Provider";
 import { useFilter } from "@react-aria/i18n";
 
 import { OverlayTriggerStateContext } from "@components/Overlays/OverlayTrigger/context";
-import { PopoverContext } from "@components/Overlays/Popover/context";
+import { PopoverContext } from "@components/Overlays/Popover/Popover.context";
 import {
   ListBoxContext,
   ListStateContext,
@@ -34,7 +34,7 @@ import { ComboInputContext } from "../ComboInput";
 import { ComboBoxFieldWrapper } from "./ComboBoxField.styles";
 
 export const ComboBoxField = forwardRef(function ComboBoxField<
-  T extends object,
+  T extends object
 >(props: ComboBoxFieldProps<T>, ref: React.Ref<HTMLInputElement>) {
   [props, ref] = useContextPropsV2(ComboBoxFieldContext, props, ref);
 
diff --git a/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx b/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx
index 91f90bbd..24c796c9 100644
--- a/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx
@@ -7,7 +7,7 @@ import { useRenderProps } from "@hooks/useRenderProps";
 import { useContextPropsV2 } from "@hooks/useContextProps";
 import { BaseCollection, CollectionBuilder } from "@react-aria/collections";
 import { Provider } from "@components/Internal/Provider";
-import { PopoverContext } from "@components/Overlays/Popover/context";
+import { PopoverContext } from "@components/Overlays/Popover/Popover.context";
 import { ButtonContext } from "@components/Buttons/Button/Button.context";
 import { DropdownButton } from "@components/Internal/DropdownButton";
 import {
diff --git a/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx b/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx
index 53c6505f..213a09cd 100644
--- a/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx
@@ -18,7 +18,7 @@ import {
   FieldLabelContext,
   FieldMessageContext,
 } from "@components/Fields";
-import { PopoverContext } from "@components/Overlays/Popover/context";
+import { PopoverContext } from "@components/Overlays/Popover/Popover.context";
 import { ButtonContext } from "@components/Buttons/Button/Button.context";
 
 import { ButtonText, SelectFieldWrapper } from "./SelectField.styles";
diff --git a/packages/atomic-elements/src/components/Overlays/OverlayTrigger/OverlayTrigger.component.tsx b/packages/atomic-elements/src/components/Overlays/OverlayTrigger/OverlayTrigger.component.tsx
index ed6f59b3..fa64bfd5 100644
--- a/packages/atomic-elements/src/components/Overlays/OverlayTrigger/OverlayTrigger.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/OverlayTrigger/OverlayTrigger.component.tsx
@@ -6,7 +6,7 @@ import {
   useOverlayTriggerState,
 } from "react-stately";
 
-import { PopoverContext } from "../Popover/context";
+import { PopoverContext } from "../Popover/Popover.context";
 import { Provider } from "../../Internal/Provider";
 import { HasChildren } from "../../../types";
 import { OverlayTriggerStateContext } from "./context";
diff --git a/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx b/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx
index ce957606..24982298 100644
--- a/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx
@@ -10,18 +10,19 @@ import {
   OverlayTriggerState,
   useOverlayTriggerState,
 } from "react-stately";
+import { useIsHidden } from "@react-aria/collections";
+
 import { mergeProps } from "@react-aria/utils";
 
 import { RenderBaseProps, HasVariant } from "../../../types";
 import { useRenderProps } from "@hooks";
-import { useContextProps } from "@hooks/useContextProps";
+import { useContextPropsV2 } from "@hooks/useContextProps";
 import { useResizeObserver } from "@hooks/useResizeObserver";
 import { useForwardedRef } from "@hooks/useForwardedRef";
 import { PopoverUnderlay, PopoverContent } from "./Popover.styles";
-import { PopoverContext } from "./context";
+import { PopoverContext } from "./Popover.context";
 import { OverlayTriggerStateContext } from "../OverlayTrigger/context";
 import { invertPlacementAxis } from "@utils/placement";
-import { useIsHidden } from "@react-aria/collections";
 
 export interface PopoverRenderProps {
   /** Width in pixels of the triggering element that opened this popover  */
@@ -41,8 +42,9 @@ export interface PopoverProps
 /** A popover is an overlay element positioned relative to a target. */
 export const Popover = React.forwardRef<HTMLDivElement, PopoverProps>(
   (props: PopoverProps, ref) => {
-    const contextProps = useContextProps(PopoverContext, props);
-    const { triggerRef, ...rest } = contextProps;
+    [props, ref] = useContextPropsV2(PopoverContext, props, ref);
+
+    const { triggerRef, ...rest } = props;
 
     const contextState = useContext(OverlayTriggerStateContext);
     const localState = useOverlayTriggerState(props);
diff --git a/packages/atomic-elements/src/components/Overlays/Popover/Popover.context.ts b/packages/atomic-elements/src/components/Overlays/Popover/Popover.context.ts
new file mode 100644
index 00000000..b14802f7
--- /dev/null
+++ b/packages/atomic-elements/src/components/Overlays/Popover/Popover.context.ts
@@ -0,0 +1,4 @@
+import { PopoverProps } from ".";
+import { createComponentContext } from "@utils/index";
+
+export const PopoverContext = createComponentContext<PopoverProps>();
diff --git a/packages/atomic-elements/src/components/Overlays/Popover/context.ts b/packages/atomic-elements/src/components/Overlays/Popover/context.ts
deleted file mode 100644
index 4f790ec6..00000000
--- a/packages/atomic-elements/src/components/Overlays/Popover/context.ts
+++ /dev/null
@@ -1,4 +0,0 @@
-import React from "react";
-import { PopoverProps } from ".";
-
-export const PopoverContext = React.createContext<PopoverProps>(null as any);
diff --git a/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx b/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx
index 1f569dee..1bca56f1 100644
--- a/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx
@@ -1,5 +1,4 @@
 import { useContext, useRef } from "react";
-import classNames from "classnames";
 import {
   Overlay,
   useOverlayPosition,
@@ -9,16 +8,10 @@ import {
 import { useTooltip } from "@react-aria/tooltip";
 import { OverlayTriggerProps } from "react-stately";
 import { Transition } from "react-transition-group";
-
 import { ToolTipArrow, ToolTipOverlay } from "./ToolTip.styles";
-import {
-  HasChildren,
-  HasClassName,
-  PresentationProps,
-  RenderBaseProps,
-} from "../../../types";
+import { RenderBaseProps } from "../../../types";
 import { TooltipContext, TooltipStateContext } from "./ToolTip.context";
-import { useContextProps } from "../../../hooks/useContextProps";
+import { useContextPropsV2 } from "../../../hooks/useContextProps";
 import { mergeProps } from "@react-aria/utils";
 import { useRenderProps } from "@hooks";
 
@@ -43,13 +36,11 @@ export interface ToolTipProps
 
 /** A ToolTip component displays a popup with additional information when a user hovers over or focuses on an element. */
 export function ToolTip(props: ToolTipProps) {
-  const {
-    triggerRef,
-    transitionDuration = 300,
-    ...rest
-  } = useContextProps(TooltipContext, props);
+  let ref = useRef(null);
+  [props, ref] = useContextPropsV2(TooltipContext, props, ref);
+  const { triggerRef, transitionDuration = 300, ...rest } = props;
+
   const state = useContext(TooltipStateContext);
-  const ref = useRef(null);
 
   const { tooltipProps } = useTooltip({ ...rest }, state);
 
diff --git a/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.context.ts b/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.context.ts
index 4f5bc8f5..04f4fb24 100644
--- a/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.context.ts
+++ b/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.context.ts
@@ -1,8 +1,9 @@
 import React from "react";
 import { TooltipTriggerState } from "react-stately";
 import { ToolTipProps } from "./ToolTip.component";
+import { createComponentContext } from "@utils/index";
 
-export const TooltipContext = React.createContext<ToolTipProps>(null as any);
+export const TooltipContext = createComponentContext<ToolTipProps>();
 
 export const TooltipStateContext = React.createContext<TooltipTriggerState>(
   null as any

From 26586492027ed19ddce0bd674906ddc7699c6ac1 Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:22:44 -0700
Subject: [PATCH 07/11] Renamed useContextPropsV2 => useContextProps

---
 .../LiveMessage/LiveMessage.component.tsx     |  4 ++--
 .../Buttons/Button/Button.component.tsx       |  8 ++-----
 .../IconButton/IconButton.component.tsx       |  4 ++--
 .../Dropdowns/ListBox/ListBox.component.tsx   |  4 ++--
 .../Dropdowns/Menu/Menu.component.tsx         |  4 ++--
 .../ErrorMessage/ErrorMessage.component.tsx   |  4 ++--
 .../Fields/Atoms/Input/Input.component.tsx    |  4 ++--
 .../Fields/Atoms/Label/Label.component.tsx    |  4 ++--
 .../Atoms/Message/Message.component.tsx       |  4 ++--
 .../Atoms/TextArea/TextArea.component.tsx     |  4 ++--
 .../ComboBoxField/ComboBoxField.component.tsx |  4 ++--
 .../ComboInput/ComboInput.component.tsx       |  4 ++--
 .../MultiSelectField.component.tsx            |  4 ++--
 .../SelectField/SelectField.component.tsx     |  4 ++--
 .../Inputs/Checkbox/Checkbox.component.tsx    |  4 ++--
 .../components/Layout/Tabs/Tabs.component.tsx |  6 +++---
 .../Overlays/Modal/Modal.component.tsx        |  4 ++--
 .../Overlays/Popover/Popover.component.tsx    |  4 ++--
 .../Overlays/ToolTip/ToolTip.component.tsx    |  4 ++--
 .../src/hooks/useContextProps.ts              | 21 +++++++------------
 20 files changed, 46 insertions(+), 57 deletions(-)

diff --git a/packages/atomic-elements/src/components/Accessibility/LiveMessage/LiveMessage.component.tsx b/packages/atomic-elements/src/components/Accessibility/LiveMessage/LiveMessage.component.tsx
index 16ba7405..f56577da 100644
--- a/packages/atomic-elements/src/components/Accessibility/LiveMessage/LiveMessage.component.tsx
+++ b/packages/atomic-elements/src/components/Accessibility/LiveMessage/LiveMessage.component.tsx
@@ -1,6 +1,6 @@
 import { useEffect, useRef } from "react";
 import { useAnnouncer } from "../LiveAnnouncer";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { LiveMessageContext } from "./LiveMessage.context";
 
 export interface LiveMessageProps {
@@ -19,7 +19,7 @@ export interface LiveMessageProps {
  */
 export function LiveMessage(props: LiveMessageProps) {
   let ref = useRef<HTMLDivElement>(null);
-  [props, ref] = useContextPropsV2(LiveMessageContext, props, ref);
+  [props, ref] = useContextProps(LiveMessageContext, props, ref);
 
   const { message, polite = true, assertive = false, timeout } = props;
   const { announceAssertive, announcePolite } = useAnnouncer();
diff --git a/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx b/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
index 74f6867c..9c97bd78 100644
--- a/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
+++ b/packages/atomic-elements/src/components/Buttons/Button/Button.component.tsx
@@ -13,7 +13,7 @@ import {
 import { useFocusRing } from "@hooks/useFocusRing";
 import { useRenderProps } from "@hooks/useRenderProps";
 import { useButtonLink } from "@hooks/useButtonLink";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { StyledButton } from "./Button.styles";
 import { ButtonContext } from "./Button.context";
 import { SlotProps } from "@hooks/useSlottedContext";
@@ -53,11 +53,7 @@ export interface ButtonProps
  */
 export const Button = forwardRef<HTMLButtonElement, ButtonProps>(
   function Button(props, forwardedRef) {
-    [props, forwardedRef] = useContextPropsV2(
-      ButtonContext,
-      props,
-      forwardedRef
-    );
+    [props, forwardedRef] = useContextProps(ButtonContext, props, forwardedRef);
 
     const ref = useObjectRef(forwardedRef);
 
diff --git a/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx b/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx
index 51e94414..591befd0 100644
--- a/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx
+++ b/packages/atomic-elements/src/components/Buttons/IconButton/IconButton.component.tsx
@@ -3,7 +3,7 @@ import { HasIcon } from "../../../types";
 import { MaterialIcon } from "../../Icons/MaterialIcon";
 import { StyledIconButton } from "./IconButton.styles";
 import { ButtonProps } from "../Button";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { ButtonContext } from "../Button/Button.context";
 
 export interface IconButtonProps
@@ -17,7 +17,7 @@ export interface IconButtonProps
  * */
 export const IconButton = forwardRef<HTMLButtonElement, IconButtonProps>(
   function IconButton(props, forwardedRef) {
-    [props, forwardedRef] = useContextPropsV2(
+    [props, forwardedRef] = useContextProps(
       ButtonContext as any,
       props,
       forwardedRef
diff --git a/packages/atomic-elements/src/components/Dropdowns/ListBox/ListBox.component.tsx b/packages/atomic-elements/src/components/Dropdowns/ListBox/ListBox.component.tsx
index 2c373304..bc527929 100644
--- a/packages/atomic-elements/src/components/Dropdowns/ListBox/ListBox.component.tsx
+++ b/packages/atomic-elements/src/components/Dropdowns/ListBox/ListBox.component.tsx
@@ -17,7 +17,7 @@ import { Label } from "@components/Fields";
 import { Divider } from "@components/Layout/Divider";
 import { Provider } from "@components/Internal/Provider";
 import { ItemContext, ItemProps, SectionContext } from "@components/Collection";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 
 import { List, ListItem, SectionTitle, SubList } from "./ListBox.styles";
 import { ListBoxProps } from "./ListBox.types";
@@ -35,7 +35,7 @@ export const ListBox = forwardRef(function ListBox<T extends object>(
   ref: React.Ref<HTMLUListElement>
 ) {
   const state = useContext(ListStateContext);
-  [props, ref] = useContextPropsV2(ListBoxContext, props, ref);
+  [props, ref] = useContextProps(ListBoxContext, props, ref);
 
   // When used within a SelectField or similar component,
   // the state will be provided by the parent component
diff --git a/packages/atomic-elements/src/components/Dropdowns/Menu/Menu.component.tsx b/packages/atomic-elements/src/components/Dropdowns/Menu/Menu.component.tsx
index 054992b4..8362a5b1 100644
--- a/packages/atomic-elements/src/components/Dropdowns/Menu/Menu.component.tsx
+++ b/packages/atomic-elements/src/components/Dropdowns/Menu/Menu.component.tsx
@@ -24,7 +24,7 @@ import { Provider } from "@components/Internal/Provider";
 import { useCollectionRenderer } from "@hooks/useCollectionRenderer";
 import { useRenderProps } from "@hooks";
 import { useFocusRing } from "@hooks/useFocusRing";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { ItemContext, ItemProps, SectionContext } from "@components/Collection";
 
 import { MenuContext, MenuStateContext } from "./Menu.context";
@@ -58,7 +58,7 @@ export const Menu = forwardRef(function Menu<T extends object>(
   props: MenuProps<T>,
   ref: React.Ref<HTMLUListElement>
 ) {
-  [props, ref] = useContextPropsV2(MenuContext, props, ref);
+  [props, ref] = useContextProps(MenuContext, props, ref);
 
   return (
     <CollectionBuilder content={<Collection {...props} />}>
diff --git a/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.component.tsx b/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.component.tsx
index 45f92454..2084d12a 100644
--- a/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/Atoms/ErrorMessage/ErrorMessage.component.tsx
@@ -2,7 +2,7 @@ import React from "react";
 import styled from "styled-components";
 import mixins from "../../../../styles/mixins";
 import { ElementWrapperProps } from "../../../../types";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { useRenderProps } from "@hooks";
 import { ErrorMessageContext } from "./ErrorMessage.context";
 
@@ -25,7 +25,7 @@ export const ErrorMessage = React.forwardRef(function ErrorMessage(
   props: ErrorMessageProps,
   ref: React.Ref<HTMLParagraphElement>
 ) {
-  [props, ref] = useContextPropsV2(ErrorMessageContext, props, ref);
+  [props, ref] = useContextProps(ErrorMessageContext, props, ref);
   const { className, size = "medium", as = "p", isInvalid, ...rest } = props;
 
   const renderProps = useRenderProps({
diff --git a/packages/atomic-elements/src/components/Fields/Atoms/Input/Input.component.tsx b/packages/atomic-elements/src/components/Fields/Atoms/Input/Input.component.tsx
index ec0bd7eb..192a7cf4 100644
--- a/packages/atomic-elements/src/components/Fields/Atoms/Input/Input.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/Atoms/Input/Input.component.tsx
@@ -2,7 +2,7 @@ import React, { forwardRef } from "react";
 import styled from "styled-components";
 import mixins from "../../../../styles/mixins";
 import { ElementWrapperProps } from "../../../../types";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { useRenderProps } from "@hooks";
 import { InputContext } from "./Input.context";
 import { SlotProps } from "@hooks/useSlottedContext";
@@ -25,7 +25,7 @@ export const Input = forwardRef(function Input(
   props: InputProps,
   ref: React.Ref<HTMLInputElement>
 ) {
-  [props, ref] = useContextPropsV2(InputContext, props, ref);
+  [props, ref] = useContextProps(InputContext, props, ref);
 
   const { className, size, style, ...rest } = props;
 
diff --git a/packages/atomic-elements/src/components/Fields/Atoms/Label/Label.component.tsx b/packages/atomic-elements/src/components/Fields/Atoms/Label/Label.component.tsx
index 6975e3ce..3503d557 100644
--- a/packages/atomic-elements/src/components/Fields/Atoms/Label/Label.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/Atoms/Label/Label.component.tsx
@@ -2,7 +2,7 @@ import React from "react";
 import styled from "styled-components";
 import mixins from "../../../../styles/mixins";
 import { ElementWrapperProps } from "../../../../types";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { LabelContext } from "./Label.context";
 import { useRenderProps } from "@hooks";
 
@@ -28,7 +28,7 @@ export const Label = React.forwardRef(function Label(
   props: LabelProps,
   ref: React.Ref<HTMLLabelElement>
 ) {
-  [props, ref] = useContextPropsV2(LabelContext, props, ref);
+  [props, ref] = useContextProps(LabelContext, props, ref);
 
   const { className, size, as = "label", ...rest } = props;
 
diff --git a/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.component.tsx b/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.component.tsx
index 20da9f37..3f324e00 100644
--- a/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/Atoms/Message/Message.component.tsx
@@ -2,7 +2,7 @@ import React from "react";
 import styled from "styled-components";
 import mixins from "../../../../styles/mixins";
 import { ElementWrapperProps } from "../../../../types";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { useRenderProps } from "@hooks";
 import { MessageContext } from "./Message.context";
 
@@ -24,7 +24,7 @@ export const Message = React.forwardRef(function Message(
   props: MessageProps,
   ref: React.Ref<HTMLParagraphElement>
 ) {
-  [props, ref] = useContextPropsV2(MessageContext, props, ref);
+  [props, ref] = useContextProps(MessageContext, props, ref);
   const { as, className, size = "auto", ...rest } = props;
 
   const renderProps = useRenderProps({
diff --git a/packages/atomic-elements/src/components/Fields/Atoms/TextArea/TextArea.component.tsx b/packages/atomic-elements/src/components/Fields/Atoms/TextArea/TextArea.component.tsx
index 54c06010..343710a8 100644
--- a/packages/atomic-elements/src/components/Fields/Atoms/TextArea/TextArea.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/Atoms/TextArea/TextArea.component.tsx
@@ -1,7 +1,7 @@
 import React, { forwardRef } from "react";
 import { ElementWrapperProps } from "../../../../types";
 import { StyledTextArea } from "./TextArea.styles";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { useRenderProps } from "@hooks";
 import { filterDOMProps } from "@react-aria/utils";
 import { TextAreaContext } from "./TextArea.context";
@@ -16,7 +16,7 @@ export const TextArea = forwardRef(function TextArea(
   props: TextAreaProps,
   ref: React.Ref<HTMLTextAreaElement>
 ) {
-  [props, ref] = useContextPropsV2(TextAreaContext, props, ref);
+  [props, ref] = useContextProps(TextAreaContext, props, ref);
 
   const { className, size, style, ...rest } = props;
 
diff --git a/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx b/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx
index d7dbed87..23fb4820 100644
--- a/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/ComboBoxField/ComboBoxField.component.tsx
@@ -1,5 +1,5 @@
 import React, { forwardRef, useRef } from "react";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { PressResponder } from "@react-aria/interactions";
 import { CollectionBuilder } from "@react-aria/collections";
 import { useComboBoxState } from "react-stately";
@@ -36,7 +36,7 @@ import { ComboBoxFieldWrapper } from "./ComboBoxField.styles";
 export const ComboBoxField = forwardRef(function ComboBoxField<
   T extends object
 >(props: ComboBoxFieldProps<T>, ref: React.Ref<HTMLInputElement>) {
-  [props, ref] = useContextPropsV2(ComboBoxFieldContext, props, ref);
+  [props, ref] = useContextProps(ComboBoxFieldContext, props, ref);
 
   return (
     <CollectionBuilder content={props.children}>
diff --git a/packages/atomic-elements/src/components/Fields/ComboInput/ComboInput.component.tsx b/packages/atomic-elements/src/components/Fields/ComboInput/ComboInput.component.tsx
index 5404f483..33ccf2f1 100644
--- a/packages/atomic-elements/src/components/Fields/ComboInput/ComboInput.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/ComboInput/ComboInput.component.tsx
@@ -2,7 +2,7 @@ import React from "react";
 import classNames from "classnames";
 import { ComboInputProps } from "./ComboInput.types";
 import { StyledComboInput } from "./ComboInput.styles";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { ComboInputContext } from "./ComboInput.context";
 
 /**
@@ -15,7 +15,7 @@ export const ComboInput = React.forwardRef(function ComboInput(
   props: ComboInputProps,
   ref: React.Ref<HTMLDivElement>
 ) {
-  [props, ref] = useContextPropsV2(ComboInputContext, props, ref);
+  [props, ref] = useContextProps(ComboInputContext, props, ref);
 
   const { className, padding = [], children, inputRef, ...rest } = props;
 
diff --git a/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx b/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx
index 24c796c9..972120a1 100644
--- a/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/MultiSelectField/MultiSelectField.component.tsx
@@ -4,7 +4,7 @@ import { PressResponder } from "@react-aria/interactions";
 import { ListBoxContext, ListStateContext } from "../../Dropdowns/ListBox";
 import { OverlayTriggerStateContext } from "@components/Overlays/OverlayTrigger/context";
 import { useRenderProps } from "@hooks/useRenderProps";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { BaseCollection, CollectionBuilder } from "@react-aria/collections";
 import { Provider } from "@components/Internal/Provider";
 import { PopoverContext } from "@components/Overlays/Popover/Popover.context";
@@ -38,7 +38,7 @@ export interface MultiSelectFieldComponent
 export const MultiSelectField = forwardRef(function MultiSelectField<
   T extends object
 >(props: MultiSelectFieldProps<T>, ref: React.Ref<HTMLButtonElement>) {
-  [props, ref] = useContextPropsV2(MultiSelectFieldContext, props, ref);
+  [props, ref] = useContextProps(MultiSelectFieldContext, props, ref);
 
   return (
     <CollectionBuilder content={props.children}>
diff --git a/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx b/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx
index 213a09cd..3d5779a8 100644
--- a/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/SelectField/SelectField.component.tsx
@@ -11,7 +11,7 @@ import {
 } from "@components/Dropdowns/ListBox";
 import { OverlayTriggerStateContext } from "../../Overlays/OverlayTrigger/context";
 import { useRenderProps } from "@hooks/useRenderProps";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { Provider } from "@components/Internal/Provider";
 import {
   FieldErrorContext,
@@ -51,7 +51,7 @@ export const SelectField = forwardRef(function SelectField<T extends object>(
   props: SelectFieldProps<T>,
   ref: React.Ref<HTMLButtonElement>
 ) {
-  [props, ref] = useContextPropsV2(SelectFieldContext, props, ref);
+  [props, ref] = useContextProps(SelectFieldContext, props, ref);
 
   return (
     <CollectionBuilder content={props.children}>
diff --git a/packages/atomic-elements/src/components/Inputs/Checkbox/Checkbox.component.tsx b/packages/atomic-elements/src/components/Inputs/Checkbox/Checkbox.component.tsx
index e38b9ba9..bac5dc24 100644
--- a/packages/atomic-elements/src/components/Inputs/Checkbox/Checkbox.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/Checkbox/Checkbox.component.tsx
@@ -7,7 +7,7 @@ import { AriaProps, FieldInputProps } from "../../../types";
 import { ChooseInput, ChooseLabel } from "../Inputs.styles";
 import { CheckboxWrapper } from "./Checkbox.styles";
 import { ErrorMessage, Message } from "../../Fields";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { CheckBoxContext } from "./Checkbox.context";
 import { SlotProps } from "@hooks/useSlottedContext";
 
@@ -19,7 +19,7 @@ export interface CheckBoxProps
 /** Checkbox Component. Accepts a `ref` */
 export const CheckBox = React.forwardRef<HTMLInputElement, CheckBoxProps>(
   (props, ref) => {
-    [props, ref] = useContextPropsV2(CheckBoxContext, props, ref);
+    [props, ref] = useContextProps(CheckBoxContext, props, ref);
 
     const {
       children,
diff --git a/packages/atomic-elements/src/components/Layout/Tabs/Tabs.component.tsx b/packages/atomic-elements/src/components/Layout/Tabs/Tabs.component.tsx
index 522b5646..8983cb09 100644
--- a/packages/atomic-elements/src/components/Layout/Tabs/Tabs.component.tsx
+++ b/packages/atomic-elements/src/components/Layout/Tabs/Tabs.component.tsx
@@ -15,7 +15,7 @@ import {
 } from "@react-aria/collections";
 import { useObjectRef } from "@react-aria/utils";
 
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { useRenderProps } from "@hooks";
 import { useCollectionRenderer } from "@hooks/useCollectionRenderer";
 import { Provider } from "@components/Internal/Provider";
@@ -48,7 +48,7 @@ export const Tabs = forwardRef(function Tabs<T extends object>(
   props: TabsProps<T>,
   ref: React.ForwardedRef<HTMLDivElement>
 ) {
-  [props, ref] = useContextPropsV2(TabsContext, props, ref);
+  [props, ref] = useContextProps(TabsContext, props, ref);
 
   return (
     <CollectionBuilder content={props.children}>
@@ -129,7 +129,7 @@ function Tab<T extends object>(
   ref: React.ForwardedRef<HTMLDivElement>,
   item: Node<T>
 ) {
-  [props, ref] = useContextPropsV2(TabContext, props, ref);
+  [props, ref] = useContextProps(TabContext, props, ref);
   const state = useContext(TabsStateContext)!;
 
   const { key } = item;
diff --git a/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx b/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx
index c275ef28..a46824bf 100644
--- a/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/Modal/Modal.component.tsx
@@ -19,7 +19,7 @@ import {
   ModalTitle,
 } from "./Modal.styles";
 import classNames from "classnames";
-import { useContextPropsV2 } from "../../../hooks/useContextProps";
+import { useContextProps } from "../../../hooks/useContextProps";
 import { ModalContext } from "./Modal.context";
 import { OverlayTriggerStateContext } from "../OverlayTrigger/context";
 import { useRenderProps } from "@hooks";
@@ -45,7 +45,7 @@ export interface ModalProps extends BaseModalProps {
  */
 export function Modal(props: ModalProps) {
   let ref = useRef(null);
-  [props, ref] = useContextPropsV2(ModalContext, props, ref);
+  [props, ref] = useContextProps(ModalContext, props, ref);
 
   const { children, centered = false, variant = "default", ...rest } = props;
 
diff --git a/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx b/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx
index 24982298..7b686620 100644
--- a/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/Popover/Popover.component.tsx
@@ -16,7 +16,7 @@ import { mergeProps } from "@react-aria/utils";
 
 import { RenderBaseProps, HasVariant } from "../../../types";
 import { useRenderProps } from "@hooks";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { useResizeObserver } from "@hooks/useResizeObserver";
 import { useForwardedRef } from "@hooks/useForwardedRef";
 import { PopoverUnderlay, PopoverContent } from "./Popover.styles";
@@ -42,7 +42,7 @@ export interface PopoverProps
 /** A popover is an overlay element positioned relative to a target. */
 export const Popover = React.forwardRef<HTMLDivElement, PopoverProps>(
   (props: PopoverProps, ref) => {
-    [props, ref] = useContextPropsV2(PopoverContext, props, ref);
+    [props, ref] = useContextProps(PopoverContext, props, ref);
 
     const { triggerRef, ...rest } = props;
 
diff --git a/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx b/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx
index 1bca56f1..eacdfc29 100644
--- a/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx
+++ b/packages/atomic-elements/src/components/Overlays/ToolTip/ToolTip.component.tsx
@@ -11,7 +11,7 @@ import { Transition } from "react-transition-group";
 import { ToolTipArrow, ToolTipOverlay } from "./ToolTip.styles";
 import { RenderBaseProps } from "../../../types";
 import { TooltipContext, TooltipStateContext } from "./ToolTip.context";
-import { useContextPropsV2 } from "../../../hooks/useContextProps";
+import { useContextProps } from "../../../hooks/useContextProps";
 import { mergeProps } from "@react-aria/utils";
 import { useRenderProps } from "@hooks";
 
@@ -37,7 +37,7 @@ export interface ToolTipProps
 /** A ToolTip component displays a popup with additional information when a user hovers over or focuses on an element. */
 export function ToolTip(props: ToolTipProps) {
   let ref = useRef(null);
-  [props, ref] = useContextPropsV2(TooltipContext, props, ref);
+  [props, ref] = useContextProps(TooltipContext, props, ref);
   const { triggerRef, transitionDuration = 300, ...rest } = props;
 
   const state = useContext(TooltipStateContext);
diff --git a/packages/atomic-elements/src/hooks/useContextProps.ts b/packages/atomic-elements/src/hooks/useContextProps.ts
index ac556120..6a658a0b 100644
--- a/packages/atomic-elements/src/hooks/useContextProps.ts
+++ b/packages/atomic-elements/src/hooks/useContextProps.ts
@@ -1,28 +1,21 @@
-import { useContext } from "react";
 import { mergeProps, mergeRefs, useObjectRef } from "@react-aria/utils";
-import { SlotProps, SlottedContextValue, useSlottedContext } from './useSlottedContext';
+import {
+  SlotProps,
+  SlottedContextValue,
+  useSlottedContext,
+} from "./useSlottedContext";
 
 type WithRef<T, R> = T & { ref?: React.Ref<R> };
 export type ContextValue<T, R> = SlottedContextValue<WithRef<T, R>>;
 
-export function useContextProps<T extends {}>(
-  context: React.Context<T>,
-  props: T
-) {
-  const contextProps = useContext(context);
-
-  return mergeProps(contextProps, props);
-}
-
-
 // from: https://github.com/adobe/react-spectrum/blob/c6bd2cb0808838a9f1f850b6c1ffe88465254222/packages/react-aria-components/src/utils.tsx#L180
 /** Consume a value from a context & merge it with the provided props */
-export function useContextPropsV2<T extends object, R>(
+export function useContextProps<T extends object, R>(
   context: React.Context<ContextValue<T, R>>,
   props: T,
   ref: React.Ref<R>
 ): [T, React.RefObject<R>] {
-  const ctx = useSlottedContext(context, (props as SlotProps).slot) || {}
+  const ctx = useSlottedContext(context, (props as SlotProps).slot) || {};
   const { ref: contextRef, ...contextProps } = ctx as WithRef<T, R>;
 
   const mergedRef = useObjectRef(mergeRefs(ref, contextRef!));

From 6ee0a360d04a1081ca7e12ead7e0b675bb910297 Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:28:41 -0700
Subject: [PATCH 08/11] cleanup

---
 .../Inputs/DateAndTimes/Calendar/Calendar.component.tsx    | 6 +++++-
 .../Inputs/DateAndTimes/Calendar/Calendar.types.ts         | 7 -------
 .../DateAndTimes/Calendar/components/CalendarCell.tsx      | 1 -
 .../DateAndTimes/Calendar/components/CalendarTitle.tsx     | 7 +++++--
 4 files changed, 10 insertions(+), 11 deletions(-)
 delete mode 100644 packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts

diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
index 4c83ec70..b3f9cada 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
@@ -77,8 +77,12 @@ export function Calendar<T extends DateValue>(props: CalendarProps<T>) {
 
   const renderProps = useRenderProps({
     componentClassName: "aje-calendar",
-    size,
     ...props,
+    size,
+    selectors: {
+      "data-disabled": props.isDisabled,
+      "data-invalid": props.isInvalid,
+    },
   });
 
   return (
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts
deleted file mode 100644
index c230efc2..00000000
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.types.ts
+++ /dev/null
@@ -1,7 +0,0 @@
-import { CalendarDate } from '@internationalized/date';
-
-export interface CalendarRenderProps {
-  focusedDate: CalendarDate;
-  isDisabled: boolean;
-  isFocused: boolean;
-}
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx
index 1f04b329..a7183427 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarCell.tsx
@@ -5,7 +5,6 @@ import { useCalendarCell } from "@react-aria/calendar";
 import { CalendarCellButton, StyledCalendarCell } from "../Calendar.styles";
 import { useRenderProps } from "@hooks";
 import { RenderBaseProps } from "../../../../../types";
-import { CalendarRenderProps } from "../Calendar.types";
 
 interface CalendarCellRenderProps {
   focusedDate: CalendarDate;
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx
index a3e14aad..b0b0334c 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/components/CalendarTitle.tsx
@@ -7,9 +7,12 @@ import {
   CalendarTitleContext,
 } from "../Calendar.context";
 import { CalendarHeaderTitle } from "../Calendar.styles";
-import { CalendarRenderProps } from "../Calendar.types";
+import { CalendarDate } from "@internationalized/date";
 
-interface CalendarTitleRenderProps extends CalendarRenderProps {
+interface CalendarTitleRenderProps {
+  focusedDate: CalendarDate;
+  isDisabled: boolean;
+  isFocused: boolean;
   title: string;
 }
 

From e21c8b9ed29bc41e4aea2b5c145db40e6e5cfa22 Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:35:07 -0700
Subject: [PATCH 09/11] Added changeset

---
 .changeset/curvy-pans-obey.md | 5 +++++
 1 file changed, 5 insertions(+)
 create mode 100644 .changeset/curvy-pans-obey.md

diff --git a/.changeset/curvy-pans-obey.md b/.changeset/curvy-pans-obey.md
new file mode 100644
index 00000000..bebe90df
--- /dev/null
+++ b/.changeset/curvy-pans-obey.md
@@ -0,0 +1,5 @@
+---
+"@atomicjolt/atomic-elements": major
+---
+
+Implemented a Composition API for the Calendar component

From 4cfc9374410586e1cb45f97057d024e027bcac01 Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:51:38 -0700
Subject: [PATCH 10/11] Updated readme

---
 packages/atomic-elements/README.md | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/packages/atomic-elements/README.md b/packages/atomic-elements/README.md
index 1d520812..7d003eb0 100644
--- a/packages/atomic-elements/README.md
+++ b/packages/atomic-elements/README.md
@@ -29,13 +29,11 @@ $ yarn add @atomicjolt/atomic-elements
 Include the following in your project
 
 ```js
-import { LoadFonts, CssVariables, CssGlobalDefaults } from "@atomicjolt/atomic-elements";
+import { CssVariables } from "@atomicjolt/atomic-elements";
 
 const App = () => (
   <>
-    <LoadFonts />
     <CssVariables />
-    <CssGlobalDefaults />
     <YourApp />
   </>
 );

From 61be30c3bc6e8610fad62feeec223a635e23bdca Mon Sep 17 00:00:00 2001
From: Sean Collings <sean.collings@atomicjolt.com>
Date: Mon, 2 Dec 2024 17:55:05 -0700
Subject: [PATCH 11/11] fixes

---
 .../src/components/Chips/Chip/Chip.component.tsx       |  4 ++--
 .../Fields/ChipGroupField/ChipGroupField.component.tsx | 10 +++++-----
 .../Fields/ChipGroupField/ChipGroupField.context.ts    |  2 +-
 .../DateAndTimes/Calendar/Calendar.component.tsx       |  6 +++---
 4 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx b/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx
index 48bd29e0..3ee81383 100644
--- a/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx
+++ b/packages/atomic-elements/src/components/Chips/Chip/Chip.component.tsx
@@ -6,7 +6,7 @@ import { createLeafComponent } from "@react-aria/collections";
 import { IconButton } from "@components/Buttons/IconButton";
 import { useConditionalPress } from "@hooks/useConditionalPress";
 import { useFocusRing } from "@hooks/useFocusRing";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { useRenderProps } from "@hooks";
 import { ChipGroupStateContext } from "@components/Fields/ChipGroupField/ChipGroupField.context";
 
@@ -15,7 +15,7 @@ import { ChipContent, ChipWrapper } from "./Chip.styles";
 import { ChipContext } from "./Chip.context";
 
 export function ChipLeaf<T>(...args: ChipArgs<T>) {
-  const [props, ref] = useContextPropsV2(ChipContext, args[0], args[1]);
+  const [props, ref] = useContextProps(ChipContext, args[0], args[1]);
 
   // We're being rendered standalone
   if (args.length === 2) {
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx
index 1db70709..7944943c 100644
--- a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.component.tsx
@@ -26,7 +26,7 @@ import {
   FieldProps,
 } from "@components/Fields";
 import { useRenderProps } from "@hooks";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { Provider } from "@components/Internal/Provider";
 import { useCollectionRenderer } from "@hooks/useCollectionRenderer";
 
@@ -71,9 +71,9 @@ interface ForwardedChipGroupField {
 
 /** Building blocks for constructing a ChipGroup field. */
 export const ChipGroupField = forwardRef(function ChipGroupField<
-  T extends object,
+  T extends object
 >(props: ChipGroupFieldProps, ref: React.ForwardedRef<HTMLDivElement>) {
-  [props, ref] = useContextPropsV2(ChipGroupFieldContext, props, ref);
+  [props, ref] = useContextProps(ChipGroupFieldContext, props, ref);
 
   return (
     <CollectionBuilder content={props.children}>
@@ -151,9 +151,9 @@ interface ForwardedChipGroupFieldList {
 }
 
 const ChipGroupFieldList = forwardRef(function ChipGroupFieldList<
-  T extends object,
+  T extends object
 >(props: ChipGroupFieldListProps<T>, ref: React.ForwardedRef<HTMLDivElement>) {
-  [props, ref] = useContextPropsV2(ChipGroupListContext, props, ref);
+  [props, ref] = useContextProps(ChipGroupListContext, props, ref);
   const state = useContext(ChipGroupStateContext);
 
   // We're building the collection in the ChipGroupField component
diff --git a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts
index 43abcc62..58193fb0 100644
--- a/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts
+++ b/packages/atomic-elements/src/components/Fields/ChipGroupField/ChipGroupField.context.ts
@@ -7,7 +7,7 @@ import {
 import { createContext } from "react";
 
 export const ChipGroupFieldContext =
-  createComponentContext<ChipGroupFieldProps<any>>();
+  createComponentContext<ChipGroupFieldProps>();
 
 export const ChipGroupStateContext = createContext<ListState<any> | null>(null);
 
diff --git a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
index b3f9cada..9de19c2b 100644
--- a/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
+++ b/packages/atomic-elements/src/components/Inputs/DateAndTimes/Calendar/Calendar.component.tsx
@@ -13,7 +13,7 @@ import { useRenderProps } from "@hooks";
 import { Provider } from "@components/Internal/Provider";
 import { ButtonContext } from "@components/Buttons/Button/Button.context";
 import { DEFAULT_SLOT } from "@hooks/useSlottedContext";
-import { useContextPropsV2 } from "@hooks/useContextProps";
+import { useContextProps } from "@hooks/useContextProps";
 import { ErrorMessageContext } from "@components/Fields/Atoms/ErrorMessage";
 
 import { CalendarTitle } from "./components/CalendarTitle";
@@ -56,8 +56,8 @@ export interface CalendarProps<T extends DateValue>
 export function Calendar<T extends DateValue>(props: CalendarProps<T>) {
   let ref = useRef(null);
   // TS is getting confused because react-aria uses some conditional types that
-  // it doesn't understand how to resolve when using useContextPropsV2
-  [props, ref] = useContextPropsV2(CalendarContext as any, props, ref);
+  // it doesn't understand how to resolve when using useContextProps
+  [props, ref] = useContextProps(CalendarContext as any, props, ref);
 
   const { size = "medium" } = props;
   const { locale } = useLocale();