支持自定义bark server

zc2638 · Apr 17, 2022 · 312b681 · 312b681
1 parent 6f6dc08
commit 312b681
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/config/config.yaml b/config/config.yaml
@@ -1,5 +1,6 @@
 bark:
-  key: ""               # Bark 通知推送的 Key
+  server: "https://api.day.app" # 自定义 Bark Server 地址, 不填默认为 https://api.day.app
+  key: ""                       # Bark 通知推送的 Key
 push_plus:
   token: ""             # Push Plus 通知推送的 Token
 regular:

diff --git a/pkg/notice/bark.go b/pkg/notice/bark.go
@@ -20,23 +20,25 @@ import (
 	"fmt"
 	"io/ioutil"
 	"net/http"
+	"path"
 )
 
 const barkURL = "https://api.day.app/push"
 
 type BarkConfig struct {
-	Key string `json:"key"`
+	Server string `json:"server"`
+	Key    string `json:"key"`
 }
 
 func NewBark(cfg *BarkConfig) Engine {
 	if cfg.Key == "" {
 		return nil
 	}
-	return &bark{key: cfg.Key}
+	return &bark{cfg: cfg}
 }
 
 type bark struct {
-	key string
+	cfg *BarkConfig
 }
 
 func (b *bark) Name() string {
@@ -45,7 +47,7 @@ func (b *bark) Name() string {
 
 func (b *bark) Send(title, body string) error {
 	data := &barkData{
-		DeviceKey: b.key,
+		DeviceKey: b.cfg.Key,
 		Title:     title,
 		Body:      body,
 		Sound:     "alarm.caf",
@@ -55,7 +57,11 @@ func (b *bark) Send(title, body string) error {
 		return err
 	}
 
-	resp, err := http.Post(barkURL, "application/json; charset=utf-8", bytes.NewReader(bs))
+	uri := barkURL
+	if b.cfg.Server != "" {
+		uri = path.Join(b.cfg.Server, "push")
+	}
+	resp, err := http.Post(uri, "application/json; charset=utf-8", bytes.NewReader(bs))
 	if err != nil {
 		return err
 	}