tftsr-devops_investigation/src-tauri/src/db/models.rs

use serde::{Deserialize, Serialize};
use uuid::Uuid;

// ─── Issue ──────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Issue {
    pub id: String,
    pub title: String,
    pub description: String,
    pub severity: String,
    pub status: String,
    pub category: String,
    pub source: String,
    pub created_at: String,
    pub updated_at: String,
    pub resolved_at: Option<String>,
    pub assigned_to: String,
    pub tags: String,
}

impl Issue {
    pub fn new(title: String, description: String, severity: String, category: String) -> Self {
        let now = chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string();
        Issue {
            id: Uuid::now_v7().to_string(),
            title,
            description,
            severity,
            status: "open".to_string(),
            category,
            source: "manual".to_string(),
            created_at: now.clone(),
            updated_at: now,
            resolved_at: None,
            assigned_to: String::new(),
            tags: "[]".to_string(),
        }
    }
}

/// Full detail view returned by get_issue.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct IssueDetail {
    pub issue: Issue,
    pub log_files: Vec<LogFile>,
    pub resolution_steps: Vec<ResolutionStep>,
    pub conversations: Vec<AiConversation>,
}

/// Lightweight row returned by list/search commands.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct IssueSummary {
    pub id: String,
    pub title: String,
    pub severity: String,
    pub status: String,
    pub category: String,
    pub created_at: String,
    pub updated_at: String,
    pub log_count: i64,
    pub step_count: i64,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct IssueListItem {
    pub id: String,
    pub title: String,
    pub domain: String,
    pub status: String,
    pub severity: String,
    pub created_at: i64,
    pub updated_at: i64,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct IssueFilter {
    pub status: Option<String>,
    pub severity: Option<String>,
    pub category: Option<String>,
    pub domain: Option<String>,
    pub search: Option<String>,
    pub limit: Option<i64>,
    pub offset: Option<i64>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct NewIssue {
    pub title: String,
    pub domain: String,
    pub description: Option<String>,
    pub severity: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct IssueUpdate {
    pub title: Option<String>,
    pub description: Option<String>,
    pub severity: Option<String>,
    pub status: Option<String>,
    pub category: Option<String>,
    pub domain: Option<String>,
    pub assigned_to: Option<String>,
    pub tags: Option<String>,
}

// ─── 5-Whys ─────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct FiveWhyEntry {
    pub id: String,
    pub why_number: i32,
    pub question: String,
    pub answer: Option<String>,
    pub created_at: i64,
}

// ─── Timeline ────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TimelineEvent {
    pub id: String,
    pub event_type: String,
    pub description: String,
    pub created_at: i64,
}

// ─── Log File ───────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct LogFile {
    pub id: String,
    pub issue_id: String,
    pub file_name: String,
    pub file_path: String,
    pub file_size: i64,
    pub mime_type: String,
    pub content_hash: String,
    pub uploaded_at: String,
    pub redacted: bool,
}

impl LogFile {
    pub fn new(issue_id: String, file_name: String, file_path: String, file_size: i64) -> Self {
        LogFile {
            id: Uuid::now_v7().to_string(),
            issue_id,
            file_name,
            file_path,
            file_size,
            mime_type: "text/plain".to_string(),
            content_hash: String::new(),
            uploaded_at: chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
            redacted: false,
        }
    }
}

// ─── PII ────────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PiiSpanRecord {
    pub id: String,
    pub log_file_id: String,
    pub pii_type: String,
    pub start_offset: i64,
    pub end_offset: i64,
    pub original_value: String,
    pub replacement: String,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PiiDetectionResult {
    pub log_file_id: String,
    pub detections: Vec<crate::pii::PiiSpan>,
    pub total_pii_found: usize,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RedactedLogFile {
    pub id: String,
    pub original_file_id: String,
    pub file_name: String,
    pub file_hash: String,
    pub redaction_count: usize,
}

// ─── AI Conversation ─────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AiConversation {
    pub id: String,
    pub issue_id: String,
    pub provider: String,
    pub model: String,
    pub created_at: String,
    pub title: String,
}

impl AiConversation {
    pub fn new(issue_id: String, provider: String, model: String) -> Self {
        AiConversation {
            id: Uuid::now_v7().to_string(),
            issue_id,
            provider,
            model,
            created_at: chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
            title: "Untitled".to_string(),
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AiMessage {
    pub id: String,
    pub conversation_id: String,
    pub role: String,
    pub content: String,
    pub token_count: i64,
    pub created_at: String,
}

impl AiMessage {
    pub fn new(conversation_id: String, role: String, content: String) -> Self {
        AiMessage {
            id: Uuid::now_v7().to_string(),
            conversation_id,
            role,
            content,
            token_count: 0,
            created_at: chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
        }
    }
}

// ─── Resolution Step ─────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ResolutionStep {
    pub id: String,
    pub issue_id: String,
    pub step_order: i64,
    pub why_question: String,
    pub answer: String,
    pub evidence: String,
    pub created_at: String,
}

impl ResolutionStep {
    pub fn new(
        issue_id: String,
        step_order: i64,
        why_question: String,
        answer: String,
        evidence: String,
    ) -> Self {
        ResolutionStep {
            id: Uuid::now_v7().to_string(),
            issue_id,
            step_order,
            why_question,
            answer,
            evidence,
            created_at: chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
        }
    }
}

// ─── Document ────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Document {
    pub id: String,
    pub issue_id: String,
    pub doc_type: String,
    pub title: String,
    pub content_md: String,
    pub created_at: i64,
    pub updated_at: i64,
}

impl Document {
    pub fn new(issue_id: String, doc_type: String, title: String, content_md: String) -> Self {
        let now = chrono::Utc::now().timestamp_millis();
        Document {
            id: Uuid::now_v7().to_string(),
            issue_id,
            doc_type,
            title,
            content_md,
            created_at: now,
            updated_at: now,
        }
    }
}

// ─── Audit ──────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AuditEntry {
    pub id: String,
    pub timestamp: String,
    pub action: String,
    pub entity_type: String,
    pub entity_id: String,
    pub user_id: String,
    pub details: String,
}

impl AuditEntry {
    pub fn new(action: String, entity_type: String, entity_id: String, details: String) -> Self {
        AuditEntry {
            id: Uuid::now_v7().to_string(),
            timestamp: chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
            action,
            entity_type,
            entity_id,
            user_id: "local".to_string(),
            details,
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct AuditFilter {
    pub action: Option<String>,
    pub entity_type: Option<String>,
    pub entity_id: Option<String>,
    pub from_date: Option<String>,
    pub to_date: Option<String>,
    pub limit: Option<i64>,
    pub offset: Option<i64>,
}

// ─── Settings ───────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SettingRecord {
    pub key: String,
    pub value: String,
    pub updated_at: String,
}