fix: add layer from dataset explorer

goat-community · Sep 17, 2024 · 90c0353 · 90c0353
1 parent 3fc965e
commit 90c0353
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/apps/web/components/modals/DatasetExplorer.tsx b/apps/web/components/modals/DatasetExplorer.tsx
@@ -82,8 +82,18 @@ const DatasetExplorerModal: React.FC<DatasetExplorerProps> = ({ open, onClose, p
                 <Paper elevation={0} sx={{ backgroundImage: "none" }}>
                   <FoldersTreeView
                     queryParams={datasetSchema}
-                    setQueryParams={setDatasetSchema}
-                    enableActions={false}
+                    setQueryParams={(params, teamId, organizationId) => {
+                      const newQueryParams = { ...queryParams, page: 1 };
+                      delete newQueryParams.team_id;
+                      delete newQueryParams.organization_id;
+                      if (teamId) {
+                        newQueryParams.team_id = teamId;
+                      } else if (organizationId) {
+                        newQueryParams.organization_id = organizationId;
+                      }
+                      setQueryParams(newQueryParams);
+                      setDatasetSchema(params);
+                    }}
                   />
                 </Paper>
               </Grid>