Create CSS using SASS and add subfilters

1 files changed, 10 insertions, 2 deletions

diff --git a/country.go b/country.go
index d5bbe48..f3d49b0 100644
--- a/country.go
+++ b/country.go
@@ -2,6 +2,7 @@ package sws
 
 import (
 	"net"
+	"sort"
 
 	maxminddb "github.com/oschwald/maxminddb-golang"
 )
@@ -13,7 +14,7 @@ type Country struct {
 
 type CountrySet []*Country
 
-func NewCountrySet(hs *HitSet) CountrySet {
+func NewCountrySet(hs *HitSet) *CountrySet {
 	tmp := make(map[string]*Country)
 	for _, h := range hs.Hits() {
 		if h.CountryCode == nil {
@@ -43,7 +44,14 @@ func NewCountrySet(hs *HitSet) CountrySet {
 		out[i] = b
 		i++
 	}
-	return CountrySet(out)
+	cs := CountrySet(out)
+	return &cs
+}
+
+func (cs *CountrySet) SortByHits() {
+	sort.Slice(*cs, func(i, j int) bool {
+		return (*cs)[i].hitSet.Count() > (*cs)[j].hitSet.Count()
+	})
 }
 
 func (c Country) Label() string {