1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
use std::{env, error::Error, collections::HashMap, process::exit, sync::Arc};
use reqwest::ClientBuilder;
use serde::Deserialize;
use axum::{Router, Extension};
use tera::Tera;
mod routes;
pub struct State {
pub templates: Tera,
pub pages: HashMap<String, (String, String)>
}
#[derive(Deserialize)]
pub struct Page {
name: String,
url: String
}
async fn load_state() -> Result<State, Box<dyn Error>> {
let mut templates = Tera::new("public/templates/**")?;
templates.autoescape_on(vec![]);
let client = ClientBuilder::new()
.use_rustls_tls()
.build()?;
let json = client.get("https://raw.githubusercontent.com/bucketfish/bucket-webring/master/webring.json")
.send()
.await?
.json::<Vec<Page>>()
.await?;
let mut pages = HashMap::new();
for (i, page) in json.iter().enumerate() {
let prev = json[(i + json.len() - 1) % json.len()].url.clone();
let next = json[(i + json.len() + 1) % json.len()].url.clone();
pages.insert(page.name.clone(), (prev, next));
}
Ok(State{ templates, pages })
}
#[tokio::main]
async fn main() {
let port: u16 = env::var("PORT")
.unwrap_or_else(|_| String::new())
.parse::<u16>()
.unwrap_or(8080);
let addr = format!("[::]:{port}")
.parse::<std::net::SocketAddr>()
.expect("Failed to create port binding");
let state = match load_state().await {
Ok(s) => s,
Err(e) => {
eprintln!("Parsing error: {e}");
exit(1);
}
};
let router = Router::new()
.fallback_service(routes::router())
.layer(Extension(Arc::new(state)));
axum::Server::bind(&addr)
.serve(router.into_make_service())
.await
.expect("Failed to launch web server");
}
|
