from django.urls import path

from api.views.coins import (
    CoinsDatasetsView,
    CoinsEntitiesView,
    CoinsModelsView,
    CoinsPredictView,
    CoinsQueryStructuresView,
    CoinsRelationsView,
    CoinsSampleQueryView,
    CoinsSampleTriplesView,
)
from api.views.graph_generation import (
    GraphGenContinueView, GraphGenDatasetsView, GraphGenGenerateView, GraphGenSamplingModesView,
)
from api.views.health import ApiRootView, ForceUnlockView, HealthView, MethodsView
from api.views.kg_anomaly import (
    KgAnomalyContinueView,
    KgAnomalyCorrectView,
    KgAnomalyDatasetsView,
    KgAnomalySampleSubgraphsView,
)

urlpatterns = [
    # Health & discovery
    path("", ApiRootView.as_view()),
    path("health", HealthView.as_view()),
    path("methods", MethodsView.as_view()),
    path("debug/force-unlock", ForceUnlockView.as_view()),
    # COINs
    path("coins/datasets", CoinsDatasetsView.as_view()),
    path("coins/datasets/<str:dataset_id>/entities", CoinsEntitiesView.as_view()),
    path("coins/datasets/<str:dataset_id>/relations", CoinsRelationsView.as_view()),
    path("coins/datasets/<str:dataset_id>/sample-triples", CoinsSampleTriplesView.as_view()),
    path("coins/datasets/<str:dataset_id>/sample-query", CoinsSampleQueryView.as_view()),
    path("coins/models", CoinsModelsView.as_view()),
    path("coins/query-structures", CoinsQueryStructuresView.as_view()),
    path("coins/predict", CoinsPredictView.as_view()),
    # Graph generation
    path("graph-generation/datasets", GraphGenDatasetsView.as_view()),
    path("graph-generation/sampling-modes", GraphGenSamplingModesView.as_view()),
    path("graph-generation/generate", GraphGenGenerateView.as_view()),
    path("graph-generation/continue", GraphGenContinueView.as_view()),
    # KG anomaly
    path("kg-anomaly/datasets", KgAnomalyDatasetsView.as_view()),
    path("kg-anomaly/datasets/<str:dataset_id>/sample-subgraphs", KgAnomalySampleSubgraphsView.as_view()),
    path("kg-anomaly/correct", KgAnomalyCorrectView.as_view()),
    path("kg-anomaly/continue", KgAnomalyContinueView.as_view()),
]