fix: persist app ID with session

DeepSourceCorp · Oct 16, 2023 · 899a55b · 899a55b
1 parent 968b0ce
commit 899a55b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/auth/oauth/handler.go b/auth/oauth/handler.go
@@ -41,7 +41,7 @@ func (h *Handler) HandleAuthorize(c echo.Context) error {
 		return err
 	}
 
-	return c.Redirect(302, url)
+	return c.Redirect(http.StatusTemporaryRedirect, url)
 }
 
 func (h *Handler) HandleCallback(c echo.Context) error {

diff --git a/auth/oauth/service.go b/auth/oauth/service.go
@@ -51,7 +51,7 @@ func (s *Service) CreateSession(ctx context.Context, req *contract.CallbackReque
 		return nil, httperror.ErrUnknown(err) // TODO: Handle upstream error types.
 	}
 
-	session, err := s.sessionService.CreateSession(token)
+	session, err := s.sessionService.CreateSession(req.AppID, token)
 	if err != nil {
 		return nil, err
 	}

diff --git a/auth/session/service.go b/auth/session/service.go
@@ -33,12 +33,14 @@ func NewService(runner *common.Runner, deepsource *common.DeepSource, sessionSto
 	}
 }
 
-func (s *Service) CreateSession(token interface{}) (*Session, error) {
+func (s *Service) CreateSession(appID string, token interface{}) (*Session, error) {
 	session := NewSession()
 	if err := session.SetBackendToken(token); err != nil {
 		return nil, fmt.Errorf("failed to set backend token, %w", err)
 	}
 
+	session.AppID = appID
+
 	if err := session.SetRunnerToken(s.Runner); err != nil {
 		return nil, fmt.Errorf("failed to set runner token, %w", err)
 	}