nexus_memory_hooks/agents/

qwen.rs

//! Qwen hook implementation
//!
//! Process-monitor detection with optional session directory scanning.

use async_trait::async_trait;
use std::path::PathBuf;

use crate::base::{AgentHook, BaseHook, LifecycleCapabilities, SessionEndCallback};
use crate::error::Result;
use crate::monitor::ProcessMonitor;
use crate::session::SessionContext;
use crate::types::{AgentType, SessionActivity};

/// Qwen hook using process monitoring and session directory detection
pub struct QwenHook {
    /// Base hook functionality
    base: BaseHook,

    /// Config path
    config_path: PathBuf,

    /// Process monitor for fallback detection
    process_monitor: ProcessMonitor,
}

impl QwenHook {
    /// Config directory
    pub const CONFIG_DIR: &'static str = ".qwen";

    /// Create a new Qwen hook
    pub fn new() -> Self {
        let config_path = dirs::home_dir()
            .unwrap_or_else(|| PathBuf::from("."))
            .join(Self::CONFIG_DIR);

        Self {
            base: BaseHook::new("qwen"),
            config_path,
            process_monitor: ProcessMonitor::new(),
        }
    }

    /// Read session data
    fn read_session_data(&self) -> Option<serde_json::Value> {
        let session_file = self.config_path.join("session.json");

        if session_file.exists() {
            let content = std::fs::read_to_string(&session_file).ok()?;
            serde_json::from_str(&content).ok()
        } else {
            None
        }
    }
}

impl Default for QwenHook {
    fn default() -> Self {
        Self::new()
    }
}

#[async_trait]
impl AgentHook for QwenHook {
    fn agent_type(&self) -> &str {
        &self.base.agent_type
    }

    async fn install_session_end_hook(&mut self, callback: SessionEndCallback) -> Result<()> {
        self.base.add_callback(callback);
        self.base.installed = true;

        Ok(())
    }

    async fn detect_session_activity(&self) -> Result<SessionActivity> {
        let mut monitor = self.process_monitor.clone();
        let processes = monitor.find_agent_processes(AgentType::Qwen);

        let mut activity = SessionActivity::new(AgentType::Qwen);

        if !processes.is_empty() {
            activity.is_active = true;
            activity.processes = processes;
        }

        // Check session directory for recent activity files
        let session_dir = self.config_path.join("sessions");
        if session_dir.exists() {
            if let Ok(entries) = std::fs::read_dir(&session_dir) {
                let most_recent = entries
                    .filter_map(|e| e.ok())
                    .filter(|e| {
                        e.path()
                            .extension()
                            .map(|ext| ext == "json")
                            .unwrap_or(false)
                    })
                    .max_by_key(|e| e.metadata().ok().and_then(|m| m.modified().ok()));

                if let Some(entry) = most_recent {
                    if let Ok(metadata) = entry.metadata() {
                        if let Ok(modified) = metadata.modified() {
                            let age = std::time::SystemTime::now()
                                .duration_since(modified)
                                .unwrap_or(std::time::Duration::MAX);

                            // Consider active if modified in last 5 minutes
                            if age.as_secs() < 300 {
                                activity.is_active = true;
                                activity.session_id = Some(
                                    entry
                                        .path()
                                        .file_stem()
                                        .unwrap()
                                        .to_string_lossy()
                                        .to_string(),
                                );
                            }
                        }
                    }
                }
            }
        }

        Ok(activity)
    }

    async fn extract_session_context(&self) -> Result<SessionContext> {
        let mut context = SessionContext::new("qwen")
            .with_source("monitor")
            .with_reliability(0.95);

        if let Some(session) = self.read_session_data() {
            if let Some(messages) = session.get("messages").and_then(|m| m.as_array()) {
                for msg in messages {
                    let role = msg
                        .get("role")
                        .and_then(|r| r.as_str())
                        .unwrap_or("unknown");
                    let content = msg.get("content").and_then(|c| c.as_str()).unwrap_or("");
                    context.add_message(role, content);
                }
            }

            if let Some(commands) = session.get("commands").and_then(|c| c.as_array()) {
                for cmd in commands {
                    if let Some(cmd_str) = cmd.as_str() {
                        context.add_command(cmd_str);
                    }
                }
            }
        }

        // Try to get git status for modified files
        let git_status = std::process::Command::new("git")
            .args(["status", "--porcelain"])
            .output()
            .ok();

        if let Some(output) = git_status {
            if output.status.success() {
                let status = String::from_utf8_lossy(&output.stdout);
                for line in status.lines() {
                    if line.len() > 3 {
                        let status_char = line.chars().next().unwrap_or(' ');
                        let file_path = &line[3..];
                        let action = match status_char {
                            '?' => crate::session::FileAction::Created,
                            'D' => crate::session::FileAction::Deleted,
                            _ => crate::session::FileAction::Modified,
                        };
                        context.add_file(crate::session::FileInfo::new(file_path, action));
                    }
                }
            }
        }

        context.complete();
        Ok(context)
    }

    fn reliability_score(&self) -> f32 {
        0.95
    }

    fn lifecycle_capabilities(&self) -> LifecycleCapabilities {
        LifecycleCapabilities::monitor_only()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_qwen_hook_new() {
        let hook = QwenHook::new();
        assert_eq!(hook.agent_type(), "qwen");
    }

    #[tokio::test]
    async fn test_qwen_hook_detect_activity() {
        let hook = QwenHook::new();
        let activity = hook.detect_session_activity().await.unwrap();

        assert_eq!(activity.agent_type, AgentType::Qwen);
    }

    #[test]
    fn test_qwen_hook_lifecycle_capabilities() {
        let hook = QwenHook::new();
        let caps = hook.lifecycle_capabilities();

        assert!(!caps.session_start, "Qwen does not support session_start");
        assert!(!caps.session_end, "Qwen is monitor-only");
        assert!(!caps.checkpoint, "Qwen does not support checkpoint");
    }
}