From e0420f4045facaed733b5d0685320db7f8f11c9f Mon Sep 17 00:00:00 2001
From: Derek McGowan <derek@mcgstyle.net>
Date: Fri, 4 Mar 2016 14:32:51 -0800
Subject: [PATCH] Add offline token option

Login needs to add an offline token flag to ensure a refresh token is returned by the token endpoint.

Signed-off-by: Derek McGowan <derek@mcgstyle.net> (github: dmcgowan)
---
 docs/client/auth/session.go | 31 +++++++++++++++++++------------
 1 file changed, 19 insertions(+), 12 deletions(-)

diff --git a/docs/client/auth/session.go b/docs/client/auth/session.go
index 35ccabf1..b2811351 100644
--- a/docs/client/auth/session.go
+++ b/docs/client/auth/session.go
@@ -116,9 +116,10 @@ type tokenHandler struct {
 	transport http.RoundTripper
 	clock     clock
 
-	forceOAuth bool
-	clientID   string
-	scopes     []Scope
+	offlineAccess bool
+	forceOAuth    bool
+	clientID      string
+	scopes        []Scope
 
 	tokenLock       sync.Mutex
 	tokenCache      string
@@ -149,9 +150,10 @@ type TokenHandlerOptions struct {
 	Transport   http.RoundTripper
 	Credentials CredentialStore
 
-	ForceOAuth bool
-	ClientID   string
-	Scopes     []Scope
+	OfflineAccess bool
+	ForceOAuth    bool
+	ClientID      string
+	Scopes        []Scope
 }
 
 // An implementation of clock for providing real time data.
@@ -180,12 +182,13 @@ func NewTokenHandler(transport http.RoundTripper, creds CredentialStore, scope s
 // options structure.
 func NewTokenHandlerWithOptions(options TokenHandlerOptions) AuthenticationHandler {
 	handler := &tokenHandler{
-		transport:  options.Transport,
-		creds:      options.Credentials,
-		forceOAuth: options.ForceOAuth,
-		clientID:   options.ClientID,
-		scopes:     options.Scopes,
-		clock:      realClock{},
+		transport:     options.Transport,
+		creds:         options.Credentials,
+		offlineAccess: options.OfflineAccess,
+		forceOAuth:    options.ForceOAuth,
+		clientID:      options.ClientID,
+		scopes:        options.Scopes,
+		clock:         realClock{},
 	}
 
 	return handler
@@ -346,6 +349,10 @@ func (th *tokenHandler) fetchTokenWithBasicAuth(realm *url.URL, service string,
 		reqParams.Add("scope", scope)
 	}
 
+	if th.offlineAccess {
+		reqParams.Add("offline_token", "true")
+	}
+
 	if th.creds != nil {
 		username, password := th.creds.Basic(realm)
 		if username != "" && password != "" {