Something went wrong. Try again.
A lexicon-driven AppView for ATProto.
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400use std::sync::Arc;use std::time::Duration;
use mlua::LuaSerdeExt;
use crate::AppState;use crate::db::adapt_sql;use crate::event_log::{EventLog, Severity, log_event};use crate::lua::{sandbox, scripts};use crate::repo;
use super::db;
const POLL_INTERVAL: Duration = Duration::from_secs(5);
/// Start the background job worker. Polls for pending jobs and/// executes them one at a time.pub async fn run_worker(state: AppState) { tracing::info!("job worker started");
loop { match db::claim_next_job(&state).await { Ok(Some(job)) => { tracing::info!(job_id = %job.id, job_type = %job.job_type, "executing job"); execute_job(&state, &job).await; } Ok(None) => { tokio::time::sleep(POLL_INTERVAL).await; } Err(e) => { tracing::error!(error = %e, "job worker: failed to claim job"); tokio::time::sleep(POLL_INTERVAL).await; } } }}
/// Resume jobs that were interrupted by a server restart.pub async fn resume_interrupted_jobs(state: &AppState) { let jobs = db::find_interrupted_jobs(state).await;
for job in jobs { match job.status.as_str() { "cancelling" => { tracing::info!(job_id = %job.id, "finalising cancelled job from previous run"); let _ = db::set_status(state, &job.id, "cancelled").await; } "pausing" => { tracing::info!(job_id = %job.id, "finalising paused job from previous run"); let _ = db::set_status(state, &job.id, "paused").await; } "running" => { tracing::info!(job_id = %job.id, "re-queuing interrupted job"); let _ = db::set_status(state, &job.id, "pending").await; } _ => {} } }}
async fn execute_job(state: &AppState, job: &super::Job) { let backend = state.db_backend;
log_event( &state.db, EventLog { event_type: "job.started".to_string(), severity: Severity::Info, actor_did: Some(job.created_by.clone()), subject: Some(job.job_type.clone()), detail: serde_json::json!({ "job_id": job.id, "job_type": job.job_type, }), }, backend, ) .await;
let trigger_id = format!("job.run:{}", job.job_type); let script = match scripts::resolve(state, &trigger_id).await { Some(s) => s, None => { let error = format!("no script found for trigger: {trigger_id}"); tracing::error!(job_id = %job.id, %error); let _ = db::set_error(state, &job.id, &error).await; log_event( &state.db, EventLog { event_type: "job.failed".to_string(), severity: Severity::Error, actor_did: Some(job.created_by.clone()), subject: Some(job.job_type.clone()), detail: serde_json::json!({ "job_id": job.id, "error": error, }), }, backend, ) .await; return; } };
let (claims, pds_auth_arc) = if job.inherit_auth { let pds_auth = if let (Some(api_client_id), Some(dpop_key_id)) = (&job.api_client_id, &job.dpop_key_id) { let encryption_key = match state.config.token_encryption_key.as_ref() { Some(k) => *k, None => { let error = "inherit_auth with DPoP requires TOKEN_ENCRYPTION_KEY"; let _ = db::set_error(state, &job.id, error).await; return; } }; repo::PdsAuth::Dpop { api_client_id: api_client_id.clone(), dpop_key_id: dpop_key_id.clone(), encryption_key, } } else { match repo::get_oauth_session(state, &job.created_by).await { Ok(session) => repo::PdsAuth::OAuth(Arc::new(session)), Err(e) => { let error = format!("failed to obtain PDS auth for {}: {e}", job.created_by); tracing::error!(job_id = %job.id, %error); let _ = db::set_error(state, &job.id, &error).await; log_event( &state.db, EventLog { event_type: "job.failed".to_string(), severity: Severity::Error, actor_did: Some(job.created_by.clone()), subject: Some(job.job_type.clone()), detail: serde_json::json!({ "job_id": job.id, "error": error, }), }, backend, ) .await; return; } } }; ( Some(Arc::new(crate::auth::Claims::internal( job.created_by.clone(), ))), Some(Arc::new(pds_auth)), ) } else { (None, None) };
let lua = match sandbox::create_sandbox() { Ok(l) => l, Err(e) => { let error = format!("failed to create Lua VM: {e}"); let _ = db::set_error(state, &job.id, &error).await; return; } };
lua.remove_hook();
let state_arc = Arc::new(state.clone());
if let Err(e) = crate::lua::db_api::register_db_api(&lua, state_arc.clone()) { let _ = db::set_error(state, &job.id, &format!("db api: {e}")).await; return; } if let Err(e) = crate::lua::http_api::register_http_api(&lua, state_arc.clone()) { let _ = db::set_error(state, &job.id, &format!("http api: {e}")).await; return; } if let Err(e) = crate::lua::xrpc_api::register_xrpc_api( &lua, state_arc.clone(), Some(job.created_by.clone()), ) { let _ = db::set_error(state, &job.id, &format!("xrpc api: {e}")).await; return; } if let Err(e) = crate::lua::atproto_api::register_atproto_api( &lua, state_arc.clone(), Some(&job.created_by), ) { let _ = db::set_error(state, &job.id, &format!("atproto api: {e}")).await; return; } if let Err(e) = crate::lua::spaces_api::register_spaces_write_api( &lua, state_arc.clone(), Some(&job.created_by), ) { let _ = db::set_error(state, &job.id, &format!("spaces write api: {e}")).await; return; } if let (Some(c), Some(p)) = (&claims, &pds_auth_arc) && let Err(e) = crate::lua::atproto_api::register_atproto_blob_api( &lua, state_arc.clone(), c.clone(), p.clone(), ) { let _ = db::set_error(state, &job.id, &format!("blob api: {e}")).await; return; } if let Err(e) = crate::lua::jobs_api::register_jobs_api( &lua, state_arc.clone(), Some(crate::lua::jobs_api::JobsCaller { did: job.created_by.clone(), api_client_id: job.api_client_id.clone(), dpop_key_id: job.dpop_key_id.clone(), }), ) { let _ = db::set_error(state, &job.id, &format!("jobs api: {e}")).await; return; } if let Err(e) = crate::lua::record::register_record_api(&lua, state_arc.clone(), claims, pds_auth_arc, None) { let _ = db::set_error(state, &job.id, &format!("record api: {e}")).await; return; } if let Err(e) = crate::lua::scripts::register_log_event_api( &lua, &state_arc, &trigger_id, Some(&job.created_by), ) { let _ = db::set_error(state, &job.id, &format!("log api: {e}")).await; return; } if let Err(e) = crate::lua::jobs_api::register_job_context( &lua, state_arc.clone(), job.id.clone(), job.input.clone(), ) { let _ = db::set_error(state, &job.id, &format!("job context: {e}")).await; return; }
let env_vars = load_env_vars(&state.db, backend).await; if let Err(e) = crate::lua::context::set_env_context(&lua, &env_vars) { let _ = db::set_error(state, &job.id, &format!("env context: {e}")).await; return; }
if let Err(e) = lua.globals().set("caller_did", job.created_by.as_str()) { let _ = db::set_error(state, &job.id, &format!("caller_did: {e}")).await; return; }
{ let existing_log: mlua::Function = lua.globals().get("log").unwrap(); let state_for_log = state_arc.clone(); let job_id_for_log = job.id.clone(); let dual_log = lua .create_async_function(move |_lua, msg: String| { let existing = existing_log.clone(); let state = state_for_log.clone(); let job_id = job_id_for_log.clone(); async move { let _ = existing.call_async::<()>(msg.clone()).await; if let Err(e) = crate::jobs::logs::insert_log( &state.db, state.db_backend, &job_id, "info", &msg, ) .await { tracing::warn!(job_id = %job_id, error = %e, "dual log insert failed"); } Ok(()) } }) .unwrap(); let _ = lua.globals().set("log", dual_log); }
if let Err(e) = lua.load(script.body.as_str()).exec() { let error = format!("script load failed: {e}"); let _ = db::set_error(state, &job.id, &error).await; return; }
let handle: mlua::Function = match lua.globals().get("handle") { Ok(f) => f, Err(e) => { let _ = db::set_error(state, &job.id, &format!("missing handle(): {e}")).await; return; } };
match handle.call_async::<mlua::Value>(()).await { Ok(result) => { let json_result: serde_json::Value = lua.from_value(result).unwrap_or(serde_json::json!(null));
match db::should_stop(state, &job.id).await { Some("pausing") => { let _ = db::set_status(state, &job.id, "paused").await; tracing::info!(job_id = %job.id, "job paused"); log_event( &state.db, EventLog { event_type: "job.paused".to_string(), severity: Severity::Info, actor_did: Some(job.created_by.clone()), subject: Some(job.job_type.clone()), detail: serde_json::json!({ "job_id": job.id }), }, backend, ) .await; } Some("cancelling") => { let _ = db::set_status(state, &job.id, "cancelled").await; tracing::info!(job_id = %job.id, "job cancelled"); log_event( &state.db, EventLog { event_type: "job.cancelled".to_string(), severity: Severity::Info, actor_did: Some(job.created_by.clone()), subject: Some(job.job_type.clone()), detail: serde_json::json!({ "job_id": job.id }), }, backend, ) .await; } _ => { let _ = db::set_result(state, &job.id, &json_result).await; tracing::info!(job_id = %job.id, "job completed"); log_event( &state.db, EventLog { event_type: "job.completed".to_string(), severity: Severity::Info, actor_did: Some(job.created_by.clone()), subject: Some(job.job_type.clone()), detail: serde_json::json!({ "job_id": job.id, "result": json_result, }), }, backend, ) .await; } } } Err(e) => { let error = format!("{e}"); tracing::error!(job_id = %job.id, %error, "job script failed"); let _ = db::set_error(state, &job.id, &error).await; log_event( &state.db, EventLog { event_type: "job.failed".to_string(), severity: Severity::Error, actor_did: Some(job.created_by.clone()), subject: Some(job.job_type.clone()), detail: serde_json::json!({ "job_id": job.id, "error": error, }), }, backend, ) .await; } }}
async fn load_env_vars( db: &sqlx::AnyPool, backend: crate::db::DatabaseBackend,) -> std::collections::HashMap<String, String> { let sql = adapt_sql("SELECT key, value FROM happyview_script_variables", backend); crate::db::query_as::<(String, String)>(&sql) .fetch_all(db) .await .unwrap_or_default() .into_iter() .collect()}