luk/demo_notification_compression.py

#!/usr/bin/env python3
"""
Demo script for notification email compression.

This script demonstrates how notification emails are detected and compressed
into terminal-friendly summaries.
"""

from src.mail.notification_detector import (
    is_notification_email,
    classify_notification,
    extract_notification_summary,
    NOTIFICATION_TYPES,
)
from src.mail.notification_compressor import NotificationCompressor, DetailedCompressor


def demo_detection():
    """Demonstrate notification detection for various email types."""

    test_emails = [
        {
            "from": {"addr": "notifications@gitlab.com", "name": "GitLab"},
            "subject": "Pipeline #12345 failed by john.doe",
        },
        {
            "from": {"addr": "noreply@github.com", "name": "GitHub"},
            "subject": "[GitHub] PR #42: Add new feature",
        },
        {
            "from": {"addr": "jira@atlassian.net", "name": "Jira"},
            "subject": "[Jira] ABC-123: Fix login bug",
        },
        {
            "from": {"addr": "confluence@atlassian.net", "name": "Confluence"},
            "subject": "[Confluence] New comment on page",
        },
        {
            "from": {"addr": "alerts@datadoghq.com", "name": "Datadog"},
            "subject": "[Datadog] Alert: High CPU usage",
        },
        {
            "from": {"addr": "renovate@renovatebot.com", "name": "Renovate"},
            "subject": "[Renovate] Update dependency to v2.0.0",
        },
        {
            "from": {"addr": "john.doe@example.com", "name": "John Doe"},
            "subject": "Let's meet for lunch",
        },
    ]

    print("=" * 70)
    print("NOTIFICATION DETECTION DEMO")
    print("=" * 70)
    print()

    for i, envelope in enumerate(test_emails, 1):
        from_addr = envelope.get("from", {}).get("addr", "")
        subject = envelope.get("subject", "")

        print(f"Email {i}: {subject}")
        print(f"  From: {from_addr}")

        # Check if notification
        is_notif = is_notification_email(envelope)
        print(f"  Is Notification: {is_notif}")

        if is_notif:
            notif_type = classify_notification(envelope)
            if notif_type:
                print(f"  Type: {notif_type.name}")
                print(f"  Icon: {notif_type.icon}")

        print()


def demo_compression():
    """Demonstrate notification compression."""

    # GitLab pipeline email content (simplified)
    gitlab_content = """
    Pipeline #12345 failed by john.doe

    The pipeline failed on stage: build
    Commit: abc123def

    View pipeline: https://gitlab.com/project/pipelines/12345
    """

    # GitHub PR email content (simplified)
    github_content = """
    PR #42: Add new feature

    @john.doe requested your review

    View PR: https://github.com/repo/pull/42
    """

    gitlab_envelope = {
        "from": {"addr": "notifications@gitlab.com", "name": "GitLab"},
        "subject": "Pipeline #12345 failed",
        "date": "2025-12-28T15:00:00Z",
    }

    github_envelope = {
        "from": {"addr": "noreply@github.com", "name": "GitHub"},
        "subject": "[GitHub] PR #42: Add new feature",
        "date": "2025-12-28T15:00:00Z",
    }

    print("=" * 70)
    print("NOTIFICATION COMPRESSION DEMO")
    print("=" * 70)
    print()

    # GitLab compression - summary mode
    print("1. GitLab Pipeline (Summary Mode)")
    print("-" * 70)
    compressor = NotificationCompressor(mode="summary")
    compressed, notif_type = compressor.compress(gitlab_content, gitlab_envelope)
    print(compressed)
    print()

    # GitLab compression - detailed mode
    print("2. GitLab Pipeline (Detailed Mode)")
    print("-" * 70)
    detailed_compressor = DetailedCompressor(mode="detailed")
    compressed, notif_type = detailed_compressor.compress(
        gitlab_content, gitlab_envelope
    )
    print(compressed)
    print()

    # GitHub PR - summary mode
    print("3. GitHub PR (Summary Mode)")
    print("-" * 70)
    compressor = NotificationCompressor(mode="summary")
    compressed, notif_type = compressor.compress(github_content, github_envelope)
    print(compressed)
    print()


def demo_summary_extraction():
    """Demonstrate structured summary extraction."""

    test_content = """
    ABC-123: Fix login bug

    Status changed from In Progress to Done

    View issue: https://jira.atlassian.net/browse/ABC-123
    """

    print("=" * 70)
    print("SUMMARY EXTRACTION DEMO")
    print("=" * 70)
    print()

    notif_type = NOTIFICATION_TYPES[2]  # jira
    summary = extract_notification_summary(test_content, notif_type)

    print("Extracted Summary:")
    print(f"  Title: {summary.get('title')}")
    print(f"  Metadata: {summary.get('metadata')}")
    print(f"  Action Items: {summary.get('action_items')}")
    print()


def main():
    """Run all demos."""

    print()
    print("╔" + "═" * 68 + "╗")
    print("║" + " " * 68 + "║")
    print("║" + "  LUK Notification Email Compression - Feature Demo".center(68) + "║")
    print("║" + " " * 68 + "║")
    print("╚" + "═" * 68 + "╝")
    print()

    # Run demos
    demo_detection()
    print()
    demo_compression()
    print()
    demo_summary_extraction()

    print()
    print("=" * 70)
    print("DEMO COMPLETE")
    print("=" * 70)
    print()
    print("The notification compression feature is now integrated into the mail app.")
    print("Configure it in ~/.config/luk/mail.toml:")
    print()
    print("  [content_display]")
    print("  compress_notifications = true")
    print("  notification_compression_mode = 'summary'  # or 'detailed' or 'off'")
    print()


if __name__ == "__main__":
    main()