# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
from dataclasses import dataclass, field
from typing import Optional
from .taggers import (
    TaggerInterface,
    SHATagger,
    DateTagger,
    UbuntuVersionTagger,
    PythonVersionTagger,
    JupyterNotebookVersionTagger,
    JupyterLabVersionTagger,
    JupyterHubVersionTagger,
    RVersionTagger,
    TensorflowVersionTagger,
    JuliaVersionTagger,
    SparkVersionTagger,
    HadoopVersionTagger,
    JavaVersionTagger,
)
from .manifests import (
    ManifestInterface,
    CondaEnvironmentManifest,
    AptPackagesManifest,
    RPackagesManifest,
    JuliaPackagesManifest,
    SparkInfoManifest,
)


@dataclass
class ImageDescription:
    parent_image: Optional[str]
    taggers: list[TaggerInterface] = field(default_factory=list)
    manifests: list[ManifestInterface] = field(default_factory=list)


ALL_IMAGES = {
    "base-notebook": ImageDescription(
        parent_image=None,
        taggers=[
            SHATagger,
            DateTagger,
            UbuntuVersionTagger,
            PythonVersionTagger,
            JupyterNotebookVersionTagger,
            JupyterLabVersionTagger,
            JupyterHubVersionTagger,
        ],
        manifests=[CondaEnvironmentManifest, AptPackagesManifest],
    ),
    "minimal-notebook": ImageDescription(parent_image="base-notebook"),
    "scipy-notebook": ImageDescription(parent_image="minimal-notebook"),
    "r-notebook": ImageDescription(
        parent_image="minimal-notebook",
        taggers=[RVersionTagger],
        manifests=[RPackagesManifest],
    ),
    "tensorflow-notebook": ImageDescription(
        parent_image="scipy-notebook", taggers=[TensorflowVersionTagger]
    ),
    "datascience-notebook": ImageDescription(
        parent_image="scipy-notebook",
        taggers=[RVersionTagger, JuliaVersionTagger],
        manifests=[RPackagesManifest, JuliaPackagesManifest],
    ),
    "pyspark-notebook": ImageDescription(
        parent_image="scipy-notebook",
        taggers=[SparkVersionTagger, HadoopVersionTagger, JavaVersionTagger],
        manifests=[SparkInfoManifest],
    ),
    "all-spark-notebook": ImageDescription(
        parent_image="pyspark-notebook",
        taggers=[RVersionTagger],
        manifests=[RPackagesManifest],
    ),
}