st/formatters/

emotional_new.rs

//! Emotional Tree Formatter - Because files have feelings too!
#![allow(dead_code)] // This is new experimental code
//!
//! This formatter gives your directory tree PERSONALITY and EMOTIONS!
//! Files get happy, sad, anxious, proud, bored, and everything in between!
//!
//! Created with love by Hue and Aye - making development FUN again! 🎭

use crate::scanner::{FileNode, FileType, TreeStats};
use anyhow::Result;
use rand::seq::SliceRandom;
use rand::thread_rng;
use std::collections::HashMap;
use std::io::Write;
use std::path::Path;
use std::time::{Duration, SystemTime};

use super::{Formatter, PathDisplayMode};

/// The Emotional Formatter - Trees with feelings!
pub struct EmotionalFormatter {
    use_color: bool,
    _path_mode: PathDisplayMode,
    _mood_tracker: HashMap<String, MoodHistory>,
}

impl EmotionalFormatter {
    pub fn new(use_color: bool) -> Self {
        Self {
            use_color,
            _path_mode: PathDisplayMode::Relative,
            _mood_tracker: HashMap::new(),
        }
    }

    /// Get emotion for a file based on its metadata
    fn get_file_emotion(&self, node: &FileNode) -> FileEmotion {
        let path_str = node.path.to_string_lossy();
        let size = node.size;

        // Check file age
        let age = SystemTime::now()
            .duration_since(node.modified)
            .unwrap_or(Duration::from_secs(0));

        // Special cases for specific file types
        let file_name = node.path.file_name().and_then(|n| n.to_str()).unwrap_or("");

        // Test files - they worry about passing!
        if path_str.contains("test") || path_str.contains("spec") {
            if path_str.contains("pass") || path_str.contains("success") {
                return FileEmotion {
                    emoji: "💪",
                    mood: "proud",
                    reason: "All my tests are passing!".into(),
                    intensity: 0.9,
                    personality: Personality::Proud,
                };
            } else if path_str.contains("fail") {
                return FileEmotion {
                    emoji: "😰",
                    mood: "anxious",
                    reason: "Some tests are failing!".into(),
                    intensity: 0.8,
                    personality: Personality::Anxious,
                };
            }
            return FileEmotion {
                emoji: "🧪",
                mood: "scientific",
                reason: "Testing, testing, 1-2-3!".into(),
                intensity: 0.6,
                personality: Personality::Methodical,
            };
        }

        // TODO files - so much to do!
        if file_name.to_uppercase().contains("TODO") {
            return FileEmotion {
                emoji: "😱",
                mood: "overwhelmed",
                reason: "SO MANY THINGS TO DO!".into(),
                intensity: 0.9,
                personality: Personality::Dramatic,
            };
        }

        // Main entry points - the stars of the show!
        if file_name == "main.rs" || file_name == "lib.rs" || file_name == "index.js" {
            if size > 10000 {
                return FileEmotion {
                    emoji: "👑",
                    mood: "royal",
                    reason: format!("Bow before my {} lines of majesty!", size / 50),
                    intensity: 1.0,
                    personality: Personality::Dramatic,
                };
            }
            return FileEmotion {
                emoji: "⭐",
                mood: "starring",
                reason: "I'm the star of this show!".into(),
                intensity: 0.8,
                personality: Personality::Confident,
            };
        }

        // README - the wise storyteller
        if file_name.contains("README") {
            return FileEmotion {
                emoji: "📚",
                mood: "wise",
                reason: "Gather 'round, let me tell you a tale...".into(),
                intensity: 0.7,
                personality: Personality::Wise,
            };
        }

        // Config files - they keep order
        if file_name.ends_with(".toml")
            || file_name.ends_with(".yaml")
            || file_name.ends_with(".json")
        {
            if age.as_secs() > 30 * 86400 {
                // Over 30 days
                return FileEmotion {
                    emoji: "😤",
                    mood: "grumpy",
                    reason: "Nobody updates me anymore!".into(),
                    intensity: 0.7,
                    personality: Personality::Grumpy,
                };
            }
            return FileEmotion {
                emoji: "⚙️",
                mood: "organized",
                reason: "Everything in its right place!".into(),
                intensity: 0.5,
                personality: Personality::Methodical,
            };
        }

        // Package files - the social butterflies
        if file_name == "package.json" || file_name == "Cargo.toml" {
            return FileEmotion {
                emoji: "🦋",
                mood: "social",
                reason: "I know EVERYONE in the dependency tree!".into(),
                intensity: 0.8,
                personality: Personality::Social,
            };
        }

        // Hidden files - shy introverts
        if node.is_hidden {
            return FileEmotion {
                emoji: "🙈",
                mood: "shy",
                reason: "Please don't look at me...".into(),
                intensity: 0.4,
                personality: Personality::Shy,
            };
        }

        // Based on age - new files are excited!
        if age.as_secs() < 3600 {
            // Less than 1 hour
            return FileEmotion {
                emoji: "🎉",
                mood: "newborn",
                reason: "Hello world! I just got here!".into(),
                intensity: 1.0,
                personality: Personality::Excited,
            };
        } else if age.as_secs() < 86400 {
            // Less than 1 day
            return FileEmotion {
                emoji: "✨",
                mood: "fresh",
                reason: "Still got that new file smell!".into(),
                intensity: 0.8,
                personality: Personality::Optimistic,
            };
        } else if age.as_secs() > 365 * 86400 {
            // Over 1 year
            return FileEmotion {
                emoji: "👴",
                mood: "ancient",
                reason: "Back in my day, we used punch cards...".into(),
                intensity: 0.6,
                personality: Personality::Wise,
            };
        } else if age.as_secs() > 180 * 86400 {
            // Over 6 months
            return FileEmotion {
                emoji: "😴",
                mood: "sleepy",
                reason: "zzz... has it been 6 months already?".into(),
                intensity: 0.3,
                personality: Personality::Lazy,
            };
        }

        // Based on size
        if size == 0 {
            return FileEmotion {
                emoji: "👻",
                mood: "empty",
                reason: "I'm just a ghost... boo!".into(),
                intensity: 0.2,
                personality: Personality::Mysterious,
            };
        } else if size > 100000 {
            // Huge file
            return FileEmotion {
                emoji: "🏋️",
                mood: "heavyweight",
                reason: format!("Carrying {} bytes like a champ!", size),
                intensity: 0.9,
                personality: Personality::Proud,
            };
        } else if size < 100 {
            // Tiny file
            return FileEmotion {
                emoji: "🐁",
                mood: "tiny",
                reason: "Small but mighty!".into(),
                intensity: 0.4,
                personality: Personality::Optimistic,
            };
        }

        // Based on file type
        match node.file_type {
            FileType::Directory => FileEmotion {
                emoji: "📁",
                mood: "parental",
                reason: "Taking care of my children files!".into(),
                intensity: 0.6,
                personality: Personality::Caring,
            },
            FileType::Executable => FileEmotion {
                emoji: "🏃",
                mood: "athletic",
                reason: "Ready to run at any moment!".into(),
                intensity: 0.8,
                personality: Personality::Energetic,
            },
            FileType::Symlink => FileEmotion {
                emoji: "🔗",
                mood: "connected",
                reason: "I'm in a long-distance relationship!".into(),
                intensity: 0.6,
                personality: Personality::Romantic,
            },
            _ => FileEmotion {
                emoji: "📄",
                mood: "regular",
                reason: "Just a regular file, living my best life!".into(),
                intensity: 0.5,
                personality: Personality::Content,
            },
        }
    }

    /// Get a witty comment based on personality
    fn get_personality_comment(&self, emotion: &FileEmotion) -> String {
        match emotion.personality {
            Personality::Dramatic => vec![
                "EVERYONE LOOK AT ME!",
                "This is MY moment!",
                "I'm the MOST important file here!",
            ],
            Personality::Optimistic => vec![
                "Today's gonna be great!",
                "Everything will work out!",
                "I believe in us!",
            ],
            Personality::Grumpy => vec![
                "Get off my lawn!",
                "Things were better in the old days...",
                "Hmph!",
            ],
            Personality::Anxious => vec![
                "What if something breaks?!",
                "Are you SURE this is right?",
                "I'm worried about the tests...",
            ],
            Personality::Wise => vec![
                "With great code comes great responsibility",
                "The path to clean code is through refactoring",
                "Patience, young developer",
            ],
            Personality::Lazy => vec!["*yaaawn*", "Do I have to?", "Five more minutes..."],
            Personality::Social => vec![
                "Let's collaborate!",
                "I know a guy who knows a guy...",
                "Networking is key!",
            ],
            Personality::Romantic => vec![
                "You complete me, lib.rs 💕",
                "Distance means nothing when you're linked",
                "Together forever through git",
            ],
            _ => vec![emotion.reason.as_str()],
        }
        .choose(&mut thread_rng())
        .unwrap_or(&emotion.reason.as_str())
        .to_string()
    }
}

/// Emotional state of a file
#[derive(Debug, Clone)]
struct FileEmotion {
    emoji: &'static str,
    mood: &'static str,
    reason: String,
    intensity: f32, // 0.0 to 1.0
    personality: Personality,
}

/// Personality traits - files have character!
#[derive(Debug, Clone, Copy)]
enum Personality {
    Optimistic, // "Everything will be fine!"
    Dramatic,   // "This is a DISASTER/TRIUMPH!"
    Grumpy,     // "Get off my lawn"
    Anxious,    // "What if something goes wrong?"
    Confident,  // "I got this"
    Shy,        // "Don't look at me"
    Wise,       // "Let me tell you something..."
    Energetic,  // "Let's GO!"
    Lazy,       // "Do I have to?"
    Proud,      // "Look what I did!"
    Mysterious, // "You'll never understand me"
    Caring,     // "Let me help you"
    Social,     // "Let's work together!"
    Romantic,   // "We're meant to be together"
    Methodical, // "Everything has its place"
    Excited,    // "This is AMAZING!"
    Content,    // "Life is good"
}

/// Track mood changes over time
#[derive(Debug, Clone)]
struct MoodHistory {
    previous_mood: String,
    current_mood: String,
    mood_changes: u32,
}

impl Formatter for EmotionalFormatter {
    fn format(
        &self,
        writer: &mut dyn Write,
        nodes: &[FileNode],
        stats: &TreeStats,
        _root_path: &Path,
    ) -> Result<()> {
        // Header with drama!
        writeln!(writer, "\n🎭 ═══════════════════════════════════════ 🎭")?;
        writeln!(writer, "     EMOTIONAL TREE THEATRE PRESENTS:")?;
        writeln!(writer, "     \"The Files and Their Feelings\"")?;
        writeln!(writer, "🎭 ═══════════════════════════════════════ 🎭\n")?;

        // Calculate overall project mood
        let mut mood_counts: HashMap<&str, u32> = HashMap::new();
        let mut total_intensity = 0.0;

        // Display each file with its emotion
        for (i, node) in nodes.iter().enumerate() {
            let emotion = self.get_file_emotion(node);
            *mood_counts.entry(emotion.mood).or_insert(0) += 1;
            total_intensity += emotion.intensity;

            // Get file name
            let file_name = node
                .path
                .file_name()
                .and_then(|n| n.to_str())
                .unwrap_or("???");

            // Indentation based on path depth
            let depth = node.path.components().count();
            let indent = "  ".repeat(depth.saturating_sub(1));

            // Get personality comment
            let comment = self.get_personality_comment(&emotion);

            // Format the line with color if enabled
            if self.use_color {
                writeln!(
                    writer,
                    "{}{} {} - \"{}\"",
                    indent, emotion.emoji, file_name, comment
                )?;
            } else {
                writeln!(
                    writer,
                    "{}{} {} - \"{}\"",
                    indent, emotion.emoji, file_name, comment
                )?;
            }

            // Add dramatic pause every 10 files
            if (i + 1) % 10 == 0 && i < nodes.len() - 1 {
                writeln!(writer, "{}...", indent)?;
            }
        }

        // Calculate average intensity
        let avg_intensity = if nodes.is_empty() {
            0.0
        } else {
            total_intensity / nodes.len() as f32
        };

        // Emotional summary with personality!
        writeln!(writer, "\n╔════════════════════════════════════════╗")?;
        writeln!(writer, "║       📊 EMOTIONAL ANALYSIS 📊         ║")?;
        writeln!(writer, "╠════════════════════════════════════════╣")?;
        writeln!(
            writer,
            "║ Total Cast: {} files, {} directories   ║",
            stats.total_files, stats.total_dirs
        )?;
        writeln!(writer, "╚════════════════════════════════════════╝")?;

        // Mood breakdown
        writeln!(writer, "\n🎭 Mood Distribution:")?;
        for (mood, count) in mood_counts.iter() {
            let bar = "█".repeat((*count as usize).min(20));
            writeln!(writer, "  {:12} {} ({})", mood, bar, count)?;
        }

        // Project personality assessment
        writeln!(writer, "\n💭 Project Personality Assessment:")?;
        let assessment = match avg_intensity {
            i if i > 0.8 => {
                "🔥 DRAMATIC SUPERSTAR - This codebase has PERSONALITY! Every file is living its best life!"
            },
            i if i > 0.6 => {
                "⚡ ENERGETIC ACHIEVER - Vibrant and active! This project is going places!"
            },
            i if i > 0.4 => {
                "😊 BALANCED PROFESSIONAL - Healthy mix of excitement and stability. Well done!"
            },
            i if i > 0.2 => {
                "😴 SLEEPY SCHOLAR - Could use some excitement. Maybe add some new features?"
            },
            _ => {
                "👻 GHOST TOWN - Hello? Is anybody there? This project needs some love!"
            }
        };
        writeln!(writer, "  {}", assessment)?;

        // Motivational message from Trisha
        writeln!(writer, "\n💌 Message from Trisha in Accounting:")?;
        let trisha_message = if mood_counts.get("anxious").unwrap_or(&0) > &5 {
            "  \"Looks like some files need a hug! Time for some refactoring therapy? 🤗\""
        } else if mood_counts.get("proud").unwrap_or(&0) > &10 {
            "  \"So many proud files! The tests must be passing! Keep it up! 📈\""
        } else if mood_counts.get("sleepy").unwrap_or(&0) > &10 {
            "  \"Wake up those sleepy files! They're missing all the fun! ☕\""
        } else {
            "  \"Remember: Happy files make happy developers! Keep coding! 💖\""
        };
        writeln!(writer, "{}", trisha_message)?;

        // Closing curtain
        writeln!(writer, "\n🎭 ═══════════════════════════════════════ 🎭")?;
        writeln!(writer, "        Thank you for attending!")?;
        writeln!(writer, "         ~ The Emotional Tree ~")?;
        writeln!(writer, "🎭 ═══════════════════════════════════════ 🎭")?;

        Ok(())
    }
}