from app.db import database
from pymongo import ReturnDocument
from enum import Enum

def get_next_sequence_value(sequence_name: str, db: database.MongoDB) -> int:
    ret = db["counters"].find_one_and_update(
        {"_id": sequence_name},
        {"$inc": {"sequence_value": 1}},
        upsert=True,
        return_document=ReturnDocument.AFTER
    )
    # If the sequence is less than 100001, set it to 100001.
    if ret["sequence_value"] < 100001:
        db["counters"].update_one(
            {"_id": sequence_name},
            {"$set": {"sequence_value": 100001}}
        )
        return 100001
    return ret["sequence_value"]

def convert_enums(obj):
    # Recursively convert enum values in dict or list objects
    if isinstance(obj, dict):
        return {k: convert_enums(v) for k, v in obj.items()}
    elif isinstance(obj, list):
        return [convert_enums(v) for v in obj]
    elif isinstance(obj, Enum):
        return obj.value
    else:
        return obj