From 5f5ec3858529821c0cf72a66d372274da3bd6091 Mon Sep 17 00:00:00 2001
From: bytedream <bytedream@protonmail.com>
Date: Sun, 6 Mar 2022 01:04:27 +0100
Subject: [PATCH] Added function to sort episode by audio

---
 utils/sort.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/utils/sort.go b/utils/sort.go
index 16ce41a..90e05c7 100644
--- a/utils/sort.go
+++ b/utils/sort.go
@@ -5,6 +5,7 @@ import (
 	"sort"
 	"strconv"
 	"strings"
+	"sync"
 )
 
 // SortEpisodesBySeason sorts the given episodes by their seasons.
@@ -42,6 +43,44 @@ func SortEpisodesBySeason(episodes []*crunchyroll.Episode) [][]*crunchyroll.Epis
 	return eps
 }
 
+// SortEpisodesByAudio sort the given episodes by their audio locale
+func SortEpisodesByAudio(episodes []*crunchyroll.Episode) (map[crunchyroll.LOCALE][]*crunchyroll.Episode, error) {
+	eps := map[crunchyroll.LOCALE][]*crunchyroll.Episode{}
+
+	errChan := make(chan error)
+
+	var wg sync.WaitGroup
+	var lock sync.Mutex
+	for _, episode := range episodes {
+		episode := episode
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			audioLocale, err := episode.AudioLocale()
+			if err != nil {
+				errChan <- err
+				return
+			}
+			lock.Lock()
+			defer lock.Unlock()
+
+			if _, ok := eps[audioLocale]; !ok {
+				eps[audioLocale] = make([]*crunchyroll.Episode, 0)
+			}
+			eps[audioLocale] = append(eps[audioLocale], episode)
+		}()
+	}
+	go func() {
+		wg.Wait()
+		errChan <- nil
+	}()
+
+	if err := <-errChan; err != nil {
+		return nil, err
+	}
+	return eps, nil
+}
+
 // MovieListingsByDuration sorts movie listings by their duration
 type MovieListingsByDuration []*crunchyroll.MovieListing