quivr/backend/models/databases/supabase/api_key_handler.py

from datetime import datetime
from uuid import UUID

from models.databases.repository import Repository


class ApiKeyHandler(Repository):
    def __init__(self, supabase_client):
        self.db = supabase_client  # type: ignore

    def create_api_key(self, new_key_id, new_api_key, user_id):
        response = (
            self.db.table("api_keys")
            .insert(
                [
                    {
                        "key_id": str(new_key_id),
                        "user_id": str(user_id),
                        "api_key": str(new_api_key),
                        "creation_time": datetime.utcnow().strftime(
                            "%Y-%m-%d %H:%M:%S"
                        ),
                        "is_active": True,
                    }
                ]
            )
            .execute()
        )
        return response

    def delete_api_key(self, key_id: str, user_id: UUID):
        return (
            self.db.table("api_keys")
            .update(
                {
                    "is_active": False,
                    "deleted_time": datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S"),
                }
            )
            .match({"key_id": key_id, "user_id": user_id})
            .execute()
        )

    def get_active_api_key(self, api_key: str):
        response = (
            self.db.table("api_keys")
            .select("api_key", "creation_time")
            .filter("api_key", "eq", api_key)
            .filter("is_active", "eq", True)
            .execute()
        )
        return response

    def get_user_id_by_api_key(self, api_key: str):
        response = (
            self.db.table("api_keys")
            .select("user_id")
            .filter("api_key", "eq", api_key)
            .execute()
        )
        return response

    def get_user_api_keys(self, user_id: UUID):
        response = (
            self.db.table("api_keys")
            .select("key_id, creation_time")
            .filter("user_id", "eq", user_id)
            .filter("is_active", "eq", True)
            .execute()
        )
        return response.data