refactor: simplify variation & spacing declaration

abelflopes · Nov 2, 2023 · 038a1c9 · 038a1c9
1 parent f3701be
commit 038a1c9
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/packages/components/all/src/container/index.tsx b/packages/components/all/src/container/index.tsx
@@ -2,25 +2,31 @@ import React from "react";
 import styles from "./index.module.scss";
 import classNames from "classnames";
 
-type ContainerVariation = "horizontal" | "vertical" | "small";
+type ContainerVariation = "small";
 
-interface ContainerProps extends React.HTMLAttributes<HTMLDivElement> {
+export interface ContainerProps extends React.HTMLAttributes<HTMLDivElement> {
   variation?: ContainerVariation | ContainerVariation[];
+  spacingX?: boolean;
+  spacingY?: boolean;
 }
 
 export const Container = ({
-  variation = "horizontal",
+  variation,
+  spacingX = true,
+  spacingY,
   className,
   ...otherProps
 }: Readonly<ContainerProps>): React.ReactElement => {
-  const computedVariations = Array.isArray(variation) ? variation : [variation];
+  const computedVariations =
+    Array.isArray(variation) || variation === undefined ? variation : [variation];
 
   return (
     <div
       className={classNames(
         styles.root,
+        { [`${styles.horizontal}`]: spacingX, [`${styles.vertical}`]: spacingY },
+        computedVariations?.map((i) => styles[i]),
         className,
-        computedVariations.map((i) => styles[i]),
       )}
       {...otherProps}
     />