bbgo: implement parseCommand

pkg/bbgo/interact.go

@@ -1,11 +1,10 @@
 package bbgo
 
 import (
-	"errors"
-	"go/scanner"
-	"go/token"
 	"reflect"
 	"strconv"
+	"strings"
+	"text/scanner"
 
 	"gopkg.in/tucnak/telebot.v2"
 )
@@ -19,31 +18,19 @@ func (i *Interact) HandleTelegramMessage(msg *telebot.Message) {
 
 }
 
-func parseCommand(text string) (args []string, err error) {
-	src := []byte(text)
-
-	// Initialize the scanner.
-	var errHandler scanner.ErrorHandler = func(pos token.Position, msg string) {
-		err = errors.New(msg)
-	}
-
+func parseCommand(src string) (args []string) {
 	var s scanner.Scanner
-	fset := token.NewFileSet()                      // positions are relative to fset
-	file := fset.AddFile("", fset.Base(), len(src)) // register input "file"
-	s.Init(file, src, errHandler, 0)
-
-	// Repeated calls to Scan yield the token sequence found in the input.
-	for {
-		_, tok, lit := s.Scan()
-		if tok == token.EOF {
-			break
+	s.Init(strings.NewReader(src))
+	s.Filename = "command"
+	for tok := s.Scan(); tok != scanner.EOF; tok = s.Scan() {
+		text := s.TokenText()
+		if text[0] == '"' && text[len(text) - 1] == '"' {
+			text, _ = strconv.Unquote(text)
+		}
+		args = append(args,text)
 	}
 
-		// we are not using the token type right now, but we will use them later
-		args = append(args, lit)
-	}
-
-	return args, nil
+	return args
 }
 
 func parseFuncArgsAndCall(f interface{}, args []string) error {

pkg/bbgo/interact_test.go

@@ -30,6 +30,15 @@ func Test_parseFuncArgsAndCall_ErrorFunction(t *testing.T) {
 }
 
 func Test_parseCommand(t *testing.T) {
-
+	args := parseCommand(`closePosition "BTC USDT" 3.1415926 market`)
+	t.Logf("args: %+v", args)
+	for i, a := range args {
+		t.Logf("args(%d): %#v", i, a)
 	}
 
+	assert.Equal(t, 4, len(args))
+	assert.Equal(t, "closePosition", args[0])
+	assert.Equal(t, "BTC USDT", args[1])
+	assert.Equal(t, "3.1415926", args[2])
+	assert.Equal(t, "market", args[3])
+}