refactor: remove legacy kms

backend/api/models.py

@@ -5,10 +5,7 @@
     PermissionsMixin,
 )
 from uuid import uuid4
-from backend.api.kv import write
-import json
 from django.utils import timezone
-from django.conf import settings
 from api.services import Providers, ServiceConfig
 from api.tasks.syncing import trigger_sync_tasks
 from backend.quotas import (
@@ -19,7 +16,6 @@
 )
 from django.core.exceptions import ValidationError
 
-CLOUD_HOSTED = settings.APP_HOST == "cloud"
 
 
 class CustomUserManager(BaseUserManager):
@@ -156,17 +152,6 @@ class App(models.Model):
 
     objects = AppManager()
 
-    def save(self, *args, **kwargs):
-        super().save(*args, **kwargs)  # Call the "real" save() method.
-        if CLOUD_HOSTED:
-            key = self.app_token
-            value = self.wrapped_key_share
-            meta = {"appId": self.id, "appName": self.name, "live": True}
-            try:
-                write(key, value, json.dumps(meta))
-            except:
-                pass
-
     def __str__(self):
         return self.name
 

backend/backend/graphene/mutations/app.py

@@ -1,4 +1,3 @@
-from backend.api.kv import delete, purge
 from backend.graphene.mutations.environment import EnvironmentKeyInput
 from api.utils.access.permissions import (
     user_can_access_app,
@@ -16,11 +15,8 @@
     ServiceAccount,
 )
 from backend.graphene.types import AppType, MemberType
-from django.conf import settings
 from django.db.models import Q
 
-CLOUD_HOSTED = settings.APP_HOST == "cloud"
-
 
 class CreateAppMutation(graphene.Mutation):
     class Arguments:
@@ -88,41 +84,6 @@ def mutate(
         return CreateAppMutation(app=app)
 
 
-class RotateAppKeysMutation(graphene.Mutation):
-    class Arguments:
-        id = graphene.ID(required=True)
-        app_token = graphene.String(required=True)
-        wrapped_key_share = graphene.String(required=True)
-
-    app = graphene.Field(AppType)
-
-    @classmethod
-    def mutate(cls, root, info, id, app_token, wrapped_key_share):
-        user = info.context.user
-        app = App.objects.get(id=id)
-
-        if not user_can_access_app(user.userId, app.id):
-            raise GraphQLError("You don't have access to this app")
-
-        if CLOUD_HOSTED:
-            # delete current keys from cloudflare KV
-            deleted = delete(app.app_token)
-
-            # purge keys from cloudflare cache
-            purged = purge(
-                f"phApp:v{app.app_version}:{app.identity_key}/{app.app_token}"
-            )
-
-            if not deleted or not purged:
-                raise GraphQLError("Failed to delete app keys. Please try again.")
-
-        app.app_token = app_token
-        app.wrapped_key_share = wrapped_key_share
-        app.save()
-
-        return RotateAppKeysMutation(app=app)
-
-
 class UpdateAppNameMutation(graphene.Mutation):
     class Arguments:
         id = graphene.ID(required=True)
@@ -176,18 +137,6 @@ def mutate(cls, root, info, id):
         ):
             raise GraphQLError("You don't have permission to delete Apps")
 
-        if CLOUD_HOSTED:
-            # delete current keys from cloudflare KV
-            deleted = delete(app.app_token)
-
-            # purge keys from cloudflare cache
-            purged = purge(
-                f"phApp:v{app.app_version}:{app.identity_key}/{app.app_token}"
-            )
-
-            if not deleted or not purged:
-                raise GraphQLError("Failed to delete app keys. Please try again.")
-
         app.wrapped_key_share = ""
         app.save()
         app.delete()

backend/backend/graphene/types.py

@@ -8,7 +8,6 @@
 from ee.integrations.secrets.dynamic.graphene.types import DynamicSecretType
 from backend.quotas import PLAN_CONFIG
 import graphene
-from enum import Enum
 from graphene import ObjectType, relay, NonNull
 from graphene_django import DjangoObjectType
 from api.models import (
@@ -40,7 +39,6 @@
     UserToken,
     Identity,
 )
-from logs.dynamodb_models import KMSLog
 from django.utils import timezone
 from api.utils.access.roles import default_roles
 from graphql import GraphQLError
@@ -906,61 +904,6 @@ class Meta:
         fields = ("id", "name", "color")
 
 
-class KMSLogType(ObjectType):
-    class Meta:
-        model = KMSLog
-        fields = (
-            "id",
-            "app_id",
-            "timestamp",
-            "phase_node",
-            "event_type",
-            "ip_address",
-            "ph_size",
-            "edge_location",
-            "country",
-            "city",
-            "latitude",
-            "longitude",
-        )
-        interfaces = (relay.Node,)
-
-    id = graphene.ID(required=True)
-    timestamp = graphene.BigInt()
-    app_id = graphene.String()
-    phase_node = graphene.String()
-    event_type = graphene.String()
-    ip_address = graphene.String()
-    ph_size = graphene.Int()
-    asn = graphene.Int()
-    isp = graphene.String()
-    edge_location = graphene.String()
-    country = graphene.String()
-    city = graphene.String()
-    latitude = graphene.Float()
-    longitude = graphene.Float()
-
-
-class ChartDataPointType(graphene.ObjectType):
-    index = graphene.Int()
-    date = graphene.BigInt()
-    data = graphene.Int()
-
-
-class TimeRange(Enum):
-    HOUR = "hour"
-    DAY = "day"
-    WEEK = "week"
-    MONTH = "month"
-    YEAR = "year"
-    ALL_TIME = "allTime"
-
-
-class KMSLogsResponseType(ObjectType):
-    logs = graphene.List(KMSLogType)
-    count = graphene.Int()
-
-
 class SecretLogsResponseType(ObjectType):
     logs = graphene.List(SecretEventType)
     count = graphene.Int()