feat: 增加推送通知配置
This commit is contained in:
@@ -0,0 +1,69 @@
|
||||
package push
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const defaultBarkServer = "https://api.day.app"
|
||||
|
||||
// BarkConfig 是 iOS Bark 推送的配置。
|
||||
type BarkConfig struct {
|
||||
DeviceKey string
|
||||
Server string // 可选,默认 https://api.day.app
|
||||
}
|
||||
|
||||
// BarkProvider 实现了通过 Bark 发送 iOS 推送。
|
||||
type BarkProvider struct {
|
||||
deviceKey string
|
||||
server string
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
// NewBarkProvider 创建 Bark 推送 Provider。
|
||||
// DeviceKey 为空时返回 ErrProviderConfigInvalid。
|
||||
func NewBarkProvider(cfg BarkConfig) (*BarkProvider, error) {
|
||||
if strings.TrimSpace(cfg.DeviceKey) == "" {
|
||||
return nil, ErrProviderConfigInvalid
|
||||
}
|
||||
server := strings.TrimSpace(cfg.Server)
|
||||
if server == "" {
|
||||
server = defaultBarkServer
|
||||
}
|
||||
server = strings.TrimRight(server, "/")
|
||||
return &BarkProvider{
|
||||
deviceKey: strings.TrimSpace(cfg.DeviceKey),
|
||||
server: server,
|
||||
client: &http.Client{Timeout: 10 * time.Second},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (p *BarkProvider) Name() string { return "bark" }
|
||||
|
||||
func (p *BarkProvider) Send(ctx context.Context, msg Message) error {
|
||||
title := url.PathEscape(msg.Title)
|
||||
body := url.PathEscape(msg.Content)
|
||||
reqURL := fmt.Sprintf("%s/%s/%s/%s", p.server, p.deviceKey, title, body)
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, reqURL, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("bark: create request: %w", err)
|
||||
}
|
||||
|
||||
resp, err := p.client.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("bark: send failed: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode >= 400 {
|
||||
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||
return fmt.Errorf("bark: http %d: %s", resp.StatusCode, string(bodyBytes))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,12 @@
|
||||
package push
|
||||
|
||||
import "context"
|
||||
|
||||
// NoopProvider 是一个空实现,未配置任何推送渠道时使用。
|
||||
type NoopProvider struct{}
|
||||
|
||||
func NewNoopProvider() *NoopProvider { return &NoopProvider{} }
|
||||
|
||||
func (p *NoopProvider) Name() string { return "noop" }
|
||||
|
||||
func (p *NoopProvider) Send(_ context.Context, _ Message) error { return nil }
|
||||
@@ -0,0 +1,23 @@
|
||||
package push
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
)
|
||||
|
||||
var ErrProviderConfigInvalid = errors.New("push provider config invalid")
|
||||
|
||||
// Message 描述一条待发送的推送消息。
|
||||
type Message struct {
|
||||
Title string
|
||||
Content string
|
||||
}
|
||||
|
||||
// Provider 是站外推送渠道的统一接口。
|
||||
// 实现方负责具体的 HTTP 调用(Bark、WPush 等)。
|
||||
type Provider interface {
|
||||
// Name 返回渠道标识,用于日志。
|
||||
Name() string
|
||||
// Send 发送一条推送消息。失败时返回 error。
|
||||
Send(ctx context.Context, msg Message) error
|
||||
}
|
||||
@@ -0,0 +1,73 @@
|
||||
package push
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const wpushAPIURL = "https://api.wpush.cn/api/v1/send"
|
||||
|
||||
// WPushConfig 是 WPush 推送的配置。
|
||||
type WPushConfig struct {
|
||||
APIKey string
|
||||
}
|
||||
|
||||
// WPushProvider 实现了通过 WPush 发送推送。
|
||||
// 支持微信公众号、飞书、钉钉、企业微信等多种渠道,取决于用户在 WPush 侧的配置。
|
||||
type WPushProvider struct {
|
||||
apiKey string
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
// NewWPushProvider 创建 WPush 推送 Provider。
|
||||
// APIKey 为空时返回 ErrProviderConfigInvalid。
|
||||
func NewWPushProvider(cfg WPushConfig) (*WPushProvider, error) {
|
||||
if strings.TrimSpace(cfg.APIKey) == "" {
|
||||
return nil, ErrProviderConfigInvalid
|
||||
}
|
||||
return &WPushProvider{
|
||||
apiKey: strings.TrimSpace(cfg.APIKey),
|
||||
client: &http.Client{Timeout: 10 * time.Second},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (p *WPushProvider) Name() string { return "wpush" }
|
||||
|
||||
func (p *WPushProvider) Send(ctx context.Context, msg Message) error {
|
||||
form := url.Values{}
|
||||
form.Set("apikey", p.apiKey)
|
||||
form.Set("title", msg.Title)
|
||||
form.Set("content", msg.Content)
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, wpushAPIURL, strings.NewReader(form.Encode()))
|
||||
if err != nil {
|
||||
return fmt.Errorf("wpush: create request: %w", err)
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
|
||||
resp, err := p.client.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("wpush: send failed: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||
|
||||
var result struct {
|
||||
Code int `json:"code"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
if err := json.Unmarshal(bodyBytes, &result); err != nil {
|
||||
return fmt.Errorf("wpush: parse response: %w (body: %s)", err, string(bodyBytes))
|
||||
}
|
||||
if result.Code != 0 {
|
||||
return fmt.Errorf("wpush: api error code=%d: %s", result.Code, result.Message)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user