From 7524c9eaea088154d6e8b9e80e6c692c1f0ee522 Mon Sep 17 00:00:00 2001
From: dark <nndaa@qq.com>
Date: Mon, 14 Jul 2025 04:21:17 +0800
Subject: [PATCH] =?UTF-8?q?=E9=9D=99=E6=80=81swagger?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/main.tpl | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/api/main.tpl b/api/main.tpl
index b64b7d5..6c4de4a 100644
--- a/api/main.tpl
+++ b/api/main.tpl
@@ -3,6 +3,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"net/http"
 
 	{{.importPackages}}
 )
@@ -14,9 +15,16 @@ func main() {
 
 	var c config.Config
 	conf.MustLoad(*configFile, &c)
-		
-	server := rest.MustNewServer(c.RestConf)
+	server := *rest.Server
+	// dev 模式下开启swagger
+	if c.Mode == "dev" {
+		server = rest.MustNewServer(c.RestConf, rest.WithCors(), rest.WithCorsHeaders("*"), rest.WithFileServer("/swagger", http.Dir("./swagger")))
+	} else {
+		server = rest.MustNewServer(c.RestConf, rest.WithCors())
+	}
 	defer server.Stop()
+	// rest.MustNewServer(c.RestConf)
+	// defer server.Stop()
 
 	ctx := svc.NewServiceContext(c)
 	handler.RegisterHandlers(server, ctx)