restore main.go

main.go

@@ -1,23 +1,28 @@
 package main
 
 import (
-"log"
-"net/http"
-"os"
+	"zai-proxy/internal/config"
+	"zai-proxy/internal/handler"
+	"zai-proxy/internal/logger"
+	"zai-proxy/internal/proxy"
+	"zai-proxy/internal/version"
+	"net/http"
 )
 
 func main() {
-// Servir la interfaz de HOI4
-http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-http.ServeFile(w, r, "index.html")
-})
-
-// Dejamos que los otros archivos se registren solos
-port := os.Getenv("PORT")
-if port == "" { port = "7860" }
-
-log.Printf("Servidor recuperado en puerto %s", port)
-if err := http.ListenAndServe(":"+port, nil); err != nil {
-log.Fatal(err)
-}
+	config.LoadConfig()
+	logger.InitLogger()
+	proxy.LoadProxies("proxies.txt")
+	version.StartVersionUpdater()
+	http.HandleFunc("/", handler.HandleIndex)
+	http.HandleFunc("/genkey", handler.HandleGenKey)
+	http.HandleFunc("/stats", handler.HandleStats)
+	http.HandleFunc("/v1/models", handler.HandleModels)
+	http.HandleFunc("/v1/chat/completions", handler.HandleChatCompletions)
+	http.HandleFunc("/v1/messages", handler.HandleMessages)
+	addr := ":" + config.Cfg.Port
+	logger.LogInfo("Server starting on %s", addr)
+	if err := http.ListenAndServe(addr, nil); err != nil {
+		logger.LogError("Server failed: %v", err)
+	}
 }