basic test

parser/telegram.go

@@ -1,22 +1,44 @@
 package parser
 
-import "strings"
+import (
+	"regexp"
+	"strconv"
+	"strings"
+)
 
 type Telegram struct {
 	Text     string
 	Complete bool
 	Parsed   bool
 	Fields   []string
-	Sequence uint16
+	Sequence int
+	Type string
 }
 
-const Start = "ZCZC"
-const End = "NNN"
+
+
+const (
+	Start = "ZCZC"
+	End = "NNNN"
+	TmqRegex = `TMQ(?P<tmq>\d{4})`
+	FlopRegex = `^\((?P<flop>\w{3})-.*`
+
+	TypePLN = "PLN"
+)
 
 //Parse
 func Parse(text string) (*Telegram, error) {
 	var telegram = &Telegram{Text: text}
 	telegram.Complete = isStartAndEnd(text)
+	telegram.Fields = strings.Split(strings.Replace(text, "\r\n", "\n", -1), "\n")
+	telegram.Fields = clean(telegram.Fields)
+	seq := getFields(TmqRegex, telegram.Fields[0])
+	telegram.Sequence, _ = strconv.Atoi( seq["tmq"])
+	if isPln(*telegram) {
+		telegram.Type = TypePLN
+	} else if t := getFields(FlopRegex, telegram.Fields[3]); t != nil {
+		telegram.Type = t["flop"]
+	}
 	return telegram, nil
 }
 
@@ -24,3 +46,35 @@ func isStartAndEnd(s string) bool {
 	text := strings.TrimSpace(s)
 	return strings.HasPrefix(text, Start) && strings.HasSuffix(text, End)
 }
+
+func clean(fields []string) []string {
+	var values []string
+	for _, s := range fields {
+		values = append(values, strings.TrimSpace(s))
+	}
+	return values[:len(values)-1]
+}
+
+
+func getFields(regEx, field string) (fields map[string]string) {
+
+	var compRegEx = regexp.MustCompile(regEx)
+	match := compRegEx.FindStringSubmatch(field)
+
+	fields = make(map[string]string)
+	for i, name := range compRegEx.SubexpNames() {
+		if i > 0 && i <= len(match) {
+			fields[name] = match[i]
+		}
+	}
+	return
+}
+
+func isPln(telegram Telegram) bool {
+	for _, value := range telegram.Fields {
+		if strings.HasPrefix(value, TypePLN) {
+			return true
+		}
+	}
+	return false
+}