Proxy v2 (#189)

passbook/outposts/views.py

@@ -0,0 +1,78 @@
+"""passbook outpost views"""
+from typing import Any, Dict, List
+
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.db.models import Model
+from django.http import HttpRequest, HttpResponse
+from django.shortcuts import get_object_or_404
+from django.views import View
+from django.views.generic import TemplateView
+from guardian.shortcuts import get_objects_for_user
+from structlog import get_logger
+
+from passbook.core.models import User
+from passbook.outposts.controllers.compose import DockerComposeController
+from passbook.outposts.models import Outpost, OutpostType
+from passbook.providers.proxy.controllers.kubernetes import ProxyKubernetesController
+
+LOGGER = get_logger()
+
+
+def get_object_for_user_or_404(user: User, perm: str, **filters) -> Model:
+    """Wrapper that combines get_objects_for_user and get_object_or_404"""
+    return get_object_or_404(get_objects_for_user(user, perm), **filters)
+
+
+class DockerComposeView(LoginRequiredMixin, View):
+    """Generate docker-compose yaml"""
+
+    def get(self, request: HttpRequest, outpost_pk: str) -> HttpResponse:
+        """Render docker-compose file"""
+        outpost: Outpost = get_object_for_user_or_404(
+            request.user, "passbook_outposts.view_outpost", pk=outpost_pk,
+        )
+        manifest = ""
+        if outpost.type == OutpostType.PROXY:
+            controller = DockerComposeController(outpost_pk)
+            manifest = controller.get_static_deployment()
+
+        return HttpResponse(manifest, content_type="text/vnd.yaml")
+
+
+class KubernetesManifestView(LoginRequiredMixin, View):
+    """Generate Kubernetes Deployment and SVC for proxy"""
+
+    def get(self, request: HttpRequest, outpost_pk: str) -> HttpResponse:
+        """Render deployment template"""
+        outpost: Outpost = get_object_for_user_or_404(
+            request.user, "passbook_outposts.view_outpost", pk=outpost_pk,
+        )
+        manifest = ""
+        if outpost.type == OutpostType.PROXY:
+            controller = ProxyKubernetesController(outpost_pk)
+            manifest = controller.get_static_deployment()
+
+        return HttpResponse(manifest, content_type="text/vnd.yaml")
+
+
+class SetupView(LoginRequiredMixin, TemplateView):
+    """Setup view"""
+
+    def get_template_names(self) -> List[str]:
+        allowed = ["dc", "custom", "k8s_manual", "k8s_integration"]
+        setup_type = self.request.GET.get("type", "dc")
+        if setup_type not in allowed:
+            setup_type = allowed[0]
+        return [f"outposts/setup_{setup_type}.html"]
+
+    def get_context_data(self, **kwargs: Any) -> Dict[str, Any]:
+        kwargs = super().get_context_data(**kwargs)
+        outpost: Outpost = get_object_for_user_or_404(
+            self.request.user,
+            "passbook_outposts.view_outpost",
+            pk=self.kwargs["outpost_pk"],
+        )
+        kwargs.update(
+            {"host": self.request.build_absolute_uri("/"), "outpost": outpost}
+        )
+        return kwargs