feat(internal/nazha): use env

configs/config.go

@@ -1,6 +0,0 @@
-package configs
-
-const (
-	NEZHA_HOST = "192.178.7.123:8008"
-	SCHEME     = "ws"
-)

internal/export.go

@@ -0,0 +1 @@
+package internal

main.go

@@ -11,13 +11,18 @@ import (
 )
 
 func main() {
+	// init
 	internal.Recoder()
+
+	// CRON
 	c := cron.New()
-
 	c.AddFunc("@every 1h", internal.Recoder)
-
 	c.Start()
 
+	// Web API
+	// gin := gin.Default()
+
+	// 阻塞
 	sig := make(chan os.Signal, 1)
 	signal.Notify(sig, os.Interrupt, syscall.SIGTERM)
 	<-sig // 阻塞直到收到信号

pkg/nezha.go

@@ -4,8 +4,8 @@ import (
 	"encoding/json"
 	"log"
 	"net/url"
+	"os"
 
-	"git.mmeiblog.cn/mei/nezha-recoder/configs"
 	"github.com/gorilla/websocket"
 )
 
@@ -40,7 +40,13 @@ type State struct {
 }
 
 func GetData() (servers Message) {
-	u := url.URL{Scheme: configs.SCHEME, Host: configs.NEZHA_HOST, Path: "/api/v1/ws/server"}
+	nazha_host := os.Getenv("NEZHA_HOST")
+	if nazha_host == "" {
+		log.Fatal("NEZHA_HOST environment variable is not set")
+		return
+	}
+
+	u := url.URL{Scheme: "ws", Host: nazha_host, Path: "/api/v1/ws/server"}
 
 	conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
 	if err != nil {