mirror of
https://github.com/yusing/godoxy.git
synced 2025-06-01 09:32:35 +02:00
4a8bd48ad5
Some checks are pending
Docker Image CI (socket-proxy) / build (push) Waiting to run
* refactor: simplify io code and make utils module independent * fix(docker): agent and socket-proxy docker event flushing with modified reverse proxy handler * refactor: remove unused code * refactor: remove the use of logging module in most code * refactor: streamline domain mismatch check in certState function * tweak: use ecdsa p-256 for autocert * fix(tests): update health check tests for invalid host and add case for port in host * feat(acme): custom acme directory * refactor: code refactor and improved context and error handling * tweak: optimize memory usage under load * fix(oidc): restore old user matching behavior * docs: add ChatGPT assistant to README --------- Co-authored-by: yusing <yusing@6uo.me>
86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
package notif
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/yusing/go-proxy/internal/gperr"
|
|
"github.com/yusing/go-proxy/internal/serialization"
|
|
)
|
|
|
|
type (
|
|
Provider interface {
|
|
serialization.CustomValidator
|
|
|
|
GetName() string
|
|
GetURL() string
|
|
GetToken() string
|
|
GetMethod() string
|
|
GetMIMEType() string
|
|
|
|
MarshalMessage(logMsg *LogMessage) ([]byte, error)
|
|
SetHeaders(logMsg *LogMessage, headers http.Header)
|
|
|
|
fmtError(respBody io.Reader) error
|
|
}
|
|
ProviderCreateFunc func(map[string]any) (Provider, gperr.Error)
|
|
ProviderConfig map[string]any
|
|
)
|
|
|
|
const (
|
|
ProviderGotify = "gotify"
|
|
ProviderNtfy = "ntfy"
|
|
ProviderWebhook = "webhook"
|
|
)
|
|
|
|
func (msg *LogMessage) notify(ctx context.Context, provider Provider) error {
|
|
body, err := provider.MarshalMessage(msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(ctx, 2*time.Second)
|
|
defer cancel()
|
|
|
|
req, err := http.NewRequestWithContext(
|
|
ctx,
|
|
http.MethodPost,
|
|
provider.GetURL(),
|
|
bytes.NewReader(body),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req.Header.Set("Content-Type", provider.GetMIMEType())
|
|
if provider.GetToken() != "" {
|
|
req.Header.Set("Authorization", "Bearer "+provider.GetToken())
|
|
}
|
|
provider.SetHeaders(msg, req.Header)
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
|
|
switch resp.StatusCode {
|
|
case http.StatusOK, http.StatusCreated, http.StatusAccepted, http.StatusNoContent:
|
|
body, _ := io.ReadAll(resp.Body)
|
|
log.Debug().
|
|
Str("provider", provider.GetName()).
|
|
Str("url", provider.GetURL()).
|
|
Str("status", resp.Status).
|
|
RawJSON("resp_body", body).
|
|
Msg("notification sent")
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("http status %d: %w", resp.StatusCode, provider.fmtError(resp.Body))
|
|
}
|
|
}
|