"""
Cultural Adaptation Module
Author: RSK World (https://rskworld.in)
Founder: Molla Samser
Designer & Tester: Rima Khatun
Contact: help@rskworld.in, +91 93305 39277
Year: 2026
Description: Cultural context adaptation for different regions
"""

import logging
from datetime import datetime
import pytz
import re

logger = logging.getLogger(__name__)

class CulturalAdapter:
    def __init__(self):
        self.cultural_contexts = {
            'en': {
                'greeting_style': 'casual',
                'formality_level': 'medium',
                'date_format': '%Y-%m-%d',
                'time_format': '%I:%M %p',
                'timezone': 'UTC',
                'cultural_notes': []
            },
            'hi': {
                'greeting_style': 'respectful',
                'formality_level': 'high',
                'date_format': '%d-%m-%Y',
                'time_format': '%I:%M %p',
                'timezone': 'Asia/Kolkata',
                'cultural_notes': ['Use respectful terms like "ji"', 'Avoid direct refusal', 'Consider family context']
            },
            'bn': {
                'greeting_style': 'respectful',
                'formality_level': 'high',
                'date_format': '%d-%m-%Y',
                'time_format': '%I:%M %p',
                'timezone': 'Asia/Dhaka',
                'cultural_notes': ['Use formal address', 'Show respect for elders', 'Consider cultural festivals']
            },
            'es': {
                'greeting_style': 'warm',
                'formality_level': 'medium',
                'date_format': '%d/%m/%Y',
                'time_format': '%H:%M',
                'timezone': 'Europe/Madrid',
                'cultural_notes': ['Use warm greetings', 'Consider siesta time', 'Family-oriented responses']
            },
            'fr': {
                'greeting_style': 'formal',
                'formality_level': 'high',
                'date_format': '%d/%m/%Y',
                'time_format': '%H:%M',
                'timezone': 'Europe/Paris',
                'cultural_notes': ['Use formal language', 'Proper greetings important', 'Cultural sophistication']
            },
            'de': {
                'greeting_style': 'formal',
                'formality_level': 'high',
                'date_format': '%d.%m.%Y',
                'time_format': '%H:%M',
                'timezone': 'Europe/Berlin',
                'cultural_notes': ['Be direct and precise', 'Use formal titles', 'Punctuality important']
            },
            'zh': {
                'greeting_style': 'respectful',
                'formality_level': 'high',
                'date_format': '%Y年%m月%d日',
                'time_format': '%H:%M',
                'timezone': 'Asia/Shanghai',
                'cultural_notes': ['Show respect for hierarchy', 'Avoid direct confrontation', 'Consider face-saving']
            },
            'ja': {
                'greeting_style': 'very_formal',
                'formality_level': 'very_high',
                'date_format': '%Y年%m月%d日',
                'time_format': '%H:%M',
                'timezone': 'Asia/Tokyo',
                'cultural_notes': ['Use honorific language', 'Be polite and indirect', 'Consider seasonal context']
            },
            'ar': {
                'greeting_style': 'formal',
                'formality_level': 'high',
                'date_format': '%d/%m/%Y',
                'time_format': '%I:%M %p',
                'timezone': 'Asia/Riyadh',
                'cultural_notes': ['Use Islamic greetings', 'Consider prayer times', 'Family and tradition important']
            },
            'pt': {
                'greeting_style': 'warm',
                'formality_level': 'medium',
                'date_format': '%d/%m/%Y',
                'time_format': '%H:%M',
                'timezone': 'America/Sao_Paulo',
                'cultural_notes': ['Warm and friendly', 'Personal connections', 'Consider regional differences']
            },
            'ru': {
                'greeting_style': 'formal',
                'formality_level': 'high',
                'date_format': '%d.%m.%Y',
                'time_format': '%H:%M',
                'timezone': 'Europe/Moscow',
                'cultural_notes': ['Use formal address', 'Be direct but polite', 'Consider cultural heritage']
            },
            'it': {
                'greeting_style': 'warm',
                'formality_level': 'medium',
                'date_format': '%d/%m/%Y',
                'time_format': '%H:%M',
                'timezone': 'Europe/Rome',
                'cultural_notes': ['Expressive communication', 'Personal relationships', 'Cultural references']
            }
        }
        
        self.regional_greetings = {
            'en': {
                'morning': ['Good morning', 'Hello'],
                'afternoon': ['Good afternoon', 'Hello'],
                'evening': ['Good evening', 'Hello'],
                'general': ['Hello', 'Hi', 'Hey']
            },
            'hi': {
                'morning': ['सुप्रभात', 'नमस्ते'],
                'afternoon': ['नमस्ते'],
                'evening': ['शुभ संध्या', 'नमस्ते'],
                'general': ['नमस्ते', 'प्रणाम']
            },
            'bn': {
                'morning': ['সুপ্রভাত', 'আসসালামু আলাইকুম'],
                'afternoon': ['আসসালামু আলাইকুম'],
                'evening': ['সুভ সন্ধ্যা', 'আসসালামু আলাইকুম'],
                'general': ['আসসালামু আলাইকুম', 'অ্যাডাব']
            },
            'es': {
                'morning': ['Buenos días', 'Hola'],
                'afternoon': ['Buenas tardes', 'Hola'],
                'evening': ['Buenas noches', 'Hola'],
                'general': ['Hola', '¿Qué tal?']
            },
            'fr': {
                'morning': ['Bonjour', 'Salut'],
                'afternoon': ['Bonjour', 'Salut'],
                'evening': ['Bonsoir', 'Salut'],
                'general': ['Bonjour', 'Salut']
            }
        }
    
    def adapt(self, text: str, target_language: str) -> str:
        """
        Adapt text for cultural context
        """
        try:
            if target_language not in self.cultural_contexts:
                return text
            
            context = self.cultural_contexts[target_language]
            adapted_text = text
            
            # Adapt formality level
            adapted_text = self._adapt_formality(adapted_text, context['formality_level'])
            
            # Add cultural context if appropriate
            adapted_text = self._add_cultural_context(adapted_text, target_language)
            
            # Adapt time/date references
            adapted_text = self._adapt_datetime_references(adapted_text, context)
            
            # Adapt greetings
            adapted_text = self._adapt_greetings(adapted_text, target_language)
            
            return adapted_text
            
        except Exception as e:
            logger.error(f"Cultural adaptation error: {str(e)}")
            return text
    
    def _adapt_formality(self, text: str, formality_level: str) -> str:
        """Adapt text formality level"""
        if formality_level in ['high', 'very_high']:
            # Make text more formal
            text = re.sub(r'\b(hi|hey)\b', 'hello', text, flags=re.IGNORECASE)
            text = re.sub(r'\b(you\'re|you are)\b', 'you are', text, flags=re.IGNORECASE)
            text = re.sub(r'\b(can\'t|cannot)\b', 'cannot', text, flags=re.IGNORECASE)
        elif formality_level == 'casual':
            # Make text more casual
            text = re.sub(r'\byou are\b', "you're", text, flags=re.IGNORECASE)
            text = re.sub(r'\bcannot\b', "can't", text, flags=re.IGNORECASE)
        
        return text
    
    def _add_cultural_context(self, text: str, language: str) -> str:
        """Add cultural context to responses"""
        cultural_additions = {
            'hi': [' ji', ' सर', ' मैम'],
            'bn': [' স্যার', ' ম্যাম', ' ভাই'],
            'ja': [' さん', ' 様'],
            'ar': [' أخي', ' أختي', ' سيدي'],
            'es': [' señor', ' señora', ' amigo'],
            'fr': [' monsieur', ' madame', ' ami']
        }
        
        # This is a simplified implementation
        # In practice, you'd want more sophisticated context-aware additions
        return text
    
    def _adapt_datetime_references(self, text: str, context: dict) -> str:
        """Adapt date and time references"""
        try:
            # Get current time in target timezone
            timezone = pytz.timezone(context['timezone'])
            now = datetime.now(timezone)
            
            # Replace generic time references with specific ones
            current_hour = now.hour
            
            if 'morning' in text.lower() and current_hour >= 12:
                text = text.replace('morning', 'day')
            elif 'evening' in text.lower() and current_hour < 17:
                text = text.replace('evening', 'afternoon')
            
            return text
            
        except Exception as e:
            logger.error(f"Datetime adaptation error: {str(e)}")
            return text
    
    def _adapt_greetings(self, text: str, language: str) -> str:
        """Adapt greetings based on time of day and culture"""
        try:
            if language not in self.regional_greetings:
                return text
            
            greetings = self.regional_greetings[language]
            timezone = pytz.timezone(self.cultural_contexts[language]['timezone'])
            now = datetime.now(timezone)
            hour = now.hour
            
            # Determine time of day
            if 5 <= hour < 12:
                time_period = 'morning'
            elif 12 <= hour < 17:
                time_period = 'afternoon'
            else:
                time_period = 'evening'
            
            # Replace generic greetings with time-appropriate ones
            if 'hello' in text.lower() or 'hi' in text.lower():
                appropriate_greetings = greetings.get(time_period, greetings['general'])
                if appropriate_greetings:
                    text = re.sub(r'\b(hello|hi)\b', appropriate_greetings[0], text, flags=re.IGNORECASE)
            
            return text
            
        except Exception as e:
            logger.error(f"Greeting adaptation error: {str(e)}")
            return text
    
    def get_cultural_tips(self, language: str) -> list:
        """Get cultural tips for a language"""
        context = self.cultural_contexts.get(language, {})
        return context.get('cultural_notes', [])
    
    def get_local_time(self, language: str) -> str:
        """Get current local time for language region"""
        try:
            if language not in self.cultural_contexts:
                return datetime.now().strftime('%H:%M')
            
            context = self.cultural_contexts[language]
            timezone = pytz.timezone(context['timezone'])
            now = datetime.now(timezone)
            
            return now.strftime(context['time_format'])
            
        except Exception as e:
            logger.error(f"Local time error: {str(e)}")
            return datetime.now().strftime('%H:%M')
    
    def get_local_date(self, language: str) -> str:
        """Get current local date for language region"""
        try:
            if language not in self.cultural_contexts:
                return datetime.now().strftime('%Y-%m-%d')
            
            context = self.cultural_contexts[language]
            timezone = pytz.timezone(context['timezone'])
            now = datetime.now(timezone)
            
            return now.strftime(context['date_format'])
            
        except Exception as e:
            logger.error(f"Local date error: {str(e)}")
            return datetime.now().strftime('%Y-%m-%d')