import logging
from dataclasses import asdict, dataclass, field
from typing import ClassVar, Dict, List

from jsonschema2ddl.types import COLUMNS_TYPES, COLUMNS_TYPES_PREFERENCE, FK_TYPES
from jsonschema2ddl.utils import db_column_name, get_one_schema


@dataclass
class Column:
    """Object to encapsulate a Column.

    Attributes:
        name (str): name of the Column.
        database_flavor (str): postgres or redshift. Defaults to postgres.
        comment (str): comment of the Column. Defaults to None.
        constraints (Dict): other columns constraints (not implemented).
        jsonschema_fields (Dict): Original fields in the jsonschema.
    """

    name: str
    database_flavor: str = "postgres"
    comment: str = field(default_factory=str, repr=False)
    constraints: Dict = field(default_factory=dict, repr=False)
    jsonschema_type: str = field(default_factory=str, repr=False)
    jsonschema_fields: Dict = field(default_factory=dict, repr=False)

    logger: ClassVar[logging.Logger] = field(default=logging.getLogger("Column"), repr=False)

    @property
    def max_lenght(self) -> int:
        return self.jsonschema_fields.get("maxLength", 256)

    @property
    def data_type(self) -> str:
        """Data type of the columns.

        It accounts of the mapping of the original type to the db types.

        Returns:
            str: data type of the column.
        """
        if "format" in self.jsonschema_fields and self.jsonschema_fields["format"] in COLUMNS_TYPES_PREFERENCE:
            self.jsonschema_type = self.jsonschema_fields["format"]

        return COLUMNS_TYPES[self.database_flavor][self.jsonschema_type].format(self.max_lenght)

    @property
    def is_pk(self) -> bool:
        return self.jsonschema_fields.get("pk", False)

    @property
    def is_index(self) -> bool:
        """Returns true if the column is a index.

        Returns:
            bool: True if it is index.
        """
        return self.jsonschema_fields.get("index", False)

    @property
    def is_unique(self) -> bool:
        """Returns true if the column is a unique.

        Returns:
            bool: True if it is unique.
        """
        return self.jsonschema_fields.get("unique", False)

    @staticmethod
    def is_fk() -> bool:
        """Returns true if the column is a foreign key.

        Returns:
            bool: True if it is foreign key
        """
        return False

    def __hash__(self):
        return hash(self.name)

    # NOTE: Overwrite dataclass method to show data_type property
    def __repr__(self):
        return f"Column(name={self.name} data_type={self.data_type})"


@dataclass
class Table:
    """Object to encapsulate a Table.

    Attributes:
        ref (str): id or reference to the table in the jsonschema.
        name (str): name of the table.
        database_flavor (str): postgres or redshift. Defaults to postgres.
        columns (List[Column]): columns of the table.
        primary_key (Column): Primary key column of the table.
        comment (str): comment of the table. Defaults to None.
        indexes (List[str]): Table indexeses (not implemented).
        unique_columns (List[str]): Table unique constraints (not implemented).
        jsonschema_fields (Dict): Original fields in the jsonschema.
    """

    ref: str
    name: str
    database_flavor: str = "postgres"
    columns: List[Column] = field(default_factory=list)
    primary_key: Column = field(default=None)
    comment: str = field(default=None)
    indexes: List[str] = field(default_factory=list)
    unique_columns: List[str] = field(default_factory=list)
    jsonschema_fields: Dict = field(default_factory=dict, repr=False)

    logger: ClassVar[logging.Logger] = field(default=logging.getLogger("Table"), repr=False)
    _expanded: bool = field(default=False, repr=False)

    def expand_columns(
        self,
        table_definitions: Dict = dict(),
        columns_definitions: Dict = dict(),
        referenced: bool = False,
    ):
        """Expand the columns definitions of the

        Args:
            table_definitions (Dict, optional): Dictionary with the rest of the
                tables definitions. It is used for recursive calls to get the
                foreign keys. Defaults to dict().
            columns_definitions (Dict, optional): Dictionary with the definition
                of columns outside the main properties field. Defaults to dict().
            referenced (bool, optional): Whether or not the table is referenced
                by others. Used to make sure there is a Primary Key defined.
                Defaults to False.
        """
        if self._expanded:
            self.logger.info("Already expanded table. Skiping...")
            return self
        for col_name, col_object in self.jsonschema_fields.get("properties").items():
            self.logger.debug(f"Creating column {col_name}")
            col_name = db_column_name(col_name)
            self.logger.debug(f"Renamed column to {col_name}")
            if "$ref" in col_object:
                self.logger.debug(f"Expanding {col_name} reference {col_object['$ref']}")
                self.logger.debug(table_definitions)
                if col_object["$ref"] in table_definitions:
                    ref = col_object["$ref"]
                    self.logger.debug(f"Column is a FK! Expanding {ref} before continue...")
                    table_definitions[ref] = table_definitions[ref].expand_columns(
                        table_definitions=table_definitions, referenced=True
                    )
                    col = FKColumn(
                        table_ref=table_definitions[ref],
                        name=col_name,
                        database_flavor=self.database_flavor,
                    )
                elif col_object["$ref"] in columns_definitions:
                    self.logger.debug("Column ref a type that is not a object. " "Copy Column from columns definitions")
                    ref = col_object["$ref"]
                    ref_col = columns_definitions[ref]
                    col_as_dict = {**asdict(ref_col), "name": col_name}
                    col = Column(**col_as_dict)
                else:
                    self.logger.debug("Skipping ref as it is not in table definitions neither in columns definitions")
                    continue
            else:
                if "type" not in col_object:
                    col_object = get_one_schema(col_object)
                col = Column(
                    name=col_name,
                    database_flavor=self.database_flavor,
                    jsonschema_type=col_object["type"],
                    jsonschema_fields=col_object,
                )
            self.columns.append(col)
            if col.is_pk:
                self.primary_key = col

            self.logger.info(f"New created column {col}")

        if referenced and not self.primary_key:
            self.logger.info("Creating id column for the table in order to reference it as PK")
            col = Column(
                name="id",
                database_flavor=self.database_flavor,
                jsonschema_type="id",
            )
            self.columns.append(col)
            self.primary_key = col

        self.columns = self._deduplicate_columns(self.columns)

        return self

    @staticmethod
    def _deduplicate_columns(columns) -> list:
        return list({c: None for c in columns})


@dataclass(eq=False)
class FKColumn(Column):
    """Special type of Column object to represent a foreign key

    Attributes:
        table_ref (Table): Pointer to the foreing table object
    """

    table_ref: Table = None

    @property
    def data_type(self) -> str:
        """Data type of the foreign key.

        Accounts of the data type of the primary key of the foreing table.

        Returns:
            str: the column data type.
        """
        data_type_ref = self.table_ref.primary_key.data_type
        if "varchar" in data_type_ref:
            return data_type_ref
        else:
            return FK_TYPES.get(data_type_ref, "bigint")

    @staticmethod
    def is_fk() -> bool:
        """Returns true if the column is a foreign key.

        Returns:
            bool: True if it is foreign key.
        """
        return True

    # FIXME: Avoid overwritting the the repr method
    # NOTE: Overwrite dataclass method to show data_type property
    def __repr__(self):
        return f"FKColumn(name={self.name} data_type={self.data_type} table_ref.name={self.table_ref.name})"
