chg: [API] Improved proxy API.

proxy/api/serializers.py

@@ -12,7 +12,7 @@ class Meta:
 class ModuleSerializer(serializers.ModelSerializer):
     class Meta:
         model = Module
-        fields = ["name", "path", "upstream", "authentication_required"]
+        fields = ["id", "name", "path", "upstream", "authentication_required"]
 
 
 class ModuleInputSerializer(serializers.ModelSerializer):

proxy/api/urls.py

@@ -8,6 +8,7 @@
 from .views import (
     ExternalTokenApiElemView,
     ExternalTokenApiView,
+    ModuleApiElemView,
     ModuleApiView,
     UserApiView,
 )
@@ -24,4 +25,5 @@
     path("externaltoken/", ExternalTokenApiView.as_view()),
     path("externaltoken/<int:id>", ExternalTokenApiElemView.as_view()),
     path("module/", ModuleApiView.as_view()),
+    path("module/<int:id>", ModuleApiElemView.as_view()),
 ]

proxy/api/views.py

@@ -158,3 +158,18 @@ def post(self, request, *args, **kwargs):
         )
         serializer = ModuleSerializer(new_module)
         return Response(serializer.data, status=status.HTTP_200_OK)
+
+
+class ModuleApiElemView(GenericAPIView):
+    # add permission to check if user is authenticated
+    authentication_classes = [SessionAuthentication, BasicAuthentication]
+    permission_classes = [IsAuthenticated, IsAdminUser]
+    serializer_class = ModuleSerializer
+
+    def delete(self, request, id=None):
+        """
+        Delete a module.
+        """
+        module = Module.objects.filter(id=id)
+        module.delete()
+        return Response(status=status.HTTP_204_NO_CONTENT)

proxy/views.py

@@ -17,6 +17,7 @@ def get_proxy_request_headers(self, request):
             external_token = ExternalToken.objects.get(
                 user=request.user, module__path=module_path
             )
+            print(external_token.token)
         except ExternalToken.DoesNotExist:
             # return the headers without the authentication token
             # users should be blocked by the proxified module
@@ -25,4 +26,7 @@ def get_proxy_request_headers(self, request):
             return headers
 
         headers["Proxy-Token"] = external_token.token
+        headers["X-Forwarded-Prefix"] = module_path
+        headers["X-Script-Name"] = module_path
+        print(headers)
         return headers