diff options
| author | Fuwn <[email protected]> | 2022-04-19 23:31:53 -0700 |
|---|---|---|
| committer | Fuwn <[email protected]> | 2022-04-19 23:31:53 -0700 |
| commit | 3eef79af97dffc814f034e78674921b4f398b9f9 (patch) | |
| tree | d82542439015be0399072b198af633c1ee3fc0db /src/main.rs | |
| parent | refactor(modules): move sitemap to modules module (diff) | |
| download | locus-3eef79af97dffc814f034e78674921b4f398b9f9.tar.xz locus-3eef79af97dffc814f034e78674921b4f398b9f9.zip | |
refactor(modules): move search to modules module
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 103 |
1 files changed, 1 insertions, 102 deletions
diff --git a/src/main.rs b/src/main.rs index c625b2e..accffe9 100644 --- a/src/main.rs +++ b/src/main.rs @@ -41,7 +41,6 @@ use std::{collections::HashMap, lazy::SyncLazy, sync::Mutex}; use pickledb::PickleDb; use route::track_mount; -use search::{INDEX, SCHEMA}; use tokio::time::Instant; use windmark::{Response, Router}; use yarte::Template; @@ -156,107 +155,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> { ); router.attach_stateless(modules::sitemap::module); - - track_mount( - &mut router, - "/search", - "A search engine for this Gemini capsule", - Box::new(|context| { - let mut response = - String::from("# SEARCH\n\n=> /search?action=go Search!"); - - if let Some(query) = context.url.query_pairs().next() { - if query.0 == "action" && query.1 == "go" { - return Response::Input( - "What would you like to search for?".to_string(), - ); - } - - { - let path = (*SCHEMA.lock().unwrap()).get_field("path").unwrap(); - let description = - (*SCHEMA.lock().unwrap()).get_field("description").unwrap(); - let content = (*SCHEMA.lock().unwrap()).get_field("content").unwrap(); - let mut results = String::new(); - - let searcher = (*INDEX.lock().unwrap()) - .reader_builder() - .reload_policy(tantivy::ReloadPolicy::OnCommit) - .try_into() - .unwrap() - .searcher(); - let top_docs = searcher - .search( - &tantivy::query::QueryParser::for_index( - &(*INDEX.lock().unwrap()), - vec![path, description, content], - ) - .parse_query(&query.0.to_string()) - .unwrap(), - &tantivy::collector::TopDocs::with_limit(SEARCH_SIZE), - ) - .unwrap(); - - for (_score, document_address) in top_docs { - let retrieved_document = searcher.doc(document_address).unwrap(); - - macro_rules! text { - ($field:ident) => {{ - retrieved_document - .get_first($field) - .unwrap() - .as_text() - .unwrap() - }}; - ($document:ident, $field:ident) => {{ - $document.get_first($field).unwrap().as_text().unwrap() - }}; - } - - results += - &format!("=> {} {}{}\n", text!(path), text!(description), { - let mut lines = retrieved_document - .get_first(content) - .unwrap() - .as_text() - .unwrap() - .lines() - .skip(2); - - lines.next().map_or_else( - || "".to_string(), - |first_line| { - format!( - "\n> ... {}\n> {}\n> {} ...", - first_line, - lines.next().unwrap_or(""), - lines.next().unwrap_or("") - ) - }, - ) - }); - } - - response += &format!( - "\n\nYou searched for \"{}\"!\n\n## RESULTS\n\n{}\n\nIn need of \ - more results? This search engine populates its index with route \ - paths and route descriptions on startup. However, route content \ - isn't populated until the route is first visited. After a \ - route's first visit, it is updated after every five minutes, at \ - time of visit.", - query.0, - if results.is_empty() { - "There are no results for your query...".to_string() - } else { - results.trim_end().to_string() - }, - ); - } - } - - success!(response, context) - }), - ); + router.attach_stateless(modules::search::module); track_mount( &mut router, |