Server: Have separate implementations for refreshing endpoints

1 files changed, 22 insertions, 0 deletions

diff --git a/internal/server/institute/institute.go b/internal/server/institute/institute.go
index e0a52b7..46977ac 100644
--- a/internal/server/institute/institute.go
+++ b/internal/server/institute/institute.go
@@ -3,6 +3,7 @@ package institute
 import (
 	"context"
 
+	"github.com/eduvpn/eduvpn-common/internal/discovery"
 	"github.com/eduvpn/eduvpn-common/internal/oauth"
 	"github.com/eduvpn/eduvpn-common/internal/server/api"
 	"github.com/eduvpn/eduvpn-common/internal/server/base"
@@ -98,6 +99,27 @@ func (s *Server) NeedsLocation() bool {
 	return false
 }
 
+func (s *Server) RefreshEndpoints(ctx context.Context, _ *discovery.Discovery) error {
+	// Re-initialize the endpoints
+	b, err := s.Base()
+	if err != nil {
+		return err
+	}
+
+	err = api.Endpoints(ctx, b)
+	if err != nil {
+		return err
+	}
+
+	// update OAuth
+	auth := s.OAuth()
+	if auth != nil {
+		auth.BaseAuthorizationURL = b.Endpoints.API.V3.Authorization
+		auth.TokenURL = b.Endpoints.API.V3.Token
+	}
+	return nil
+}
+
 func (s *Server) Public() (interface{}, error) {
 	return &server.Server{
 		DisplayName: s.Basic.DisplayName,