From a5155aa6eb6a409bfe05d0d73f7d1dc8932b0623 Mon Sep 17 00:00:00 2001
From: colinmcneil <colin.mcneil@docker.com>
Date: Thu, 30 Jan 2025 16:13:40 -0500
Subject: [PATCH] Bugfix pass 1

---
 src/extension/ui/src/App.tsx                  |  4 +-
 .../src/components/ClaudeConfigSyncStatus.tsx |  3 +-
 .../ui/src/components/PromptCard.tsx          | 81 ++++++++++---------
 3 files changed, 46 insertions(+), 42 deletions(-)

diff --git a/src/extension/ui/src/App.tsx b/src/extension/ui/src/App.tsx
index 5ba0ea6..3fc0a59 100644
--- a/src/extension/ui/src/App.tsx
+++ b/src/extension/ui/src/App.tsx
@@ -81,7 +81,7 @@ export function App() {
         }]
       })
       await client.docker.cli.exec('run', ['--rm', '-v', 'docker-prompts:/docker-prompts', '--workdir', '/docker-prompts', 'vonwig/function_write_files:latest', `'${payload}'`])
-      client.desktopUI.toast.success('Prompt registered successfully');
+      client.desktopUI.toast.success('Prompt registered successfully. Restart Claude Desktop to apply.');
       await loadRegistry();
     }
     catch (error) {
@@ -100,7 +100,7 @@ export function App() {
         }]
       })
       await client.docker.cli.exec('run', ['--rm', '-v', 'docker-prompts:/docker-prompts', '--workdir', '/docker-prompts', 'vonwig/function_write_files:latest', `'${payload}'`])
-      client.desktopUI.toast.success('Prompt unregistered successfully');
+      client.desktopUI.toast.success('Prompt unregistered successfully. Restart Claude Desktop to apply.');
       await loadRegistry();
     }
     catch (error) {
diff --git a/src/extension/ui/src/components/ClaudeConfigSyncStatus.tsx b/src/extension/ui/src/components/ClaudeConfigSyncStatus.tsx
index 89929b6..8a53ba6 100644
--- a/src/extension/ui/src/components/ClaudeConfigSyncStatus.tsx
+++ b/src/extension/ui/src/components/ClaudeConfigSyncStatus.tsx
@@ -146,7 +146,8 @@ export const ClaudeConfigSyncStatus = ({ client, setHasConfig }: { client: v1.Do
             </DialogContent>
         </Dialog >
 
-        <Dialog open={showRestartModal} onClose={() => setShowRestartModal(false)} >
+        {/* Keep this closed for now. */}
+        <Dialog open={showRestartModal && false} onClose={() => setShowRestartModal(false)} >
             <DialogTitle>Restart Claude Desktop</DialogTitle>
             <DialogContent sx={{ padding: 5, mt: 2 }}>
                 <Stack direction="column" spacing={3}>
diff --git a/src/extension/ui/src/components/PromptCard.tsx b/src/extension/ui/src/components/PromptCard.tsx
index 01ef2da..d3268c3 100644
--- a/src/extension/ui/src/components/PromptCard.tsx
+++ b/src/extension/ui/src/components/PromptCard.tsx
@@ -19,45 +19,48 @@ export function CatalogItemCard({ openUrl, item, canRegister, registered, regist
     const [isRegistering, setIsRegistering] = useState(false)
     return (
         <Card sx={{ height: '100%' }}>
-            <CardContent>
-                <a href="">
-                    <Stack onClick={openUrl} direction="row" spacing={2} justifyContent="space-between" sx={{ cursor: 'pointer' }}>
-                        <Typography gutterBottom variant="h5" component="div">
-                            {item.name}
-                        </Typography>
-                        <CardMedia
-                            component="img"
-                            sx={{ maxWidth: 100, padding: 1, background: 'white', borderRadius: 1 }}
-                            alt={`Icon for ${item.name}`}
-                            image={item.icon}
-                        />
-                    </Stack>
-                </a>
-                <Typography variant="body2" sx={{ color: 'text.secondary', mt: 2 }}>
-                    {item.description}
-                </Typography>
-            </CardContent>
-            <CardActions>
-                <Button
-                    size="small"
-                    onClick={() => {
-                        trackEvent('registry-changed', { name: item.name, ref: item.ref, action: registered ? 'remove' : 'add' });
-                        setIsRegistering(true)
-                        if (registered) {
-                            unregister(item).then(() => {
-                                setIsRegistering(false)
-                            })
-                        } else {
-                            register(item).then(() => {
-                                setIsRegistering(false)
-                            })
-                        }
-                    }}
-                    disabled={!canRegister || isRegistering}
-                >
-                    {isRegistering ? <CircularProgress size={20} /> : registered ? 'Remove' : 'Add'}
-                </Button>
-            </CardActions>
+            <Stack direction="column" height="100%" sx={{ justifyContent: 'space-between' }}>
+                <CardContent>
+                    <a href="">
+                        <Stack onClick={openUrl} direction="row" spacing={2} justifyContent="space-between" sx={{ cursor: 'pointer' }}>
+
+                            <Typography gutterBottom variant="h5" component="div">
+                                {item.name}
+                            </Typography>
+                            <CardMedia
+                                component="img"
+                                sx={{ maxWidth: 100, padding: 1, background: 'white', borderRadius: 1 }}
+                                alt={`Icon for ${item.name}`}
+                                image={item.icon}
+                            />
+                        </Stack>
+                    </a>
+                    <Typography variant="body2" sx={{ color: 'text.secondary', mt: 2 }}>
+                        {item.description}
+                    </Typography>
+                </CardContent>
+                <CardActions>
+                    <Button
+                        size="small"
+                        onClick={() => {
+                            trackEvent('registry-changed', { name: item.name, ref: item.ref, action: registered ? 'remove' : 'add' });
+                            setIsRegistering(true)
+                            if (registered) {
+                                unregister(item).then(() => {
+                                    setIsRegistering(false)
+                                })
+                            } else {
+                                register(item).then(() => {
+                                    setIsRegistering(false)
+                                })
+                            }
+                        }}
+                        disabled={!canRegister || isRegistering}
+                    >
+                        {isRegistering ? <CircularProgress size={20} /> : registered ? 'Remove' : 'Add'}
+                    </Button>
+                </CardActions>
+            </Stack>
         </Card >
     )
 }
\ No newline at end of file