forked from Plume/Plume
449641d158
* Add search engine to the model Add a Tantivy based search engine to the model Implement most required functions for it * Implement indexing and plm subcommands Implement indexation on insert, update and delete Modify func args to get the indexer where required Add subcommand to initialize, refill and unlock search db * Move to a new threadpool engine allowing scheduling * Autocommit search index every half an hour * Implement front part of search Add default fields for search Add new routes and templates for search and result Implement FromFormValue for Page to reuse it on search result pagination Add optional query parameters to paginate template's macro Update to newer rocket_csrf, don't get csrf token on GET forms * Handle process termination to release lock Handle process termination Add tests to search * Add proper support for advanced search Add an advanced search form to /search, in template and route Modify Tantivy schema, add new tokenizer for some properties Create new String query parser Create Tantivy query AST from our own * Split search.rs, add comment and tests Split search.rs into multiple submodules Add comments and tests for Query Make user@domain be treated as one could assume
84 lines
2.6 KiB
Rust
84 lines
2.6 KiB
Rust
use chrono::offset::Utc;
|
|
use rocket_contrib::Template;
|
|
use serde_json;
|
|
|
|
use plume_models::{
|
|
db_conn::DbConn, users::User,
|
|
search::Query};
|
|
use routes::Page;
|
|
use Searcher;
|
|
|
|
#[get("/search")]
|
|
fn index(conn: DbConn, user: Option<User>) -> Template {
|
|
Template::render("search/index", json!({
|
|
"account": user.map(|u| u.to_json(&*conn)),
|
|
"now": format!("{}", Utc::today().format("%Y-%m-d")),
|
|
}))
|
|
}
|
|
|
|
#[derive(FromForm)]
|
|
struct SearchQuery {
|
|
q: Option<String>,
|
|
title: Option<String>,
|
|
subtitle: Option<String>,
|
|
content: Option<String>,
|
|
instance: Option<String>,
|
|
author: Option<String>,
|
|
tag: Option<String>,
|
|
blog: Option<String>,
|
|
lang: Option<String>,
|
|
license: Option<String>,
|
|
after: Option<String>,
|
|
before: Option<String>,
|
|
page: Option<Page>,
|
|
}
|
|
|
|
macro_rules! param_to_query {
|
|
( $query:ident, $parsed_query:ident; normal: $($field:ident),*; date: $($date:ident),*) => {
|
|
$(
|
|
if let Some(field) = $query.$field {
|
|
let mut rest = field.as_str();
|
|
while !rest.is_empty() {
|
|
let (token, r) = Query::get_first_token(rest);
|
|
rest = r;
|
|
$parsed_query.$field(token, None);
|
|
}
|
|
}
|
|
)*
|
|
$(
|
|
if let Some(field) = $query.$date {
|
|
let mut rest = field.as_str();
|
|
while !rest.is_empty() {
|
|
use chrono::naive::NaiveDate;
|
|
let (token, r) = Query::get_first_token(rest);
|
|
rest = r;
|
|
if let Ok(token) = NaiveDate::parse_from_str(token, "%Y-%m-%d") {
|
|
$parsed_query.$date(&token);
|
|
}
|
|
}
|
|
}
|
|
)*
|
|
}
|
|
}
|
|
|
|
|
|
#[get("/search?<query>")]
|
|
fn query(query: SearchQuery, conn: DbConn, searcher: Searcher, user: Option<User>) -> Template {
|
|
let page = query.page.unwrap_or(Page::first());
|
|
let mut parsed_query = Query::from_str(&query.q.unwrap_or_default());
|
|
|
|
param_to_query!(query, parsed_query; normal: title, subtitle, content, tag,
|
|
instance, author, blog, lang, license;
|
|
date: before, after);
|
|
|
|
let str_q = parsed_query.to_string();
|
|
let res = searcher.search_document(&conn, parsed_query, page.limits());
|
|
|
|
Template::render("search/result", json!({
|
|
"query":str_q,
|
|
"account": user.map(|u| u.to_json(&*conn)),
|
|
"next_page": if res.is_empty() { 0 } else { page.page+1 },
|
|
"posts": res.into_iter().map(|p| p.to_json(&*conn)).collect::<Vec<serde_json::Value>>(),
|
|
"page": page.page,
|
|
}))
|
|
}
|