Compare commits
13 Commits
Author | SHA1 | Date |
---|---|---|
|
a25fdc474a | |
|
c76b172a3a | |
|
eb024fd00c | |
|
d2b252d82a | |
|
250cfe9d33 | |
|
65928f4c0b | |
|
678438ace4 | |
|
a1fddd32af | |
|
5af327af4a | |
|
7462961556 | |
|
f0a01461fa | |
|
dd7ee3f703 | |
|
f38f14b926 |
|
@ -160,7 +160,7 @@ func makeSendResponseFromPart(part *multipart.Part) (SendResponse, error) {
|
|||
}
|
||||
defer response.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(response.Body)
|
||||
body, err := io.ReadAll(response.Body)
|
||||
if err != nil {
|
||||
return SendResponse{}, errors.WithStack(err)
|
||||
}
|
||||
|
|
38
fcm.go
38
fcm.go
|
@ -15,17 +15,19 @@ import (
|
|||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"git.bit5.ru/backend/errors"
|
||||
|
||||
"github.com/go-logr/logr"
|
||||
"golang.org/x/oauth2"
|
||||
|
||||
"go.opentelemetry.io/otel"
|
||||
"go.opentelemetry.io/otel/attribute"
|
||||
"go.opentelemetry.io/otel/codes"
|
||||
"go.opentelemetry.io/otel/trace"
|
||||
)
|
||||
|
||||
const tracerName = "git.bit5.ru/backend/fcm"
|
||||
|
@ -45,7 +47,7 @@ var (
|
|||
AuthScopes = []string{"https://www.googleapis.com/auth/firebase.messaging"}
|
||||
)
|
||||
|
||||
func MakeSendEndpoint(projectId string) string {
|
||||
func makeSendEndpoint(projectId string) string {
|
||||
return fmt.Sprintf("https://fcm.googleapis.com/v1/projects/%s/messages:send", projectId)
|
||||
}
|
||||
|
||||
|
@ -64,7 +66,7 @@ type Credentials struct {
|
|||
}
|
||||
|
||||
func ReadCredentialsFromFile(filename string) (Credentials, error) {
|
||||
data, err := ioutil.ReadFile(filename)
|
||||
data, err := os.ReadFile(filename)
|
||||
if err != nil {
|
||||
return Credentials{}, errors.WithStack(err)
|
||||
}
|
||||
|
@ -77,23 +79,19 @@ func ReadCredentialsFromFile(filename string) (Credentials, error) {
|
|||
return c, nil
|
||||
}
|
||||
|
||||
type ClientConfig struct {
|
||||
SendEndpoint string
|
||||
}
|
||||
|
||||
type Client struct {
|
||||
cfg ClientConfig
|
||||
ts oauth2.TokenSource
|
||||
hc *http.Client
|
||||
logger logr.Logger
|
||||
sendEndpoint string
|
||||
ts oauth2.TokenSource
|
||||
hc *http.Client
|
||||
logger logr.Logger
|
||||
}
|
||||
|
||||
func NewClient(cfg ClientConfig, ts oauth2.TokenSource, hc *http.Client, logger logr.Logger) *Client {
|
||||
func NewClient(projectId string, ts oauth2.TokenSource, hc *http.Client, logger logr.Logger) *Client {
|
||||
return &Client{
|
||||
cfg: cfg,
|
||||
ts: ts,
|
||||
hc: hc,
|
||||
logger: logger,
|
||||
sendEndpoint: makeSendEndpoint(projectId),
|
||||
ts: ts,
|
||||
hc: hc,
|
||||
logger: logger,
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -165,7 +163,11 @@ func (c *Client) Validate(ctx context.Context, message Message) (string, error)
|
|||
|
||||
func (c *Client) doSendRequest(ctx context.Context, req SendRequest, loggerEnabled bool) (SendResponse, error) {
|
||||
|
||||
_, span := tracer.Start(ctx, "Client.doSendRequest")
|
||||
spanAttrs := trace.WithAttributes(
|
||||
attribute.String("token", req.Message.Token),
|
||||
)
|
||||
|
||||
_, span := tracer.Start(ctx, "Client.doSendRequest", spanAttrs)
|
||||
defer span.End()
|
||||
|
||||
accessToken, err := c.ts.Token()
|
||||
|
@ -188,7 +190,7 @@ func (c *Client) doSendRequest(ctx context.Context, req SendRequest, loggerEnabl
|
|||
c.logger.Info("sending", "message", data)
|
||||
}
|
||||
|
||||
request, err := http.NewRequest(http.MethodPost, c.cfg.SendEndpoint, bytes.NewReader(data))
|
||||
request, err := http.NewRequest(http.MethodPost, c.sendEndpoint, bytes.NewReader(data))
|
||||
if err != nil {
|
||||
span.SetStatus(codes.Error, err.Error())
|
||||
span.RecordError(err)
|
||||
|
|
16
message.go
16
message.go
|
@ -1,9 +1,14 @@
|
|||
package fcm
|
||||
|
||||
// Message to send by Firebase Cloud Messaging Service.
|
||||
// https://firebase.google.com/docs/reference/fcm/rest/v1/projects.messages#resource:-message
|
||||
type Message struct {
|
||||
Token string `json:"token,omitempty"`
|
||||
Topic string `json:"topic,omitempty"`
|
||||
Condition string `json:"condition,omitempty"`
|
||||
// Union field target can be only one of the following:
|
||||
Token string `json:"token,omitempty"`
|
||||
Topic string `json:"topic,omitempty"`
|
||||
Condition string `json:"condition,omitempty"`
|
||||
// End of list of possible types for union field target.
|
||||
|
||||
Data map[string]string `json:"data,omitempty"`
|
||||
Notification *Notification `json:"notification,omitempty"`
|
||||
FcmOptions *FcmOptions `json:"fcm_options,omitempty"`
|
||||
|
@ -13,6 +18,7 @@ type Message struct {
|
|||
}
|
||||
|
||||
// Basic notification template to use across all platforms.
|
||||
// https://firebase.google.com/docs/reference/fcm/rest/v1/projects.messages#notification
|
||||
type Notification struct {
|
||||
Title string `json:"title,omitempty"`
|
||||
Body string `json:"body,omitempty"`
|
||||
|
@ -170,8 +176,8 @@ type ApnsPayload struct {
|
|||
Aps ApnsPayloadKeys `json:"aps"`
|
||||
|
||||
// TODO: add support for custom keys
|
||||
// https://developer.apple.com/documentation/usernotifications/setting_up_a_remote_notification_server/generating_a_remote_notification
|
||||
customKeys map[string]interface{}
|
||||
// https://developer.apple.com/documentation/usernotifications/generating-a-remote-notification
|
||||
customKeys map[string]any
|
||||
}
|
||||
|
||||
type ApnsPayloadKeys struct {
|
||||
|
|
Loading…
Reference in New Issue