mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
a70a0cd41e
For my own use case (and anyone in a similar situation), the Jira module is more useful if I can specify a list of projects to display issues from, rather than no project (which selects all projects) or one specific project. New supported syntax: ` project: ["PROJA", "PROJB"]` If this is merged, documentation for the Jira module should be updated accordingly. (Sorry, I would have done this myself but I'm not sure what the right place is - when grepping I find multiple places in the repo containing documentation strings and I'm not sure what the "master" location is) Specifying a single project (or no project) is still supported - behavior shouldn't change for anyone who doesn't change their config.
107 lines
2.1 KiB
Go
107 lines
2.1 KiB
Go
package jira
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func IssuesFor(username string, projects []string, jql string) (*SearchResult, error) {
|
|
query := []string{}
|
|
|
|
var projQuery = getProjectQuery(projects)
|
|
if projQuery != "" {
|
|
query = append(query, projQuery)
|
|
}
|
|
|
|
if username != "" {
|
|
query = append(query, buildJql("assignee", username))
|
|
}
|
|
|
|
if jql != "" {
|
|
query = append(query, jql)
|
|
}
|
|
|
|
v := url.Values{}
|
|
|
|
v.Set("jql", strings.Join(query, " AND "))
|
|
|
|
url := fmt.Sprintf("/rest/api/2/search?%s", v.Encode())
|
|
|
|
resp, err := jiraRequest(url)
|
|
if err != nil {
|
|
return &SearchResult{}, err
|
|
}
|
|
|
|
searchResult := &SearchResult{}
|
|
parseJson(searchResult, resp.Body)
|
|
|
|
return searchResult, nil
|
|
}
|
|
|
|
func buildJql(key string, value string) string {
|
|
return fmt.Sprintf("%s = \"%s\"", key, value)
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func jiraRequest(path string) (*http.Response, error) {
|
|
url := fmt.Sprintf("%s%s", Config.UString("wtf.mods.jira.domain"), path)
|
|
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.SetBasicAuth(Config.UString("wtf.mods.jira.email"), os.Getenv("WTF_JIRA_API_KEY"))
|
|
|
|
httpClient := &http.Client{}
|
|
resp, err := httpClient.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if resp.StatusCode < 200 || resp.StatusCode > 299 {
|
|
return nil, fmt.Errorf(resp.Status)
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
func parseJson(obj interface{}, text io.Reader) {
|
|
jsonStream, err := ioutil.ReadAll(text)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
decoder := json.NewDecoder(bytes.NewReader(jsonStream))
|
|
|
|
for {
|
|
if err := decoder.Decode(obj); err == io.EOF {
|
|
break
|
|
} else if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func getProjectQuery(projects []string) string {
|
|
singleEmptyProject := len(projects) == 1 && len(projects[0]) == 0
|
|
if len(projects) == 0 || singleEmptyProject {
|
|
return ""
|
|
} else if len(projects) == 1 {
|
|
return buildJql("project", projects[0])
|
|
}
|
|
|
|
quoted := make([]string, len(projects))
|
|
for i := range projects {
|
|
quoted[i] = fmt.Sprintf("\"%s\"", projects[i])
|
|
}
|
|
return fmt.Sprintf("project in (%s)", strings.Join(quoted, ", "))
|
|
}
|