feat: add version headers

src/server/core.rs

@@ -1,8 +1,9 @@
+use super::compress;
+use super::execute;
+use super::judge;
+use super::jwt;
+use super::version;
 use crate::config::*;
-use crate::server::compress;
-use crate::server::execute;
-use crate::server::judge;
-use crate::server::jwt;
 use rocket::data::ByteUnit;
 use rocket::serde::{json::Json, Deserialize, Serialize};
 use rocket::Build;
@@ -51,6 +52,8 @@ pub fn rocket() -> Rocket<Build> {
             routes![index, info, jwt::validate, execute::execute, judge::judge],
         );
 
+    let server = server.attach(version::fairing());
+
     if cfg!(debug_assertions) {
         server
     } else {

src/server/execute.rs

@@ -1,6 +1,6 @@
+use super::jwt;
 use crate::config::*;
 use crate::run;
-use crate::server::jwt;
 use base64::{engine::general_purpose, Engine as _};
 use rocket::serde::{
     json::{Error, Json},

src/server/judge.rs

@@ -1,6 +1,6 @@
+use super::jwt;
 use crate::judger::{Input, JudgeSpec, Judger, Output};
 use crate::run;
-use crate::server::jwt;
 use base64::engine::general_purpose;
 use base64::Engine;
 use reqwest::Client;

src/server/mod.rs

@@ -3,3 +3,4 @@ pub mod core;
 mod execute;
 mod judge;
 mod jwt;
+mod version;

src/server/version.rs

@@ -0,0 +1,13 @@
+use rocket::{fairing::AdHoc, http::Header};
+
+pub fn fairing() -> AdHoc {
+    AdHoc::on_response("Version Headers", |_req, res| {
+        Box::pin(async move {
+            res.set_header(Header::new(
+                "X-Powered-By",
+                "WARK (https://github.com/wasm-oj/wark)",
+            ));
+            res.set_header(Header::new("X-Version", env!("VERGEN_GIT_DESCRIBE")));
+        })
+    })
+}