libreddit/src/main.rs

88 lines
2.8 KiB
Rust
Raw Normal View History

2020-10-25 21:25:59 +01:00
// Import Crates
2020-12-20 04:54:46 +01:00
use actix_web::{get, middleware::NormalizePath, web, App, HttpResponse, HttpServer};
2020-10-25 21:25:59 +01:00
// Reference local files
mod post;
2020-11-30 03:50:29 +01:00
mod proxy;
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
// Create Services
2020-11-19 01:31:46 +01:00
async fn style() -> HttpResponse {
2020-11-29 22:46:53 +01:00
HttpResponse::Ok().content_type("text/css").body(include_str!("../static/style.css"))
2020-10-25 21:25:59 +01:00
}
2020-11-21 04:33:38 +01:00
async fn robots() -> HttpResponse {
2021-01-06 23:19:10 +01:00
HttpResponse::Ok()
.header("Cache-Control", "public, max-age=1209600, s-maxage=86400")
.body(include_str!("../static/robots.txt"))
2020-11-21 04:33:38 +01:00
}
2020-10-25 21:25:59 +01:00
#[get("/favicon.ico")]
async fn favicon() -> HttpResponse {
HttpResponse::Ok().body("")
}
#[actix_web::main]
async fn main() -> std::io::Result<()> {
2020-11-23 04:21:07 +01:00
let args: Vec<String> = std::env::args().collect();
let mut address = "0.0.0.0:8080".to_string();
if args.len() > 1 {
2020-11-24 03:25:22 +01:00
for arg in args {
if arg.starts_with("--address=") || arg.starts_with("-a=") {
2021-01-01 21:33:57 +01:00
let split: Vec<&str> = arg.split('=').collect();
2020-11-24 03:25:22 +01:00
address = split[1].to_string();
}
2020-11-23 04:21:07 +01:00
}
}
2020-10-26 04:57:19 +01:00
// start http server
2021-01-01 21:55:09 +01:00
println!("Running Libreddit v{} on {}!", env!("CARGO_PKG_VERSION"), &address);
2020-11-19 03:50:59 +01:00
2020-10-25 21:25:59 +01:00
HttpServer::new(|| {
2020-10-26 04:57:19 +01:00
App::new()
2020-12-15 01:35:04 +01:00
// TRAILING SLASH MIDDLEWARE
.wrap(NormalizePath::default())
2021-01-01 06:03:44 +01:00
// DEFAULT SERVICE
.default_service(web::get().to(utils::error))
2020-10-26 04:57:19 +01:00
// GENERAL SERVICES
2020-12-15 01:35:04 +01:00
.route("/style.css/", web::get().to(style))
2021-01-01 21:33:57 +01:00
.route("/favicon.ico/", web::get().to(HttpResponse::Ok))
2020-12-15 01:35:04 +01:00
.route("/robots.txt/", web::get().to(robots))
2021-01-03 05:50:23 +01:00
// SETTINGS SERVICE
2021-01-06 03:04:49 +01:00
.route("/settings/", web::get().to(settings::get))
.route("/settings/", web::post().to(settings::set))
2020-11-23 05:22:51 +01:00
// PROXY SERVICE
2020-12-15 01:35:04 +01:00
.route("/proxy/{url:.*}/", web::get().to(proxy::handler))
2021-01-01 00:54:13 +01:00
// SEARCH SERVICES
2021-01-01 06:03:44 +01:00
.route("/search/", web::get().to(search::find))
.route("r/{sub}/search/", web::get().to(search::find))
2020-12-15 01:35:04 +01:00
// USER SERVICES
2021-01-01 00:54:13 +01:00
.route("/u/{username}/", web::get().to(user::profile))
.route("/user/{username}/", web::get().to(user::profile))
2021-01-02 07:21:43 +01:00
// WIKI SERVICES
2021-01-02 19:58:21 +01:00
.route("/wiki/", web::get().to(subreddit::wiki))
.route("/wiki/{page}/", web::get().to(subreddit::wiki))
2021-01-02 07:21:43 +01:00
.route("/r/{sub}/wiki/", web::get().to(subreddit::wiki))
.route("/r/{sub}/wiki/{page}/", web::get().to(subreddit::wiki))
2020-10-26 04:57:19 +01:00
// SUBREDDIT SERVICES
2020-12-15 01:35:04 +01:00
.route("/r/{sub}/", web::get().to(subreddit::page))
2021-01-04 06:00:36 +01:00
.route("/r/{sub}/{sort:hot|new|top|rising|controversial}/", web::get().to(subreddit::page))
2020-10-26 04:57:19 +01:00
// POPULAR SERVICES
2021-01-01 00:54:13 +01:00
.route("/", web::get().to(subreddit::page))
2021-01-04 06:00:36 +01:00
.route("/{sort:best|hot|new|top|rising|controversial}/", web::get().to(subreddit::page))
2020-12-15 01:35:04 +01:00
// POST SERVICES
2021-01-01 00:54:13 +01:00
.route("/{id:.{5,6}}/", web::get().to(post::item))
.route("/r/{sub}/comments/{id}/{title}/", web::get().to(post::item))
.route("/r/{sub}/comments/{id}/{title}/{comment_id}/", web::get().to(post::item))
2020-10-26 04:57:19 +01:00
})
2021-01-01 21:55:09 +01:00
.bind(&address)
2021-01-01 21:33:57 +01:00
.unwrap_or_else(|_| panic!("Cannot bind to the address: {}", address))
2020-10-26 04:57:19 +01:00
.run()
.await
}