summaryrefslogtreecommitdiff
path: root/internal/processing/push
diff options
context:
space:
mode:
Diffstat (limited to 'internal/processing/push')
-rw-r--r--internal/processing/push/create.go65
-rw-r--r--internal/processing/push/delete.go39
-rw-r--r--internal/processing/push/get.go47
-rw-r--r--internal/processing/push/push.go85
-rw-r--r--internal/processing/push/update.go63
5 files changed, 299 insertions, 0 deletions
diff --git a/internal/processing/push/create.go b/internal/processing/push/create.go
new file mode 100644
index 000000000..42a67dc19
--- /dev/null
+++ b/internal/processing/push/create.go
@@ -0,0 +1,65 @@
+// GoToSocial
+// Copyright (C) GoToSocial Authors admin@gotosocial.org
+// SPDX-License-Identifier: AGPL-3.0-or-later
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+package push
+
+import (
+ "context"
+
+ apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
+ "github.com/superseriousbusiness/gotosocial/internal/gtserror"
+ "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+ "github.com/superseriousbusiness/gotosocial/internal/id"
+)
+
+// CreateOrReplace creates a Web Push subscription for the given access token,
+// or entirely replaces the previously existing subscription for that token.
+func (p *Processor) CreateOrReplace(
+ ctx context.Context,
+ accountID string,
+ accessToken string,
+ request *apimodel.WebPushSubscriptionCreateRequest,
+) (*apimodel.WebPushSubscription, gtserror.WithCode) {
+ tokenID, errWithCode := p.getTokenID(ctx, accessToken)
+ if errWithCode != nil {
+ return nil, errWithCode
+ }
+
+ // Clear any previous subscription.
+ if err := p.state.DB.DeleteWebPushSubscriptionByTokenID(ctx, tokenID); err != nil {
+ err := gtserror.Newf("couldn't delete Web Push subscription for token ID %s: %w", tokenID, err)
+ return nil, gtserror.NewErrorInternalError(err)
+ }
+
+ // Insert a new one.
+ subscription := &gtsmodel.WebPushSubscription{
+ ID: id.NewULID(),
+ AccountID: accountID,
+ TokenID: tokenID,
+ Endpoint: request.Subscription.Endpoint,
+ Auth: request.Subscription.Keys.Auth,
+ P256dh: request.Subscription.Keys.P256dh,
+ NotificationFlags: alertsToNotificationFlags(request.Data.Alerts),
+ }
+
+ if err := p.state.DB.PutWebPushSubscription(ctx, subscription); err != nil {
+ err := gtserror.Newf("couldn't create Web Push subscription for token ID %s: %w", tokenID, err)
+ return nil, gtserror.NewErrorInternalError(err)
+ }
+
+ return p.apiSubscription(ctx, subscription)
+}
diff --git a/internal/processing/push/delete.go b/internal/processing/push/delete.go
new file mode 100644
index 000000000..6f5c61444
--- /dev/null
+++ b/internal/processing/push/delete.go
@@ -0,0 +1,39 @@
+// GoToSocial
+// Copyright (C) GoToSocial Authors admin@gotosocial.org
+// SPDX-License-Identifier: AGPL-3.0-or-later
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+package push
+
+import (
+ "context"
+
+ "github.com/superseriousbusiness/gotosocial/internal/gtserror"
+)
+
+// Delete deletes the Web Push subscription for the given access token, if there is one.
+func (p *Processor) Delete(ctx context.Context, accessToken string) gtserror.WithCode {
+ tokenID, errWithCode := p.getTokenID(ctx, accessToken)
+ if errWithCode != nil {
+ return errWithCode
+ }
+
+ if err := p.state.DB.DeleteWebPushSubscriptionByTokenID(ctx, tokenID); err != nil {
+ err := gtserror.Newf("couldn't delete Web Push subscription for token ID %s: %w", tokenID, err)
+ return gtserror.NewErrorInternalError(err)
+ }
+
+ return nil
+}
diff --git a/internal/processing/push/get.go b/internal/processing/push/get.go
new file mode 100644
index 000000000..542f08862
--- /dev/null
+++ b/internal/processing/push/get.go
@@ -0,0 +1,47 @@
+// GoToSocial
+// Copyright (C) GoToSocial Authors admin@gotosocial.org
+// SPDX-License-Identifier: AGPL-3.0-or-later
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+package push
+
+import (
+ "context"
+ "errors"
+
+ apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
+ "github.com/superseriousbusiness/gotosocial/internal/db"
+ "github.com/superseriousbusiness/gotosocial/internal/gtserror"
+)
+
+// Get returns the Web Push subscription for the given access token.
+func (p *Processor) Get(ctx context.Context, accessToken string) (*apimodel.WebPushSubscription, gtserror.WithCode) {
+ tokenID, errWithCode := p.getTokenID(ctx, accessToken)
+ if errWithCode != nil {
+ return nil, errWithCode
+ }
+
+ subscription, err := p.state.DB.GetWebPushSubscriptionByTokenID(ctx, tokenID)
+ if err != nil && !errors.Is(err, db.ErrNoEntries) {
+ err := gtserror.Newf("couldn't get Web Push subscription for token ID %s: %w", tokenID, err)
+ return nil, gtserror.NewErrorInternalError(err)
+ }
+ if subscription == nil {
+ err := errors.New("no Web Push subscription exists for this access token")
+ return nil, gtserror.NewErrorNotFound(err)
+ }
+
+ return p.apiSubscription(ctx, subscription)
+}
diff --git a/internal/processing/push/push.go b/internal/processing/push/push.go
new file mode 100644
index 000000000..f46280386
--- /dev/null
+++ b/internal/processing/push/push.go
@@ -0,0 +1,85 @@
+// GoToSocial
+// Copyright (C) GoToSocial Authors admin@gotosocial.org
+// SPDX-License-Identifier: AGPL-3.0-or-later
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+package push
+
+import (
+ "context"
+
+ apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
+ "github.com/superseriousbusiness/gotosocial/internal/gtserror"
+ "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+ "github.com/superseriousbusiness/gotosocial/internal/state"
+ "github.com/superseriousbusiness/gotosocial/internal/typeutils"
+)
+
+type Processor struct {
+ state *state.State
+ converter *typeutils.Converter
+}
+
+func New(state *state.State, converter *typeutils.Converter) Processor {
+ return Processor{
+ state: state,
+ converter: converter,
+ }
+}
+
+// getTokenID returns the token ID for a given access token.
+// Since all push API calls require authentication, this should always be available.
+func (p *Processor) getTokenID(ctx context.Context, accessToken string) (string, gtserror.WithCode) {
+ token, err := p.state.DB.GetTokenByAccess(ctx, accessToken)
+ if err != nil {
+ err := gtserror.Newf("couldn't find token ID for access token: %w", err)
+ return "", gtserror.NewErrorInternalError(err)
+ }
+
+ return token.ID, nil
+}
+
+// apiSubscription is a shortcut to return the API version of the given Web Push subscription,
+// or return an appropriate error if conversion fails.
+func (p *Processor) apiSubscription(ctx context.Context, subscription *gtsmodel.WebPushSubscription) (*apimodel.WebPushSubscription, gtserror.WithCode) {
+ apiSubscription, err := p.converter.WebPushSubscriptionToAPIWebPushSubscription(ctx, subscription)
+ if err != nil {
+ err := gtserror.Newf("error converting Web Push subscription %s to API representation: %w", subscription.ID, err)
+ return nil, gtserror.NewErrorInternalError(err)
+ }
+
+ return apiSubscription, nil
+}
+
+// alertsToNotificationFlags turns the alerts section of a push subscription API request into a packed bitfield.
+func alertsToNotificationFlags(alerts *apimodel.WebPushSubscriptionAlerts) gtsmodel.WebPushSubscriptionNotificationFlags {
+ var n gtsmodel.WebPushSubscriptionNotificationFlags
+
+ n.Set(gtsmodel.NotificationFollow, alerts.Follow)
+ n.Set(gtsmodel.NotificationFollowRequest, alerts.FollowRequest)
+ n.Set(gtsmodel.NotificationFavourite, alerts.Favourite)
+ n.Set(gtsmodel.NotificationMention, alerts.Mention)
+ n.Set(gtsmodel.NotificationReblog, alerts.Reblog)
+ n.Set(gtsmodel.NotificationPoll, alerts.Poll)
+ n.Set(gtsmodel.NotificationStatus, alerts.Status)
+ n.Set(gtsmodel.NotificationUpdate, alerts.Update)
+ n.Set(gtsmodel.NotificationAdminSignup, alerts.AdminSignup)
+ n.Set(gtsmodel.NotificationAdminReport, alerts.AdminReport)
+ n.Set(gtsmodel.NotificationPendingFave, alerts.PendingFavourite)
+ n.Set(gtsmodel.NotificationPendingReply, alerts.PendingReply)
+ n.Set(gtsmodel.NotificationPendingReblog, alerts.PendingReblog)
+
+ return n
+}
diff --git a/internal/processing/push/update.go b/internal/processing/push/update.go
new file mode 100644
index 000000000..370536f9b
--- /dev/null
+++ b/internal/processing/push/update.go
@@ -0,0 +1,63 @@
+// GoToSocial
+// Copyright (C) GoToSocial Authors admin@gotosocial.org
+// SPDX-License-Identifier: AGPL-3.0-or-later
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+package push
+
+import (
+ "context"
+ "errors"
+
+ apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
+ "github.com/superseriousbusiness/gotosocial/internal/db"
+ "github.com/superseriousbusiness/gotosocial/internal/gtserror"
+)
+
+// Update updates the Web Push subscription for the given access token.
+func (p *Processor) Update(
+ ctx context.Context,
+ accessToken string,
+ request *apimodel.WebPushSubscriptionUpdateRequest,
+) (*apimodel.WebPushSubscription, gtserror.WithCode) {
+ tokenID, errWithCode := p.getTokenID(ctx, accessToken)
+ if errWithCode != nil {
+ return nil, errWithCode
+ }
+
+ // Get existing subscription.
+ subscription, err := p.state.DB.GetWebPushSubscriptionByTokenID(ctx, tokenID)
+ if err != nil && !errors.Is(err, db.ErrNoEntries) {
+ err := gtserror.Newf("couldn't get Web Push subscription for token ID %s: %w", tokenID, err)
+ return nil, gtserror.NewErrorInternalError(err)
+ }
+ if subscription == nil {
+ err := errors.New("no Web Push subscription exists for this access token")
+ return nil, gtserror.NewErrorNotFound(err)
+ }
+
+ // Update it.
+ subscription.NotificationFlags = alertsToNotificationFlags(request.Data.Alerts)
+ if err = p.state.DB.UpdateWebPushSubscription(
+ ctx,
+ subscription,
+ "notification_flags",
+ ); err != nil {
+ err := gtserror.Newf("couldn't update Web Push subscription for token ID %s: %w", tokenID, err)
+ return nil, gtserror.NewErrorInternalError(err)
+ }
+
+ return p.apiSubscription(ctx, subscription)
+}
generated by cgit v1.2.3 (git 2.46.0) at 2025-07-25 08:09:09 +0000