forked from NdoleStudio/httpsms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathemulator_fcm_client.go
More file actions
100 lines (84 loc) · 2.75 KB
/
Copy pathemulator_fcm_client.go
File metadata and controls
100 lines (84 loc) · 2.75 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package services
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"net/http"
"firebase.google.com/go/messaging"
"github.com/NdoleStudio/httpsms/pkg/telemetry"
"github.com/palantir/stacktrace"
)
// EmulatorFCMClient sends FCM messages to the phone emulator via HTTP.
type EmulatorFCMClient struct {
httpClient *http.Client
endpoint string
logger telemetry.Logger
}
// NewEmulatorFCMClient creates a new EmulatorFCMClient.
func NewEmulatorFCMClient(httpClient *http.Client, endpoint string, logger telemetry.Logger) *EmulatorFCMClient {
return &EmulatorFCMClient{
httpClient: httpClient,
endpoint: endpoint,
logger: logger,
}
}
// emulatorFCMRequest is the payload sent to the emulator's FCM endpoint.
type emulatorFCMRequest struct {
Message *emulatorFCMMessage `json:"message"`
}
type emulatorFCMMessage struct {
Token string `json:"token"`
Data map[string]string `json:"data,omitempty"`
Android *emulatorAndroid `json:"android,omitempty"`
}
type emulatorAndroid struct {
Priority string `json:"priority,omitempty"`
}
// emulatorFCMResponse is the response from the emulator.
type emulatorFCMResponse struct {
Name string `json:"name"`
}
// Send sends a message to the emulator's FCM endpoint.
func (c *EmulatorFCMClient) Send(ctx context.Context, message *messaging.Message) (string, error) {
payload := &emulatorFCMRequest{
Message: &emulatorFCMMessage{
Token: message.Token,
Data: message.Data,
},
}
if message.Android != nil {
payload.Message.Android = &emulatorAndroid{
Priority: message.Android.Priority,
}
}
body, err := json.Marshal(payload)
if err != nil {
return "", stacktrace.Propagate(err, "cannot marshal FCM request for emulator")
}
url := fmt.Sprintf("%s/v1/projects/httpsms-test/messages:send", c.endpoint)
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(body))
if err != nil {
return "", stacktrace.Propagate(err, "cannot create HTTP request for emulator FCM")
}
req.Header.Set("Content-Type", "application/json")
resp, err := c.httpClient.Do(req)
if err != nil {
return "", stacktrace.Propagate(err, fmt.Sprintf("cannot send FCM to emulator at [%s]", url))
}
defer resp.Body.Close()
respBody, err := io.ReadAll(resp.Body)
if err != nil {
return "", stacktrace.Propagate(err, "cannot read emulator FCM response body")
}
if resp.StatusCode != http.StatusOK {
return "", stacktrace.NewError("emulator FCM returned status %d: %s", resp.StatusCode, string(respBody))
}
var result emulatorFCMResponse
if err = json.Unmarshal(respBody, &result); err != nil {
return "", stacktrace.Propagate(err, "cannot decode emulator FCM response")
}
c.logger.Info(fmt.Sprintf("emulator FCM sent successfully: %s", result.Name))
return result.Name, nil
}