2022-11-08 10:50:05 +01:00

54 lines
1.8 KiB
Python

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))