diff options
Diffstat (limited to 'internal/typeutils')
-rw-r--r-- | internal/typeutils/converter_test.go | 3 | ||||
-rw-r--r-- | internal/typeutils/internaltofrontend.go | 45 | ||||
-rw-r--r-- | internal/typeutils/internaltofrontend_test.go | 13 |
3 files changed, 59 insertions, 2 deletions
diff --git a/internal/typeutils/converter_test.go b/internal/typeutils/converter_test.go index a5fe5201b..3ed2828f2 100644 --- a/internal/typeutils/converter_test.go +++ b/internal/typeutils/converter_test.go @@ -540,8 +540,9 @@ func (suite *TypeUtilsTestSuite) GetProcessor() *processing.Processor { mediaManager := testrig.NewTestMediaManager(&suite.state) federator := testrig.NewTestFederator(&suite.state, transportController, mediaManager) emailSender := testrig.NewEmailSender("../../web/template/", nil) + webPushSender := testrig.NewNoopWebPushSender() - processor := testrig.NewTestProcessor(&suite.state, federator, emailSender, mediaManager) + processor := testrig.NewTestProcessor(&suite.state, federator, emailSender, webPushSender, mediaManager) testrig.StartWorkers(&suite.state, processor.Workers()) return processor diff --git a/internal/typeutils/internaltofrontend.go b/internal/typeutils/internaltofrontend.go index cdc250f98..6739d0540 100644 --- a/internal/typeutils/internaltofrontend.go +++ b/internal/typeutils/internaltofrontend.go @@ -616,6 +616,11 @@ func (c *Converter) AccountToAdminAPIAccount(ctx context.Context, a *gtsmodel.Ac } func (c *Converter) AppToAPIAppSensitive(ctx context.Context, a *gtsmodel.Application) (*apimodel.Application, error) { + vapidKeyPair, err := c.state.DB.GetVAPIDKeyPair(ctx) + if err != nil { + return nil, gtserror.Newf("error getting VAPID public key: %w", err) + } + return &apimodel.Application{ ID: a.ID, Name: a.Name, @@ -623,6 +628,7 @@ func (c *Converter) AppToAPIAppSensitive(ctx context.Context, a *gtsmodel.Applic RedirectURI: a.RedirectURI, ClientID: a.ClientID, ClientSecret: a.ClientSecret, + VapidKey: vapidKeyPair.Public, }, nil } @@ -1878,6 +1884,12 @@ func (c *Converter) InstanceToAPIV2Instance(ctx context.Context, i *gtsmodel.Ins instance.Configuration.Emojis.EmojiSizeLimit = int(config.GetMediaEmojiLocalMaxSize()) // #nosec G115 -- Already validated. instance.Configuration.OIDCEnabled = config.GetOIDCEnabled() + vapidKeyPair, err := c.state.DB.GetVAPIDKeyPair(ctx) + if err != nil { + return nil, gtserror.Newf("error getting VAPID public key: %w", err) + } + instance.Configuration.VAPID.PublicKey = vapidKeyPair.Public + // registrations instance.Registrations.Enabled = config.GetAccountsRegistrationOpen() instance.Registrations.ApprovalRequired = true // always required @@ -2985,3 +2997,36 @@ func (c *Converter) InteractionReqToAPIInteractionReq( URI: req.URI, }, nil } + +func (c *Converter) WebPushSubscriptionToAPIWebPushSubscription( + ctx context.Context, + subscription *gtsmodel.WebPushSubscription, +) (*apimodel.WebPushSubscription, error) { + vapidKeyPair, err := c.state.DB.GetVAPIDKeyPair(ctx) + if err != nil { + return nil, gtserror.Newf("error getting VAPID key pair: %w", err) + } + + return &apimodel.WebPushSubscription{ + ID: subscription.ID, + Endpoint: subscription.Endpoint, + ServerKey: vapidKeyPair.Public, + Alerts: apimodel.WebPushSubscriptionAlerts{ + Follow: subscription.NotificationFlags.Get(gtsmodel.NotificationFollow), + FollowRequest: subscription.NotificationFlags.Get(gtsmodel.NotificationFollowRequest), + Favourite: subscription.NotificationFlags.Get(gtsmodel.NotificationFavourite), + Mention: subscription.NotificationFlags.Get(gtsmodel.NotificationMention), + Reblog: subscription.NotificationFlags.Get(gtsmodel.NotificationReblog), + Poll: subscription.NotificationFlags.Get(gtsmodel.NotificationPoll), + Status: subscription.NotificationFlags.Get(gtsmodel.NotificationStatus), + Update: subscription.NotificationFlags.Get(gtsmodel.NotificationUpdate), + AdminSignup: subscription.NotificationFlags.Get(gtsmodel.NotificationAdminSignup), + AdminReport: subscription.NotificationFlags.Get(gtsmodel.NotificationAdminReport), + PendingFavourite: subscription.NotificationFlags.Get(gtsmodel.NotificationPendingFave), + PendingReply: subscription.NotificationFlags.Get(gtsmodel.NotificationPendingReply), + PendingReblog: subscription.NotificationFlags.Get(gtsmodel.NotificationPendingReblog), + }, + Policy: apimodel.WebPushNotificationPolicyAll, + Standard: true, + }, nil +} diff --git a/internal/typeutils/internaltofrontend_test.go b/internal/typeutils/internaltofrontend_test.go index 005abf4eb..1ca0840a5 100644 --- a/internal/typeutils/internaltofrontend_test.go +++ b/internal/typeutils/internaltofrontend_test.go @@ -21,6 +21,7 @@ import ( "bytes" "context" "encoding/json" + "strings" "testing" "github.com/stretchr/testify/suite" @@ -2061,6 +2062,13 @@ func (suite *InternalToFrontendTestSuite) TestInstanceV2ToFrontend() { b, err := json.MarshalIndent(instance, "", " ") suite.NoError(err) + // The VAPID public key changes from run to run. + vapidKeyPair, err := suite.db.GetVAPIDKeyPair(ctx) + if err != nil { + suite.FailNow(err.Error()) + } + s := strings.Replace(string(b), vapidKeyPair.Public, "VAPID_PUBLIC_KEY_PLACEHOLDER", 1) + suite.Equal(`{ "domain": "localhost:8080", "account_domain": "localhost:8080", @@ -2140,6 +2148,9 @@ func (suite *InternalToFrontendTestSuite) TestInstanceV2ToFrontend() { }, "emojis": { "emoji_size_limit": 51200 + }, + "vapid": { + "public_key": "VAPID_PUBLIC_KEY_PLACEHOLDER" } }, "registrations": { @@ -2184,7 +2195,7 @@ func (suite *InternalToFrontendTestSuite) TestInstanceV2ToFrontend() { "rules": [], "terms": "\u003cp\u003eThis is where a list of terms and conditions might go.\u003c/p\u003e\u003cp\u003eFor example:\u003c/p\u003e\u003cp\u003eIf you want to sign up on this instance, you oughta know that we:\u003c/p\u003e\u003col\u003e\u003cli\u003eWill sell your data to whoever offers.\u003c/li\u003e\u003cli\u003eSecure the server with password \u003ccode\u003epassword\u003c/code\u003e wherever possible.\u003c/li\u003e\u003c/ol\u003e", "terms_text": "This is where a list of terms and conditions might go.\n\nFor example:\n\nIf you want to sign up on this instance, you oughta know that we:\n\n1. Will sell your data to whoever offers.\n2. Secure the server with password `+"`"+`password`+"`"+` wherever possible." -}`, string(b)) +}`, s) } func (suite *InternalToFrontendTestSuite) TestEmojiToFrontend() { |