ari/mcstatusface
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix some sloppy unwraps, teensy refactor

Browse source
This commit is contained in:
ari melody 2025-06-16 22:05:01 +01:00
parent 75bd0000bb
commit c765db8148
Signed by: ari
GPG key ID: CF99829C92678188
2 changed files with 65 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

101
src/main.rs
View file

@ -49,17 +49,17 @@ env!("CARGO_PKG_VERSION"));
if args.len() > 2 { address = args[2].to_string() }
let trusted_proxies: Vec<IpAddr> =
match env::var("MCSTATUSFACE_TRUSTED_PROXIES") {
Err(_) => { vec![] }
Ok(envar) => {
let mut trusted_proxies: Vec<IpAddr> = Vec::new();
for addr in envar.split(",") {
match IpAddr::from_str(addr) {
Ok(addr) => { trusted_proxies.push(addr); }
Err(_) => {}
Ok(addr) => { trusted_proxies.push(addr); }
}
}
trusted_proxies
}
Err(_) => { vec![] }
};
HttpServer::new(address, 64, trusted_proxies).start(|request, mut response| {
@ -90,6 +90,9 @@ env!("CARGO_PKG_VERSION"));
response.status(StatusCode::OK);
let query_response: String;
match request.query().get("s") {
None => {
query_response = String::from("");
}
Some(query_address) => {
let mut address = query_address.to_string();
if !address.contains(":") { address.push_str(":25565"); }
@ -103,17 +106,25 @@ env!("CARGO_PKG_VERSION"));
Ok(mut addrs_iter) => {
let address = addrs_iter.next().unwrap();
let status = MinecraftStatus::fetch(address).unwrap();
match MinecraftStatus::fetch(address) {
Err(_) => {
query_response = format!(
"<hr/>
<h2>Server Details</h2>
<pre><code>Failed to connect to {}.</pre></code>",
sanitize_html(&address.to_string()),
);
}
Ok(status) => {
let minecraft_status = MinecraftStatusResponse{
version: &status.version.name,
players: status.players.online,
max_players: status.players.max,
motd: status.parse_description(),
};
let minecraft_status = MinecraftStatusResponse{
version: &status.version.name,
players: status.players.online,
max_players: status.players.max,
motd: status.parse_description(),
};
query_response = format!(
"<hr/>
query_response = format!(
"<hr/>
<h2>Server Details</h2>
<p>
<strong>Version:</strong> <code>{}</code><br/>
@ -125,20 +136,19 @@ env!("CARGO_PKG_VERSION"));
minecraft_status.players,
minecraft_status.max_players,
sanitize_html(&minecraft_status.motd).to_string(),
);
);
}
}
}
}
}
None => {
query_response = String::from("");
}
}
let response_content = content
.replace("{{response}}", &query_response)
.replace("{{host}}", match request.headers().get("host") {
Some(host) => { host }
None => { "mcq.bliss.town" }
Some(host) => { host }
});
response.body(response_content.to_string());
return response.send();
@ -154,31 +164,44 @@ env!("CARGO_PKG_VERSION"));
Some(query_address) => {
let mut address = query_address.to_string();
if !address.contains(":") { address.push_str(":25565"); }
let mut addrs_iter = address.to_socket_addrs().unwrap();
let address = addrs_iter.next().unwrap();
let status = MinecraftStatus::fetch(address).unwrap();
let minecraft_status = MinecraftStatusResponse{
version: &status.version.name,
players: status.players.online,
max_players: status.players.max,
motd: status.parse_description(),
};
match serde_json::to_string(&minecraft_status) {
Ok(json) => {
response.status(StatusCode::OK);
response.set_header("Content-Type", "application/json".to_string());
response.body(json + "\n");
}
Err(e) => {
eprintln!("Request to {address} failed: {e}");
match address.to_socket_addrs() {
Err(_) => {
response.status(StatusCode::InternalServerError);
response.body("Unable to reach the requested server.\n".to_string());
response.body("Invalid server address.\n".to_string());
}
Ok(mut addrs_iter) => {
let address = addrs_iter.next().unwrap();
match MinecraftStatus::fetch(address) {
Err(_) => {
response.status(StatusCode::InternalServerError);
response.body(format!("Failed to connect to {address}.\n"));
}
Ok(status) => {
let minecraft_status = MinecraftStatusResponse{
version: &status.version.name,
players: status.players.online,
max_players: status.players.max,
motd: status.parse_description(),
};
match serde_json::to_string(&minecraft_status) {
Err(e) => {
eprintln!("Failed to parse status for {address}: {e}");
response.status(StatusCode::InternalServerError);
response.body(format!("Failed to parse response from {address}.\n"));
}
Ok(json) => {
response.status(StatusCode::OK);
response.set_header("Content-Type", "application/json".to_string());
response.body(json + "\n");
}
}
}
}
}
}
return response.send()
}
}

4
src/status.rs
View file

@ -51,8 +51,10 @@ impl MinecraftStatus {
pub fn fetch(address: SocketAddr) -> Result<MinecraftStatus> {
// println!("Connecting to {address}...");
let mut stream = TcpStream::connect(address.to_string()).unwrap();
let stream = TcpStream::connect(address.to_string());
if stream.is_err() { return Err(stream.unwrap_err()); }
// println!("Connected!");
let mut stream = stream.unwrap();
let mut send_buffer: Vec<u8> = Vec::new();