systemprompt-scheduler 0.2.2

use anyhow::Result;
use async_trait::async_trait;
use systemprompt_database::DbPool;
use systemprompt_traits::{Job, JobContext, JobResult};
use tracing::info;

#[derive(Debug, Clone, Copy)]
pub struct NoJsCleanupJob;

#[async_trait]
impl Job for NoJsCleanupJob {
    fn name(&self) -> &'static str {
        "no_js_cleanup"
    }

    fn description(&self) -> &'static str {
        "Marks sessions with HTTP requests but no JavaScript engagement as behavioral bots"
    }

    fn schedule(&self) -> &'static str {
        "0 */15 * * * *"
    }

    async fn execute(&self, ctx: &JobContext) -> Result<JobResult> {
        let start_time = std::time::Instant::now();

        let db_pool = std::sync::Arc::clone(
            ctx.db_pool::<DbPool>()
                .ok_or_else(|| anyhow::anyhow!("DbPool not available in job context"))?,
        );

        let pool = db_pool.write_pool_arc()?;

        let result = sqlx::query_scalar::<_, i64>(
            r"
            WITH cleaned AS (
                UPDATE user_sessions
                SET is_behavioral_bot = true,
                    behavioral_bot_reason = 'no_javascript',
                    behavioral_bot_score = 20
                WHERE is_bot = false
                  AND is_scanner = false
                  AND is_behavioral_bot = false
                  AND request_count > 0
                  AND started_at < NOW() - INTERVAL '10 minutes'
                  AND session_id NOT IN (
                    SELECT DISTINCT session_id FROM engagement_events
                    WHERE time_on_page_ms > 0
                  )
                RETURNING 1
            )
            SELECT COUNT(*)::BIGINT FROM cleaned
            ",
        )
        .fetch_one(pool.as_ref())
        .await
        .unwrap_or(0);

        let marked = result as u64;
        let duration_ms = start_time.elapsed().as_millis() as u64;

        if marked > 0 {
            info!(
                marked = marked,
                duration_ms = duration_ms,
                "No-JS session cleanup completed"
            );
        }

        Ok(JobResult::success()
            .with_stats(marked, 0)
            .with_duration(duration_ms))
    }
}

systemprompt_provider_contracts::submit_job!(&NoJsCleanupJob);