71 lines
2.1 KiB
Rust
71 lines
2.1 KiB
Rust
use axum::{Router, response::IntoResponse, routing::get};
|
|
use reqwest::StatusCode;
|
|
use serde_json::json;
|
|
use std::env;
|
|
|
|
// Define an asynchronous main function with the tokio runtime macro
|
|
#[tokio::main]
|
|
async fn main() {
|
|
// Build our application with one route: a GET request to "/"
|
|
let app = Router::new().route("/", get(trigger_runner));
|
|
|
|
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
|
|
axum::serve(listener, app).await.unwrap();
|
|
}
|
|
// Handler that returns a simple string
|
|
async fn trigger_runner() -> impl IntoResponse {
|
|
let token = match env::var("RAILWAY_TOKEN") {
|
|
Ok(t) => t,
|
|
Err(_) => {
|
|
return (
|
|
StatusCode::INTERNAL_SERVER_ERROR,
|
|
"Missing Railway project token",
|
|
);
|
|
}
|
|
};
|
|
|
|
let service_id = match env::var("RUNNER_SERVICE_ID") {
|
|
Ok(t) => t,
|
|
Err(_) => {
|
|
return (
|
|
StatusCode::INTERNAL_SERVER_ERROR,
|
|
"Missing Runner Service ID",
|
|
);
|
|
}
|
|
};
|
|
|
|
let env_id = match env::var("RAILWAY_ENVIRONMENT_ID") {
|
|
Ok(id) => id,
|
|
Err(_) => return (StatusCode::INTERNAL_SERVER_ERROR, "Missing Railway env ID"),
|
|
};
|
|
|
|
let client = reqwest::Client::new();
|
|
let query = json!({
|
|
"query": "mutation Wake($sid: String!, $eid: String!) { serviceInstanceRedeploy(serviceId: $sid, environmentId: $eid) }",
|
|
"variables": {
|
|
"sid": service_id,
|
|
"eid": env_id
|
|
}
|
|
});
|
|
|
|
let res = client
|
|
.post("https://backboard.railway.com/graphql/v2")
|
|
.header("Project-Access-Token", token)
|
|
.json(&query)
|
|
.send()
|
|
.await;
|
|
|
|
match res {
|
|
Ok(res) if res.status().is_success() => (StatusCode::OK, "Runner is waking up"),
|
|
Ok(res) => {
|
|
let err_test = res.text().await.unwrap_or_default();
|
|
println!("Railway API error: {}", err_test);
|
|
(StatusCode::BAD_GATEWAY, "Railway API rejected the request")
|
|
}
|
|
Err(_) => (
|
|
StatusCode::SERVICE_UNAVAILABLE,
|
|
"Could not connect to Railway",
|
|
),
|
|
}
|
|
}
|