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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// This file is part of Windmark <https://github.com/gemrest/windmark>.
// Copyright (C) 2022-2022 Fuwn <[email protected]>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, version 3.
//
// This program is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// Copyright (C) 2022-2022 Fuwn <[email protected]>
// SPDX-License-Identifier: GPL-3.0-only
//! `cargo run --example windmark --features logger`
#[macro_use]
extern crate log;
use windmark::Response;
#[windmark::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
windmark::Router::new()
.set_private_key_file("windmark_private.pem")
.set_certificate_chain_file("windmark_pair.pem")
.enable_default_logger(true)
.set_error_handler(|_| Response::PermanentFailure("error...".to_string()))
.attach(|r| {
r.mount("/module", |_| Response::Success("This is a module!".into()));
})
.set_pre_route_callback(|stream, url, _| {
info!(
"accepted connection from {} to {}",
stream.peer_addr().unwrap().ip(),
url.to_string()
)
})
.set_post_route_callback(|stream, _url, _| {
info!(
"closed connection from {}",
stream.peer_addr().unwrap().ip()
)
})
.set_header(|_| "```\nART IS COOL\n```".to_string())
.set_footer(|_| "Copyright 2022".to_string())
.mount("/", |_| {
Response::Success(
"# INDEX\n\nWelcome!\n\n=> /test Test Page\n=> /time Unix Epoch\n"
.to_string(),
)
})
.mount("/ip", |context| {
Response::Success(
{ format!("Hello, {}", context.tcp.peer_addr().unwrap().ip()) }.into(),
)
})
.mount("/test", |_| {
Response::Success("hi there\n=> / back".to_string())
})
.mount("/temporary-failure", |_| {
Response::TemporaryFailure("Woops, temporarily...".into())
})
.mount("/time", |_| {
Response::Success(
std::time::UNIX_EPOCH
.elapsed()
.unwrap()
.as_nanos()
.to_string(),
)
})
.mount("/query", |context| {
Response::Success(format!(
"queries: {:?}",
windmark::utilities::queries_from_url(&context.url)
))
})
.mount("/param/:lang", |context| {
Response::Success(format!(
"Parameter lang is {}",
context.params.get("lang").unwrap()
))
})
.mount("/names/:first/:last", |context| {
Response::Success(format!(
"{} {}",
context.params.get("first").unwrap(),
context.params.get("last").unwrap()
))
})
.mount("/input", |context| {
if let Some(name) = context.url.query() {
Response::Success(format!("Your name is {}!", name))
} else {
Response::Input("What is your name?".into())
}
})
.mount("/sensitive-input", |context| {
if let Some(password) = context.url.query() {
Response::Success(format!("Your password is {}!", password))
} else {
Response::SensitiveInput("What is your password?".into())
}
})
.mount("/error", |_| Response::CertificateNotValid("no".into()))
.mount("/redirect", |_| {
Response::PermanentRedirect("gemini://localhost/test".into())
})
.mount("/file", |_| {
Response::SuccessFile(include_bytes!("../LICENSE"))
})
.run()
.await
}
|