coder · BrunoQuaresma · Jul 18, 2023 · Jul 13, 2023 · Jul 14, 2023 · Jul 14, 2023
diff --git a/site/src/api/api.ts b/site/src/api/api.ts
@@ -519,11 +519,13 @@ export const startWorkspace = (
   workspaceId: string,
   templateVersionId: string,
   logLevel?: TypesGen.CreateWorkspaceBuildRequest["log_level"],
+  buildParameters?: TypesGen.WorkspaceBuildParameter[],
 ) =>
   postWorkspaceBuild(workspaceId, {
     transition: "start",
     template_version_id: templateVersionId,
     log_level: logLevel,
+    rich_parameter_values: buildParameters,
   })
 export const stopWorkspace = (
   workspaceId: string,
@@ -552,7 +554,13 @@ export const cancelWorkspaceBuild = async (
   return response.data
 }
 
-export const restartWorkspace = async (workspace: TypesGen.Workspace) => {
+export const restartWorkspace = async ({
+  workspace,
+  buildParameters,
+}: {
+  workspace: TypesGen.Workspace
+  buildParameters?: TypesGen.WorkspaceBuildParameter[]
+}) => {
   const stopBuild = await stopWorkspace(workspace.id)
   const awaitedStopBuild = await waitForBuild(stopBuild)
 
@@ -564,6 +572,8 @@ export const restartWorkspace = async (workspace: TypesGen.Workspace) => {
   const startBuild = await startWorkspace(
     workspace.id,
     workspace.latest_build.template_version_id,
+    undefined,
+    buildParameters,
   )
   await waitForBuild(startBuild)
 }
@@ -1346,3 +1356,15 @@ export const issueReconnectingPTYSignedToken = async (
   )
   return response.data
 }
+
+export const getWorkspaceParameters = async (workspace: TypesGen.Workspace) => {
+  const latestBuild = workspace.latest_build
+  const [templateVersionRichParameters, buildParameters] = await Promise.all([
+    getTemplateVersionRichParameters(latestBuild.template_version_id),
+    getWorkspaceBuildParameters(latestBuild.id),
+  ])
+  return {
+    templateVersionRichParameters,
+    buildParameters,
+  }
+}
diff --git a/site/src/components/LoadingButton/LoadingButton.tsx b/site/src/components/LoadingButton/LoadingButton.tsx
@@ -1,21 +1,25 @@
-import { FC } from "react"
+import { forwardRef } from "react"
 import MuiLoadingButton, {
   LoadingButtonProps as MuiLoadingButtonProps,
 } from "@mui/lab/LoadingButton"
 
 export type LoadingButtonProps = MuiLoadingButtonProps
 
-export const LoadingButton: FC<LoadingButtonProps> = ({
-  children,
-  loadingIndicator,
-  ...buttonProps
-}) => {
+export const LoadingButton = forwardRef<
+  HTMLButtonElement,
+  MuiLoadingButtonProps
+>(({ children, loadingIndicator, ...buttonProps }, ref) => {
   return (
-    <MuiLoadingButton variant="outlined" color="neutral" {...buttonProps}>
+    <MuiLoadingButton
+      variant="outlined"
+      color="neutral"
+      ref={ref}
+      {...buttonProps}
+    >
       {/* known issue: https://github.com/mui/material-ui/issues/27853 */}
       <span>
         {buttonProps.loading && loadingIndicator ? loadingIndicator : children}
       </span>
     </MuiLoadingButton>
   )
-}
+})
diff --git a/site/src/components/Markdown/Markdown.tsx b/site/src/components/Markdown/Markdown.tsx
@@ -12,17 +12,21 @@ import { Prism as SyntaxHighlighter } from "react-syntax-highlighter"
 import gfm from "remark-gfm"
 import { colors } from "theme/colors"
 import { darcula } from "react-syntax-highlighter/dist/cjs/styles/prism"
+import { combineClasses } from "utils/combineClasses"
 
 export interface MarkdownProps {
   children: string
 }
 
-export const Markdown: FC<{ children: string }> = ({ children }) => {
+export const Markdown: FC<{ children: string; className?: string }> = ({
+  children,
+  className,
+}) => {
   const styles = useStyles()
 
   return (
     <ReactMarkdown
-      className={styles.markdown}
+      className={combineClasses([styles.markdown, className])}
       remarkPlugins={[gfm]}
       components={{
         a: ({ href, target, children }) => (