summaryrefslogtreecommitdiff
path: root/internal/webpush
diff options
context:
space:
mode:
authorLibravatar kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>2025-01-27 15:54:59 +0000
committerLibravatar GitHub <noreply@github.com>2025-01-27 15:54:59 +0000
commit726d2ba483b58402fa5e8d5a99d5cbd290bdeb67 (patch)
treeec597aa14d4c5cd8f36661750ff71a3ffbc2eb2d /internal/webpush
parentbumps uptrace/bun deps to v1.2.8 (#3698) (diff)
downloadgotosocial-726d2ba483b58402fa5e8d5a99d5cbd290bdeb67.tar.xz
[chore] some tidy ups (#3677)
* small formatting changes (no logic) * improve code comments * fix import cycle * shutup stinky linter
Diffstat (limited to 'internal/webpush')
-rw-r--r--internal/webpush/realsender.go24
-rw-r--r--internal/webpush/realsender_test.go8
-rw-r--r--internal/webpush/sender.go34
3 files changed, 33 insertions, 33 deletions
diff --git a/internal/webpush/realsender.go b/internal/webpush/realsender.go
index 8b3a1bd66..4c4657957 100644
--- a/internal/webpush/realsender.go
+++ b/internal/webpush/realsender.go
@@ -33,30 +33,20 @@ import (
"github.com/superseriousbusiness/gotosocial/internal/filter/usermute"
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
- "github.com/superseriousbusiness/gotosocial/internal/httpclient"
"github.com/superseriousbusiness/gotosocial/internal/log"
"github.com/superseriousbusiness/gotosocial/internal/state"
"github.com/superseriousbusiness/gotosocial/internal/text"
"github.com/superseriousbusiness/gotosocial/internal/typeutils"
)
-// realSender is the production Web Push sender, backed by an HTTP client, DB, and worker pool.
+// realSender is the production Web Push sender,
+// backed by an HTTP client, DB, and worker pool.
type realSender struct {
httpClient *http.Client
state *state.State
converter *typeutils.Converter
}
-// NewRealSender creates a Sender from an http.Client instead of an httpclient.Client.
-// This should only be used by NewSender and in tests.
-func NewRealSender(httpClient *http.Client, state *state.State, converter *typeutils.Converter) Sender {
- return &realSender{
- httpClient: httpClient,
- state: state,
- converter: converter,
- }
-}
-
func (r *realSender) Send(
ctx context.Context,
notification *gtsmodel.Notification,
@@ -329,13 +319,3 @@ func formatNotificationBody(apiNotification *apimodel.Notification) string {
func firstNBytesTrimSpace(s string, n int) string {
return strings.TrimSpace(text.FirstNBytesByWords(strings.TrimSpace(s), n))
}
-
-// gtsHTTPClientRoundTripper helps wrap a GtS HTTP client back into a regular HTTP client,
-// so that webpush-go can use our IP filters, bad hosts list, and retries.
-type gtsHTTPClientRoundTripper struct {
- httpClient *httpclient.Client
-}
-
-func (r *gtsHTTPClientRoundTripper) RoundTrip(request *http.Request) (*http.Response, error) {
- return r.httpClient.Do(request)
-}
diff --git a/internal/webpush/realsender_test.go b/internal/webpush/realsender_test.go
index c94bbbb8e..8446fc47d 100644
--- a/internal/webpush/realsender_test.go
+++ b/internal/webpush/realsender_test.go
@@ -24,6 +24,9 @@ import (
"testing"
"time"
+ // for go:linkname
+ _ "unsafe"
+
"github.com/stretchr/testify/suite"
"github.com/superseriousbusiness/gotosocial/internal/cleaner"
"github.com/superseriousbusiness/gotosocial/internal/db"
@@ -120,7 +123,7 @@ func (suite *RealSenderStandardTestSuite) SetupTest() {
suite.oauthServer = testrig.NewTestOauthServer(suite.db)
suite.emailSender = testrig.NewEmailSender("../../web/template/", nil)
- suite.webPushSender = webpush.NewRealSender(
+ suite.webPushSender = newSenderWith(
&http.Client{
Transport: suite,
},
@@ -261,3 +264,6 @@ func (suite *RealSenderStandardTestSuite) TestServerError() {
func TestRealSenderStandardTestSuite(t *testing.T) {
suite.Run(t, &RealSenderStandardTestSuite{})
}
+
+//go:linkname newSenderWith github.com/superseriousbusiness/gotosocial/internal/webpush.newSenderWith
+func newSenderWith(*http.Client, *state.State, *typeutils.Converter) webpush.Sender
diff --git a/internal/webpush/sender.go b/internal/webpush/sender.go
index 5331f049a..060118019 100644
--- a/internal/webpush/sender.go
+++ b/internal/webpush/sender.go
@@ -30,7 +30,9 @@ import (
// Sender can send Web Push notifications.
type Sender interface {
- // Send queues up a notification for delivery to all of an account's Web Push subscriptions.
+
+ // Send queues up a notification for delivery to
+ // all of an account's Web Push subscriptions.
Send(
ctx context.Context,
notification *gtsmodel.Notification,
@@ -41,14 +43,26 @@ type Sender interface {
// NewSender creates a new sender from an HTTP client, DB, and worker pool.
func NewSender(httpClient *httpclient.Client, state *state.State, converter *typeutils.Converter) Sender {
- return NewRealSender(
- &http.Client{
- Transport: &gtsHTTPClientRoundTripper{
- httpClient: httpClient,
- },
- // Other fields are already set on the http.Client inside the httpclient.Client.
+ return &realSender{
+ httpClient: &http.Client{
+ // Pass in our wrapped httpclient.Client{}
+ // type as http.Transport{} in order to take
+ // advantage of retries, SSF protection etc.
+ Transport: httpClient,
+
+ // Other http.Client{} fields are already
+ // set in embedded httpclient.Client{}.
},
- state,
- converter,
- )
+ state: state,
+ converter: converter,
+ }
+}
+
+// an internal function purely existing for the webpush test package to link to and use a custom http.Client{}.
+func newSenderWith(client *http.Client, state *state.State, converter *typeutils.Converter) Sender { //nolint:unused
+ return &realSender{
+ httpClient: client,
+ state: state,
+ converter: converter,
+ }
}