mirror of https://github.com/spikecodes/libreddit
266 lines
8.9 KiB
Rust
266 lines
8.9 KiB
Rust
// Global specifiers
|
|
#![forbid(unsafe_code)]
|
|
#![warn(clippy::pedantic, clippy::all)]
|
|
#![allow(
|
|
clippy::needless_pass_by_value,
|
|
clippy::match_wildcard_for_single_variants,
|
|
clippy::cast_possible_truncation,
|
|
clippy::similar_names,
|
|
clippy::cast_possible_wrap,
|
|
clippy::find_map
|
|
)]
|
|
|
|
// Reference local files
|
|
mod post;
|
|
mod search;
|
|
mod settings;
|
|
mod subreddit;
|
|
mod user;
|
|
mod utils;
|
|
|
|
// Import Crates
|
|
use clap::{App as cli, Arg};
|
|
|
|
use futures_lite::FutureExt;
|
|
use hyper::{header::HeaderValue, Body, Request, Response};
|
|
|
|
mod client;
|
|
use client::proxy;
|
|
use server::RequestExt;
|
|
use utils::{error, redirect};
|
|
|
|
mod server;
|
|
|
|
// Create Services
|
|
|
|
// Required for the manifest to be valid
|
|
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
|
|
async fn iphone_logo() -> Result<Response<Body>, String> {
|
|
Ok(
|
|
Response::builder()
|
|
.status(200)
|
|
.header("content-type", "image/png")
|
|
.body(include_bytes!("../static/apple-touch-icon.png").as_ref().into())
|
|
.unwrap_or_default(),
|
|
)
|
|
}
|
|
|
|
async fn favicon() -> Result<Response<Body>, String> {
|
|
Ok(
|
|
Response::builder()
|
|
.status(200)
|
|
.header("content-type", "image/vnd.microsoft.icon")
|
|
.header("Cache-Control", "public, max-age=1209600, s-maxage=86400")
|
|
.body(include_bytes!("../static/favicon.ico").as_ref().into())
|
|
.unwrap_or_default(),
|
|
)
|
|
}
|
|
|
|
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();
|
|
|
|
if cache {
|
|
if let Ok(val) = HeaderValue::from_str("public, max-age=1209600, s-maxage=86400") {
|
|
res.headers_mut().insert("Cache-Control", val);
|
|
}
|
|
}
|
|
|
|
Ok(res)
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let matches = cli::new("Libreddit")
|
|
.version(env!("CARGO_PKG_VERSION"))
|
|
.about("Private front-end for Reddit written in Rust ")
|
|
.arg(
|
|
Arg::with_name("address")
|
|
.short("a")
|
|
.long("address")
|
|
.value_name("ADDRESS")
|
|
.help("Sets address to listen on")
|
|
.default_value("0.0.0.0")
|
|
.takes_value(true),
|
|
)
|
|
.arg(
|
|
Arg::with_name("port")
|
|
.short("p")
|
|
.long("port")
|
|
.value_name("PORT")
|
|
.help("Port to listen on")
|
|
.default_value("8080")
|
|
.takes_value(true),
|
|
)
|
|
.arg(
|
|
Arg::with_name("redirect-https")
|
|
.short("r")
|
|
.long("redirect-https")
|
|
.help("Redirect all HTTP requests to HTTPS (no longer functional)")
|
|
.takes_value(false),
|
|
)
|
|
.arg(
|
|
Arg::with_name("hsts")
|
|
.short("H")
|
|
.long("hsts")
|
|
.value_name("EXPIRE_TIME")
|
|
.help("HSTS header to tell browsers that this site should only be accessed over HTTPS")
|
|
.default_value("604800")
|
|
.takes_value(true),
|
|
)
|
|
.get_matches();
|
|
|
|
let address = matches.value_of("address").unwrap_or("0.0.0.0");
|
|
let port = matches.value_of("port").unwrap_or("8080");
|
|
let hsts = matches.value_of("hsts");
|
|
|
|
let listener = format!("{}:{}", address, port);
|
|
|
|
println!("Starting Libreddit...");
|
|
|
|
// Begin constructing a server
|
|
let mut app = server::Server::new();
|
|
|
|
// Define default headers (added to all responses)
|
|
app.default_headers = headers! {
|
|
"Referrer-Policy" => "no-referrer",
|
|
"X-Content-Type-Options" => "nosniff",
|
|
"X-Frame-Options" => "DENY",
|
|
"Content-Security-Policy" => "default-src 'none'; 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:;"
|
|
};
|
|
|
|
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(|_| resource(include_str!("../static/style.css"), "text/css", false).boxed());
|
|
app
|
|
.at("/manifest.json")
|
|
.get(|_| resource(include_str!("../static/manifest.json"), "application/json", false).boxed());
|
|
app.at("/robots.txt").get(|_| resource("User-agent: *\nAllow: /", "text/plain", true).boxed());
|
|
app.at("/favicon.ico").get(|_| favicon().boxed());
|
|
app.at("/logo.png").get(|_| pwa_logo().boxed());
|
|
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
|
|
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/:id").get(|r| proxy(r, "https://i.redd.it/{id}").boxed());
|
|
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/:id").get(|r| proxy(r, "https://{loc}view.redd.it/{id}").boxed());
|
|
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
|
|
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());
|
|
|
|
app.at("/user/[deleted]").get(|req| error(req, "User has deleted their account".to_string()).boxed());
|
|
app.at("/user/:name").get(|r| user::profile(r).boxed());
|
|
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
|
|
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());
|
|
|
|
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(r).boxed());
|
|
app.at("/r/:sub/unsubscribe").post(|r| subreddit::subscriptions(r).boxed());
|
|
|
|
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("/r/:sub/search").get(|r| search::find(r).boxed());
|
|
|
|
app
|
|
.at("/r/:sub/w")
|
|
.get(|r| async move { Ok(redirect(format!("/r/{}/wiki", r.param("sub").unwrap_or_default()))) }.boxed());
|
|
app
|
|
.at("/r/:sub/w/*page")
|
|
.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());
|
|
app.at("/r/:sub/wiki/*page").get(|r| subreddit::wiki(r).boxed());
|
|
|
|
app.at("/r/:sub/about/sidebar").get(|r| subreddit::sidebar(r).boxed());
|
|
|
|
app.at("/r/:sub/:sort").get(|r| subreddit::community(r).boxed());
|
|
|
|
// Comments handler
|
|
app.at("/comments/:id").get(|r| post::item(r).boxed());
|
|
|
|
// Front page
|
|
app.at("/").get(|r| subreddit::community(r).boxed());
|
|
|
|
// View Reddit wiki
|
|
app.at("/w").get(|_| async { Ok(redirect("/wiki".to_string())) }.boxed());
|
|
app
|
|
.at("/w/*page")
|
|
.get(|r| async move { Ok(redirect(format!("/wiki/{}", r.param("page").unwrap_or_default()))) }.boxed());
|
|
app.at("/wiki").get(|r| subreddit::wiki(r).boxed());
|
|
app.at("/wiki/*page").get(|r| subreddit::wiki(r).boxed());
|
|
|
|
// Search all of Reddit
|
|
app.at("/search").get(|r| search::find(r).boxed());
|
|
|
|
// Handle about pages
|
|
app.at("/about").get(|req| error(req, "About pages aren't added yet".to_string()).boxed());
|
|
|
|
app.at("/:id").get(|req: Request<Body>| match req.param("id").as_deref() {
|
|
// Sort front page
|
|
Some("best") | Some("hot") | Some("new") | Some("top") | Some("rising") | Some("controversial") => subreddit::community(req).boxed(),
|
|
// Short link for post
|
|
Some(id) if id.len() > 4 && id.len() < 7 => post::item(req).boxed(),
|
|
// Error message for unknown pages
|
|
_ => error(req, "Nothing here".to_string()).boxed(),
|
|
});
|
|
|
|
// Default service in case no routes match
|
|
app.at("/*").get(|req| error(req, "Nothing here".to_string()).boxed());
|
|
|
|
println!("Running Libreddit v{} on {}!", env!("CARGO_PKG_VERSION"), listener);
|
|
|
|
let server = app.listen(listener);
|
|
|
|
// Run this server for... forever!
|
|
if let Err(e) = server.await {
|
|
eprintln!("Server error: {}", e);
|
|
}
|
|
}
|