mirror of
https://github.com/christianselig/apollo-backend
synced 2024-11-22 19:57:43 +00:00
117 lines
2.8 KiB
Go
117 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/dustin/go-humanize/english"
|
|
"github.com/gorilla/mux"
|
|
"github.com/sideshow/apns2"
|
|
"github.com/sideshow/apns2/payload"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/christianselig/apollo-backend/internal/domain"
|
|
)
|
|
|
|
const notificationTitle = "📣 Hello, is this thing on?"
|
|
|
|
func (a *api) upsertDeviceHandler(w http.ResponseWriter, r *http.Request) {
|
|
ctx := context.Background()
|
|
|
|
d := &domain.Device{}
|
|
if err := json.NewDecoder(r.Body).Decode(d); err != nil {
|
|
a.errorResponse(w, r, 500, err.Error())
|
|
return
|
|
}
|
|
|
|
d.ActiveUntil = time.Now().Unix() + domain.DeviceGracePeriodDuration
|
|
d.GracePeriodUntil = d.ActiveUntil + domain.DeviceGracePeriodAfterReceiptExpiry
|
|
|
|
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())
|
|
return
|
|
}
|
|
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 {
|
|
_ = a.accountRepo.Disassociate(ctx, &acc, &dev)
|
|
}
|
|
|
|
_ = a.deviceRepo.Delete(ctx, vars["apns"])
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|