apollo-backend/internal/api/devices.go

118 lines
2.8 KiB
Go
Raw Normal View History

package api
2021-05-10 00:51:15 +00:00
import (
2021-08-08 18:19:47 +00:00
"context"
2021-05-10 00:51:15 +00:00
"encoding/json"
2021-08-08 18:19:47 +00:00
"fmt"
2021-05-10 00:51:15 +00:00
"net/http"
2021-08-08 18:19:47 +00:00
"strings"
2021-08-14 15:51:27 +00:00
"time"
2021-05-10 00:51:15 +00:00
2021-08-08 18:19:47 +00:00
"github.com/dustin/go-humanize/english"
"github.com/gorilla/mux"
"github.com/sideshow/apns2"
"github.com/sideshow/apns2/payload"
"github.com/sirupsen/logrus"
2021-08-14 17:56:03 +00:00
"github.com/christianselig/apollo-backend/internal/domain"
2021-05-10 00:51:15 +00:00
)
2021-08-08 18:19:47 +00:00
const notificationTitle = "📣 Hello, is this thing on?"
func (a *api) upsertDeviceHandler(w http.ResponseWriter, r *http.Request) {
ctx := context.Background()
d := &domain.Device{}
2021-05-10 00:51:15 +00:00
if err := json.NewDecoder(r.Body).Decode(d); err != nil {
a.errorResponse(w, r, 500, err.Error())
2021-05-10 00:51:15 +00:00
return
}
2021-09-11 14:53:19 +00:00
d.ActiveUntil = time.Now().Unix() + domain.DeviceGracePeriodDuration
d.GracePeriodUntil = d.ActiveUntil + domain.DeviceGracePeriodAfterReceiptExpiry
2021-08-14 15:51:27 +00:00
2021-08-08 18:19:47 +00:00
if err := a.deviceRepo.CreateOrUpdate(ctx, d); err != nil {
a.errorResponse(w, r, 500, err.Error())
return
}
w.WriteHeader(http.StatusOK)
}
func (a *api) testDeviceHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
ctx := context.Background()
tok := vars["apns"]
d, err := a.deviceRepo.GetByAPNSToken(ctx, tok)
if err != nil {
a.logger.WithFields(logrus.Fields{
"err": err,
}).Info("failed fetching device from database")
a.errorResponse(w, r, 500, err.Error())
return
}
accs, err := a.accountRepo.GetByAPNSToken(ctx, tok)
if err != nil {
a.errorResponse(w, r, 500, err.Error())
return
}
users := make([]string, len(accs))
for i := range accs {
users[i] = accs[i].Username
}
body := fmt.Sprintf("Active usernames are: %s. Tap me for more info!", english.OxfordWordSeries(users, "and"))
notification := &apns2.Notification{}
notification.Topic = "com.christianselig.Apollo"
notification.DeviceToken = d.APNSToken
notification.Payload = payload.
NewPayload().
Category("test-notification").
Custom("test_accounts", strings.Join(users, ",")).
AlertTitle(notificationTitle).
AlertBody(body)
client := apns2.NewTokenClient(a.apns)
if !d.Sandbox {
client = client.Production()
}
if _, err := client.Push(notification); err != nil {
a.logger.WithFields(logrus.Fields{
"err": err,
}).Info("failed to send test notification")
a.errorResponse(w, r, 500, err.Error())
2021-05-10 00:51:15 +00:00
return
}
2021-08-08 18:19:47 +00:00
w.WriteHeader(http.StatusOK)
}
func (a *api) deleteDeviceHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
ctx := context.Background()
dev, err := a.deviceRepo.GetByAPNSToken(ctx, vars["apns"])
if err != nil {
a.errorResponse(w, r, 500, err.Error())
return
}
accs, err := a.accountRepo.GetByAPNSToken(ctx, vars["apns"])
if err != nil {
a.errorResponse(w, r, 500, err.Error())
return
}
for _, acc := range accs {
2021-09-25 13:19:42 +00:00
_ = a.accountRepo.Disassociate(ctx, &acc, &dev)
2021-08-08 18:19:47 +00:00
}
2021-05-10 00:51:15 +00:00
2021-09-25 13:19:42 +00:00
_ = a.deviceRepo.Delete(ctx, vars["apns"])
2021-09-11 14:53:19 +00:00
2021-05-10 00:51:15 +00:00
w.WriteHeader(http.StatusOK)
}