support xml response type main

1 files changed, 7 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index 3859754..2d1d20a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,6 +2,7 @@ use core::fmt;
 use std::{
 	borrow::Cow,
 	collections::HashMap,
+	fmt::format,
 	io::{self, BufRead, BufReader, BufWriter, Write},
 	net::{IpAddr, SocketAddr},
 	path::PathBuf,
@@ -183,6 +184,7 @@ async fn form_response<'req>(
 		Err(e) => return e,
 		Ok(None) | Ok(Some("text/plain")) => ContentType::Plain,
 		Ok(Some("application/json")) => ContentType::Json,
+		Ok(Some("application/xml")) | Ok(Some("text/xml")) => ContentType::Xml,
 		Ok(Some(_)) => return error(415, "unsupported content-type"),
 	};
 
@@ -266,6 +268,9 @@ fn make_ip_response(requested_type: ContentType, response: Response) -> Vec<u8>
 	let body = match requested_type {
 		ContentType::Plain => format!("{client_addr}"),
 		ContentType::Json => format!(r#"{{"ip": "{client_addr}"}}"#),
+		ContentType::Xml => {
+			format!("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n<ip>{client_addr}</ip>")
+		}
 	};
 	let length = body.len();
 
@@ -457,6 +462,7 @@ impl Database {
 enum ContentType {
 	Plain,
 	Json,
+	Xml,
 }
 
 impl fmt::Display for ContentType {
@@ -464,6 +470,7 @@ impl fmt::Display for ContentType {
 		match self {
 			ContentType::Plain => write!(f, "text/plain"),
 			ContentType::Json => write!(f, "application/json"),
+			ContentType::Xml => write!(f, "application/xml"),
 		}
 	}
 }