#!/usr/bin/env python3
"""
Quick markdown formatting utility
"""

import sys
import os
import argparse

def format_for_whatsapp(text):
    """Quick format for WhatsApp"""
    from convert_to_whatsapp import convert_to_whatsapp
    return convert_to_whatsapp(text)

def format_for_discord(text):
    """Quick format for Discord"""
    from convert_to_discord import convert_to_discord
    return convert_to_discord(text)

def format_clean(text):
    """Quick clean format"""
    from clean_markdown import clean_markdown
    return clean_markdown(text)

def main():
    parser = argparse.ArgumentParser(description='Quick markdown formatting')
    parser.add_argument('--platform', '-p', choices=['whatsapp', 'discord', 'clean'], 
                       default='whatsapp', help='Target platform')
    parser.add_argument('--input', '-i', help='Input file (stdin if not specified)')
    parser.add_argument('--output', '-o', help='Output file (stdout if not specified)')
    
    args = parser.parse_args()
    
    # Read input
    if args.input:
        with open(args.input, 'r', encoding='utf-8') as f:
            text = f.read()
    else:
        # Read from stdin
        text = sys.stdin.read()
    
    # Format based on platform
    if args.platform == 'whatsapp':
        formatted = format_for_whatsapp(text)
    elif args.platform == 'discord':
        formatted = format_for_discord(text)
    else:  # clean
        formatted = format_clean(text)
    
    # Write output
    if args.output:
        with open(args.output, 'w', encoding='utf-8') as f:
            f.write(formatted)
        print(f"✅ Formatted for {args.platform}: {args.output}", file=sys.stderr)
    else:
        print(formatted)
    
    # Show quick stats
    print(f"📊 {args.platform}: {len(formatted)} chars, {formatted.count(chr(10)) + 1} lines", 
          file=sys.stderr)

if __name__ == "__main__":
    # Add current directory to path for imports
    sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
    main()