feat: add domainMapping in config, to allow for localModels (#5)

* feature: add domainMapping in config, to allow for localModels

* feat: so I can install my fork

* feature: add links to original repo

* feat: update config fns

* fix: tests

Author: mihaisavezi

Makefile

@@ -102,4 +102,4 @@ help:
 	@echo "Usage: make [target]"
 	@echo ""
 	@echo "Targets:"
-	@awk '/^##/{c=substr($$0,3);next}c&&/^[[:alpha:]][[:alnum:]_-]+:/{print substr($$1,1,index($$1,":")-1)":"c}1{c=""}' $(MAKEFILE_LIST) | column -t -s ":"
+	@awk '/^##/{c=substr($$0,3);next}c&&/^[[:alpha:]][[:alnum:]_-]+:/{print substr($$1,1,index($$1,":")-1)":"c}1{c=""}' $(MAKEFILE_LIST) | column -t -s ":"

README.md

@@ -374,6 +374,29 @@ router:
   web_search: openrouter,perplexity/llama-3.1-sonar-huge-128k-online
 ```
 
+### 🗺️ Domain Mappings
+
+Map custom domains (like localhost) to existing providers for local model support:
+
+```yaml
+# config.yaml
+domain_mappings:
+  localhost: openai # Use OpenAI transformations for localhost requests
+  127.0.0.1: gemini # Use Gemini transformations for 127.0.0.1 requests
+  custom.api: openrouter # Use OpenRouter transformations for custom APIs
+
+providers:
+  name: local-lmstudio
+  url: "http://localhost:1234/v1/chat/completions"
+  api_key: "not-needed"
+```
+
+**Benefits:**
+- ✅ **Local Model Support** - Route localhost to existing providers
+- ✅ **Reuse Transformations** - Leverage proven request/response logic
+- ✅ **No Custom Provider Needed** - Use existing provider implementations
+- ✅ **Flexible Mapping** - Any domain can map to any provider
+
 ### 📜 Legacy JSON Format
 
 <details>

Taskfile.yml

@@ -229,4 +229,4 @@ tasks:
     desc: Run comprehensive checks
     deps: [fmt, lint, test, security]
     cmds:
-      - echo 'All checks passed!'
+      - echo 'All checks passed!'

cmd/config.go

@@ -139,124 +139,167 @@ func runConfigInit(cmd *cobra.Command, _ []string) error {
 }
 
 func runConfigShow(cmd *cobra.Command, _ []string) error {
-	if !cfgMgr.Exists() {
-		color.Yellow("No configuration found. Run 'cco config init' or 'cco config generate' to create one.")
-		return nil
-	}
-
-	cfg, err := cfgMgr.Load()
-	if err != nil {
-		return fmt.Errorf("failed to load configuration: %w", err)
-	}
-
-	color.Blue("Current Configuration:")
-	fmt.Printf("  %-15s: %s\n", "Host", cfg.Host)
-	fmt.Printf("  %-15s: %d\n", "Port", cfg.Port)
-	fmt.Printf("  %-15s: %s\n", "API Key", maskString(cfg.APIKey))
-	fmt.Printf("  %-15s: %s\n", "Config Path", cfgMgr.GetPath())
-
-	// Show config file type
-	configType := "JSON"
-	if cfgMgr.HasYAML() {
-		configType = "YAML"
-	}
-
-	fmt.Printf("  %-15s: %s\n", "Format", configType)
-
-	fmt.Println("\nProviders:")
-
-	for _, provider := range cfg.Providers {
-		fmt.Printf("  - Name: %s\n", provider.Name)
-		fmt.Printf("    URL: %s\n", provider.APIBase)
-		fmt.Printf("    API Key: %s\n", maskString(provider.APIKey))
-
-		if len(provider.DefaultModels) > 0 {
-			fmt.Printf("    Default Models: %v\n", provider.DefaultModels)
-		}
-
-		if len(provider.ModelWhitelist) > 0 {
-			fmt.Printf("    Model Whitelist: %v\n", provider.ModelWhitelist)
-		}
-
-		if len(provider.Models) > 0 {
-			fmt.Printf("    Models: %v\n", provider.Models)
-		}
-
-		fmt.Println()
-	}
-
-	fmt.Println("Router Configuration:")
-	fmt.Printf("  %-15s: %s\n", "Default", cfg.Router.Default)
-
-	if cfg.Router.Think != "" {
-		fmt.Printf("  %-15s: %s\n", "Think", cfg.Router.Think)
-	}
-
-	if cfg.Router.Background != "" {
-		fmt.Printf("  %-15s: %s\n", "Background", cfg.Router.Background)
-	}
-
-	if cfg.Router.LongContext != "" {
-		fmt.Printf("  %-15s: %s\n", "Long Context", cfg.Router.LongContext)
-	}
-
-	if cfg.Router.WebSearch != "" {
-		fmt.Printf("  %-15s: %s\n", "Web Search", cfg.Router.WebSearch)
-	}
-
-	return nil
+    if !cfgMgr.Exists() {
+        color.Yellow("No configuration found. Run 'cco config init' or 'cco config generate' to create one.")
+        return nil
+    }
+
+    cfg, err := cfgMgr.Load()
+    if err != nil {
+        return fmt.Errorf("failed to load configuration: %w", err)
+    }
+
+    color.Blue("Current Configuration:")
+    fmt.Printf("  %-15s: %s\n", "Host", cfg.Host)
+    fmt.Printf("  %-15s: %d\n", "Port", cfg.Port)
+    fmt.Printf("  %-15s: %s\n", "API Key", maskString(cfg.APIKey))
+    fmt.Printf("  %-15s: %s\n", "Config Path", cfgMgr.GetPath())
+
+    // Show config file type
+    configType := "JSON"
+    if cfgMgr.HasYAML() {
+        configType = "YAML"
+    }
+    fmt.Printf("  %-15s: %s\n", "Format", configType)
+
+    // Show domain mappings if present
+    if len(cfg.DomainMappings) > 0 {
+        fmt.Println("\nDomain Mappings:")
+        for domain, provider := range cfg.DomainMappings {
+            fmt.Printf("  %-15s → %s\n", domain, provider)
+        }
+    }
+
+    fmt.Println("\nProviders:")
+    for _, provider := range cfg.Providers {
+        fmt.Printf("  - Name: %s\n", provider.Name)
+        fmt.Printf("    URL: %s\n", provider.APIBase)
+        fmt.Printf("    API Key: %s\n", maskString(provider.APIKey))
+
+        if len(provider.DefaultModels) > 0 {
+            fmt.Printf("    Default Models: %v\n", provider.DefaultModels)
+        }
+
+        if len(provider.ModelWhitelist) > 0 {
+            fmt.Printf("    Model Whitelist: %v\n", provider.ModelWhitelist)
+        }
+
+        if len(provider.Models) > 0 {
+            fmt.Printf("    Models: %v\n", provider.Models)
+        }
+
+        fmt.Println()
+    }
+
+    fmt.Println("Router Configuration:")
+    fmt.Printf("  %-15s: %s\n", "Default", cfg.Router.Default)
+
+    if cfg.Router.Think != "" {
+        fmt.Printf("  %-15s: %s\n", "Think", cfg.Router.Think)
+    }
+
+    if cfg.Router.Background != "" {
+        fmt.Printf("  %-15s: %s\n", "Background", cfg.Router.Background)
+    }
+
+    if cfg.Router.LongContext != "" {
+        fmt.Printf("  %-15s: %s\n", "Long Context", cfg.Router.LongContext)
+    }
+
+    if cfg.Router.WebSearch != "" {
+        fmt.Printf("  %-15s: %s\n", "Web Search", cfg.Router.WebSearch)
+    }
+
+    return nil
 }
 
-func runConfigValidate(cmd *cobra.Command, _ []string) error {
-	if !cfgMgr.Exists() {
-		return errors.New("no configuration found")
-	}
-
-	cfg, err := cfgMgr.Load()
-	if err != nil {
-		return fmt.Errorf("failed to load configuration: %w", err)
-	}
-
-	// Validation logic
-	var validationErrors []string
-
-	if len(cfg.Providers) == 0 {
-		validationErrors = append(validationErrors, "no providers configured")
-	}
-
-	for i, provider := range cfg.Providers {
-		if provider.Name == "" {
-			validationErrors = append(validationErrors, fmt.Sprintf("provider %d: name is required", i))
-		}
-
-		if provider.APIBase == "" {
-			validationErrors = append(validationErrors, fmt.Sprintf("provider %d: API base URL is required", i))
-		}
-
-		if provider.APIKey == "" {
-			validationErrors = append(validationErrors, fmt.Sprintf("provider %d: API key is required", i))
-		}
-	}
-
-	if cfg.Router.Default == "" {
-		validationErrors = append(validationErrors, "default router model is required")
-	}
-
-	if len(validationErrors) > 0 {
-		color.Red("Configuration validation failed:")
-
-		for _, err := range validationErrors {
-			fmt.Printf("  - %s\n", err)
-		}
 
-		return errors.New("configuration validation failed")
-	}
-
-	color.Green("Configuration is valid!")
-
-	return nil
+func runConfigValidate(cmd *cobra.Command, _ []string) error {
+    if !cfgMgr.Exists() {
+        return errors.New("no configuration found")
+    }
+
+    cfg, err := cfgMgr.Load()
+    if err != nil {
+        return fmt.Errorf("failed to load configuration: %w", err)
+    }
+
+    // Validation logic
+    var validationErrors []string
+
+    if len(cfg.Providers) == 0 {
+        validationErrors = append(validationErrors, "no providers configured")
+    }
+
+    // Validate providers
+    providerNames := make(map[string]bool)
+    for i, provider := range cfg.Providers {
+        if provider.Name == "" {
+            validationErrors = append(validationErrors, fmt.Sprintf("provider %d: name is required", i))
+        } else {
+            providerNames[provider.Name] = true
+        }
+
+        if provider.APIBase == "" {
+            validationErrors = append(validationErrors, fmt.Sprintf("provider %d: API base URL is required", i))
+        }
+
+        if provider.APIKey == "" {
+            validationErrors = append(validationErrors, fmt.Sprintf("provider %d: API key is required", i))
+        }
+    }
+
+    // Validate domain mappings reference valid providers
+    if len(cfg.DomainMappings) > 0 {
+        for domain, providerName := range cfg.DomainMappings {
+            if domain == "" {
+                validationErrors = append(validationErrors, "domain mapping: empty domain not allowed")
+                continue
+            }
+            
+            // Check if referenced provider exists (either in config or built-in)
+            builtInProviders := map[string]bool{
+                "openrouter": true,
+                "openai":     true,
+                "anthropic":  true,
+                "nvidia":     true,
+                "gemini":     true,
+            }
+            
+            if !providerNames[providerName] && !builtInProviders[providerName] {
+                validationErrors = append(validationErrors, 
+                    fmt.Sprintf("domain mapping '%s → %s': provider '%s' not found", 
+                        domain, providerName, providerName))
+            }
+        }
+    }
+
+    if cfg.Router.Default == "" {
+        validationErrors = append(validationErrors, "default router model is required")
+    }
+
+    if len(validationErrors) > 0 {
+        color.Red("Configuration validation failed:")
+        for _, err := range validationErrors {
+            fmt.Printf("  - %s\n", err)
+        }
+        return errors.New("configuration validation failed")
+    }
+
+    color.Green("Configuration is valid!")
+    
+    // Show additional info about domain mappings if present
+    if len(cfg.DomainMappings) > 0 {
+        color.Cyan("\nDomain mappings configured:")
+        for domain, provider := range cfg.DomainMappings {
+            fmt.Printf("  %s → %s\n", domain, provider)
+        }
+    }
+
+    return nil
 }
 
+
 func runConfigGenerate(cmd *cobra.Command, _ []string) error {
 	force, err := cmd.Flags().GetBool("force")
 	if err != nil {
@@ -276,26 +319,26 @@ func runConfigGenerate(cmd *cobra.Command, _ []string) error {
 		return nil
 	}
 
-	// Generate example YAML config
-	if err := cfgMgr.CreateExampleYAML(); err != nil {
-		return fmt.Errorf("failed to create example configuration: %w", err)
-	}
-
-	color.Green("Example YAML configuration created: %s", cfgMgr.GetYAMLPath())
-	color.Cyan("\nNext steps:")
-	fmt.Println("1. Edit the configuration file to add your API keys")
-	fmt.Println("2. Customize provider settings and model whitelists as needed")
-	fmt.Println("3. Run 'cco config validate' to check your configuration")
-	fmt.Println("4. Start the router with 'cco start'")
-
-	color.Yellow("\nNote: The configuration includes all 5 supported providers:")
-	fmt.Println("- OpenRouter (access to multiple models)")
-	fmt.Println("- OpenAI (GPT models)")
-	fmt.Println("- Anthropic (Claude models)")
-	fmt.Println("- Nvidia (Nemotron models)")
-	fmt.Println("- Google Gemini (Gemini models)")
-
-	return nil
+    // Generate example YAML config
+    if err := cfgMgr.CreateExampleYAML(); err != nil {
+        return fmt.Errorf("failed to create example configuration: %w", err)
+    }
+
+    color.Green("Example YAML configuration created: %s", cfgMgr.GetYAMLPath())
+    color.Cyan("\nNext steps:")
+    fmt.Println("1. Edit the configuration file to add your API keys")
+    fmt.Println("2. Customize provider settings and model whitelists as needed")
+    fmt.Println("3. Configure domain_mappings to route local servers to providers")
+    fmt.Println("4. Run 'cco config validate' to check your configuration")
+    fmt.Println("5. Start the router with 'cco start'")
+
+    color.Yellow("\nNote: The configuration includes:")
+    fmt.Println("- All 5 supported providers (OpenRouter, OpenAI, Anthropic, Nvidia, Gemini)")
+    fmt.Println("- Domain mappings for local model support (localhost, 127.0.0.1, etc.)")
+    fmt.Println("- Example local LM Studio provider configuration")
+    fmt.Println("- Router configuration for different use cases")
+
+    return nil
 }
 
 func maskString(s string) string {