from typing import Dict, List
from natsort import natsorted
from sqlalchemy import Column, ForeignKey, Integer, String, Boolean
from sqlalchemy.orm import relationship

from src.db.models.base import Base, BaseMixin


class Publisher(Base, BaseMixin):
    __tablename__ = "publisher"
    name = Column(String, unique=True)
    comics = relationship("Comic")

    def __repr__(self):
        return f'Publisher({self.id} {self.name})'

    def __str__(self):
        return self.__repr__()


class Comic(Base, BaseMixin):
    __tablename__ = 'comic'
    title = Column(String, unique=True)
    publisher_id = Column(String, ForeignKey('publisher.id'), nullable=False)
    publisher = relationship("Publisher", back_populates="comics")
    current_order = Column(Boolean)
    completed = Column(Boolean)
    issues = relationship("Issue", order_by="Issue.issue_number")
    story_arcs = relationship("StoryArc")
    trade_paperbacks = relationship("TradePaperback")
    volumes = relationship("Volume")
    comic_works = relationship("ComicWork")

    def __repr__(self):
        return f'Comic({self.id} {self.version} {self.title} {self.publisher.name})'

    def __str__(self):
        return f'{self.title}({self.id})'

    def get_artists(self) -> Dict[str, List[str]]:
        works: Dict[str, List[str]] = {}
        for work in self.comic_works:
            work_type = work.work_type.name
            artist = work.artist
            if work_type in works:
                works[work_type].append(artist)
            else:
                works[work_type] = [artist]
        return works

    def sorted_issues(self):
        sorted_issues = natsorted(self.issues, key=lambda x: getattr(x, 'issue_number'))
        return sorted_issues


class Volume(Base, BaseMixin):
    __tablename__ = "volume"
    name = Column(String, nullable=False)
    comic_id = Column(String, ForeignKey("comic.id"), nullable=False)
    comic = relationship("Comic", back_populates="volumes")
    issues = relationship("Issue")


class TradePaperback(Base, BaseMixin):
    __tablename__ = "trade_paperback"
    name = Column(String, nullable=False)
    issue_start = Column(Integer)
    issue_end = Column(Integer)
    comic_id = Column(String, ForeignKey("comic.id"), nullable=False)
    comic = relationship("Comic", back_populates="trade_paperbacks")


class StoryArc(Base, BaseMixin):
    __tablename__ = "story_arc"
    name = Column(String, nullable=False)
    comic_id = Column(String, ForeignKey("comic.id"), nullable=False)
    comic = relationship("Comic", back_populates="story_arcs")


class Issue(Base, BaseMixin):
    __tablename__ = "issue"
    issue_number = Column(String)
    in_stock = Column(Boolean)
    is_read = Column(Boolean)
    comic_id = Column(String, ForeignKey("comic.id"), nullable=False)
    comic = relationship("Comic", back_populates="issues")
    volume_id = Column(String, ForeignKey("volume.id"), nullable=True)
    volume = relationship("Volume", back_populates="issues")


class Artist(Base, BaseMixin):
    __tablename__ = "artist"
    name = Column(String, nullable=False)
    comic_works = relationship("ComicWork")

    def get_comics(self) -> Dict[str, List[str]]:
        works: Dict[str, List[str]] = {}
        for work in self.comic_works:
            work_type = work.work_type.name
            comic = work.comic
            if work_type in works:
                works[work_type].append(comic)
            else:
                works[work_type] = [comic]
        return works


class WorkType(Base, BaseMixin):
    __tablename__ = "worktype"
    name = Column(String, nullable=False, unique=True)
    comic_works = relationship("ComicWork")

    def __repr__(self):
        return f'Worktype({self.id} {self.version} {self.name} {len(self.comic_works)})'

    def __str__(self):
        return f'{self.name}({self.id})'


class ComicWork(Base, BaseMixin):
    __tablename__ = "comic_work"
    comic_id = Column(String, ForeignKey("comic.id"), nullable=False)
    comic = relationship("Comic", back_populates="comic_works")
    artist_id = Column(String, ForeignKey("artist.id"), nullable=False)
    artist = relationship("Artist", back_populates="comic_works")
    work_type_id = Column(String, ForeignKey("worktype.id"), nullable=False)
    work_type = relationship("WorkType", back_populates="comic_works")