2022-06-15 22:03:12 +02:00
|
|
|
package server
|
|
|
|
|
|
|
|
import (
|
2022-06-16 02:36:49 +02:00
|
|
|
"bytes"
|
2022-06-15 22:03:12 +02:00
|
|
|
"encoding/json"
|
2022-06-16 02:36:49 +02:00
|
|
|
"fmt"
|
2022-06-15 22:03:12 +02:00
|
|
|
"heckel.io/ntfy/log"
|
2022-06-16 02:36:49 +02:00
|
|
|
"heckel.io/ntfy/util"
|
2022-06-15 22:03:12 +02:00
|
|
|
"io"
|
|
|
|
"net/http"
|
2022-06-16 02:36:49 +02:00
|
|
|
"strings"
|
2022-06-15 22:03:12 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2022-06-16 02:36:49 +02:00
|
|
|
matrixPushKeyHeader = "X-Matrix-Pushkey"
|
2022-06-15 22:03:12 +02:00
|
|
|
)
|
|
|
|
|
2022-06-16 02:51:42 +02:00
|
|
|
// matrixRequest represents a Matrix message, as it is sent to a Push Gateway (as per
|
|
|
|
// this spec: https://spec.matrix.org/v1.2/push-gateway-api/).
|
|
|
|
//
|
|
|
|
// From the message, we only require the "pushkey", as it represents our target topic URL.
|
|
|
|
// A message may look like this (excerpt):
|
|
|
|
// {
|
|
|
|
// "notification": {
|
|
|
|
// "devices": [
|
|
|
|
// {
|
|
|
|
// "pushkey": "https://ntfy.sh/upDAHJKFFDFD?up=1",
|
|
|
|
// ...
|
|
|
|
// }
|
|
|
|
// ]
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
type matrixRequest struct {
|
|
|
|
Notification *struct {
|
|
|
|
Devices []*struct {
|
|
|
|
PushKey string `json:"pushkey"`
|
|
|
|
} `json:"devices"`
|
|
|
|
} `json:"notification"`
|
2022-06-15 22:03:12 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
type matrixResponse struct {
|
|
|
|
Rejected []string `json:"rejected"`
|
|
|
|
}
|
|
|
|
|
2022-06-16 02:36:49 +02:00
|
|
|
type errMatrix struct {
|
|
|
|
pushKey string
|
|
|
|
err error
|
|
|
|
}
|
|
|
|
|
|
|
|
func (e errMatrix) Error() string {
|
|
|
|
if e.err != nil {
|
|
|
|
return fmt.Sprintf("message with push key %s rejected: %s", e.pushKey, e.err.Error())
|
|
|
|
}
|
|
|
|
return fmt.Sprintf("message with push key %s rejected", e.pushKey)
|
|
|
|
}
|
|
|
|
|
2022-06-16 02:51:42 +02:00
|
|
|
// newRequestFromMatrixJSON reads the request body as a Matrix JSON message, parses the "pushkey", and creates a new
|
|
|
|
// HTTP request that looks like a normal ntfy request from it.
|
|
|
|
//
|
|
|
|
// It basically converts a Matrix push gatewqy request:
|
|
|
|
//
|
|
|
|
// POST /_matrix/push/v1/notify HTTP/1.1
|
|
|
|
// { "notification": { "devices": [ { "pushkey": "https://ntfy.sh/upDAHJKFFDFD?up=1", ... } ] } }
|
|
|
|
//
|
|
|
|
// to a ntfy request, looking like this:
|
|
|
|
//
|
|
|
|
// POST /upDAHJKFFDFD?up=1 HTTP/1.1
|
|
|
|
// { "notification": { "devices": [ { "pushkey": "https://ntfy.sh/upDAHJKFFDFD?up=1", ... } ] } }
|
|
|
|
//
|
2022-06-16 02:36:49 +02:00
|
|
|
func newRequestFromMatrixJSON(r *http.Request, baseURL string, messageLimit int) (*http.Request, error) {
|
|
|
|
if baseURL == "" {
|
|
|
|
return nil, errHTTPInternalErrorMissingBaseURL
|
|
|
|
}
|
|
|
|
body, err := util.Peek(r.Body, messageLimit)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
defer r.Body.Close()
|
2022-06-16 02:51:42 +02:00
|
|
|
var m matrixRequest
|
2022-06-16 02:36:49 +02:00
|
|
|
if err := json.NewDecoder(body).Decode(&m); err != nil {
|
|
|
|
return nil, errHTTPBadRequestMatrixMessageInvalid
|
|
|
|
} else if m.Notification == nil || len(m.Notification.Devices) == 0 || m.Notification.Devices[0].PushKey == "" {
|
|
|
|
return nil, errHTTPBadRequestMatrixMessageInvalid
|
|
|
|
}
|
|
|
|
pushKey := m.Notification.Devices[0].PushKey
|
|
|
|
if !strings.HasPrefix(pushKey, baseURL+"/") {
|
|
|
|
return nil, &errMatrix{pushKey: pushKey, err: errHTTPBadRequestMatrixPushkeyBaseURLMismatch}
|
|
|
|
}
|
|
|
|
newRequest, err := http.NewRequest(http.MethodPost, pushKey, io.NopCloser(bytes.NewReader(body.PeekedBytes)))
|
|
|
|
if err != nil {
|
|
|
|
return nil, &errMatrix{pushKey: pushKey, err: err}
|
|
|
|
}
|
|
|
|
newRequest.Header.Set(matrixPushKeyHeader, pushKey)
|
|
|
|
return newRequest, nil
|
|
|
|
}
|
|
|
|
|
2022-06-15 22:03:12 +02:00
|
|
|
func handleMatrixDiscovery(w http.ResponseWriter) error {
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
_, err := io.WriteString(w, `{"unifiedpush":{"gateway":"matrix"}}`+"\n")
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-06-16 02:36:49 +02:00
|
|
|
func writeMatrixError(w http.ResponseWriter, r *http.Request, v *visitor, err *errMatrix) error {
|
|
|
|
log.Debug("%s Matrix gateway error: %s", logHTTPPrefix(v, r), err.Error())
|
|
|
|
return writeMatrixResponse(w, err.pushKey)
|
2022-06-15 22:03:12 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func writeMatrixSuccess(w http.ResponseWriter) error {
|
2022-06-16 02:36:49 +02:00
|
|
|
return writeMatrixResponse(w, "")
|
|
|
|
}
|
|
|
|
|
|
|
|
func writeMatrixResponse(w http.ResponseWriter, rejectedPushKey string) error {
|
|
|
|
rejected := make([]string, 0)
|
|
|
|
if rejectedPushKey != "" {
|
|
|
|
rejected = append(rejected, rejectedPushKey)
|
|
|
|
}
|
2022-06-15 22:03:12 +02:00
|
|
|
response := &matrixResponse{
|
2022-06-16 02:36:49 +02:00
|
|
|
Rejected: rejected,
|
2022-06-15 22:03:12 +02:00
|
|
|
}
|
2022-06-16 02:36:49 +02:00
|
|
|
w.Header().Set("Content-Type", "application/json")
|
2022-06-15 22:03:12 +02:00
|
|
|
if err := json.NewEncoder(w).Encode(response); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|