Go implemenation of DMR parser

dmr-parser.go

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

// StringParser parses XML as strings.
type StringParser struct{}

// NewStringParser returns a DMR string parser.
func NewStringParser() *StringParser {
	return &StringParser{}
}

func getXMLVal(line string) string {
	start := strings.Index(line, ">") + 1
	end := strings.LastIndex(line, "<")
	return line[start:end]
}

// ParseFile runs the string parser.
func (p *StringParser) ParseFile(file *os.File) (*VehicleList, error) {
	var vehicles VehicleList = make(map[string]struct{})
	var isCar bool
	csv, brand, model := "", "", ""
	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		line := strings.TrimSpace(scanner.Text())
		if strings.HasPrefix(line, "<ns:KoeretoejArtNummer>") {
			isCar = strings.HasPrefix(line, "<ns:KoeretoejArtNummer>1<")
			continue
		}
		if isCar {
			if strings.HasPrefix(line, "<ns:KoeretoejMaerkeTypeNavn>") {
				brand = getXMLVal(line)
			} else if strings.HasPrefix(line, "<ns:KoeretoejModelTypeNavn>") {
				model = getXMLVal(line)
			}
			if brand != "" && model != "" {
				csv = fmt.Sprintf("%v;%v", brand, model)
				if _, ok := vehicles[csv]; !ok {
					vehicles[csv] = struct{}{}
				}
				brand, model = "", ""
			}
		}
	}
	return &vehicles, nil
}