feat: setup UI endpoint to serve the UI

closes trustbloc#39

Signed-off-by: talwinder.kaur <[email protected]>

pkg/restapi/operation/operations.go

@@ -13,6 +13,8 @@ import (
 	"errors"
 	"fmt"
 	"net/http"
+	"net/url"
+	"sync"
 
 	"github.com/coreos/go-oidc"
 	"github.com/google/uuid"
@@ -27,6 +29,7 @@ import (
 const (
 	oauth2GetRequestPath = "/oauth2/request"
 	oauth2CallbackPath   = "/oauth2/callback"
+
 	// api path params
 	scopeQueryParam = "scope"
 
@@ -108,15 +111,18 @@ type oauth2Token interface {
 
 // Operation defines handlers.
 type Operation struct {
-	client           httpClient
-	requestTokens    map[string]string
-	transientStore   storage.Store
-	oidcProvider     oidcProvider
-	oidcClientID     string
-	oidcClientSecret string
-	oidcCallbackURL  string
-	oauth2ConfigFunc func(...string) oauth2Config
-	bootstrapStore   storage.Store
+	client             httpClient
+	requestTokens      map[string]string
+	transientStore     storage.Store
+	oidcProvider       oidcProvider
+	oidcClientID       string
+	oidcClientSecret   string
+	oidcCallbackURL    string
+	uiEndpoint         string
+	userProfileRequest map[string]*user.Profile
+	presDefsLock       sync.Mutex
+	oauth2ConfigFunc   func(...string) oauth2Config
+	bootstrapStore     storage.Store
 }
 
 // Config defines configuration for rp operations.
@@ -127,6 +133,7 @@ type Config struct {
 	OIDCClientID           string
 	OIDCClientSecret       string
 	OIDCCallbackURL        string
+	UIEndpoint             string
 	TransientStoreProvider storage.Provider
 	StoreProvider          storage.Provider
 }
@@ -135,14 +142,16 @@ type createOIDCRequestResponse struct {
 	Request string `json:"request"`
 }
 
-// New returns rp operation instance.
+// New returns hub-auth operation instance.
 func New(config *Config) (*Operation, error) {
 	svc := &Operation{
-		client:           &http.Client{Transport: &http.Transport{TLSClientConfig: config.TLSConfig}},
-		requestTokens:    config.RequestTokens,
-		oidcClientID:     config.OIDCClientID,
-		oidcClientSecret: config.OIDCClientSecret,
-		oidcCallbackURL:  config.OIDCCallbackURL,
+		client:             &http.Client{Transport: &http.Transport{TLSClientConfig: config.TLSConfig}},
+		requestTokens:      config.RequestTokens,
+		oidcClientID:       config.OIDCClientID,
+		oidcClientSecret:   config.OIDCClientSecret,
+		oidcCallbackURL:    config.OIDCCallbackURL,
+		userProfileRequest: make(map[string]*user.Profile),
+		uiEndpoint:         config.UIEndpoint,
 	}
 	// TODO remove this empty string check. This is just here temporarily to allow for testing of other parts
 	// while the OIDC stuff is in development
@@ -325,7 +334,7 @@ func (c *Operation) handleOIDCCallback(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	handleAuthResult(w, r, userProfile)
+	c.handleAuthResult(w, r, userProfile)
 }
 
 // TODO onboard user at key server and SDS: https://github.com/trustbloc/hub-auth/issues/38
@@ -342,9 +351,15 @@ func (c *Operation) onboardUser(id string) (*user.Profile, error) {
 	return userProfile, nil
 }
 
-// TODO redirect to the UI: https://github.com/trustbloc/hub-auth/issues/39
-func handleAuthResult(w http.ResponseWriter, r *http.Request, _ *user.Profile) {
-	http.Redirect(w, r, "", http.StatusFound)
+func (c *Operation) handleAuthResult(w http.ResponseWriter, r *http.Request, userProfile *user.Profile) {
+	handle := url.QueryEscape(uuid.New().String())
+
+	c.setUserProfile(handle, userProfile)
+
+	redirectURL := fmt.Sprintf("%s?up=%s", c.uiEndpoint, handle)
+
+	http.Redirect(w, r, redirectURL, http.StatusFound)
+	logger.Debugf("redirected to: %s", redirectURL)
 }
 
 func handleAuthError(w http.ResponseWriter, status int, msg string) {
@@ -359,6 +374,13 @@ func handleAuthError(w http.ResponseWriter, status int, msg string) {
 	}
 }
 
+func (c *Operation) setUserProfile(handle string, r *user.Profile) {
+	c.presDefsLock.Lock()
+	defer c.presDefsLock.Unlock()
+
+	c.userProfileRequest[handle] = r
+}
+
 // writeResponse writes interface value to response.
 func (c *Operation) writeErrorResponse(rw http.ResponseWriter, status int, msg string) {
 	logger.Errorf(msg)