from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.future import select
from sqlalchemy import update, delete, func
from typing import List, Optional

from app.models.clothing import Clothing, ClothingCategory
from app.schemas.clothing import ClothingCreate, ClothingUpdate
from app.schemas.clothing import ClothingCategoryCreate, ClothingCategoryUpdate
from app.models.users import User
from fastapi import Depends

# 衣服分类服务函数
async def get_category(db: AsyncSession, category_id: int):
    """获取单个衣服分类"""
    result = await db.execute(select(ClothingCategory).filter(ClothingCategory.id == category_id))
    return result.scalars().first()

async def get_categories(db: AsyncSession, skip: int = 0, limit: int = 100):
    """获取所有衣服分类"""
    result = await db.execute(
        select(ClothingCategory)
        .offset(skip)
        .limit(limit)
    )
    return result.scalars().all()

async def create_category(db: AsyncSession, category: ClothingCategoryCreate):
    """创建衣服分类"""
    db_category = ClothingCategory(
        name=category.name
    )
    db.add(db_category)
    await db.commit()
    await db.refresh(db_category)
    return db_category

async def update_category(db: AsyncSession, category_id: int, category_update: ClothingCategoryUpdate):
    """更新衣服分类"""
    db_category = await get_category(db, category_id)
    if not db_category:
        return None
        
    if category_update.name:
        db_category.name = category_update.name
        
    await db.commit()
    await db.refresh(db_category)
    return db_category

async def delete_category(db: AsyncSession, category_id: int):
    """删除衣服分类（会级联删除相关衣服）"""
    db_category = await get_category(db, category_id)
    if db_category:
        await db.delete(db_category)
        await db.commit()
    return db_category

# 衣服服务函数
async def get_clothing(db: AsyncSession, clothing_id: int):
    """获取单个衣服"""
    result = await db.execute(select(Clothing).filter(Clothing.id == clothing_id))
    return result.scalars().first()

async def get_clothes(db: AsyncSession, skip: int = 0, limit: int = 100, user_id: int = None):
    """获取所有衣服"""
    result = await db.execute(
        select(Clothing)
        .filter(Clothing.user_id == user_id)
        .order_by(Clothing.create_time.desc())
        .offset(skip)
        .limit(limit)
    )
    return result.scalars().all()

async def get_clothes_by_category(db: AsyncSession, category_id: int, skip: int = 0, limit: int = 100):
    """根据分类获取衣服"""
    query = select(Clothing).order_by(Clothing.create_time.desc())
    if category_id > 0:
        query = query.filter(Clothing.clothing_category_id == category_id)
    result = await db.execute(
        query
        .offset(skip)
        .limit(limit)
    )
    return result.scalars().all()

async def create_clothing(db: AsyncSession, clothing: ClothingCreate):
    """创建衣服"""
    db_clothing = Clothing(
        clothing_category_id=clothing.clothing_category_id,
        image_url=clothing.image_url
    )
    db.add(db_clothing)
    await db.commit()
    await db.refresh(db_clothing)
    return db_clothing

async def update_clothing(db: AsyncSession, clothing_id: int, clothing_update: ClothingUpdate):
    """更新衣服"""
    db_clothing = await get_clothing(db, clothing_id)
    if not db_clothing:
        return None
        
    if clothing_update.clothing_category_id is not None:
        db_clothing.clothing_category_id = clothing_update.clothing_category_id
        
    if clothing_update.image_url:
        db_clothing.image_url = clothing_update.image_url
        
    await db.commit()
    await db.refresh(db_clothing)
    return db_clothing

async def delete_clothing(db: AsyncSession, clothing_id: int):
    """删除衣服"""
    db_clothing = await get_clothing(db, clothing_id)
    if db_clothing:
        await db.delete(db_clothing)
        await db.commit()
    return db_clothing