|
|
@@ -0,0 +1,152 @@
|
|
|
+package gitinstallation
|
|
|
+
|
|
|
+import (
|
|
|
+ "fmt"
|
|
|
+ "net/http"
|
|
|
+
|
|
|
+ "github.com/porter-dev/porter/api/server/handlers"
|
|
|
+ "github.com/porter-dev/porter/api/server/shared"
|
|
|
+ "github.com/porter-dev/porter/api/server/shared/apierrors"
|
|
|
+ "github.com/porter-dev/porter/api/server/shared/config"
|
|
|
+ "github.com/porter-dev/porter/api/types"
|
|
|
+ "github.com/porter-dev/porter/internal/models"
|
|
|
+ "github.com/porter-dev/porter/internal/models/integrations"
|
|
|
+ "golang.org/x/oauth2"
|
|
|
+)
|
|
|
+
|
|
|
+type GithubAppOAuthCallbackHandler struct {
|
|
|
+ handlers.PorterHandlerReadWriter
|
|
|
+}
|
|
|
+
|
|
|
+func NewGithubAppOAuthCallbackHandler(
|
|
|
+ config *config.Config,
|
|
|
+ decoderValidator shared.RequestDecoderValidator,
|
|
|
+ writer shared.ResultWriter,
|
|
|
+) *GithubAppOAuthCallbackHandler {
|
|
|
+ return &GithubAppOAuthCallbackHandler{
|
|
|
+ PorterHandlerReadWriter: handlers.NewDefaultPorterHandler(config, decoderValidator, writer),
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (c *GithubAppOAuthCallbackHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
|
+ user, _ := r.Context().Value(types.UserScope).(*models.User)
|
|
|
+
|
|
|
+ session, err := c.Config().Store.Get(r, c.Config().ServerConf.CookieName)
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ token, err := c.Config().GithubAppConf.Exchange(oauth2.NoContext, r.URL.Query().Get("code"))
|
|
|
+
|
|
|
+ if err != nil || !token.Valid() {
|
|
|
+ if session.Values["query_params"] != "" {
|
|
|
+ http.Redirect(w, r, fmt.Sprintf("/dashboard?%s", session.Values["query_params"]), 302)
|
|
|
+ } else {
|
|
|
+ http.Redirect(w, r, "/dashboard", 302)
|
|
|
+ }
|
|
|
+
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ oauthInt := &integrations.GithubAppOAuthIntegration{
|
|
|
+ SharedOAuthModel: integrations.SharedOAuthModel{
|
|
|
+ AccessToken: []byte(token.AccessToken),
|
|
|
+ RefreshToken: []byte(token.RefreshToken),
|
|
|
+ Expiry: token.Expiry,
|
|
|
+ },
|
|
|
+ UserID: user.ID,
|
|
|
+ }
|
|
|
+
|
|
|
+ oauthInt, err = c.Repo().GithubAppOAuthIntegration().CreateGithubAppOAuthIntegration(oauthInt)
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ user.GithubAppIntegrationID = oauthInt.ID
|
|
|
+
|
|
|
+ user, err = c.Repo().User().UpdateUser(user)
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if session.Values["query_params"] != "" {
|
|
|
+ http.Redirect(w, r, fmt.Sprintf("/dashboard?%s", session.Values["query_params"]), 302)
|
|
|
+ } else {
|
|
|
+ http.Redirect(w, r, "/dashboard", 302)
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// func (app *App) HandleGithubAppOAuthCallback(w http.ResponseWriter, r *http.Request) {
|
|
|
+// session, err := app.Store.Get(r, app.ServerConf.CookieName)
|
|
|
+
|
|
|
+// if err != nil {
|
|
|
+// app.handleErrorDataRead(err, w)
|
|
|
+// return
|
|
|
+// }
|
|
|
+
|
|
|
+// token, err := app.GithubAppConf.Exchange(oauth2.NoContext, r.URL.Query().Get("code"))
|
|
|
+
|
|
|
+// if err != nil || !token.Valid() {
|
|
|
+// if session.Values["query_params"] != "" {
|
|
|
+// http.Redirect(w, r, fmt.Sprintf("/dashboard?%s", session.Values["query_params"]), 302)
|
|
|
+// } else {
|
|
|
+// http.Redirect(w, r, "/dashboard", 302)
|
|
|
+// }
|
|
|
+// return
|
|
|
+// }
|
|
|
+
|
|
|
+// fmt.Println("exchange happaned")
|
|
|
+// fmt.Println(token.AccessToken)
|
|
|
+// fmt.Println(token.RefreshToken)
|
|
|
+
|
|
|
+// userID, err := app.getUserIDFromRequest(r)
|
|
|
+
|
|
|
+// if err != nil {
|
|
|
+// app.handleErrorInternal(err, w)
|
|
|
+// return
|
|
|
+// }
|
|
|
+
|
|
|
+// user, err := app.Repo.User().ReadUser(userID)
|
|
|
+
|
|
|
+// if err != nil {
|
|
|
+// app.handleErrorInternal(err, w)
|
|
|
+// return
|
|
|
+// }
|
|
|
+
|
|
|
+// oauthInt := &integrations.GithubAppOAuthIntegration{
|
|
|
+// SharedOAuthModel: integrations.SharedOAuthModel{
|
|
|
+// AccessToken: []byte(token.AccessToken),
|
|
|
+// RefreshToken: []byte(token.RefreshToken),
|
|
|
+// Expiry: token.Expiry,
|
|
|
+// },
|
|
|
+// UserID: user.ID,
|
|
|
+// }
|
|
|
+
|
|
|
+// oauthInt, err = app.Repo.GithubAppOAuthIntegration().CreateGithubAppOAuthIntegration(oauthInt)
|
|
|
+
|
|
|
+// if err != nil {
|
|
|
+// app.handleErrorInternal(err, w)
|
|
|
+// return
|
|
|
+// }
|
|
|
+
|
|
|
+// user.GithubAppIntegrationID = oauthInt.ID
|
|
|
+
|
|
|
+// user, err = app.Repo.User().UpdateUser(user)
|
|
|
+
|
|
|
+// if err != nil {
|
|
|
+// app.handleErrorInternal(err, w)
|
|
|
+// return
|
|
|
+// }
|
|
|
+
|
|
|
+// if session.Values["query_params"] != "" {
|
|
|
+// http.Redirect(w, r, fmt.Sprintf("/dashboard?%s", session.Values["query_params"]), 302)
|
|
|
+// } else {
|
|
|
+// http.Redirect(w, r, "/dashboard", 302)
|
|
|
+// }
|
|
|
+// }
|