mirror of https://github.com/matrix-org/go-neb.git
Browse Source
Merge pull request #52 from matrix-org/kegan/split-gh-service
Merge pull request #52 from matrix-org/kegan/split-gh-service
Split up github service into 2 serviceskegan/configure-client-name
Kegsay
8 years ago
committed by
GitHub
3 changed files with 350 additions and 270 deletions
-
5src/github.com/matrix-org/go-neb/api.go
-
276src/github.com/matrix-org/go-neb/services/github/github.go
-
339src/github.com/matrix-org/go-neb/services/github/github_webhook.go
@ -0,0 +1,339 @@ |
|||||
|
package services |
||||
|
|
||||
|
import ( |
||||
|
"fmt" |
||||
|
log "github.com/Sirupsen/logrus" |
||||
|
"github.com/google/go-github/github" |
||||
|
"github.com/matrix-org/go-neb/database" |
||||
|
"github.com/matrix-org/go-neb/matrix" |
||||
|
"github.com/matrix-org/go-neb/plugin" |
||||
|
"github.com/matrix-org/go-neb/services/github/client" |
||||
|
"github.com/matrix-org/go-neb/services/github/webhook" |
||||
|
"github.com/matrix-org/go-neb/types" |
||||
|
"github.com/matrix-org/go-neb/util" |
||||
|
"net/http" |
||||
|
"sort" |
||||
|
"strings" |
||||
|
) |
||||
|
|
||||
|
type githubWebhookService struct { |
||||
|
id string |
||||
|
serviceUserID string |
||||
|
webhookEndpointURL string |
||||
|
ClientUserID string // optional; required for webhooks
|
||||
|
RealmID string |
||||
|
SecretToken string |
||||
|
Rooms map[string]struct { // room_id => {}
|
||||
|
Repos map[string]struct { // owner/repo => { events: ["push","issue","pull_request"] }
|
||||
|
Events []string |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func (s *githubWebhookService) ServiceUserID() string { return s.serviceUserID } |
||||
|
func (s *githubWebhookService) ServiceID() string { return s.id } |
||||
|
func (s *githubWebhookService) ServiceType() string { return "github-webhook" } |
||||
|
func (s *githubWebhookService) Plugin(cli *matrix.Client, roomID string) plugin.Plugin { |
||||
|
return plugin.Plugin{} |
||||
|
} |
||||
|
func (s *githubWebhookService) OnReceiveWebhook(w http.ResponseWriter, req *http.Request, cli *matrix.Client) { |
||||
|
evType, repo, msg, err := webhook.OnReceiveRequest(req, s.SecretToken) |
||||
|
if err != nil { |
||||
|
w.WriteHeader(err.Code) |
||||
|
return |
||||
|
} |
||||
|
logger := log.WithFields(log.Fields{ |
||||
|
"event": evType, |
||||
|
"repo": *repo.FullName, |
||||
|
}) |
||||
|
repoExistsInConfig := false |
||||
|
|
||||
|
for roomID, roomConfig := range s.Rooms { |
||||
|
for ownerRepo, repoConfig := range roomConfig.Repos { |
||||
|
if !strings.EqualFold(*repo.FullName, ownerRepo) { |
||||
|
continue |
||||
|
} |
||||
|
repoExistsInConfig = true // even if we don't notify for it.
|
||||
|
notifyRoom := false |
||||
|
for _, notifyType := range repoConfig.Events { |
||||
|
if evType == notifyType { |
||||
|
notifyRoom = true |
||||
|
break |
||||
|
} |
||||
|
} |
||||
|
if notifyRoom { |
||||
|
logger.WithFields(log.Fields{ |
||||
|
"msg": msg, |
||||
|
"room_id": roomID, |
||||
|
}).Print("Sending notification to room") |
||||
|
if _, e := cli.SendMessageEvent(roomID, "m.room.message", msg); e != nil { |
||||
|
logger.WithError(e).WithField("room_id", roomID).Print( |
||||
|
"Failed to send notification to room.") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if !repoExistsInConfig { |
||||
|
segs := strings.Split(*repo.FullName, "/") |
||||
|
if len(segs) != 2 { |
||||
|
logger.Error("Received event with malformed owner/repo.") |
||||
|
w.WriteHeader(400) |
||||
|
return |
||||
|
} |
||||
|
if err := s.deleteHook(segs[0], segs[1]); err != nil { |
||||
|
logger.WithError(err).Print("Failed to delete webhook") |
||||
|
} else { |
||||
|
logger.Info("Deleted webhook") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
w.WriteHeader(200) |
||||
|
} |
||||
|
|
||||
|
// Register will create webhooks for the repos specified in Rooms
|
||||
|
//
|
||||
|
// The hooks made are a delta between the old service and the current configuration. If all webhooks are made,
|
||||
|
// Register() succeeds. If any webhook fails to be created, Register() fails. A delta is used to allow clients to incrementally
|
||||
|
// build up the service config without recreating the hooks every time a change is made.
|
||||
|
//
|
||||
|
// Hooks are deleted when this service receives a webhook event from Github for a repo which has no user configurations.
|
||||
|
//
|
||||
|
// Hooks can get out of sync if a user manually deletes a hook in the Github UI. In this case, toggling the repo configuration will
|
||||
|
// force NEB to recreate the hook.
|
||||
|
func (s *githubWebhookService) Register(oldService types.Service, client *matrix.Client) error { |
||||
|
if s.RealmID == "" || s.ClientUserID == "" { |
||||
|
return fmt.Errorf("RealmID and ClientUserID is required") |
||||
|
} |
||||
|
// check realm exists
|
||||
|
realm, err := database.GetServiceDB().LoadAuthRealm(s.RealmID) |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
// make sure the realm is of the type we expect
|
||||
|
if realm.Type() != "github" { |
||||
|
return fmt.Errorf("Realm is of type '%s', not 'github'", realm.Type()) |
||||
|
} |
||||
|
|
||||
|
// In order to register the GH service as a client, you must have authed with GH.
|
||||
|
cli := s.githubClientFor(s.ClientUserID, false) |
||||
|
if cli == nil { |
||||
|
return fmt.Errorf( |
||||
|
"User %s does not have a Github auth session with realm %s.", s.ClientUserID, realm.ID()) |
||||
|
} |
||||
|
// Make sure they have specified some webhooks (it makes no sense otherwise)
|
||||
|
reposForWebhooks := s.repoList() |
||||
|
if len(reposForWebhooks) == 0 { |
||||
|
return fmt.Errorf("No repos for webhooks specified") |
||||
|
} |
||||
|
|
||||
|
// Fetch the old service list and work out the difference between the two.
|
||||
|
var oldRepos []string |
||||
|
if oldService != nil { |
||||
|
old, ok := oldService.(*githubWebhookService) |
||||
|
if !ok { |
||||
|
log.WithFields(log.Fields{ |
||||
|
"service_id": oldService.ServiceID(), |
||||
|
"service_type": oldService.ServiceType(), |
||||
|
}).Print("Cannot cast old github service to GithubWebhookService") |
||||
|
// non-fatal though, we'll just make the hooks
|
||||
|
} else { |
||||
|
oldRepos = old.repoList() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// Add the repos in the new service but not the old service
|
||||
|
newRepos, _ := util.Difference(reposForWebhooks, oldRepos) |
||||
|
for _, r := range newRepos { |
||||
|
logger := log.WithField("repo", r) |
||||
|
err := s.createHook(cli, r) |
||||
|
if err != nil { |
||||
|
logger.WithError(err).Error("Failed to create webhook") |
||||
|
return err |
||||
|
} |
||||
|
logger.Info("Created webhook") |
||||
|
} |
||||
|
|
||||
|
if err := s.joinWebhookRooms(client); err != nil { |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
log.Infof("%+v", s) |
||||
|
|
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func (s *githubWebhookService) joinWebhookRooms(client *matrix.Client) error { |
||||
|
for roomID := range s.Rooms { |
||||
|
if _, err := client.JoinRoom(roomID, "", ""); err != nil { |
||||
|
// TODO: Leave the rooms we successfully joined?
|
||||
|
return err |
||||
|
} |
||||
|
} |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func (s *githubWebhookService) repoList() []string { |
||||
|
var repos []string |
||||
|
if s.Rooms == nil { |
||||
|
return repos |
||||
|
} |
||||
|
for _, roomConfig := range s.Rooms { |
||||
|
for ownerRepo := range roomConfig.Repos { |
||||
|
if strings.Count(ownerRepo, "/") != 1 { |
||||
|
log.WithField("repo", ownerRepo).Error("Bad owner/repo key in config") |
||||
|
continue |
||||
|
} |
||||
|
exists := false |
||||
|
for _, r := range repos { |
||||
|
if r == ownerRepo { |
||||
|
exists = true |
||||
|
break |
||||
|
} |
||||
|
} |
||||
|
if !exists { |
||||
|
repos = append(repos, ownerRepo) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
return repos |
||||
|
} |
||||
|
|
||||
|
func (s *githubWebhookService) createHook(cli *github.Client, ownerRepo string) error { |
||||
|
o := strings.Split(ownerRepo, "/") |
||||
|
owner := o[0] |
||||
|
repo := o[1] |
||||
|
// make a hook for all GH events since we'll filter it when we receive webhook requests
|
||||
|
name := "web" // https://developer.github.com/v3/repos/hooks/#create-a-hook
|
||||
|
cfg := map[string]interface{}{ |
||||
|
"content_type": "json", |
||||
|
"url": s.webhookEndpointURL, |
||||
|
} |
||||
|
if s.SecretToken != "" { |
||||
|
cfg["secret"] = s.SecretToken |
||||
|
} |
||||
|
events := []string{"push", "pull_request", "issues", "issue_comment", "pull_request_review_comment"} |
||||
|
_, res, err := cli.Repositories.CreateHook(owner, repo, &github.Hook{ |
||||
|
Name: &name, |
||||
|
Config: cfg, |
||||
|
Events: events, |
||||
|
}) |
||||
|
|
||||
|
if res.StatusCode == 422 { |
||||
|
errResponse, ok := err.(*github.ErrorResponse) |
||||
|
if !ok { |
||||
|
return err |
||||
|
} |
||||
|
for _, ghErr := range errResponse.Errors { |
||||
|
if strings.Contains(ghErr.Message, "already exists") { |
||||
|
log.WithField("repo", ownerRepo).Print("422 : Hook already exists") |
||||
|
return nil |
||||
|
} |
||||
|
} |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
func (s *githubWebhookService) deleteHook(owner, repo string) error { |
||||
|
logger := log.WithFields(log.Fields{ |
||||
|
"endpoint": s.webhookEndpointURL, |
||||
|
"repo": owner + "/" + repo, |
||||
|
}) |
||||
|
logger.Info("Removing hook") |
||||
|
|
||||
|
cli := s.githubClientFor(s.ClientUserID, false) |
||||
|
if cli == nil { |
||||
|
logger.WithField("user_id", s.ClientUserID).Print("Cannot delete webhook: no authenticated client exists for user ID.") |
||||
|
return fmt.Errorf("no authenticated client exists for user ID") |
||||
|
} |
||||
|
|
||||
|
// Get a list of webhooks for this owner/repo and find the one which has the
|
||||
|
// same endpoint URL which is what github uses to determine equivalence.
|
||||
|
hooks, _, err := cli.Repositories.ListHooks(owner, repo, nil) |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
var hook *github.Hook |
||||
|
for _, h := range hooks { |
||||
|
if h.Config["url"] == nil { |
||||
|
logger.Print("Ignoring nil config.url") |
||||
|
continue |
||||
|
} |
||||
|
hookURL, ok := h.Config["url"].(string) |
||||
|
if !ok { |
||||
|
logger.Print("Ignoring non-string config.url") |
||||
|
continue |
||||
|
} |
||||
|
if hookURL == s.webhookEndpointURL { |
||||
|
hook = h |
||||
|
break |
||||
|
} |
||||
|
} |
||||
|
if hook == nil { |
||||
|
return fmt.Errorf("Failed to find hook with endpoint: %s", s.webhookEndpointURL) |
||||
|
} |
||||
|
|
||||
|
_, err = cli.Repositories.DeleteHook(owner, repo, *hook.ID) |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
func sameRepos(a *githubWebhookService, b *githubWebhookService) bool { |
||||
|
getRepos := func(s *githubWebhookService) []string { |
||||
|
r := make(map[string]bool) |
||||
|
for _, roomConfig := range s.Rooms { |
||||
|
for ownerRepo := range roomConfig.Repos { |
||||
|
r[ownerRepo] = true |
||||
|
} |
||||
|
} |
||||
|
var rs []string |
||||
|
for k := range r { |
||||
|
rs = append(rs, k) |
||||
|
} |
||||
|
return rs |
||||
|
} |
||||
|
aRepos := getRepos(a) |
||||
|
bRepos := getRepos(b) |
||||
|
|
||||
|
if len(aRepos) != len(bRepos) { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
sort.Strings(aRepos) |
||||
|
sort.Strings(bRepos) |
||||
|
for i := 0; i < len(aRepos); i++ { |
||||
|
if aRepos[i] != bRepos[i] { |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
func (s *githubWebhookService) githubClientFor(userID string, allowUnauth bool) *github.Client { |
||||
|
token, err := getTokenForUser(s.RealmID, userID) |
||||
|
if err != nil { |
||||
|
log.WithFields(log.Fields{ |
||||
|
log.ErrorKey: err, |
||||
|
"user_id": userID, |
||||
|
"realm_id": s.RealmID, |
||||
|
}).Print("Failed to get token for user") |
||||
|
} |
||||
|
if token != "" { |
||||
|
return client.New(token) |
||||
|
} else if allowUnauth { |
||||
|
return client.New("") |
||||
|
} else { |
||||
|
return nil |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func init() { |
||||
|
types.RegisterService(func(serviceID, serviceUserID, webhookEndpointURL string) types.Service { |
||||
|
return &githubWebhookService{ |
||||
|
id: serviceID, |
||||
|
serviceUserID: serviceUserID, |
||||
|
webhookEndpointURL: webhookEndpointURL, |
||||
|
} |
||||
|
}) |
||||
|
} |
Write
Preview
Loading…
Cancel
Save
Reference in new issue