package service

import (
	"github.com/QuantumNous/new-api/constant"
	"github.com/QuantumNous/new-api/model"
)

const KlingAssetChannelType = constant.ChannelTypeKlingAiping

func ResolveKlingAssetChannelForModel(userId int, tokenGroup string, modelName string) (*model.Channel, error) {
	return ResolveVideoAssetChannelForModel(userId, tokenGroup, modelName, VideoAssetFamilyKling)
}

func ResolveKlingAssetChannel(userId int, tokenGroup string) (*model.Channel, error) {
	return ResolveKlingAssetChannelForModel(userId, tokenGroup, "")
}

func GetBoundKlingAssetChannelForModel(userId int, tokenGroup string, modelName string) (*model.Channel, error) {
	return GetBoundVideoAssetChannelForModel(userId, tokenGroup, modelName, VideoAssetFamilyKling)
}

func GetBoundKlingAssetChannel(userId int, tokenGroup string) (*model.Channel, error) {
	return GetBoundKlingAssetChannelForModel(userId, tokenGroup, "")
}

func AutoMatchKlingAssetChannel(tokenGroup string) (*model.Channel, error) {
	return autoMatchVideoAssetChannelForModel(tokenGroup, "", VideoAssetFamilyKling)
}

func IsUsableKlingAssetChannel(channel *model.Channel, tokenGroup string) bool {
	return IsUsableVideoAssetChannelForFamily(channel, tokenGroup, "", VideoAssetFamilyKling)
}