composite.go

package main

import (
	"log"
	"sort"
	"strings"
	"time"
)

func compositeBans() {
	var compositeBanlist []banDataType

	//Add our bans first
	for _, ban := range ourBanData {

		compositeBanlist = append(compositeBanlist, banDataType{
			UserName: strings.ToLower(ban.UserName),
			Sources:  []string{serverConfig.CommunityName},
			Reasons:  []string{ban.Reason},
			Revokes:  []bool{ban.Revoked},
			Adds:     []string{ban.Added}})
	}

	dupes := 0

	//Now add the bans from other servers
	for _, server := range serverList.ServerList {
		//Only if subscribed, and skip self
		if server.LocalData.Subscribed && !strings.EqualFold(server.CommunityName, serverConfig.CommunityName) {
			for _, ban := range server.LocalData.BanList {
				//Don't composite revoked bans
				if !ban.Revoked {
					found := false
					for ipos, iban := range compositeBanlist {
						//Duplicate, add data to existing entry
						if strings.EqualFold(iban.UserName, ban.UserName) {
							found = true
							dupes++
							compositeBanlist[ipos].Sources = append(compositeBanlist[ipos].Sources, server.CommunityName)

							if server.LocalData.StripReasons {
								compositeBanlist[ipos].Reasons = append(compositeBanlist[ipos].Reasons, "")
							} else {
								compositeBanlist[ipos].Reasons = append(compositeBanlist[ipos].Reasons, ban.Reason)
							}

							compositeBanlist[ipos].Revokes = append(compositeBanlist[ipos].Revokes, ban.Revoked)

							compositeBanlist[ipos].Adds = append(compositeBanlist[ipos].Adds, ban.Added)
							break
						}
					}
					//Strip ban reasons if set to
					if server.LocalData.StripReasons {
						ban.Reason = ""
					}
					//This isn't already in the list, so add it (avoid dupes)
					if !found {
						//If we require a reason, and there isn't one... skip
						if serverConfig.ServerPrefs.RequireReason && ban.Reason == "" {
							continue
						}
						compositeBanlist = append(compositeBanlist, banDataType{
							UserName: strings.ToLower(ban.UserName),
							Sources:  []string{server.CommunityName},
							Reasons:  []string{ban.Reason},
							Revokes:  []bool{ban.Revoked},
							Adds:     []string{ban.Added}})
					}
				}
			}
		}
	}

	if serverConfig.ServerPrefs.VerboseLogging {
		log.Printf("Composited %v bans, %v duplicates.\n", len(compositeBanlist), dupes)
	}

	//Sort by time added, new to old
	sort.Slice(compositeBanlist, func(i, j int) bool {
		//Use newest date, if there are multiple sources
		var newest_a time.Time = time.Time{}
		for _, addA := range compositeBanlist[j].Adds {
			bTime, errb := time.Parse(timeFormat, addA)
			if errb == nil {
				if bTime.After(newest_a) || newest_a.IsZero() {
					newest_a = bTime
				}
			}
		}
		var newest_b time.Time = time.Time{}
		for _, addB := range compositeBanlist[j].Adds {
			bTime, errb := time.Parse(timeFormat, addB)
			if errb == nil {
				if bTime.After(newest_b) || newest_b.IsZero() {
					newest_b = bTime
				}
			}
		}
		return newest_a.Before(newest_b)
	})

	//Cut list to size, new entries are at the start
	compBan := []banDataType{}
	for bpos, ban := range compositeBanlist {
		if bpos < serverConfig.ServerPrefs.MaxBanOutputCount {
			compBan = append(compBan, ban)
		} else {
			log.Printf(" Banlist max size exceeded (%v items), truncating...\n", serverConfig.ServerPrefs.MaxBanOutputCount)
			break
		}
	}
	if serverConfig.ServerPrefs.VerboseLogging {
		log.Printf("Composite banlist updated: %v items.\n", len(compBan))
	}

	var condList []minBanDataType
	//output as a Factorio-friendly list
	for _, ban := range compBan {
		if !ban.Revoked {
			reasonList := ""
			for rpos, reason := range ban.Reasons {
				if rpos > 0 {
					reasonList += ", "
				}
				if reason != "" {
					reasonList += ban.Sources[rpos] + ": " + reason
				} else {
					reasonList += "(" + ban.Sources[rpos] + ")"
				}
			}
			condList = append(condList, minBanDataType{
				UserName: strings.ToLower(ban.UserName), Reason: reasonList})
		}
	}
	compositeBanData = condList

	writeCompositeBanlist()
}