Add Context.SetAuthToken

mislav · mislav · commit 2aa77fb8ea8b · 2019-10-18T18:47:42.000+02:00
diff --git a/context/blank_context.go b/context/blank_context.go
@@ -26,6 +26,10 @@ func (c *blankContext) AuthToken() (string, error) {
 	return c.authToken, nil
 }
 
+func (c *blankContext) SetAuthToken(t string) {
+	c.authToken = t
+}
+
 func (c *blankContext) AuthLogin() (string, error) {
 	return c.authLogin, nil
 }
diff --git a/context/context.go b/context/context.go
@@ -10,6 +10,7 @@ import (
 // Context represents the interface for querying information about the current environment
 type Context interface {
 	AuthToken() (string, error)
+	SetAuthToken(string)
 	AuthLogin() (string, error)
 	Branch() (string, error)
 	SetBranch(string)
@@ -36,10 +37,11 @@ func InitDefaultContext() Context {
 
 // A Context implementation that queries the filesystem
 type fsContext struct {
-	config   *configEntry
-	remotes  Remotes
-	branch   string
-	baseRepo *GitHubRepository
+	config    *configEntry
+	remotes   Remotes
+	branch    string
+	baseRepo  *GitHubRepository
+	authToken string
 }
 
 func (c *fsContext) configFile() string {
@@ -54,18 +56,27 @@ func (c *fsContext) getConfig() (*configEntry, error) {
 			return nil, err
 		}
 		c.config = entry
+		c.authToken = ""
 	}
 	return c.config, nil
 }
 
 func (c *fsContext) AuthToken() (string, error) {
+	if c.authToken != "" {
+		return c.authToken, nil
+	}
+
 	config, err := c.getConfig()
 	if err != nil {
 		return "", err
 	}
 	return config.Token, nil
 }
 
+func (c *fsContext) SetAuthToken(t string) {
+	c.authToken = t
+}
+
 func (c *fsContext) AuthLogin() (string, error) {
 	config, err := c.getConfig()
 	if err != nil {

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,10 @@ func (c *blankContext) AuthToken() (string, error) {`
`26`	`26`	`return c.authToken, nil`
`27`	`27`	`}`
`28`	`28`
	`29`	`+func (c *blankContext) SetAuthToken(t string) {`
	`30`	`+ c.authToken = t`
	`31`	`+}`
	`32`	`+`
`29`	`33`	`func (c *blankContext) AuthLogin() (string, error) {`
`30`	`34`	`return c.authLogin, nil`
`31`	`35`	`}`