Plume/src/routes/search.rs

use chrono::offset::Utc;
use rocket::request::Form;
use rocket_i18n::I18n;

use plume_models::{
    db_conn::DbConn, users::User,
    search::Query};
use routes::Page;
use template_utils::Ructe;
use Searcher;

#[derive(Default, FromForm)]
pub 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(ref 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(ref 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..>")]
pub fn search(query: Form<SearchQuery>, conn: DbConn, searcher: Searcher, user: Option<User>, intl: I18n) -> Ructe {
    let page = query.page.unwrap_or(Page::first());
    let mut parsed_query = Query::from_str(&query.q.as_ref().map(|q| q.as_str()).unwrap_or_default());

    param_to_query!(query, parsed_query; normal: title, subtitle, content, tag,
              instance, author, blog, lang, license;
              date: before, after);

    let str_query = parsed_query.to_string();

    if str_query.is_empty() {
        render!(search::index(
            &(&*conn, &intl.catalog, user),
            &format!("{}", Utc::today().format("%Y-%m-d"))
        ))
    } else {
        let res = searcher.search_document(&conn, parsed_query, page.limits());
        let next_page = if res.is_empty() { 0 } else { page.0+1 };
        render!(search::result(
            &(&*conn, &intl.catalog, user),
            &str_query,
            res,
            page.0,
            next_page
        ))
    }
}