emgauwa/core
Watch
2
0
Fork
You've already forked core
 0
Code Issues Pull requests Projects Releases 3 Packages Wiki Activity Actions

Add token authentication for controllers

Browse source
This commit is contained in:
Tobias Reisinger 2025-05-31 23:19:38 +02:00
commit ebac452a86
Signed by: serguzim
GPG key ID: 13AD60C237A28DFE
3 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/handlers/v1/ws/mod.rs
View file

@ -5,10 +5,11 @@ use actix_web::{get, web, HttpRequest, HttpResponse};
use actix_web_actors::ws;
use emgauwa_common::errors::EmgauwaError;
use sqlx::{Pool, Sqlite};
use emgauwa_common::constants;
use crate::app_state::AppState;
use crate::handlers::v1::ws::controllers::ControllersWs;
use crate::handlers::v1::ws::relays::RelaysWs;
use crate::settings::Settings;
pub mod controllers;
pub mod relays;
@ -17,9 +18,23 @@ pub mod relays;
pub async fn ws_controllers(
pool: web::Data<Pool<Sqlite>>,
app_state: web::Data<Addr<AppState>>,
settings: web::Data<Settings>,
req: HttpRequest,
stream: web::Payload,
) -> Result<HttpResponse, EmgauwaError> {
let token = req
.headers()
.get(constants::CONTROLLER_WS_TOKEN_HEADER)
.ok_or(EmgauwaError::Unauthorized(
String::from("Missing or invalid token header"),
))?
.to_str()
.map_err(|_| EmgauwaError::Unauthorized(String::from("Invalid token header")))?;
if token != settings.server.token {
return Err(EmgauwaError::Unauthorized(String::from("Wrong token header")));
}
let resp = ws::start(
ControllersWs {
pool: pool.get_ref().clone(),