-
-
Save vjeantet/6d861c060aa37346823dd1187af1c9ca to your computer and use it in GitHub Desktop.
Example of using OAuth authentication with JIRA in Go
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"crypto/x509" | |
"encoding/json" | |
"encoding/pem" | |
"fmt" | |
"net/http" | |
"net/url" | |
"os" | |
"os/user" | |
"path/filepath" | |
"strings" | |
"github.com/andygrunwald/go-jira" | |
"github.com/dghubble/oauth1" | |
"golang.org/x/net/context" | |
"gopkg.in/alecthomas/kingpin.v2" | |
) | |
var ( | |
jiraURL = kingpin.Flag("jira-url", "JIRA instance to use").Default("<your jira here>").URL() | |
) | |
/* | |
$ openssl genrsa -out jira.pem 1024 | |
$ openssl rsa -in jira.pem -pubout -out jira.pub | |
*/ | |
const jiraPrivateKey = `-----BEGIN RSA PRIVATE KEY----- | |
<your private key here> | |
-----END RSA PRIVATE KEY-----` | |
const jiraConsumerKey = '<your consumer key>' | |
func getJIRAHTTPClient(ctx context.Context, config *oauth1.Config) *http.Client { | |
cacheFile, err := jiraTokenCacheFile() | |
if err != nil { | |
log.Fatalf("Unable to get path to cached credential file. %v", err) | |
} | |
tok, err := jiraTokenFromFile(cacheFile) | |
if err != nil { | |
tok = getJIRATokenFromWeb(config) | |
saveJIRAToken(cacheFile, tok) | |
} | |
return config.Client(ctx, tok) | |
} | |
func getJIRATokenFromWeb(config *oauth1.Config) *oauth1.Token { | |
requestToken, requestSecret, err := config.RequestToken() | |
if err != nil { | |
log.Fatalf("Unable to get request token. %v", err) | |
} | |
authorizationURL, err := config.AuthorizationURL(requestToken) | |
if err != nil { | |
log.Fatalf("Unable to get authorization url. %v", err) | |
} | |
fmt.Printf("Go to the following link in your browser then type the "+ | |
"authorization code: \n%v\n", authorizationURL.String()) | |
var code string | |
if _, err := fmt.Scan(&code); err != nil { | |
log.Fatalf("Unable to read authorization code. %v", err) | |
} | |
accessToken, accessSecret, err := config.AccessToken(requestToken, requestSecret, code) | |
if err != nil { | |
log.Fatalf("Unable to get access token. %v", err) | |
} | |
return oauth1.NewToken(accessToken, accessSecret) | |
} | |
func jiraTokenCacheFile() (string, error) { | |
usr, err := user.Current() | |
if err != nil { | |
return "", err | |
} | |
tokenCacheDir := filepath.Join(usr.HomeDir, ".credentials") | |
os.MkdirAll(tokenCacheDir, 0700) | |
return filepath.Join(tokenCacheDir, | |
url.QueryEscape((*jiraURL).Host+".json")), err | |
} | |
func jiraTokenFromFile(file string) (*oauth1.Token, error) { | |
f, err := os.Open(file) | |
if err != nil { | |
return nil, err | |
} | |
t := &oauth1.Token{} | |
err = json.NewDecoder(f).Decode(t) | |
defer f.Close() | |
return t, err | |
} | |
func saveJIRAToken(file string, token *oauth1.Token) { | |
fmt.Printf("Saving credential file to: %s\n", file) | |
f, err := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600) | |
if err != nil { | |
log.Fatalf("Unable to cache oauth token: %v", err) | |
} | |
defer f.Close() | |
json.NewEncoder(f).Encode(token) | |
} | |
func getJIRAClient() *jira.Client { | |
ctx := context.Background() | |
keyDERBlock, _ := pem.Decode([]byte(jiraPrivateKey)) | |
if keyDERBlock == nil { | |
log.Fatal("unable to decode key PEM block") | |
} | |
if !(keyDERBlock.Type == "PRIVATE KEY" || strings.HasSuffix(keyDERBlock.Type, " PRIVATE KEY")) { | |
log.Fatalf("unexpected key DER block type: %s", keyDERBlock.Type) | |
} | |
privateKey, err := x509.ParsePKCS1PrivateKey(keyDERBlock.Bytes) | |
if err != nil { | |
log.Fatalf("unable to parse PKCS1 private key. %v", err) | |
} | |
config := oauth1.Config{ | |
ConsumerKey: jiraConsumerKey, | |
CallbackURL: "oob", /* for command line usage */ | |
Endpoint: oauth1.Endpoint{ | |
RequestTokenURL: (*jiraURL).String() + "plugins/servlet/oauth/request-token", | |
AuthorizeURL: (*jiraURL).String() + "plugins/servlet/oauth/authorize", | |
AccessTokenURL: (*jiraURL).String() + "plugins/servlet/oauth/access-token", | |
}, | |
Signer: &oauth1.RSASigner{ | |
PrivateKey: privateKey, | |
}, | |
} | |
jiraClient, err := jira.NewClient(getJIRAHTTPClient(ctx, &config), (*jiraURL).String()) | |
if err != nil { | |
log.Fatalf("unable to create new JIRA client. %v", err) | |
} | |
return jiraClient | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment