tabby-web/backend/tabby/app/views.py

import fsspec
import os
from fsspec.implementations.local import LocalFileSystem
from django.conf import settings
from django.http.response import (
    FileResponse,
    HttpResponseNotFound,
    HttpResponseRedirect,
)
from django.views import static
from rest_framework.views import APIView
from urllib.parse import urlparse


class IndexView(APIView):
    def get(self, request, format=None):
        if settings.FRONTEND_URL:
            return HttpResponseRedirect(settings.FRONTEND_URL)
        return static.serve(
            request, "index.html", document_root=str(settings.STATIC_ROOT)
        )


class TerminalView(APIView):
    def get(self, request, format=None):
        response = static.serve(
            request, "terminal.html", document_root=str(settings.STATIC_ROOT)
        )
        response["Content-Security-Policy"] = "frame-ancestors 'self' https://tabby.sh;"
        response["X-Frame-Options"] = "SAMEORIGIN"
        return response


class DemoView(APIView):
    def get(self, request, format=None):
        response = static.serve(
            request, "demo.html", document_root=str(settings.STATIC_ROOT)
        )
        response["Content-Security-Policy"] = "frame-ancestors 'self' https://tabby.sh;"
        response['X-Frame-Options'] = 'ALLOW-FROM https://tabby.sh'
        return response


class AppDistView(APIView):
    def get(self, request, version=None, path=None, format=None):
        fs = fsspec.filesystem(urlparse(settings.APP_DIST_STORAGE).scheme)
        url = f"{settings.APP_DIST_STORAGE}/{version}/{path}"
        if isinstance(fs, LocalFileSystem):
            if not fs.exists(url):
                return HttpResponseNotFound()
            return FileResponse(fs.open(url), filename=os.path.basename(url))
        else:
            return HttpResponseRedirect(fs.url(url))