OAuth: Add a Cancel method

1 files changed, 13 insertions, 0 deletions

diff --git a/internal/oauth.go b/internal/oauth.go
index 16edd1e..a49b492 100644
--- a/internal/oauth.go
+++ b/internal/oauth.go
@@ -271,6 +271,7 @@ func (oauth *OAuth) Finish() error {
 		return errors.New("invalid state to finish oauth")
 	}
 	tokenErr := oauth.getTokensWithCallback()
+
 	if tokenErr != nil {
 		return tokenErr
 	}
@@ -278,6 +279,11 @@ func (oauth *OAuth) Finish() error {
 	return nil
 }
 
+func (oauth *OAuth) Cancel() {
+	oauth.Session.CallbackError = &OAuthCancelledCallbackError{}
+	oauth.Session.Server.Shutdown(oauth.Session.Context)
+}
+
 func (oauth *OAuth) Login(name string, authorizationURL string, tokenURL string) error {
 	authInitializeErr := oauth.start(name, authorizationURL, tokenURL)
 
@@ -320,6 +326,13 @@ func (oauth *OAuth) NeedsRelogin() bool {
 	return true
 }
 
+type OAuthCancelledCallbackError struct {
+}
+
+func (e *OAuthCancelledCallbackError) Error() string {
+	return fmt.Sprintf("Client cancelled OAuth")
+}
+
 type OAuthGenStateUnableError struct {
 	Err error
 }