from sqlalchemy import select
from sqlalchemy.orm import Session, selectinload
from ..models import Campaign, CtaBanner, DemandFeatureCard, DemandForm, DemandHero, Destination, DestinationHero, DestinationRegion, HeroSlide, HotelGroup, MapImage, ThemeCard, VehicleOption
from ..demand_recommendations import demand_recommendation_dict, demand_recommendation_query
from ..route_sections import route_section_dict, route_section_query
from ..serializers import destination_dict, hotel_group_dict, model_dict


def site_config(db: Session, active_only: bool, include_public_extras: bool = True) -> dict:
    hero_stmt = select(HeroSlide).order_by(HeroSlide.sortOrder.asc())
    destination_stmt = select(Destination).options(selectinload(Destination.aliases)).order_by(Destination.sortOrder.asc())
    destination_hero_stmt = select(DestinationHero).order_by(DestinationHero.sortOrder.asc())
    destination_region_stmt = select(DestinationRegion).order_by(DestinationRegion.sortOrder.asc())
    map_stmt = select(MapImage).order_by(MapImage.createdAt.asc())
    theme_stmt = select(ThemeCard).order_by(ThemeCard.sortOrder.asc())
    cta_stmt = select(CtaBanner).order_by(CtaBanner.sortOrder.asc())
    hotel_stmt = select(HotelGroup).order_by(HotelGroup.sortOrder.asc())
    vehicle_stmt = select(VehicleOption).order_by(VehicleOption.sortOrder.asc())
    demand_hero_stmt = select(DemandHero).order_by(DemandHero.sortOrder.asc())
    demand_feature_card_stmt = select(DemandFeatureCard).order_by(DemandFeatureCard.sortOrder.asc())
    demand_form_stmt = select(DemandForm).order_by(DemandForm.createdAt.asc())
    if active_only:
        hero_stmt = hero_stmt.where(HeroSlide.isActive.is_(True))
        destination_stmt = destination_stmt.where(Destination.isActive.is_(True))
        destination_hero_stmt = destination_hero_stmt.where(DestinationHero.isActive.is_(True))
        destination_region_stmt = destination_region_stmt.where(DestinationRegion.isActive.is_(True))
        map_stmt = map_stmt.where(MapImage.isActive.is_(True))
        theme_stmt = theme_stmt.where(ThemeCard.isActive.is_(True))
        cta_stmt = cta_stmt.where(CtaBanner.isActive.is_(True))
        hotel_stmt = hotel_stmt.where(HotelGroup.isActive.is_(True), HotelGroup.status == "published")
        vehicle_stmt = vehicle_stmt.where(VehicleOption.isActive.is_(True))
        demand_hero_stmt = demand_hero_stmt.where(DemandHero.isActive.is_(True))
        demand_feature_card_stmt = demand_feature_card_stmt.where(DemandFeatureCard.isActive.is_(True))
        demand_form_stmt = demand_form_stmt.where(DemandForm.isActive.is_(True))

    hero_slides = db.scalars(hero_stmt).all()
    destinations = db.scalars(destination_stmt).all()
    destination_hero = db.scalars(destination_hero_stmt).all()
    destination_regions = db.scalars(destination_region_stmt).all()
    map_images = db.scalars(map_stmt).all()
    themes = db.scalars(theme_stmt).all()
    cta_banners = db.scalars(cta_stmt).all()
    result = {
        "heroSlides": [model_dict(item) for item in hero_slides],
        "destinations": [destination_dict(item) for item in destinations],
        "destinationHero": [model_dict(item) for item in destination_hero],
        "destinationRegions": [model_dict(item) for item in destination_regions],
        "map": [model_dict(item) for item in map_images],
        "themes": [model_dict(item) for item in themes],
        "ctaBanners": [model_dict(item) for item in cta_banners],
    }
    if include_public_extras:
        campaigns = db.scalars(select(Campaign).where(Campaign.status == "published").order_by(Campaign.updatedAt.desc())).all()
        route_sections = db.scalars(route_section_query()).all()
        if active_only:
            route_sections = [section for section in route_sections if section.isActive]
        result["campaigns"] = [model_dict(item) for item in campaigns]
        result["routeSections"] = [route_section_dict(item, public=True) for item in route_sections]
        result["hotelGroups"] = [hotel_group_dict(item) for item in db.scalars(hotel_stmt).all()]
        result["vehicleOptions"] = [model_dict(item) for item in db.scalars(vehicle_stmt).all()]
    result["demandHero"] = [model_dict(item) for item in db.scalars(demand_hero_stmt).all()]
    result["demandFeatureCards"] = [model_dict(item) for item in db.scalars(demand_feature_card_stmt).all()]
    result["demandForm"] = [model_dict(item) for item in db.scalars(demand_form_stmt).all()]
    demand_recommendations = db.scalars(demand_recommendation_query()).all()
    if active_only:
        demand_recommendations = [item for item in demand_recommendations if item.isActive]
    result["demandRecommendations"] = [demand_recommendation_dict(item, public=active_only) for item in demand_recommendations]
    return result