From 254a8713070d322ae8c969de82eae189a9dfdb62 Mon Sep 17 00:00:00 2001
From: mei <i@mmeiblog.cn>
Date: Tue, 9 Sep 2025 22:32:08 +0800
Subject: [PATCH] feat(internal/nazha): use env

---
 configs/.gitkeep   |  0
 configs/config.go  |  6 ------
 internal/export.go |  1 +
 main.go            |  9 +++++++--
 pkg/nezha.go       | 10 ++++++++--
 5 files changed, 16 insertions(+), 10 deletions(-)
 create mode 100644 configs/.gitkeep
 delete mode 100644 configs/config.go
 create mode 100644 internal/export.go

diff --git a/configs/.gitkeep b/configs/.gitkeep
new file mode 100644
index 0000000..e69de29
diff --git a/configs/config.go b/configs/config.go
deleted file mode 100644
index 046292a..0000000
--- a/configs/config.go
+++ /dev/null
@@ -1,6 +0,0 @@
-package configs
-
-const (
-	NEZHA_HOST = "192.178.7.123:8008"
-	SCHEME     = "ws"
-)
diff --git a/internal/export.go b/internal/export.go
new file mode 100644
index 0000000..5bf0569
--- /dev/null
+++ b/internal/export.go
@@ -0,0 +1 @@
+package internal
diff --git a/main.go b/main.go
index 7e2ca13..c8d0828 100644
--- a/main.go
+++ b/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 // 阻塞直到收到信号
diff --git a/pkg/nezha.go b/pkg/nezha.go
index d84fece..b35d957 100644
--- a/pkg/nezha.go
+++ b/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 {