feat: better configuration

2026-01-31 22:37:58 +01:00
parent 077e08b887
commit f251df302d
21 changed files with 2972 additions and 2906 deletions
@@ -0,0 +1,130 @@
+package config
+
+import (
+	"os"
+
+	"github.com/leonardotrapani/hyprvoice/internal/injection"
+	"github.com/leonardotrapani/hyprvoice/internal/recording"
+	"github.com/leonardotrapani/hyprvoice/internal/transcriber"
+)
+
+func (c *Config) ToRecordingConfig() recording.Config {
+	return recording.Config{
+		SampleRate:        c.Recording.SampleRate,
+		Channels:          c.Recording.Channels,
+		Format:            c.Recording.Format,
+		BufferSize:        c.Recording.BufferSize,
+		Device:            c.Recording.Device,
+		ChannelBufferSize: c.Recording.ChannelBufferSize,
+		Timeout:           c.Recording.Timeout,
+	}
+}
+
+func (c *Config) ToTranscriberConfig() transcriber.Config {
+	config := transcriber.Config{
+		Provider: c.Transcription.Provider,
+		Language: c.Transcription.Language,
+		Model:    c.Transcription.Model,
+		Keywords: c.Keywords,
+	}
+
+	config.APIKey = c.resolveAPIKeyForProvider(c.Transcription.Provider)
+
+	return config
+}
+
+// resolveAPIKeyForProvider returns the API key for a provider from multiple sources
+func (c *Config) resolveAPIKeyForProvider(provider string) string {
+	providerName := provider
+	envVar := ""
+	switch provider {
+	case "openai":
+		providerName = "openai"
+		envVar = "OPENAI_API_KEY"
+	case "groq-transcription", "groq-translation":
+		providerName = "groq"
+		envVar = "GROQ_API_KEY"
+	case "mistral-transcription":
+		providerName = "mistral"
+		envVar = "MISTRAL_API_KEY"
+	case "elevenlabs":
+		providerName = "elevenlabs"
+		envVar = "ELEVENLABS_API_KEY"
+	}
+
+	if c.Providers != nil {
+		if pc, ok := c.Providers[providerName]; ok && pc.APIKey != "" {
+			return pc.APIKey
+		}
+	}
+
+	if c.Transcription.APIKey != "" {
+		return c.Transcription.APIKey
+	}
+
+	if envVar != "" {
+		return os.Getenv(envVar)
+	}
+
+	return ""
+}
+
+// ToLLMConfig returns the LLM adapter configuration
+func (c *Config) ToLLMConfig() LLMAdapterConfig {
+	config := LLMAdapterConfig{
+		Provider:          c.LLM.Provider,
+		Model:             c.LLM.Model,
+		RemoveStutters:    c.LLM.PostProcessing.RemoveStutters,
+		AddPunctuation:    c.LLM.PostProcessing.AddPunctuation,
+		FixGrammar:        c.LLM.PostProcessing.FixGrammar,
+		RemoveFillerWords: c.LLM.PostProcessing.RemoveFillerWords,
+		Keywords:          c.Keywords,
+	}
+
+	if c.LLM.Provider != "" {
+		config.APIKey = c.resolveAPIKeyForLLMProvider(c.LLM.Provider)
+	}
+
+	if c.LLM.CustomPrompt.Enabled && c.LLM.CustomPrompt.Prompt != "" {
+		config.CustomPrompt = c.LLM.CustomPrompt.Prompt
+	}
+
+	return config
+}
+
+// resolveAPIKeyForLLMProvider returns the API key for an LLM provider
+func (c *Config) resolveAPIKeyForLLMProvider(provider string) string {
+	envVar := ""
+	switch provider {
+	case "openai":
+		envVar = "OPENAI_API_KEY"
+	case "groq":
+		envVar = "GROQ_API_KEY"
+	}
+
+	if c.Providers != nil {
+		if pc, ok := c.Providers[provider]; ok && pc.APIKey != "" {
+			return pc.APIKey
+		}
+	}
+
+	if envVar != "" {
+		return os.Getenv(envVar)
+	}
+
+	return ""
+}
+
+// IsLLMEnabled returns true if LLM post-processing is enabled and configured
+func (c *Config) IsLLMEnabled() bool {
+	return c.LLM.Enabled && c.LLM.Provider != "" && c.LLM.Model != ""
+}
+
+func (c *Config) ToInjectionConfig() injection.Config {
+	return injection.Config{
+		Backends:         c.Injection.Backends,
+		YdotoolTimeout:   c.Injection.YdotoolTimeout,
+		WtypeTimeout:     c.Injection.WtypeTimeout,
+		ClipboardTimeout: c.Injection.ClipboardTimeout,
+	}
+}