aboutsummaryrefslogtreecommitdiffstats
path: root/bridge/github/export.go
diff options
context:
space:
mode:
Diffstat (limited to 'bridge/github/export.go')
-rw-r--r--bridge/github/export.go36
1 files changed, 33 insertions, 3 deletions
diff --git a/bridge/github/export.go b/bridge/github/export.go
index 658ef484..71790031 100644
--- a/bridge/github/export.go
+++ b/bridge/github/export.go
@@ -22,6 +22,9 @@ import (
type githubExporter struct {
conf core.Configuration
+ // export only bugs taged with one of these origins
+ onlyOrigins []string
+
// cache identities clients
identityClient map[string]*githubv4.Client
@@ -41,8 +44,8 @@ type githubExporter struct {
// Init .
func (ge *githubExporter) Init(conf core.Configuration) error {
- //TODO: initialize with multiple tokens
ge.conf = conf
+ //TODO: initialize with multiple tokens
ge.identityToken = make(map[string]string)
ge.identityClient = make(map[string]*githubv4.Client)
ge.cachedIDs = make(map[string]string)
@@ -50,18 +53,38 @@ func (ge *githubExporter) Init(conf core.Configuration) error {
return nil
}
+func (ge *githubExporter) allowOrigin(origin string) bool {
+ if ge.onlyOrigins == nil {
+ return true
+ }
+
+ for _, o := range ge.onlyOrigins {
+ if origin == o {
+ return true
+ }
+ }
+
+ return false
+}
+
func (ge *githubExporter) getIdentityClient(id string) (*githubv4.Client, error) {
client, ok := ge.identityClient[id]
if ok {
return client, nil
}
+ // get token
token, ok := ge.identityToken[id]
if !ok {
return nil, fmt.Errorf("unknown identity token")
}
- return buildClient(token), nil
+ // create client
+ client = buildClient(token)
+ // cache client
+ ge.identityClient[id] = client
+
+ return client, nil
}
// ExportAll export all event made by the current user to Github
@@ -99,8 +122,8 @@ bugLoop:
continue
}
- // if identity participated in a bug
for _, p := range snapshot.Participants {
+ // if we have a token for one of the participants
for userId := range ge.identityToken {
if p.Id() == userId {
// try to export the bug and it associated events
@@ -133,6 +156,13 @@ func (ge *githubExporter) exportBug(b *cache.BugCache, since time.Time) error {
createOp := snapshot.Operations[0].(*bug.CreateOperation)
author := createOp.GetAuthor()
+ // skip bug if origin is not allowed
+ origin, ok := createOp.GetMetadata(keyOrigin)
+ if ok && !ge.allowOrigin(origin) {
+ // TODO print a warn ?
+ return nil
+ }
+
// get github bug ID
githubID, ok := createOp.GetMetadata(keyGithubId)
if ok {