Make bot automatically fill in averages

bot.go

@@ -117,13 +117,13 @@ func getSongs() (SongPriorityQueue, error) {
 	return songs, nil
 }
 
-func task() {
+func submitSong() {
 	now := time.Now()
 	if len(songs) > 0 && songs[0].time.Before(now) {
 		fmt.Println("Time has passed for " + songs[0].song)
 		success, err := addSong("Levyraati 2018", songs[0].week, "Lamperi", songs[0].song)
 		if err != nil {
-			log.Fatal(err)
+			log.Println("Error while adding song:", err)
 		}
 		if success {
 			err := songSynced(songs[0].week)
@@ -136,6 +136,139 @@ func task() {
 	}
 }
 
+func isCurrentAuthor(line, author string) bool {
+	authorIndex := strings.Index(line, author)
+	endIndex := strings.Index(line, "-->")
+	return authorIndex != -1 && authorIndex < endIndex
+}
+
+func parseScore(line string) string {
+	parts := strings.Split(line, "-->")
+	if len(parts) < 2 {
+		return ""
+	}
+	score := strings.TrimRight(strings.Trim(parts[1], " \t\n"), "p")
+	if score == "" {
+		return score
+	}
+	number, _ := regexp.Compile("^\\d([.,]\\d)?$")
+	if number.MatchString(score) {
+		return strings.Replace(score, ",", ".", 1)
+	}
+	numberHalf, _ := regexp.Compile("^\\d½$")
+	if numberHalf.MatchString(score) {
+		return fmt.Sprintf("%c.5", score[0])
+	}
+	stars, _ := regexp.Compile("^\\*+$")
+	if stars.MatchString(score) {
+		return fmt.Sprintf("%d", len(score))
+	}
+	imageStars, _ := regexp.Compile("^(\\[\\[Image:[01]\\.png\\]\\]){5}$")
+	if imageStars.MatchString(score) {
+		return fmt.Sprintf("%d", strings.Count(score, "1"))
+	}
+	quarterScore, _ := regexp.Compile("^\\d-\\d½?$")
+	if quarterScore.MatchString(score) {
+		return fmt.Sprintf("%c.25", score[0])
+	}
+	thirdQuarterScore, _ := regexp.Compile("^\\d½?-\\d$")
+	if thirdQuarterScore.MatchString(score) {
+		return fmt.Sprintf("%c.75", score[0])
+	}
+	fmt.Printf("Could not match '%s'\n", score)
+	return ""
+}
+
+func appendAverages(wikiText string) string {
+	const AUTHOR_MARK = "<!-- Lisääjä -->"
+	const SONG_MARK = "<!-- Kappale -->"
+	const AVERAGE_MARK = "<!-- KA -->"
+	lines := strings.Split(wikiText, "\n")
+	isScore := false
+	scores := make([]string, 0)
+	count := 0
+	currentAuthor := ""
+
+	changedLines := make([]string, 0, len(lines))
+	for _, line := range lines {
+		if strings.Index(line, AUTHOR_MARK) != -1 {
+			currentAuthor = strings.Trim(strings.Split(line, AUTHOR_MARK)[1], " \t")
+		} else if strings.Index(line, SONG_MARK) != -1 {
+			isScore = true
+			scores = make([]string, 0)
+			count = 0
+		} else if isScore && strings.Index(line, AVERAGE_MARK) == -1 {
+			if !isCurrentAuthor(line, currentAuthor) {
+				score := parseScore(line)
+				if score != "" {
+					scores = append(scores, score)
+					count += 1
+				} else {
+					scores = append(scores, "0")
+				}
+
+			}
+		}
+		if strings.Index(line, AVERAGE_MARK) != -1 {
+			expression := fmt.Sprintf("'''{{#expr:(%s)/%d round 2}}'''", strings.Join(scores, "+"), count)
+			newLine := "| " + AVERAGE_MARK + " " + expression
+			changedLines = append(changedLines, newLine)
+			if newLine != line {
+				fmt.Printf("Difference for %s\n%s\n%s\n", currentAuthor, newLine, line)
+			}
+
+		} else {
+			changedLines = append(changedLines, line)
+		}
+	}
+	return strings.Join(changedLines, "\n")
+}
+
+func fixAverages(title string) error {
+	wiki := CreateWikiClient(credentials.APIURL, credentials.UserName, credentials.Password)
+
+	sections, err := wiki.GetWikiPageSections(title)
+	if err != nil {
+		return err
+	}
+
+	currentWeek := 2
+	numberReg, _ := regexp.Compile("\\d+")
+	for _, section := range sections {
+		weekStr := numberReg.FindString(section.title)
+		if weekStr != "" {
+			weekNumber, _ := strconv.Atoi(weekStr)
+			if weekNumber > currentWeek {
+				break
+			}
+
+			wikiText, err := wiki.GetWikiPageSectionText(title, section.index)
+			if err != nil {
+				return err
+			}
+			changedWikiText := appendAverages(wikiText)
+			if changedWikiText != wikiText {
+				//fmt.Println(wikiText)
+				//fmt.Println(changedWikiText)
+
+				_, err := wiki.EditWikiPageSection(title, section.index, changedWikiText,
+					fmt.Sprintf("Calculate averages for week %d", weekNumber))
+				if err != nil {
+					return err
+				}
+			}
+		}
+	}
+	return nil
+}
+
+func fixAveragesTask() {
+	err := fixAverages("Levyraati 2018")
+	if err != nil {
+		fmt.Println("Error while calculating averages:", err)
+	}
+}
+
 func initCreds() error {
 	f, err := os.Open("credentials.json")
 	if err != nil {
@@ -216,7 +349,9 @@ func main() {
 		}
 	}()
 
-	gocron.Every(1).Second().Do(task)
+	gocron.Every(1).Hour().Do(fixAveragesTask)
+
+	gocron.Every(1).Second().Do(submitSong)
 	<-gocron.Start()
 
 }