import { root } from '@repo/core'
import { TemplateSocialController } from '@repo/sdk'
import { useCallback, useState } from 'react'

import { type ApiError } from '@/lib/types'
import { handleError } from './use-error'
import { templateSocialStore } from '@/stores/templateSocialStore'

export const useTemplateLike = (templateId?: string) => {
    const [loading, setLoading] = useState<boolean>(false)
    const [error, setError] = useState<ApiError | null>(null)

    const like = useCallback(async (): Promise<{ error?: ApiError }> => {
        if (!templateId) {
            return { error: { message: 'TemplateId is required' } as ApiError }
        }

        setLoading(true)
        setError(null)

        // 乐观更新：立即更新状态
        const currentCount = templateSocialStore.getLikeCount(templateId) ?? 0
        templateSocialStore.setLiked(templateId, true)
        templateSocialStore.setLikeCount(templateId, currentCount + 1)

        const social = root.get(TemplateSocialController)
        const { data, error } = await handleError(
            async () => await social.like({ templateId })
        )

        setLoading(false)

        if (error) {
            // 回滚乐观更新
            templateSocialStore.setLiked(templateId, false)
            templateSocialStore.setLikeCount(templateId, currentCount)
            setError(error)
            return { error }
        }

        // 使用服务器返回的准确数量
        if (data?.likeCount !== undefined) {
            templateSocialStore.setLikeCount(templateId, data.likeCount)
        }

        return {}
    }, [templateId])

    const unlike = useCallback(async (): Promise<{ error?: ApiError }> => {
        if (!templateId) {
            return { error: { message: 'TemplateId is required' } as ApiError }
        }

        setLoading(true)
        setError(null)

        // 乐观更新：立即更新状态
        const currentCount = templateSocialStore.getLikeCount(templateId) ?? 0
        templateSocialStore.setLiked(templateId, false)
        templateSocialStore.setLikeCount(templateId, Math.max(0, currentCount - 1))

        const social = root.get(TemplateSocialController)
        const { data, error } = await handleError(
            async () => await social.unlike({ templateId })
        )

        setLoading(false)

        if (error) {
            // 回滚乐观更新
            templateSocialStore.setLiked(templateId, true)
            templateSocialStore.setLikeCount(templateId, currentCount)
            setError(error)
            return { error }
        }

        // 使用服务器返回的准确数量
        if (data?.likeCount !== undefined) {
            templateSocialStore.setLikeCount(templateId, data.likeCount)
        }

        return {}
    }, [templateId])

    const checkLiked = useCallback(async (): Promise<{ error?: ApiError }> => {
        if (!templateId) {
            return { error: { message: 'TemplateId is required' } as ApiError }
        }

        setLoading(true)
        setError(null)

        const social = root.get(TemplateSocialController)
        const { data, error } = await handleError(
            async () => await social.checkLiked({ templateId })
        )

        setLoading(false)

        if (error) {
            setError(error)
            return { error }
        }

        if (data?.liked !== undefined) {
            templateSocialStore.setLiked(templateId, data.liked)
        }
        return {}
    }, [templateId])

    return {
        loading,
        error,
        like,
        unlike,
        checkLiked,
    }
}