Add Register() function to Services

Fix plugin_tests and make them a bit more intelligible.

src/github.com/matrix-org/go-neb/api.go

@@ -198,6 +198,11 @@ func (s *configureServiceHandler) OnIncomingRequest(req *http.Request) (interfac
 		return nil, &errors.HTTPError{err, "Error parsing config JSON", 400}
 	}
 
+	err := service.Register()
+	if err != nil {
+		return nil, &errors.HTTPError{err, "Failed to register service", 500}
+	}
+
 	client, err := s.clients.Client(service.ServiceUserID())
 	if err != nil {
 		return nil, &errors.HTTPError{err, "Unknown matrix client", 400}

src/github.com/matrix-org/go-neb/plugin/plugin.go

@@ -31,7 +31,7 @@ type Command struct {
 // the appropriate RFC.
 type Expansion struct {
 	Regexp *regexp.Regexp
-	Expand func(roomID, matchingText string) interface{}
+	Expand func(roomID, userID, matchingText string) interface{}
 }
 
 // matches if the arguments start with the path of the command.
@@ -95,7 +95,7 @@ func runExpansionsForPlugin(plugin Plugin, event *matrix.Event, body string) []i
 				continue
 			}
 			matches[matchingText] = true
-			if response := expansion.Expand(event.RoomID, matchingText); response != nil {
+			if response := expansion.Expand(event.RoomID, event.Sender, matchingText); response != nil {
 				responses = append(responses, response)
 			}
 		}

src/github.com/matrix-org/go-neb/plugin/plugin_test.go

@@ -26,21 +26,21 @@ func makeTestEvent(msgtype, body string) *matrix.Event {
 
 type testResponse struct {
 	RoomID    string
-	Sender    string
 	Arguments []string
 }
 
 func makeTestResponse(roomID, sender string, arguments []string) interface{} {
-	return testResponse{roomID, sender, arguments}
+	return testResponse{roomID, arguments}
 }
 
 type testExpansion struct {
 	RoomID       string
+	UserID       string
 	MatchingText string
 }
 
-func makeTestExpansion(roomID, matchingText string) interface{} {
-	return testExpansion{roomID, matchingText}
+func makeTestExpansion(roomID, userID, matchingText string) interface{} {
+	return testExpansion{roomID, userID, matchingText}
 }
 
 func makeTestPlugin(paths [][]string, regexps []*regexp.Regexp) Plugin {
@@ -74,7 +74,7 @@ func TestRunCommands(t *testing.T) {
 		"arg1", "arg 2", "arg 3",
 	})}
 	if !reflect.DeepEqual(got, want) {
-		t.Errorf("runCommands(%q, %q) == %q, want %q", plugins, event, got, want)
+		t.Errorf("runCommands(\nplugins=%+v\nevent=%+v\n)\n%+v\nwanted: %+v", plugins, event, got, want)
 	}
 }
 
@@ -89,7 +89,7 @@ func TestRunCommandsBestMatch(t *testing.T) {
 		"arg1",
 	})}
 	if !reflect.DeepEqual(got, want) {
-		t.Errorf("runCommands(%q, %q) == %q, want %q", plugins, event, got, want)
+		t.Errorf("runCommands(\nplugins=%+v\nevent=%+v\n)\n%+v\nwanted: %+v", plugins, event, got, want)
 	}
 }
 
@@ -105,7 +105,7 @@ func TestRunCommandsMultiplePlugins(t *testing.T) {
 		makeTestResponse(myRoomID, mySender, []string{"first", "arg1"}),
 	}
 	if !reflect.DeepEqual(got, want) {
-		t.Errorf("runCommands(%q, %q) == %q, want %q", plugins, event, got, want)
+		t.Errorf("runCommands(\nplugins=%+v\nevent=%+v\n)\n%+v\nwanted: %+v", plugins, event, got, want)
 	}
 }
 
@@ -119,7 +119,7 @@ func TestRunCommandsInvalidShell(t *testing.T) {
 		makeTestResponse(myRoomID, mySender, []string{"'mismatched", `quotes"`}),
 	}
 	if !reflect.DeepEqual(got, want) {
-		t.Errorf("runCommands(%q, %q) == %q, want %q", plugins, event, got, want)
+		t.Errorf("runCommands(\nplugins=%+v\nevent=%+v\n)\n%+v\nwanted: %+v", plugins, event, got, want)
 	}
 }
 
@@ -133,12 +133,12 @@ func TestExpansion(t *testing.T) {
 	event := makeTestEvent("m.text", "test banana for scale")
 	got := runCommands(plugins, event)
 	want := []interface{}{
-		makeTestExpansion(myRoomID, "anana"),
-		makeTestExpansion(myRoomID, "ale"),
-		makeTestExpansion(myRoomID, "ban"),
+		makeTestExpansion(myRoomID, mySender, "anana"),
+		makeTestExpansion(myRoomID, mySender, "ale"),
+		makeTestExpansion(myRoomID, mySender, "ban"),
 	}
 	if !reflect.DeepEqual(got, want) {
-		t.Errorf("runCommands(%q, %q) == %q, want %q", plugins, event, got, want)
+		t.Errorf("runCommands(\nplugins=%+v\nevent=%+v\n)\n%+v\nwanted: %+v", plugins, event, got, want)
 	}
 }
 
@@ -151,9 +151,9 @@ func TestExpansionDuplicateMatches(t *testing.T) {
 	event := makeTestEvent("m.text", "badger badger badger")
 	got := runCommands(plugins, event)
 	want := []interface{}{
-		makeTestExpansion(myRoomID, "badger"),
+		makeTestExpansion(myRoomID, mySender, "badger"),
 	}
 	if !reflect.DeepEqual(got, want) {
-		t.Errorf("runCommands(%q, %q) == %q, want %q", plugins, event, got, want)
+		t.Errorf("runCommands(\nplugins=%+v\nevent=%+v\n)\n%+v\nwanted: %+v", plugins, event, got, want)
 	}
 }

src/github.com/matrix-org/go-neb/services/echo/echo.go

@@ -18,6 +18,7 @@ func (e *echoService) ServiceUserID() string { return e.UserID }
 func (e *echoService) ServiceID() string     { return e.id }
 func (e *echoService) ServiceType() string   { return "echo" }
 func (e *echoService) RoomIDs() []string     { return e.Rooms }
+func (e *echoService) Register() error       { return nil }
 func (e *echoService) Plugin(roomID string) plugin.Plugin {
 	return plugin.Plugin{
 		Commands: []plugin.Command{

src/github.com/matrix-org/go-neb/services/github/github.go

@@ -12,6 +12,7 @@ import (
 	"net/http"
 	"regexp"
 	"strconv"
+	"strings"
 )
 
 // Matches alphanumeric then a /, then more alphanumeric then a #, then a number.
@@ -19,29 +20,43 @@ import (
 var ownerRepoIssueRegex = regexp.MustCompile("([A-z0-9-_]+)/([A-z0-9-_]+)#([0-9]+)")
 
 type githubService struct {
-	id     string
-	UserID string
-	Rooms  map[string][]string // room_id => ["push","issue","pull_request"]
+	id           string
+	BotUserID    string
+	GithubUserID string
+	WebhookRooms map[string][]string // room_id => ["push","issue","pull_request"]
 }
 
-func (s *githubService) ServiceUserID() string { return s.UserID }
+func (s *githubService) ServiceUserID() string { return s.BotUserID }
 func (s *githubService) ServiceID() string     { return s.id }
 func (s *githubService) ServiceType() string   { return "github" }
 func (s *githubService) RoomIDs() []string {
 	var keys []string
-	for k := range s.Rooms {
+	for k := range s.WebhookRooms {
 		keys = append(keys, k)
 	}
 	return keys
 }
 func (s *githubService) Plugin(roomID string) plugin.Plugin {
 	return plugin.Plugin{
-		Commands: []plugin.Command{},
+		Commands: []plugin.Command{
+			plugin.Command{
+				Path: []string{"github", "create"},
+				Command: func(roomID, userID string, args []string) (interface{}, error) {
+					cli := githubClientFor(userID, false)
+					if cli == nil {
+						// TODO: send starter link
+						return &matrix.TextMessage{"m.notice",
+							userID + " : You have not linked your Github account."}, nil
+					}
+					return &matrix.TextMessage{"m.notice", strings.Join(args, " ")}, nil
+				},
+			},
+		},
 		Expansions: []plugin.Expansion{
 			plugin.Expansion{
 				Regexp: ownerRepoIssueRegex,
-				Expand: func(roomID, matchingText string) interface{} {
-					cli := githubClient("")
+				Expand: func(roomID, userID, matchingText string) interface{} {
+					cli := githubClientFor(userID, true)
 					owner, repo, num, err := ownerRepoNumberFromText(matchingText)
 					if err != nil {
 						log.WithError(err).WithField("text", matchingText).Print(
@@ -75,7 +90,7 @@ func (s *githubService) OnReceiveWebhook(w http.ResponseWriter, req *http.Reques
 		return
 	}
 
-	for roomID, notif := range s.Rooms {
+	for roomID, notif := range s.WebhookRooms {
 		notifyRoom := false
 		for _, notifyType := range notif {
 			if evType == notifyType {
@@ -99,6 +114,20 @@ func (s *githubService) OnReceiveWebhook(w http.ResponseWriter, req *http.Reques
 	}
 	w.WriteHeader(200)
 }
+func (s *githubService) Register() error {
+	return nil
+}
+
+func githubClientFor(userID string, allowUnauth bool) *github.Client {
+	var cli *github.Client
+
+	// TODO get the token for this user
+
+	if cli == nil && allowUnauth {
+		return githubClient("")
+	}
+	return cli
+}
 
 // githubClient returns a github Client which can perform Github API operations.
 // If `token` is empty, a non-authenticated client will be created. This should be

src/github.com/matrix-org/go-neb/types/types.go

@@ -35,6 +35,7 @@ type Service interface {
 	RoomIDs() []string
 	Plugin(roomID string) plugin.Plugin
 	OnReceiveWebhook(w http.ResponseWriter, req *http.Request, cli *matrix.Client)
+	Register() error
 }
 
 var servicesByType = map[string]func(string) Service{}