From 11af1da071251cbb3520e2659da09f8b1e323884 Mon Sep 17 00:00:00 2001
From: Timmy Huang <thuang@chanzuckerberg.com>
Date: Fri, 28 Jan 2022 17:01:49 -0800
Subject: [PATCH] idea: button color prop

---
 src/core/Button/index.tsx | 22 ++++++++--------------
 src/core/Button/style.ts  | 29 +++++++++++++++++++++++------
 2 files changed, 31 insertions(+), 20 deletions(-)

diff --git a/src/core/Button/index.tsx b/src/core/Button/index.tsx
index 8b4d0bcb2..e9ae46d2d 100644
--- a/src/core/Button/index.tsx
+++ b/src/core/Button/index.tsx
@@ -1,6 +1,7 @@
 import { ButtonProps as RawButtonProps } from "@material-ui/core";
 import React from "react";
 import {
+  ExtraProps,
   PrimaryMinimalButton,
   RoundedButton,
   SecondaryMinimalButton,
@@ -8,18 +9,11 @@ import {
   StyledButton,
 } from "./style";
 
-interface SdsProps {
-  isAllCaps?: boolean;
-  isRounded?: boolean;
-  sdsStyle?: "minimal" | "rounded" | "square";
-  sdsType?: "primary" | "secondary";
-}
-export type ButtonProps = RawButtonProps & SdsProps;
+export type ButtonProps = RawButtonProps & ExtraProps;
 
 const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
   (props: ButtonProps, ref): JSX.Element | null => {
-    const sdsStyle = props?.sdsStyle;
-    const sdsType = props?.sdsType;
+    const { color = "primary", sdsStyle, sdsType } = props;
 
     if (!sdsStyle || !sdsType) {
       // eslint-disable-next-line no-console
@@ -44,7 +38,7 @@ const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
       case sdsStyle === "rounded" && sdsType === "primary":
         return (
           <RoundedButton
-            color="primary"
+            color={color}
             ref={ref}
             variant="contained"
             {...propsWithDefault}
@@ -53,7 +47,7 @@ const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
       case sdsStyle === "rounded" && sdsType === "secondary":
         return (
           <RoundedButton
-            color="primary"
+            color={color}
             ref={ref}
             variant="outlined"
             {...propsWithDefault}
@@ -62,7 +56,7 @@ const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
       case sdsStyle === "square" && sdsType === "primary":
         return (
           <SquareButton
-            color="primary"
+            color={color}
             ref={ref}
             variant="contained"
             {...propsWithDefault}
@@ -71,7 +65,7 @@ const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
       case sdsStyle === "square" && sdsType === "secondary":
         return (
           <SquareButton
-            color="primary"
+            color={color}
             ref={ref}
             variant="outlined"
             {...propsWithDefault}
@@ -80,7 +74,7 @@ const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
       case sdsStyle === "minimal" && sdsType === "primary":
         return (
           <PrimaryMinimalButton
-            color="primary"
+            color={color}
             ref={ref}
             variant="text"
             {...propsWithDefault}
diff --git a/src/core/Button/style.ts b/src/core/Button/style.ts
index 5539df222..62c46fcff 100644
--- a/src/core/Button/style.ts
+++ b/src/core/Button/style.ts
@@ -8,16 +8,31 @@ import {
   Props,
 } from "../styles";
 
-const sdsPropNames = ["isAllCaps", "isRounded", "sdsStyle", "sdsType"];
+export interface ExtraProps {
+  isAllCaps?: boolean;
+  isRounded?: boolean;
+  sdsStyle?: "minimal" | "rounded" | "square";
+  sdsType?: "primary" | "secondary";
+  color?: "success" | "error" | "warning" | "info";
+}
+
+// Please keep this in sync with the props used in `ExtraProps`
+const doNotForwardProps = [
+  "isAllCaps",
+  "isRounded",
+  "sdsStyle",
+  "sdsType",
+  "color",
+];
 
 const ButtonBase = styled(Button, {
   shouldForwardProp: (prop) => {
-    return !sdsPropNames.includes(prop.toString());
+    return !doNotForwardProps.includes(prop.toString());
   },
-})`
+})<ExtraProps>`
   box-shadow: none;
   ${(props) => {
-    const { variant } = props;
+    const { variant, color: colorProp } = props;
     const colors = getColors(props);
     const spacings = getSpaces(props);
 
@@ -30,13 +45,15 @@ const ButtonBase = styled(Button, {
 
     const padding = variant === "outlined" ? outlinedPadding : containedPadding;
 
+    const color = (colors && colors[colorProp]) || colors?.primary;
+
     return `
       padding: ${padding};
       min-width: 120px;
       height: 34px;
       &:hover, &:focus {
         color: white;
-        background-color: ${colors?.primary[500]};
+        background-color: ${color[500]};
         box-shadow: none;
       }
       &:focus {
@@ -45,7 +62,7 @@ const ButtonBase = styled(Button, {
       }
       &:active {
         color: white;
-        background-color: ${colors?.primary[600]};
+        background-color: ${color[600]};
         box-shadow: none;
       }
       &:disabled {