增加配置功能

main.go

@@ -1,14 +1,25 @@
 package main
 
 import (
+	"fmt"
+	"log"
+	"net"
 	"net/http"
+	"os"
+	"path/filepath"
 	"strings"
 
+	"lmvpn/config"
+
 	"github.com/gin-gonic/gin"
 )
 
 func main() {
-	//启动gin服务
+	cfg, err := config.Load("data/config.yml")
+	if err != nil {
+		log.Fatalf("加载配置失败: %v", err)
+	}
+
 	r := gin.Default()
 
 	// 静态文件服务
@@ -25,5 +36,40 @@ func main() {
 		c.Abort()
 	})
 
-	r.Run()
+	if cfg.Web.Port == 0 && cfg.Web.Sock == "" {
+		log.Fatal("配置错误: port 和 sock 不能同时为空")
+	}
+
+	if cfg.Web.Port > 0 {
+		go func() {
+			log.Printf("TCP 监听 :%d", cfg.Web.Port)
+			if err := r.Run(fmt.Sprintf(":%d", cfg.Web.Port)); err != nil {
+				log.Fatalf("TCP 启动失败: %v", err)
+			}
+		}()
+	}
+
+	if cfg.Web.Sock != "" {
+		if err := os.Remove(cfg.Web.Sock); err != nil && !os.IsNotExist(err) {
+			log.Fatalf("删除残留 sock 文件失败: %v", err)
+		}
+		if err := os.MkdirAll(filepath.Dir(cfg.Web.Sock), 0755); err != nil {
+			log.Fatalf("创建 sock 目录失败: %v", err)
+		}
+		listener, err := net.Listen("unix", cfg.Web.Sock)
+		if err != nil {
+			log.Fatalf("Unix socket 监听失败: %v", err)
+		}
+		if err := os.Chmod(cfg.Web.Sock, 0666); err != nil {
+			log.Fatalf("设置 sock 权限失败: %v", err)
+		}
+		go func() {
+			log.Printf("Unix socket 监听 %s", cfg.Web.Sock)
+			if err := r.RunListener(listener); err != nil {
+				log.Fatalf("Unix socket 启动失败: %v", err)
+			}
+		}()
+	}
+
+	select {}
 }