Merge pull request #40 from rauls5382/abbrev

Add support for abbreviated commands on interative command like

internal/driver/interactive.go

@@ -17,6 +17,7 @@ package driver
 import (
 	"fmt"
 	"io"
+	"regexp"
 	"sort"
 	"strconv"
 	"strings"
@@ -27,6 +28,7 @@ import (
 )
 
 var commentStart = "//:" // Sentinel for comments on options
+var tailDigitsRE = regexp.MustCompile("[0-9]+$")
 
 // interactive starts a shell to read pprof commands.
 func interactive(p *profile.Profile, o *plugin.Options) error {
@@ -218,7 +220,15 @@ func parseCommandLine(input []string) ([]string, variables, error) {
 	cmd, args := input[:1], input[1:]
 	name := cmd[0]
 
-	c := pprofCommands[cmd[0]]
+	c := pprofCommands[name]
+	if c == nil {
+		// Attempt splitting digits on abbreviated commands (eg top10)
+		if d := tailDigitsRE.FindString(name); d != "" && d != name {
+			name = name[:len(name)-len(d)]
+			cmd[0], args = name, append([]string{d}, args...)
+			c = pprofCommands[name]
+		}
+	}
 	if c == nil {
 		return nil, nil, fmt.Errorf("Unrecognized command: %q", name)
 	}

internal/driver/interactive_test.go

@@ -247,6 +247,16 @@ func TestInteractiveCommands(t *testing.T) {
 				"ignore":    "ignore",
 			},
 		},
+		{
+			"top10 --cum focus1 -ignore focus2",
+			map[string]string{
+				"functions": "true",
+				"nodecount": "10",
+				"cum":       "true",
+				"focus":     "focus1|focus2",
+				"ignore":    "ignore",
+			},
+		},
 		{
 			"dot",
 			map[string]string{
@@ -288,14 +298,23 @@ func TestInteractiveCommands(t *testing.T) {
 				"output":    "out",
 			},
 		},
+		{
+			"999",
+			nil, // Error
+		},
 	}
 
 	for _, tc := range testcases {
 		cmd, vars, err := parseCommandLine(strings.Fields(tc.input))
-		vars = applyCommandOverrides(cmd, vars)
+		if tc.want == nil && err != nil {
+			// Error expected
+			continue
+		}
 		if err != nil {
 			t.Errorf("failed on %q: %v", tc.input, err)
+			continue
 		}
+		vars = applyCommandOverrides(cmd, vars)
 
 		for n, want := range tc.want {
 			if got := vars[n].stringValue(); got != want {