scott/crabfit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix logic and document CRON_KEY env variable

Browse source
This commit is contained in:
Ben Grant 2023-05-16 02:18:13 +10:00
parent 68cf43164d
commit bca67d2f06
2 changed files with 11 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
api/src/routes/tasks.rs
View file

@ -24,13 +24,13 @@ pub async fn cleanup<A: Adaptor>(
headers: HeaderMap,
) -> Result<(), ApiError<A>> {
// Check cron key
let cron_key_header = headers.get("X-Cron-Key");
if let Some(cron_key) = cron_key_header {
if let Ok(key) = env::var("CRON_KEY") {
if !key.is_empty() && *cron_key != key {
return Err(ApiError::NotAuthorized);
}
}
let cron_key_header: String = headers
.get("X-Cron-Key")
.map(|k| k.to_str().unwrap_or_default().into())
.unwrap_or_default();
let env_key = env::var("CRON_KEY").unwrap_or_default();
if !env_key.is_empty() && cron_key_header != env_key {
return Err(ApiError::NotAuthorized);
}
info!("Running cleanup task");