#!/usr/bin/env python

import re
import sys

from datetime import datetime

import scrapy
from scrapy.crawler import CrawlerProcess
from scrapy.spiders import CrawlSpider, Rule
from scrapy.linkextractors import LinkExtractor

from sqlalchemy import create_engine, func, select, Engine, Integer, String, Text, DateTime
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
from sqlalchemy.orm import sessionmaker


class Base(DeclarativeBase):
    """Base déclarative pour SQLAlchemy 2.0."""
    pass


class TvShowDB(Base):
    """Modèle pour le stockage des épisodes (SQLAlchemy 2.0)."""

    __tablename__: str = "tvshows"

    id: Mapped[int] = mapped_column(
        Integer,
        primary_key=True,
        autoincrement=True
    )
    post_id: Mapped[str] = mapped_column(
        String(length=255),
        nullable=False,
        unique=True,
        index=True
    )
    post_title: Mapped[str] = mapped_column(
        String(255),
        nullable=False
    )
    title: Mapped[str] = mapped_column(
        String(255),
        nullable=False,
        index=True
    )
    date: Mapped[datetime] = mapped_column(
        DateTime,
        nullable=False,
        index=True
    )
    summary: Mapped[str | None] = mapped_column(
        Text,
        nullable=True
    )
    image_url: Mapped[str | None] = mapped_column(
        String(255),
        nullable=True
    )
    created_at: Mapped[datetime] = mapped_column(
        DateTime,
        server_default=func.datetime('now'),
        nullable=False
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime,
        server_default=func.datetime('now'),
        onupdate=func.datetime('now'),
        nullable=False
    )
    links: Mapped[list["LinkDB"]] = relationship(back_populates="show")


class LinkDB(Base):
    """Modèle pour le stockage des liens de téléchargement (SQLAlchemy 2.0)."""

    __tablename__: str = "links"

    id: Mapped[int] = mapped_column(
        Integer,
        primary_key=True,
        autoincrement=True
    )
    link: Mapped[str] = mapped_column(
        String(255),
        nullable=False
    )
    is_downloaded: Mapped[bool] = mapped_column(
        Boolean,
        default=False
    )
    show_id: Mapped[int] = mapped_column(ForeignKey("tvshows.id"))
    show: Mapped["TvShowDB"] = relationship(back_populates="links")


class TvShowItem(scrapy.Item):
    post_id: scrapy.Field = scrapy.Field()
    post_title: scrapy.Field = scrapy.Field()
    title:scrapy.Field = scrapy.Field()
    date: scrapy.Field = scrapy.Field()
    summary: scrapy.Field = scrapy.Field()
    image_url: scrapy.Field = scrapy.Field()
    download_url: scrapy.Field = scrapy.Field()


class SQLAlchemyPipeline:
    def __init__(self):
        self.engine: Engine = create_engine('sqlite:///tvshows.db', echo=True)
        Base.metadata.create_all(self.engine)
        self.Session = sessionmaker(bind=self.engine)

    def process_item(self, item, spider):
        session = self.Session()
        try:
            stmt = select(TvShowDB).where(TvShowDB.post_id == item["post_id"])
            show = session.scalars(stmt).first()
            print(f"{show=}")
            if not show:
                show = TvShowDB(**item)
                session.add(show)
            else:
                for key, value in item.items():
                    setattr(show, key, value)
            session.commit()
        except Exception as e:
            session.rollback()
            raise
        finally:
            session.close()
        return item


class TvShow(CrawlSpider):
    name: str = "rlsb_tvshow"
    allowed_domains: list[str] = ["rlsbb.ru"]
    start_urls: list[str] = ["https://rlsbb.ru/category/tv-shows/"]
    custom_settings = {
        'USER_AGENT': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36',
        'AUTOTHROTTLE_ENABLED': True,
        'DOWNLOAD_DELAY': 10,
        'ITEM_PIPELINES': {
            '__main__.SQLAlchemyPipeline': 300,
        },
    }
    rules: list[Rule] = [
        Rule(LinkExtractor(allow=r"/tv-shows/page/"), callback="parse", follow=True)
    ]

    def parse(self, response):
        for article in response.css("article"):
            item = TvShowItem()
            item['post_id'] = article.attrib['id']
            item['post_title'] = article.css('h1.entry-title > a::text').get()
            item['title'] = article.css('.entry-summary > p:nth-child(4) > strong::text').get()
            item['date'] = self.parse_date(article.css('.entry-meta-header-before::text').getall()[1].strip())
            item['summary'] = "".join(article.xpath('.//div[@class="entry-summary"]/node()').extract())
            item['image_url'] = article.css('.entry-summary > p > img::attr(src)').get()
            item['download_url'] = "#".join(article.css('.entry-summary > p > a[href ^= "https://rapidgator"]::attr(href)').extract())
            yield item

    def parse_date(self, formatted_date: str):
        formatted_date = re.sub(r'(\d)(st|nd|rd|th)', r'\1', formatted_date)
        return datetime.strptime(formatted_date, "Posted on %B %d, %Y at %H:%M in")


def main():
    process = CrawlerProcess()
    _ = process.crawl(TvShow)
    process.start()
    return 0


if __name__ == "__main__":
    sys.exit(main())