from datetime import datetime
from enum import Enum
from typing import List, Optional, Set

from pydantic import BaseModel


class AppReview(BaseModel):
    uid: str
    rated_at: datetime
    score: float
    review: str
    username: Optional[str] = None
    response: Optional[str] = None
    responded_at: Optional[datetime] = None

    @classmethod
    def from_json(cls, json_data: dict):
        return cls(
            uid=json_data['uid'],
            ratedAt=datetime.fromisoformat(json_data['rated_at']),
            score=json_data['score'],
            review=json_data['review'],
            username=json_data.get('username'),
            response=json_data.get('response'),
            responded_at=datetime.fromisoformat(json_data['responded_at']) if json_data.get('responded_at') else None
        )


class AuthStep(BaseModel):
    name: str
    url: str


class ActionType(str, Enum):
    CREATE_MEMORY = "create_conversation"
    CREATE_FACTS = "create_facts"
    READ_MEMORIES = "read_memories"
    READ_CONVERSATIONS = "read_conversations"


class Action(BaseModel):
    action: ActionType


class ExternalIntegration(BaseModel):
    triggers_on: Optional[str] = None
    webhook_url: Optional[str] = None
    setup_completed_url: Optional[str] = None
    setup_instructions_file_path: Optional[str]
    is_instructions_url: bool = True
    auth_steps: Optional[List[AuthStep]] = []
    app_home_url: Optional[str] = None
    actions: Optional[List[Action]] = []


class ProactiveNotification(BaseModel):
    scopes: Set[str]


class ApiKey(BaseModel):
    id: str
    hashed: str
    label: str
    created_at: Optional[datetime] = None


class App(BaseModel):
    id: str
    name: str
    uid: Optional[str] = None
    private: bool = False
    approved: bool = False
    status: str = 'approved'
    category: str
    email: Optional[str] = None
    author: str
    description: str
    image: str
    capabilities: Set[str]
    memory_prompt: Optional[str] = None
    chat_prompt: Optional[str] = None
    persona_prompt: Optional[str] = None
    username: Optional[str] = None
    connected_accounts: List[str] = []
    twitter: Optional[dict] = None
    external_integration: Optional[ExternalIntegration] = None
    reviews: List[AppReview] = []
    user_review: Optional[AppReview] = None
    rating_avg: Optional[float] = 0
    rating_count: int = 0
    enabled: bool = False
    deleted: bool = False
    trigger_workflow_memories: bool = True  # default true
    installs: int = 0
    proactive_notification: Optional[ProactiveNotification] = None
    created_at: Optional[datetime] = None
    money_made: Optional[float] = None
    usage_count: Optional[int] = None
    is_paid: Optional[bool] = False
    price: Optional[float] = 0.0  # cents/100
    payment_plan: Optional[str] = None
    payment_product_id: Optional[str] = None
    payment_price_id: Optional[str] = None
    payment_link_id: Optional[str] = None
    payment_link: Optional[str] = None
    is_user_paid: Optional[bool] = False
    thumbnails: Optional[List[str]] = []  # List of thumbnail IDs
    thumbnail_urls: Optional[List[str]] = []  # List of thumbnail URLs
    is_influencer: Optional[bool] = False
    is_popular: Optional[bool] = False

    def get_rating_avg(self) -> Optional[str]:
        return f'{self.rating_avg:.1f}' if self.rating_avg is not None else None

    def has_capability(self, capability: str) -> bool:
        return capability in self.capabilities

    def works_with_memories(self) -> bool:
        return self.has_capability('memories')

    def works_with_chat(self) -> bool:
        return self.has_capability('chat') or self.has_capability('persona')

    def is_a_persona(self) -> bool:
        return self.has_capability('persona')

    def works_externally(self) -> bool:
        return self.has_capability('external_integration')

    def triggers_on_conversation_creation(self) -> bool:
        return self.works_externally() and self.external_integration.triggers_on == 'memory_creation'

    def triggers_realtime(self) -> bool:
        return self.works_externally() and self.external_integration.triggers_on == 'transcript_processed'

    def triggers_realtime_audio_bytes(self) -> bool:
        return self.works_externally() and self.external_integration.triggers_on == 'audio_bytes'

    def filter_proactive_notification_scopes(self, params: [str]) -> []:
        if not self.proactive_notification:
            return []
        return [param for param in params if param in self.proactive_notification.scopes]

    def get_image_url(self) -> str:
        return f'https://raw.githubusercontent.com/BasedHardware/Omi/main{self.image}'

class AppCreate(BaseModel):
    id: str
    name: str
    uid: Optional[str] = None
    private: bool = False
    approved: bool = False
    status: str = 'approved'
    category: str
    email: Optional[str] = None
    author: str
    description: str
    image: str
    capabilities: Set[str]
    memory_prompt: Optional[str] = None
    chat_prompt: Optional[str] = None
    persona_prompt: Optional[str] = None
    username: Optional[str] = None
    connected_accounts: List[str] = []
    twitter: Optional[dict] = None
    external_integration: Optional[ExternalIntegration] = None
    deleted: bool = False
    proactive_notification: Optional[ProactiveNotification] = None
    created_at: Optional[datetime] = None
    is_paid: Optional[bool] = False
    price: Optional[float] = 0.0  # cents/100
    payment_plan: Optional[str] = None
    thumbnails: Optional[List[str]] = []  # List of thumbnail IDs

class AppUpdate(BaseModel):
    id: str
    name: Optional[str] = None
    uid: Optional[str] = None
    private: Optional[bool] = None
    category: Optional[str] = None
    email: Optional[str] = None
    author: Optional[str] = None
    description: Optional[str] = None
    image: Optional[str] = None
    capabilities: Optional[Set[str]] = None
    memory_prompt: Optional[str] = None
    chat_prompt: Optional[str] = None
    persona_prompt: Optional[str] = None
    username: Optional[str] = None
    connected_accounts: Optional[List[str]] = None
    twitter: Optional[dict] = None
    external_integration: Optional[ExternalIntegration] = None
    deleted: Optional[bool] = None
    proactive_notification: Optional[ProactiveNotification] = None
    created_at: Optional[datetime] = None
    is_paid: Optional[bool] = None
    price: Optional[float] = None  # cents/100
    payment_plan: Optional[str] = None
    thumbnails: Optional[List[str]] = None  # List of thumbnail IDs
    updated_at: Optional[datetime] = None

class UsageHistoryType(str, Enum):
    memory_created_external_integration = 'memory_created_external_integration'
    transcript_processed_external_integration = 'transcript_processed_external_integration'
    memory_created_prompt = 'memory_created_prompt'
    chat_message_sent = 'chat_message_sent'


class UsageHistoryItem(BaseModel):
    uid: str
    memory_id: Optional[str] = None
    timestamp: datetime
    type: UsageHistoryType