2021-01-01 00:54:13 +01:00
|
|
|
// CRATES
|
2021-05-15 22:59:42 +02:00
|
|
|
use crate::utils::{catch_random, error, format_num, format_url, param, setting, template, val, Post, Preferences};
|
2021-05-16 17:53:39 +02:00
|
|
|
use crate::{
|
|
|
|
|
client::json,
|
|
|
|
|
subreddit::{can_access_quarantine, quarantine},
|
|
|
|
|
RequestExt,
|
|
|
|
|
};
|
2021-01-01 00:54:13 +01:00
|
|
|
use askama::Template;
|
2021-03-17 23:30:33 +01:00
|
|
|
use hyper::{Body, Request, Response};
|
2021-01-01 00:54:13 +01:00
|
|
|
|
|
|
|
|
// STRUCTS
|
2021-01-03 07:37:54 +01:00
|
|
|
struct SearchParams {
|
|
|
|
|
q: String,
|
|
|
|
|
sort: String,
|
|
|
|
|
t: String,
|
|
|
|
|
before: String,
|
|
|
|
|
after: String,
|
|
|
|
|
restrict_sr: String,
|
|
|
|
|
}
|
|
|
|
|
|
2021-01-14 19:22:50 +01:00
|
|
|
// STRUCTS
|
|
|
|
|
struct Subreddit {
|
|
|
|
|
name: String,
|
|
|
|
|
url: String,
|
2021-03-19 05:32:54 +01:00
|
|
|
icon: String,
|
2021-01-14 19:22:50 +01:00
|
|
|
description: String,
|
2021-03-20 23:42:47 +01:00
|
|
|
subscribers: (String, String),
|
2021-01-14 19:22:50 +01:00
|
|
|
}
|
|
|
|
|
|
2021-01-01 00:54:13 +01:00
|
|
|
#[derive(Template)]
|
|
|
|
|
#[template(path = "search.html", escape = "none")]
|
|
|
|
|
struct SearchTemplate {
|
|
|
|
|
posts: Vec<Post>,
|
2021-01-14 19:22:50 +01:00
|
|
|
subreddits: Vec<Subreddit>,
|
2021-01-01 00:54:13 +01:00
|
|
|
sub: String,
|
2021-01-03 07:37:54 +01:00
|
|
|
params: SearchParams,
|
2021-01-09 02:35:04 +01:00
|
|
|
prefs: Preferences,
|
2021-05-10 03:25:52 +02:00
|
|
|
url: String,
|
2021-01-01 00:54:13 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// SERVICES
|
2021-03-17 23:30:33 +01:00
|
|
|
pub async fn find(req: Request<Body>) -> Result<Response<Body>, String> {
|
2021-05-15 22:59:42 +02:00
|
|
|
let nsfw_results = if setting(&req, "show_nsfw") == "on" { "&include_over_18=on" } else { "" };
|
2021-03-17 23:30:33 +01:00
|
|
|
let path = format!("{}.json?{}{}", req.uri().path(), req.uri().query().unwrap_or_default(), nsfw_results);
|
|
|
|
|
let sub = req.param("sub").unwrap_or_default();
|
2021-05-16 17:53:39 +02:00
|
|
|
let quarantined = can_access_quarantine(&req, &sub);
|
2021-05-09 17:40:49 +02:00
|
|
|
// Handle random subreddits
|
|
|
|
|
if let Ok(random) = catch_random(&sub, "/find").await {
|
|
|
|
|
return Ok(random);
|
|
|
|
|
}
|
2021-05-16 17:53:39 +02:00
|
|
|
let query = param(&path, "q").unwrap_or_default();
|
2021-01-14 20:45:04 +01:00
|
|
|
|
2021-05-16 17:53:39 +02:00
|
|
|
let sort = param(&path, "sort").unwrap_or("relevance".to_string());
|
2021-01-14 19:22:50 +01:00
|
|
|
|
2021-05-16 17:53:39 +02:00
|
|
|
let subreddits = match param(&path, "restrict_sr") {
|
|
|
|
|
None => search_subreddits(&query).await,
|
|
|
|
|
Some(_) => Vec::new()
|
2021-01-14 20:45:04 +01:00
|
|
|
};
|
2021-01-01 00:54:13 +01:00
|
|
|
|
2021-05-10 03:25:52 +02:00
|
|
|
let url = String::from(req.uri().path_and_query().map_or("", |val| val.as_str()));
|
|
|
|
|
|
2021-05-16 17:53:39 +02:00
|
|
|
match Post::fetch(&path, String::new(), quarantined).await {
|
2021-02-09 18:38:52 +01:00
|
|
|
Ok((posts, after)) => template(SearchTemplate {
|
|
|
|
|
posts,
|
|
|
|
|
subreddits,
|
|
|
|
|
sub,
|
|
|
|
|
params: SearchParams {
|
2021-02-20 22:59:16 +01:00
|
|
|
q: query.replace('"', """),
|
2021-02-09 18:38:52 +01:00
|
|
|
sort,
|
2021-05-16 17:53:39 +02:00
|
|
|
t: param(&path, "t").unwrap_or_default(),
|
|
|
|
|
before: param(&path, "after").unwrap_or_default(),
|
2021-02-09 18:38:52 +01:00
|
|
|
after,
|
2021-05-16 17:53:39 +02:00
|
|
|
restrict_sr: param(&path, "restrict_sr").unwrap_or_default(),
|
2021-02-09 18:38:52 +01:00
|
|
|
},
|
2021-02-25 06:29:23 +01:00
|
|
|
prefs: Preferences::new(req),
|
2021-05-10 03:25:52 +02:00
|
|
|
url,
|
2021-02-09 18:38:52 +01:00
|
|
|
}),
|
2021-05-16 17:53:39 +02:00
|
|
|
Err(msg) => {
|
|
|
|
|
if msg == "quarantined" {
|
|
|
|
|
let sub = req.param("sub").unwrap_or_default();
|
|
|
|
|
quarantine(req, sub)
|
|
|
|
|
} else {
|
|
|
|
|
error(req, msg).await
|
|
|
|
|
}
|
|
|
|
|
}
|
2021-01-01 00:54:13 +01:00
|
|
|
}
|
|
|
|
|
}
|
2021-01-14 20:45:04 +01:00
|
|
|
|
2021-02-20 22:59:16 +01:00
|
|
|
async fn search_subreddits(q: &str) -> Vec<Subreddit> {
|
2021-01-14 20:45:04 +01:00
|
|
|
let subreddit_search_path = format!("/subreddits/search.json?q={}&limit=3", q.replace(' ', "+"));
|
|
|
|
|
|
|
|
|
|
// Send a request to the url
|
2021-05-16 17:53:39 +02:00
|
|
|
match json(subreddit_search_path, false).await {
|
2021-01-14 20:45:04 +01:00
|
|
|
// If success, receive JSON in response
|
|
|
|
|
Ok(response) => {
|
|
|
|
|
match response["data"]["children"].as_array() {
|
|
|
|
|
// For each subreddit from subreddit list
|
|
|
|
|
Some(list) => list
|
|
|
|
|
.iter()
|
2021-03-19 05:32:54 +01:00
|
|
|
.map(|subreddit| {
|
|
|
|
|
// Fetch subreddit icon either from the community_icon or icon_img value
|
|
|
|
|
let community_icon: &str = subreddit["data"]["community_icon"].as_str().map_or("", |s| s.split('?').collect::<Vec<&str>>()[0]);
|
|
|
|
|
let icon = if community_icon.is_empty() {
|
|
|
|
|
val(&subreddit, "icon_img")
|
|
|
|
|
} else {
|
|
|
|
|
community_icon.to_string()
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
Subreddit {
|
|
|
|
|
name: val(subreddit, "display_name_prefixed"),
|
|
|
|
|
url: val(subreddit, "url"),
|
|
|
|
|
icon: format_url(&icon),
|
|
|
|
|
description: val(subreddit, "public_description"),
|
|
|
|
|
subscribers: format_num(subreddit["data"]["subscribers"].as_f64().unwrap_or_default() as i64),
|
|
|
|
|
}
|
2021-01-14 20:45:04 +01:00
|
|
|
})
|
|
|
|
|
.collect::<Vec<Subreddit>>(),
|
|
|
|
|
_ => Vec::new(),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
// If the Reddit API returns an error, exit this function
|
|
|
|
|
_ => Vec::new(),
|
|
|
|
|
}
|
|
|
|
|
}
|
