import os
import datetime

from dotenv import load_dotenv
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
from dto.post import Post
from dto.comment import Comment
from server.connectors.base import BaseConnector

load_dotenv()

API_KEY = os.getenv("YOUTUBE_API_KEY")


class YouTubeAPI(BaseConnector):
    source_name: str = "youtube"
    display_name: str = "YouTube"
    search_enabled: bool = True
    categories_enabled: bool = False

    def __init__(self):
        self.youtube = build("youtube", "v3", developerKey=API_KEY)

    def get_new_posts_by_search(
        self, search: str, category: str, post_limit: int
    ) -> list[Post]:
        videos = self._search_videos(search, post_limit)
        posts = []

        for video in videos:
            video_id = video["id"]["videoId"]
            snippet = video["snippet"]
            title = snippet["title"]
            description = snippet["description"]
            published_at = datetime.datetime.strptime(
                snippet["publishedAt"], "%Y-%m-%dT%H:%M:%SZ"
            ).timestamp()
            channel_title = snippet["channelTitle"]

            comments = []
            comments_data = self._get_video_comments(video_id)
            for comment_thread in comments_data:
                comment_snippet = comment_thread["snippet"]["topLevelComment"][
                    "snippet"
                ]
                comment = Comment(
                    id=comment_thread["id"],
                    post_id=video_id,
                    content=comment_snippet["textDisplay"],
                    author=comment_snippet["authorDisplayName"],
                    timestamp=datetime.datetime.strptime(
                        comment_snippet["publishedAt"], "%Y-%m-%dT%H:%M:%SZ"
                    ).timestamp(),
                    reply_to=None,
                    source=self.source_name,
                )

                comments.append(comment)

            post = Post(
                id=video_id,
                content=f"{title}\n\n{description}",
                author=channel_title,
                timestamp=published_at,
                url=f"https://www.youtube.com/watch?v={video_id}",
                title=title,
                source=self.source_name,
                comments=comments,
            )

            posts.append(post)

        return posts

    def category_exists(self, category):
        return True

    def _search_videos(self, query, limit):
        request = self.youtube.search().list(
            q=query, part="snippet", type="video", maxResults=limit
        )
        response = request.execute()
        return response.get("items", [])

    def _get_video_comments(self, video_id):
        request = self.youtube.commentThreads().list(
            part="snippet", videoId=video_id, textFormat="plainText"
        )

        try:
            response = request.execute()
        except HttpError as e:
            print(f"Error fetching comments for video {video_id}: {e}")
            return []
        return response.get("items", [])