search: add support for streaming search

This adds the flag `-stream` to the `search` sub-command. Searches
submitted with `-stream` are routed to the endpoint `search/stream`
instead of the GraphQL api.

In addition, users can set `-display <int>` to set the display limit.

We will add support for JSON in a follow-up PR.

Author: stefanhengl

cmd/src/format.go

@@ -60,6 +60,9 @@ func parseTemplate(text string) (*template.Template, error) {
 		"addFloat": func(x, y float64) float64 {
 			return x + y
 		},
+		"addInt32": func(x, y int32) int32 {
+			return x + y
+		},
 		"debug": func(v interface{}) string {
 			data, _ := marshalIndent(v)
 			fmt.Println(string(data))
@@ -91,6 +94,13 @@ func parseTemplate(text string) (*template.Template, error) {
 		"buildVersionHasNewSearchInterface": searchTemplateFuncs["buildVersionHasNewSearchInterface"],
 		"renderResult":                      searchTemplateFuncs["renderResult"],
 
+		// Register stream-search specific template functions.
+		"streamSearchSequentialLineNumber": streamSearchTemplateFuncs["streamSearchSequentialLineNumber"],
+		"streamSearchHighlightMatch":       streamSearchTemplateFuncs["streamSearchHighlightMatch"],
+		"streamSearchHighlightCommit":      streamSearchTemplateFuncs["streamSearchHighlightCommit"],
+		"streamSearchRenderCommitLabel":    streamSearchTemplateFuncs["streamSearchRenderCommitLabel"],
+		"matchOrMatches":                   streamSearchTemplateFuncs["matchOrMatches"],
+
 		// Alert rendering
 		"searchAlertRender": func(alert searchResultsAlert) string {
 			if content, err := alert.Render(); err != nil {

cmd/src/search.go

@@ -54,13 +54,29 @@ Other tips:
 		explainJSONFlag = flagSet.Bool("explain-json", false, "Explain the JSON output schema and exit.")
 		apiFlags        = api.NewFlags(flagSet)
 		lessFlag        = flagSet.Bool("less", true, "Pipe output to 'less -R' (only if stdout is terminal, and not json flag)")
+		streamFlag      = flagSet.Bool("stream", false, "Consume results as stream.")
+
+		// Streaming.
+		_ = flagSet.Int("display", -1, "Limit the number of results shown. Only supported for streaming.")
 	)
 
 	handler := func(args []string) error {
 		if err := flagSet.Parse(args); err != nil {
 			return err
 		}
 
+		if *streamFlag {
+			// Remove -stream from args.
+			argsWOStream := make([]string, 0, len(args)-1)
+			for _, a := range args {
+				if a == "-stream" {
+					continue
+				}
+				argsWOStream = append(argsWOStream, a)
+			}
+			return streamHandler(argsWOStream)
+		}
+
 		if *explainJSONFlag {
 			fmt.Println(searchJSONExplanation)
 			return nil

cmd/src/search_alert.go

@@ -19,15 +19,18 @@ func init() {
 	}
 }
 
+// ProposedQuery is a suggested query to run when we emit an alert.
+type ProposedQuery struct {
+	Description string
+	Query       string
+}
+
 // searchResultsAlert is a type that can be used to unmarshal values returned by
 // the searchResultsAlertFragment GraphQL fragment below.
 type searchResultsAlert struct {
 	Title           string
 	Description     string
-	ProposedQueries []struct {
-		Description string
-		Query       string
-	}
+	ProposedQueries []ProposedQuery
 }
 
 // Render renders an alert to a string ready to be output to a console,

cmd/src/search_alert_test.go

@@ -13,10 +13,7 @@ func TestRender(t *testing.T) {
 	full := &searchResultsAlert{
 		Title:       "foo",
 		Description: "bar",
-		ProposedQueries: []struct {
-			Description string
-			Query       string
-		}{
+		ProposedQueries: []ProposedQuery{
 			{
 				Description: "quux",
 				Query:       "xyz:abc",