summaryrefslogtreecommitdiff
path: root/internal/tts/registry.go
blob: 0b38a5edb19be79bc96e099e24cf2043b3ca5df6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package tts

import (
	"fmt"
	"reflect"

	"codeberg.org/snonux/comicforge/internal/provider"
)

// Config exposes the configured TTS provider settings.
type Config interface {
	provider.TTSConfig
	GoogleAPIKey() string
	TTSModel() string
}

type registryConfig struct {
	Config
	voice string
}

// Registry resolves provider names to factories.
type Registry struct {
	*provider.Registry[provider.TTSProvider, registryConfig]
}

// NewRegistry creates an empty TTS-provider registry.
func NewRegistry() *Registry {
	return &Registry{
		Registry: provider.NewRegistry[provider.TTSProvider, registryConfig](),
	}
}

// DefaultRegistry returns the built-in TTS provider registry.
func DefaultRegistry() *Registry {
	registry := NewRegistry()
	registry.Register(provider.Gemini, func(cfg registryConfig) (provider.TTSProvider, error) {
		geminiProvider := NewGeminiProvider(&GeminiConfig{
			APIKey: cfg.GoogleAPIKey(),
			Model:  cfg.TTSModel(),
			Voice:  cfg.voice,
		})
		if err := geminiProvider.IsAvailable(); err != nil {
			return nil, err
		}
		return geminiProvider, nil
	})
	registry.Register(provider.OpenAI, func(registryConfig) (provider.TTSProvider, error) {
		return nil, fmt.Errorf("tts provider %q is not implemented", provider.OpenAI)
	})
	return registry
}

// Has reports whether a factory is registered for name.
func (r *Registry) Has(name string) bool {
	if r == nil || r.Registry == nil {
		return false
	}
	_, ok := r.Resolve(name)
	return ok
}

// NewFromConfig resolves the provider name from cfg and constructs it.
func (r *Registry) NewFromConfig(cfg Config, voice string) (provider.TTSProvider, error) {
	if r == nil || r.Registry == nil {
		return nil, fmt.Errorf("tts registry is required")
	}
	if isNilValue(cfg) {
		return nil, fmt.Errorf("tts config is required")
	}
	return r.New(cfg.TTSProviderName(), registryConfig{
		Config: cfg,
		voice:  voice,
	})
}

func isNilValue[T any](value T) bool {
	v := reflect.ValueOf(value)
	if !v.IsValid() {
		return true
	}

	switch v.Kind() {
	case reflect.Chan, reflect.Func, reflect.Interface, reflect.Map, reflect.Pointer, reflect.Slice:
		return v.IsNil()
	default:
		return false
	}
}