feat: added hook for intercepting api calls made to the core

supertokens · Nov 2, 2023 · 962f684 · 962f684
1 parent 583b953
commit 962f684
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/supertokens/models.go b/supertokens/models.go
@@ -51,8 +51,9 @@ type TypeInput struct {
 }
 
 type ConnectionInfo struct {
-	ConnectionURI string
-	APIKey        string
+	ConnectionURI      string
+	APIKey             string
+	NetworkInterceptor func(*http.Request, UserContext) *http.Request
 }
 
 type APIHandled struct {

diff --git a/supertokens/querier.go b/supertokens/querier.go
@@ -44,6 +44,7 @@ var (
 	querierLastTriedIndex int
 	querierLock           sync.Mutex
 	querierHostLock       sync.Mutex
+	querierInterceptor    func(*http.Request, UserContext) *http.Request
 )
 
 func SetQuerierApiVersionForTests(version string) {
@@ -100,7 +101,7 @@ func GetNewQuerierInstanceOrThrowError(rIDToCore string) (*Querier, error) {
 	return &Querier{RIDToCore: rIDToCore}, nil
 }
 
-func initQuerier(hosts []QuerierHost, APIKey string) {
+func initQuerier(hosts []QuerierHost, APIKey string, interceptor func(*http.Request, UserContext) *http.Request) {
 	if !querierInitCalled {
 		querierInitCalled = true
 		QuerierHosts = hosts
@@ -109,6 +110,7 @@ func initQuerier(hosts []QuerierHost, APIKey string) {
 		}
 		querierAPIVersion = ""
 		querierLastTriedIndex = 0
+		querierInterceptor = interceptor
 	}
 }
 
@@ -129,6 +131,7 @@ func (q *Querier) SendPostRequest(path string, data map[string]interface{}) (map
 		if err != nil {
 			return nil, err
 		}
+		req = querierInterceptor(req, nil)
 
 		apiVerion, querierAPIVersionError := q.GetQuerierAPIVersion()
 		if querierAPIVersionError != nil {

diff --git a/supertokens/supertokens.go b/supertokens/supertokens.go
@@ -85,7 +85,7 @@ func supertokensInit(config TypeInput) error {
 					BasePath: basePath,
 				})
 			}
-			initQuerier(hosts, config.Supertokens.APIKey)
+			initQuerier(hosts, config.Supertokens.APIKey, config.Supertokens.NetworkInterceptor)
 			superTokens.SuperTokens = *config.Supertokens
 		} else {
 			return errors.New("please provide 'ConnectionURI' value. If you do not want to provide a connection URI, then set config.Supertokens to nil")