saved_search.py

"""Saved Search API."""


# Standard Python Libraries
from datetime import datetime, timezone
import uuid

# Third-Party Libraries
from django.http import JsonResponse
from fastapi import HTTPException

from ..models import SavedSearch, User


def validate_name(value: str):
    """Validate name."""
    name = value.strip()
    if name == "":
        raise HTTPException(status_code=400, detail="Name cannot be empty")

    all_saved_searches = SavedSearch.objects.all()
    for search in all_saved_searches:
        if search.name.strip() == name:
            raise HTTPException(status_code=400, detail="Name already exists")


def create_saved_search(request):
    """Create saved search."""
    validate_name(request.get("name"))
    try:
        # Process filter values when selecting organizations
        def process_filter_values(values):
            processed_values = []
            for value in values:
                if isinstance(value, dict):
                    # Include only the required fields
                    processed_values.append(
                        {
                            "id": value.get("id"),
                            "name": value.get("name"),
                            "regionId": value.get("regionId"),
                            "rootDomains": value.get("rootDomains", []),
                        }
                    )
                else:
                    processed_values.append(value)
            return processed_values

        filters = [
            {
                "type": f.type,
                "field": f.field,
                "values": process_filter_values(f.values),
            }
            for f in request.get("filters", [])
        ]

        search = SavedSearch.objects.create(
            name=request.get("name"),
            count=request.get("count", 0),  # Default to 0 if count does not exist
            sortDirection=request.get("sortDirection", ""),
            sortField=request.get("sortField", ""),
            searchTerm=request.get("searchTerm", ""),
            searchPath=request.get("searchPath", ""),
            filters=filters,
            createdById=request.get("createdById"),
        )

        response = {
            "id": str(search.id),
            "createdAt": search.createdAt,
            "updatedAt": search.updatedAt,
            "name": search.name,
            "searchTerm": search.searchTerm,
            "sortDirection": search.sortDirection,
            "sortField": search.sortField,
            "count": search.count,
            "filters": search.filters,
            "searchPath": search.searchPath,
            "createdById": search.createdById.id,
        }

        search.save()
        return response

    except User.DoesNotExist:
        raise HTTPException(status_code=404, detail="User not found")

    except Exception as e:
        raise HTTPException(status_code=404, detail=str(e))


def list_saved_searches(user):
    """List all saved searches."""
    try:
        all_saved_searches = SavedSearch.objects.all()
        saved_search_list = []
        for search in all_saved_searches:
            if search.createdById != user:
                continue
            response = {
                "id": str(search.id),
                "createdAt": search.createdAt,
                "updatedAt": search.updatedAt,
                "name": search.name,
                "searchTerm": search.searchTerm,
                "sortDirection": search.sortDirection,
                "sortField": search.sortField,
                "count": search.count,
                "filters": search.filters,
                "searchPath": search.searchPath,
                "createdById": search.createdById.id,
            }
            saved_search_list.append(response)
        return {
            "result": list(saved_search_list),
            "count": len(list(saved_search_list)),
        }
    except User.DoesNotExist:
        raise HTTPException(status_code=404, detail="User not found")

    except Exception as e:
        raise HTTPException(status_code=404, detail=str(e))


def get_saved_search(saved_search_id, user):
    """Get saved search."""
    if user.userType == "globalView":
        raise HTTPException(
            status_code=404, detail="Global View users cannot retrieve saved searches."
        )
    if not uuid.UUID(saved_search_id):
        raise HTTPException({"error": "Invalid UUID"})

    try:
        saved_search = SavedSearch.objects.get(id=saved_search_id)

        if saved_search.createdById.id != user.id:
            raise HTTPException(status_code=404, detail="Saved search not found")

        response = {
            "id": str(saved_search.id),
            "createdAt": saved_search.createdAt,
            "updatedAt": saved_search.updatedAt,
            "name": saved_search.name,
            "searchTerm": saved_search.searchTerm,
            "sortDirection": saved_search.sortDirection,
            "sortField": saved_search.sortField,
            "count": saved_search.count,
            "filters": saved_search.filters,
            "searchPath": saved_search.searchPath,
            "createdById": saved_search.createdById.id,
        }
        return response
    except SavedSearch.DoesNotExist as dne:
        raise HTTPException(status_code=404, detail=str(dne))
    except User.DoesNotExist:
        raise HTTPException(status_code=404, detail="User not found")
    except Exception as e:
        raise HTTPException(status_code=404, detail=str(e))


def update_saved_search(request, user):
    """Update saved search."""
    if not uuid.UUID(request["saved_search_id"]):
        raise HTTPException(status_code=404, detail={"error": "Invalid UUID"})
    try:
        # Process filter values when selecting organizations
        def process_filter_values(values):
            processed_values = []
            for value in values:
                if isinstance(value, dict):
                    # Include only the required fields
                    processed_values.append(
                        {
                            "id": value.get("id"),
                            "name": value.get("name"),
                            "regionId": value.get("regionId"),
                            "rootDomains": value.get("rootDomains", []),
                        }
                    )
                else:
                    processed_values.append(value)
            return processed_values

        filters = [
            {
                "type": f.type,
                "field": f.field,
                "values": process_filter_values(f.values),
            }
            for f in request.get("filters", [])
        ]

        saved_search = SavedSearch.objects.get(id=request["saved_search_id"])
        if saved_search.createdById.id != user.id:
            raise HTTPException(status_code=404, detail="Saved search not found")

        name = request["name"].strip()
        if name == "":
            raise HTTPException(status_code=400, detail="Name cannot be empty")

        saved_search.name = request["name"]
        saved_search.updatedAt = datetime.now(timezone.utc)
        saved_search.searchTerm = request["searchTerm"]
        saved_search.save()
        response = {
            "name": saved_search.name,
            "searchTerm": saved_search.searchTerm,
            "sortDirection": saved_search.sortDirection,
            "sortField": saved_search.sortField,
            "count": saved_search.count,
            "filters": filters,
            "searchPath": saved_search.searchPath,
        }
    except User.DoesNotExist:
        raise HTTPException(status_code=404, detail="User not found")
    except SavedSearch.DoesNotExist as dne:
        raise HTTPException(status_code=404, detail=str(dne))
    except Exception as e:
        raise HTTPException(status_code=404, detail=str(e))

    return response


def delete_saved_search(saved_search_id, user):
    """Delete saved search by id."""
    if not uuid.UUID(saved_search_id):
        raise HTTPException(status_code=404, detail={"error": "Invalid UUID"})
    try:
        search = SavedSearch.objects.get(id=saved_search_id)
        if search.createdById.id != user.id:
            raise HTTPException(status_code=404, detail="Saved search not found")
        search.delete()
        return JsonResponse(
            {
                "status": "success",
                "message": "Saved search id:{} deleted.".format(saved_search_id),
            }
        )
    except User.DoesNotExist:
        raise HTTPException(status_code=404, detail="User not found")
    except SavedSearch.DoesNotExist as dne:
        raise HTTPException(status_code=404, detail=str(dne))
    except Exception as e:
        raise HTTPException(status_code=404, detail=str(e))