fix: add handlers to lib.rs

devflowinc · Oct 2, 2024 · b6749a1 · b6749a1
1 parent 0a16c19
commit b6749a1
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 7 deletions.
diff --git a/frontends/dashboard/src/pages/dataset/CrawlingSettings.tsx b/frontends/dashboard/src/pages/dataset/CrawlingSettings.tsx
@@ -1,7 +1,20 @@
 import { createQuery } from "@tanstack/solid-query";
-import { useContext } from "solid-js";
+import { Show, useContext } from "solid-js";
 import { DatasetContext } from "../../contexts/DatasetContext";
 import { useTrieve } from "../../hooks/useTrieve";
+import { CrawlOptions } from "trieve-ts-sdk";
+
+const defaultCrawlOptions: CrawlOptions = {
+  boost_titles: false,
+  exclude_paths: [],
+  exclude_tags: [],
+  include_paths: [],
+  include_tags: [],
+  interval: "daily",
+  limit: 1000,
+  max_depth: 10,
+  site_url: "",
+};
 
 export const CrawlingSettings = () => {
   const datasetId = useContext(DatasetContext).datasetId;
@@ -21,9 +34,24 @@ export const CrawlingSettings = () => {
     },
   }));
 
+  return (
+    <Show when={crawlSettingsQuery.isSuccess}>
+      <div>success</div>
+      <RealCrawlingSettings
+        initialCrawlingSettings={crawlSettingsQuery.data || defaultCrawlOptions}
+      />
+    </Show>
+  );
+};
+
+interface RealCrawlingSettingsProps {
+  initialCrawlingSettings: CrawlOptions;
+}
+
+const RealCrawlingSettings = (props: RealCrawlingSettingsProps) => {
   return (
     <div>
-      <div>Crawling settings</div>
+      <div>Crawling Options</div>
     </div>
   );
 };
diff --git a/server/src/lib.rs b/server/src/lib.rs
@@ -783,11 +783,16 @@ pub fn main() -> std::io::Result<()> {
                                         .route(web::post().to(handlers::dataset_handler::get_all_tags)),
                                 )
                                 .service(
-                                    web::resource("/{dataset_id}")
-                                        .route(web::get().to(handlers::dataset_handler::get_dataset))
-                                        .route(
-                                            web::delete().to(handlers::dataset_handler::delete_dataset),
-                                        )
+                                    web::scope("/{dataset_id}")
+                                    .service(
+                                          web::resource("")
+                                              .route(web::get().to(handlers::dataset_handler::get_dataset))
+                                              .route(web::delete().to(handlers::dataset_handler::delete_dataset))
+                                    ).service(
+                                          web::resource("/crawl_options").route(
+                                              web::get().to(handlers::dataset_handler::get_dataset_crawl_options)
+                                          )
+                                    )
                                 )
                                 .service(
                                     web::resource("/clear/{dataset_id}")