update improvement

tscircuit · Oct 12, 2024 · ab234e2 · ab234e2
1 parent ba979af
commit ab234e2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/components/CodeAndPreview.tsx b/src/components/CodeAndPreview.tsx
@@ -188,7 +188,13 @@ export function CodeAndPreview({ snippet }: Props) {
           />
         )}
       </div>
-      <ImportSnippetDialog />
+      <ImportSnippetDialog
+        onSnippetSelected={(snippet) => {
+          setCode(
+            `import {} from "@tsci/${snippet.owner_name}.${snippet.unscoped_name}"\n${snippet.code}`,
+          )
+        }}
+      />
     </div>
   )
 }
diff --git a/src/components/dialogs/import-snippet-dialog.tsx b/src/components/dialogs/import-snippet-dialog.tsx
@@ -46,7 +46,12 @@ export const ImportSnippetDialog = ({
           <ul>
             {snippets?.map((snippet: Snippet) => (
               <li key={snippet.snippet_id}>
-                <Button onClick={() => onSnippetSelected(snippet)}>
+                <Button
+                  onClick={() => {
+                    onSnippetSelected(snippet)
+                    onOpenChange(false)
+                  }}
+                >
                   {snippet.name}
                 </Button>
               </li>