libreddit/src/main.rs

346 lines
12 KiB
Rust
Raw Permalink Normal View History

2021-03-09 03:49:06 +01:00
// Global specifiers
#![forbid(unsafe_code)]
#![allow(clippy::cmp_owned)]
2021-03-09 03:49:06 +01:00
2020-10-25 21:25:59 +01:00
// Reference local files
mod config;
2022-11-09 17:16:51 +01:00
mod duplicates;
mod instance_info;
2020-10-25 21:25:59 +01:00
mod post;
2021-01-01 00:54:13 +01:00
mod search;
2021-01-06 03:04:49 +01:00
mod settings;
2020-10-25 21:25:59 +01:00
mod subreddit;
2020-10-26 04:57:19 +01:00
mod user;
2020-11-25 22:53:30 +01:00
mod utils;
2020-10-25 21:25:59 +01:00
2021-02-14 00:02:38 +01:00
// Import Crates
use clap::{Arg, ArgAction, Command};
2021-02-14 00:02:38 +01:00
2021-03-17 23:30:33 +01:00
use futures_lite::FutureExt;
use hyper::{header::HeaderValue, Body, Request, Response};
2021-03-17 23:30:33 +01:00
mod client;
use client::{canonical_path, proxy};
use once_cell::sync::Lazy;
2021-03-17 23:30:33 +01:00
use server::RequestExt;
use utils::{error, redirect, ThemeAssets};
2021-03-17 23:30:33 +01:00
mod server;
2020-10-25 21:25:59 +01:00
// Create Services
// Required for the manifest to be valid
2021-03-17 23:30:33 +01:00
async fn pwa_logo() -> Result<Response<Body>, String> {
Ok(
Response::builder()
.status(200)
.header("content-type", "image/png")
.body(include_bytes!("../static/logo.png").as_ref().into())
.unwrap_or_default(),
)
}
// Required for iOS App Icons
2021-03-17 23:30:33 +01:00
async fn iphone_logo() -> Result<Response<Body>, String> {
Ok(
2021-03-17 23:30:33 +01:00
Response::builder()
.status(200)
.header("content-type", "image/png")
.body(include_bytes!("../static/apple-touch-icon.png").as_ref().into())
.unwrap_or_default(),
)
}
2021-03-17 23:30:33 +01:00
async fn favicon() -> Result<Response<Body>, String> {
Ok(
2021-03-17 23:30:33 +01:00
Response::builder()
.status(200)
.header("content-type", "image/vnd.microsoft.icon")
.header("Cache-Control", "public, max-age=1209600, s-maxage=86400")
2021-03-17 23:30:33 +01:00
.body(include_bytes!("../static/favicon.ico").as_ref().into())
.unwrap_or_default(),
)
2020-10-25 21:25:59 +01:00
}
2021-05-20 01:09:08 +02:00
async fn font() -> Result<Response<Body>, String> {
Ok(
Response::builder()
.status(200)
.header("content-type", "font/woff2")
.header("Cache-Control", "public, max-age=1209600, s-maxage=86400")
2021-05-20 01:09:08 +02:00
.body(include_bytes!("../static/Inter.var.woff2").as_ref().into())
.unwrap_or_default(),
)
}
2021-03-17 23:30:33 +01:00
async fn resource(body: &str, content_type: &str, cache: bool) -> Result<Response<Body>, String> {
let mut res = Response::builder()
.status(200)
.header("content-type", content_type)
.body(body.to_string().into())
.unwrap_or_default();
2021-02-24 18:26:01 +01:00
if cache {
2021-03-21 06:10:31 +01:00
if let Ok(val) = HeaderValue::from_str("public, max-age=1209600, s-maxage=86400") {
res.headers_mut().insert("Cache-Control", val);
}
2021-02-24 18:26:01 +01:00
}
Ok(res)
}
async fn style() -> Result<Response<Body>, String> {
let mut res = include_str!("../static/style.css").to_string();
for file in ThemeAssets::iter() {
2022-05-21 04:20:44 +02:00
res.push('\n');
let theme = ThemeAssets::get(file.as_ref()).unwrap();
res.push_str(std::str::from_utf8(theme.data.as_ref()).unwrap());
}
Ok(
Response::builder()
.status(200)
.header("content-type", "text/css")
.header("Cache-Control", "public, max-age=1209600, s-maxage=86400")
.body(res.to_string().into())
.unwrap_or_default(),
)
}
2021-03-17 23:30:33 +01:00
#[tokio::main]
async fn main() {
let matches = Command::new("Libreddit")
2021-02-18 20:40:10 +01:00
.version(env!("CARGO_PKG_VERSION"))
.about("Private front-end for Reddit written in Rust ")
2021-05-27 05:30:08 +02:00
.arg(
2022-01-06 01:39:56 +01:00
Arg::new("redirect-https")
.short('r')
2021-05-27 05:30:08 +02:00
.long("redirect-https")
.help("Redirect all HTTP requests to HTTPS (no longer functional)")
2022-11-01 04:23:59 +01:00
.num_args(0),
2021-05-27 05:30:08 +02:00
)
2021-02-18 20:40:10 +01:00
.arg(
2022-01-06 01:39:56 +01:00
Arg::new("address")
.short('a')
2021-02-18 20:40:10 +01:00
.long("address")
.value_name("ADDRESS")
.help("Sets address to listen on")
2021-02-18 20:49:50 +01:00
.default_value("0.0.0.0")
2022-11-01 04:23:59 +01:00
.num_args(1),
2021-02-18 20:49:50 +01:00
)
.arg(
2022-01-06 01:39:56 +01:00
Arg::new("port")
.short('p')
2021-02-18 20:49:50 +01:00
.long("port")
.value_name("PORT")
.env("PORT")
2021-02-18 20:49:50 +01:00
.help("Port to listen on")
.default_value("8080")
.action(ArgAction::Set)
2022-11-01 04:23:59 +01:00
.num_args(1),
2021-02-18 20:40:10 +01:00
)
2021-03-21 06:10:31 +01:00
.arg(
2022-01-06 01:39:56 +01:00
Arg::new("hsts")
.short('H')
2021-03-21 06:10:31 +01:00
.long("hsts")
.value_name("EXPIRE_TIME")
.help("HSTS header to tell browsers that this site should only be accessed over HTTPS")
.default_value("604800")
2022-11-01 04:23:59 +01:00
.num_args(1),
2021-03-21 06:10:31 +01:00
)
2021-02-18 20:40:10 +01:00
.get_matches();
let address = matches.get_one::<String>("address").unwrap();
let port = matches.get_one::<String>("port").unwrap();
2022-11-01 04:23:59 +01:00
let hsts = matches.get_one("hsts").map(|m: &String| m.as_str());
2021-02-18 20:40:10 +01:00
let listener = [address, ":", port].concat();
2020-11-23 04:21:07 +01:00
2021-02-24 20:00:04 +01:00
println!("Starting Libreddit...");
2021-03-17 23:30:33 +01:00
// Begin constructing a server
let mut app = server::Server::new();
// Force evaluation of statics. In instance_info case, we need to evaluate
// the timestamp so deploy date is accurate - in config case, we need to
// evaluate the configuration to avoid paying penalty at first request.
Lazy::force(&config::CONFIG);
Lazy::force(&instance_info::INSTANCE_INFO);
2021-03-17 23:30:33 +01:00
// Define default headers (added to all responses)
app.default_headers = headers! {
"Referrer-Policy" => "no-referrer",
"X-Content-Type-Options" => "nosniff",
"X-Frame-Options" => "DENY",
2021-05-20 01:09:08 +02:00
"Content-Security-Policy" => "default-src 'none'; font-src 'self'; script-src 'self' blob:; manifest-src 'self'; media-src 'self' data: blob: about:; style-src 'self' 'unsafe-inline'; base-uri 'none'; img-src 'self' data:; form-action 'self'; frame-ancestors 'none'; connect-src 'self'; worker-src blob:;"
2021-03-17 23:30:33 +01:00
};
2021-03-21 06:10:31 +01:00
if let Some(expire_time) = hsts {
if let Ok(val) = HeaderValue::from_str(&format!("max-age={}", expire_time)) {
app.default_headers.insert("Strict-Transport-Security", val);
}
}
// Read static files
app.at("/style.css").get(|_| style().boxed());
2021-02-25 18:07:45 +01:00
app
2021-03-17 23:30:33 +01:00
.at("/manifest.json")
.get(|_| resource(include_str!("../static/manifest.json"), "application/json", false).boxed());
2023-04-26 13:52:12 +02:00
app.at("/robots.txt").get(|_| {
resource(
if match config::get_setting("LIBREDDIT_ROBOTS_DISABLE_INDEXING") {
Some(val) => val == "on",
None => false,
} {
"User-agent: *\nDisallow: /"
} else {
"User-agent: *\nDisallow: /u/\nDisallow: /user/"
},
"text/plain",
true,
)
.boxed()
});
2021-03-17 23:30:33 +01:00
app.at("/favicon.ico").get(|_| favicon().boxed());
app.at("/logo.png").get(|_| pwa_logo().boxed());
2021-05-20 01:09:08 +02:00
app.at("/Inter.var.woff2").get(|_| font().boxed());
2021-03-17 23:30:33 +01:00
app.at("/touch-icon-iphone.png").get(|_| iphone_logo().boxed());
app.at("/apple-touch-icon.png").get(|_| iphone_logo().boxed());
app
.at("/playHLSVideo.js")
.get(|_| resource(include_str!("../static/playHLSVideo.js"), "text/javascript", false).boxed());
app
.at("/hls.min.js")
.get(|_| resource(include_str!("../static/hls.min.js"), "text/javascript", false).boxed());
// Proxy media through Libreddit
2021-03-17 23:30:33 +01:00
app.at("/vid/:id/:size").get(|r| proxy(r, "https://v.redd.it/{id}/DASH_{size}").boxed());
app.at("/hls/:id/*path").get(|r| proxy(r, "https://v.redd.it/{id}/{path}").boxed());
app.at("/img/*path").get(|r| proxy(r, "https://i.redd.it/{path}").boxed());
2021-03-17 23:30:33 +01:00
app.at("/thumb/:point/:id").get(|r| proxy(r, "https://{point}.thumbs.redditmedia.com/{id}").boxed());
app.at("/emoji/:id/:name").get(|r| proxy(r, "https://emoji.redditmedia.com/{id}/{name}").boxed());
app
.at("/preview/:loc/award_images/:fullname/:id")
.get(|r| proxy(r, "https://{loc}view.redd.it/award_images/{fullname}/{id}").boxed());
app.at("/preview/:loc/:id").get(|r| proxy(r, "https://{loc}view.redd.it/{id}").boxed());
2021-03-17 23:30:33 +01:00
app.at("/style/*path").get(|r| proxy(r, "https://styles.redditmedia.com/{path}").boxed());
app.at("/static/*path").get(|r| proxy(r, "https://www.redditstatic.com/{path}").boxed());
// Browse user profile
2021-03-17 23:30:33 +01:00
app
.at("/u/:name")
.get(|r| async move { Ok(redirect(format!("/user/{}", r.param("name").unwrap_or_default()))) }.boxed());
app.at("/u/:name/comments/:id/:title").get(|r| post::item(r).boxed());
app.at("/u/:name/comments/:id/:title/:comment_id").get(|r| post::item(r).boxed());
2021-03-18 05:40:55 +01:00
app.at("/user/[deleted]").get(|req| error(req, "User has deleted their account".to_string()).boxed());
2021-03-17 23:30:33 +01:00
app.at("/user/:name").get(|r| user::profile(r).boxed());
app.at("/user/:name/:listing").get(|r| user::profile(r).boxed());
2021-03-17 23:30:33 +01:00
app.at("/user/:name/comments/:id").get(|r| post::item(r).boxed());
app.at("/user/:name/comments/:id/:title").get(|r| post::item(r).boxed());
app.at("/user/:name/comments/:id/:title/:comment_id").get(|r| post::item(r).boxed());
// Configure settings
2021-03-17 23:30:33 +01:00
app.at("/settings").get(|r| settings::get(r).boxed()).post(|r| settings::set(r).boxed());
app.at("/settings/restore").get(|r| settings::restore(r).boxed());
app.at("/settings/update").get(|r| settings::update(r).boxed());
// Subreddit services
app
.at("/r/:sub")
.get(|r| subreddit::community(r).boxed())
.post(|r| subreddit::add_quarantine_exception(r).boxed());
2021-02-24 18:26:01 +01:00
2021-03-20 06:04:44 +01:00
app
.at("/r/u_:name")
.get(|r| async move { Ok(redirect(format!("/user/{}", r.param("name").unwrap_or_default()))) }.boxed());
app.at("/r/:sub/subscribe").post(|r| subreddit::subscriptions_filters(r).boxed());
app.at("/r/:sub/unsubscribe").post(|r| subreddit::subscriptions_filters(r).boxed());
app.at("/r/:sub/filter").post(|r| subreddit::subscriptions_filters(r).boxed());
app.at("/r/:sub/unfilter").post(|r| subreddit::subscriptions_filters(r).boxed());
2021-02-24 18:26:01 +01:00
2021-03-17 23:30:33 +01:00
app.at("/r/:sub/comments/:id").get(|r| post::item(r).boxed());
app.at("/r/:sub/comments/:id/:title").get(|r| post::item(r).boxed());
app.at("/r/:sub/comments/:id/:title/:comment_id").get(|r| post::item(r).boxed());
app.at("/comments/:id").get(|r| post::item(r).boxed());
app.at("/comments/:id/comments").get(|r| post::item(r).boxed());
app.at("/comments/:id/comments/:comment_id").get(|r| post::item(r).boxed());
app.at("/comments/:id/:title").get(|r| post::item(r).boxed());
app.at("/comments/:id/:title/:comment_id").get(|r| post::item(r).boxed());
2021-02-24 18:26:01 +01:00
2022-11-09 17:16:51 +01:00
app.at("/r/:sub/duplicates/:id").get(|r| duplicates::item(r).boxed());
app.at("/r/:sub/duplicates/:id/:title").get(|r| duplicates::item(r).boxed());
app.at("/duplicates/:id").get(|r| duplicates::item(r).boxed());
app.at("/duplicates/:id/:title").get(|r| duplicates::item(r).boxed());
2021-03-17 23:30:33 +01:00
app.at("/r/:sub/search").get(|r| search::find(r).boxed());
2021-02-24 18:26:01 +01:00
2021-03-18 05:40:55 +01:00
app
.at("/r/:sub/w")
.get(|r| async move { Ok(redirect(format!("/r/{}/wiki", r.param("sub").unwrap_or_default()))) }.boxed());
app
2021-04-15 06:53:17 +02:00
.at("/r/:sub/w/*page")
2021-03-18 05:40:55 +01:00
.get(|r| async move { Ok(redirect(format!("/r/{}/wiki/{}", r.param("sub").unwrap_or_default(), r.param("wiki").unwrap_or_default()))) }.boxed());
app.at("/r/:sub/wiki").get(|r| subreddit::wiki(r).boxed());
2021-04-15 06:53:17 +02:00
app.at("/r/:sub/wiki/*page").get(|r| subreddit::wiki(r).boxed());
2021-02-24 18:26:01 +01:00
2021-03-22 03:28:05 +01:00
app.at("/r/:sub/about/sidebar").get(|r| subreddit::sidebar(r).boxed());
2021-03-17 23:30:33 +01:00
app.at("/r/:sub/:sort").get(|r| subreddit::community(r).boxed());
// Front page
2021-03-17 23:30:33 +01:00
app.at("/").get(|r| subreddit::community(r).boxed());
// View Reddit wiki
app.at("/w").get(|_| async { Ok(redirect("/wiki".to_string())) }.boxed());
2021-03-18 05:40:55 +01:00
app
2021-04-15 06:53:17 +02:00
.at("/w/*page")
2021-03-18 05:40:55 +01:00
.get(|r| async move { Ok(redirect(format!("/wiki/{}", r.param("page").unwrap_or_default()))) }.boxed());
2021-03-17 23:30:33 +01:00
app.at("/wiki").get(|r| subreddit::wiki(r).boxed());
2021-04-15 06:53:17 +02:00
app.at("/wiki/*page").get(|r| subreddit::wiki(r).boxed());
// Search all of Reddit
2021-03-17 23:30:33 +01:00
app.at("/search").get(|r| search::find(r).boxed());
2021-02-21 19:13:20 +01:00
// Handle about pages
2021-03-17 23:30:33 +01:00
app.at("/about").get(|req| error(req, "About pages aren't added yet".to_string()).boxed());
// Instance info page
app.at("/info").get(|r| instance_info::instance_info(r).boxed());
app.at("/info.:extension").get(|r| instance_info::instance_info(r).boxed());
app.at("/:id").get(|req: Request<Body>| {
Box::pin(async move {
match req.param("id").as_deref() {
// Sort front page
Some("best" | "hot" | "new" | "top" | "rising" | "controversial") => subreddit::community(req).await,
// Short link for post
Some(id) if (5..8).contains(&id.len()) => match canonical_path(format!("/{}", id)).await {
Ok(path_opt) => match path_opt {
Some(path) => Ok(redirect(path)),
None => error(req, "Post ID is invalid. It may point to a post on a community that has been banned.").await,
},
Err(e) => error(req, e).await,
},
// Error message for unknown pages
_ => error(req, "Nothing here".to_string()).await,
}
})
});
// Default service in case no routes match
2021-03-17 23:30:33 +01:00
app.at("/*").get(|req| error(req, "Nothing here".to_string()).boxed());
2021-02-24 20:00:04 +01:00
println!("Running Libreddit v{} on {}!", env!("CARGO_PKG_VERSION"), listener);
2021-03-17 23:30:33 +01:00
let server = app.listen(listener);
2021-02-24 18:26:01 +01:00
2021-03-17 23:30:33 +01:00
// Run this server for... forever!
if let Err(e) = server.await {
eprintln!("Server error: {}", e);
}
2020-10-26 04:57:19 +01:00
}