1st

tennis.go

@@ -0,0 +1,144 @@
+package flashscore
+
+import (
+	"regexp"
+	"strconv"
+	"strings"
+)
+
+type Surface string
+type Sex string
+
+const (
+	SurfaceHard   Surface = "hard"
+	SurfaceClay   Surface = "clay"
+	SurfaceGrass  Surface = "grass"
+	SurfaceCarpet Surface = "carpet"
+
+	SexMen   = "men"
+	SexWomen = "women"
+)
+
+var reBreakpoints = regexp.MustCompile(`(\d+)%\s*\((\d+)/(\d+)`)
+
+type Breakpoints struct {
+	Qty       int // всего сколько было брейкпоинтов
+	Converted int // реализованных
+	Percent   int // процент реализации
+}
+
+// parseBreakpoints - извлекает статистику по брейкпоинтам из строки вида "50% (0/4)"
+func parseBreakpoints(str string) *Breakpoints {
+	groups := reBreakpoints.FindStringSubmatch(str)
+
+	if len(groups) == 4 {
+		qty, _ := strconv.Atoi(groups[3])
+		converted, _ := strconv.Atoi(groups[2])
+		percent, _ := strconv.Atoi(groups[1])
+
+		return &Breakpoints{
+			Qty:       qty,
+			Converted: converted,
+			Percent:   percent,
+		}
+	}
+	return nil
+}
+
+// /type TennisSetStat
+
+type TennisMatchStat struct {
+	// Сколько было брейкпоинтов и сколько реализовал первый теннисист
+	HomeBreakpoints *Breakpoints
+	// Сколько было брейкпоинтов и сколько реализовал второй теннисист
+	AwayBreakpoints *Breakpoints
+}
+
+func parseTennisMatchStat(str string) (stat TennisMatchStat, err error) {
+	blocks, err := ParseFlashscoreEncoding(str)
+	if err != nil {
+		return
+	}
+
+	//buf, _ := json.MarshalIndent(blocks, "", "    ")
+	//fmt.Printf("%s\n\n", buf)
+
+	var period string
+
+	for _, block := range blocks {
+		if block["SE"] != "" {
+			if block["SE"] == "Match" {
+				period = "match"
+			} else {
+				period = ""
+			}
+		} else {
+			if period != "match" {
+				continue
+			}
+
+			switch block["SG"] {
+			case "Break Points Converted":
+				stat.HomeBreakpoints = parseBreakpoints(block["SH"])
+				stat.AwayBreakpoints = parseBreakpoints(block["SI"])
+			}
+		}
+
+	}
+	return
+}
+
+func ParseTennisSurfaceFromChampName(fullChampName string) Surface {
+	substrings := strings.Split(fullChampName, ",")
+	surface := substrings[len(substrings)-1]
+
+	surface = strings.ToLower(strings.TrimSpace(surface))
+
+	switch surface {
+	case "hard", "indoor hard", "indoor":
+		return SurfaceHard
+
+	case "clay", "indoor clay", "green clay", "red clay":
+		return SurfaceClay
+
+	case "grass":
+		return SurfaceGrass
+
+	case "carpet", "indoor carpet":
+		return SurfaceCarpet
+
+	default:
+		return ""
+	}
+}
+
+// ATP - SINGLES
+// ATP - DOUBLES
+// CHALLENGER MEN - SINGLES
+// WTA - DOUBLES
+// WTA - SINGLES
+// ITF MEN - DOUBLES
+// ITF WOMEN
+// TEAMS - WOMEN
+
+var reMen = regexp.MustCompile(`\b(atp|men)\b`)
+var reWomen = regexp.MustCompile(`\b(wta|wom[ea]n)\b`)
+
+func ParseTennisSexFromChampName(fullChampName string) string {
+	fullChampName = strings.ToLower(fullChampName)
+
+	if reMen.MatchString(fullChampName) {
+		return SexMen
+	}
+
+	if reWomen.MatchString(fullChampName) {
+		return SexWomen
+	}
+	return ""
+}
+
+func IsTennisDoublesChamp(fullChampName string) bool {
+	fullChampName = strings.ToLower(fullChampName)
+
+	return strings.Contains(fullChampName, "doubles")
+}