urls.py

from __future__ import annotations

from django.conf import settings
from django.contrib import admin
from django.urls import include, path, re_path, register_converter, reverse_lazy
from django.views.generic import RedirectView
from drf_yasg import openapi
from drf_yasg.views import get_schema_view
from rest_framework import permissions
from rest_framework_extensions.routers import ExtendedSimpleRouter

from dandiapi.api.views import (
    AssetViewSet,
    DandisetViewSet,
    DashboardView,
    NestedAssetViewSet,
    VersionViewSet,
    asset_audit_events,
    auth_token_view,
    authorize_view,
    blob_read_view,
    info_view,
    mailchimp_csv_view,
    robots_txt_view,
    root_content_view,
    schema_list_view,
    schema_view,
    stats_view,
    upload_complete_view,
    upload_initialize_view,
    upload_validate_view,
    user_approval_view,
    user_questionnaire_form_view,
    users_list_view,
    users_me_view,
    users_search_view,
    webdav,
)
from dandiapi.search.views import search_genotypes, search_species
from dandiapi.zarr.views import ZarrViewSet

router = ExtendedSimpleRouter()
(
    router.register(r'dandisets', DandisetViewSet, basename='dandiset')
    .register(
        r'versions',
        VersionViewSet,
        basename='version',
        parents_query_lookups=[f'dandiset__{DandisetViewSet.lookup_field}'],
    )
    .register(
        r'assets',
        NestedAssetViewSet,
        basename='asset',
        parents_query_lookups=[
            f'versions__dandiset__{DandisetViewSet.lookup_field}',
            f'versions__{VersionViewSet.lookup_field}',
        ],
    )
)
router.register('assets', AssetViewSet, basename='asset')
router.register('zarr', ZarrViewSet, basename='zarr')

# All core API endpoints
api_urlpatterns = [
    path('api/', include(router.urls)),
    path('api/auth/token/', auth_token_view, name='auth-token'),
    path('api/stats/', stats_view),
    path('api/info/', info_view),
    path('api/blobs/digest/', blob_read_view, name='blob-read'),
    path('api/schemas/available/', schema_list_view, name='schema-list-view'),
    path('api/schemas/', schema_view, name='schema-view'),
    path('api/uploads/initialize/', upload_initialize_view, name='upload-initialize'),
    re_path(
        r'api/uploads/(?P<upload_id>[0-9a-f\-]{36})/complete/',
        upload_complete_view,
        name='upload-complete',
    ),
    re_path(
        r'^api/uploads/(?P<upload_id>[0-9a-f\-]{36})/validate/$',
        upload_validate_view,
        name='upload-validate',
    ),
    path('api/users/me/', users_me_view),
    path('api/users/search/', users_search_view),
    path('api/users/', users_list_view),
    path('api/users/questionnaire-form/', user_questionnaire_form_view, name='user-questionnaire'),
    path('api/search/genotypes/', search_genotypes),
    path('api/search/species/', search_species),
]
schema_view = get_schema_view(
    openapi.Info(
        title='DANDI Archive',
        default_version='v1',
        description='The BRAIN Initiative archive for publishing and sharing '
        'cellular neurophysiology data',
    ),
    public=True,
    permission_classes=(permissions.AllowAny,),
    patterns=api_urlpatterns,
)

# Webdav only endpoints
webdav_urlpatterns = [
    path('api/webdav/assets/atpath/', webdav.atpath),
]
webdav_schema_view = get_schema_view(
    openapi.Info(
        title='Webdav API',
        default_version='v1',
    ),
    public=True,
    permission_classes=(permissions.AllowAny,),
    patterns=webdav_urlpatterns,
)


class DandisetIDConverter:
    regex = r'\d{6}'

    def to_python(self, value):
        return value

    def to_url(self, value):
        return value


register_converter(DandisetIDConverter, 'dandiset_id')
urlpatterns = [
    path('', root_content_view),
    path('robots.txt', robots_txt_view, name='robots_txt'),
    path('api/audit/events/asset', asset_audit_events, name='asset_audit_events'),
    *api_urlpatterns,
    *webdav_urlpatterns,
    path('admin/', admin.site.urls),
    path('accounts/', include('allauth.urls')),
    path('dashboard/', DashboardView.as_view(), name='dashboard-index'),
    path('dashboard/user/<str:username>/', user_approval_view, name='user-approval'),
    path('dashboard/mailchimp/', mailchimp_csv_view, name='mailchimp-csv'),
    # this url overrides the authorize url in oauth2_provider.urls to
    # support our user signup workflow
    path('oauth/authorize/', authorize_view, name='authorize'),
    path('oauth/', include('oauth2_provider.urls')),
    # Doc page views
    path('api/docs/redoc/', schema_view.with_ui('redoc'), name='docs-redoc'),
    path('api/docs/swagger/', schema_view.with_ui('swagger'), name='docs-swagger'),
    # Doc page redirects for backwards compatibility
    path(
        'swagger/',
        RedirectView.as_view(permanent=True, url=reverse_lazy('docs-swagger')),
        name='schema-swagger-ui',
    ),
    path(
        'redoc/',
        RedirectView.as_view(permanent=True, url=reverse_lazy('docs-redoc')),
        name='schema-redoc',
    ),
    # Webdav doc page views
    path(
        'api/webdav/docs/swagger/',
        webdav_schema_view.with_ui('swagger', cache_timeout=0),
        name='webdav-schema-swagger-ui',
    ),
]

if settings.DEBUG:
    import debug_toolbar.toolbar

    urlpatterns += [
        *debug_toolbar.toolbar.debug_toolbar_urls(),
        path('__reload__/', include('django_browser_reload.urls')),
    ]