85 lines
2.6 KiB
Rust
85 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,
|
||
|
|
}))
|
||
|
|
}
|