From 3738b46ce9edc5a8f372f4f4c6cac7e20fc51af1 Mon Sep 17 00:00:00 2001
From: Eduard Urbach <e.urbach@gmail.com>
Date: Sat, 7 Nov 2020 08:24:27 +0900
Subject: [PATCH] Added episode count filter for editors

---
 pages/editor/editor.pixy                 |  1 +
 pages/editor/filteranime/episodecount.go | 18 ++++++++++++++++++
 pages/index/staffroutes/staffroutes.go   |  1 +
 3 files changed, 20 insertions(+)
 create mode 100644 pages/editor/filteranime/episodecount.go

diff --git a/pages/editor/editor.pixy b/pages/editor/editor.pixy
index 6024f26e..99b29bd1 100644
--- a/pages/editor/editor.pixy
+++ b/pages/editor/editor.pixy
@@ -66,6 +66,7 @@ component EditorTabs(url string, user *arn.User)
 			Tab("Start date", "calendar", "/editor/anime/details/startdate" + user.Settings().Editor.Filter.Suffix())
 			Tab("End date", "calendar", "/editor/anime/details/enddate" + user.Settings().Editor.Filter.Suffix())
 			Tab("Ep. Length", "clock-o", "/editor/anime/details/episodelength" + user.Settings().Editor.Filter.Suffix())
+			Tab("Ep. Count", "clock-o", "/editor/anime/details/episodecount" + user.Settings().Editor.Filter.Suffix())
 			Tab("Source", "book", "/editor/anime/details/source" + user.Settings().Editor.Filter.Suffix())
 	
 	if strings.Contains(url, "/editor/anime/companies")
diff --git a/pages/editor/filteranime/episodecount.go b/pages/editor/filteranime/episodecount.go
new file mode 100644
index 00000000..15f76eb2
--- /dev/null
+++ b/pages/editor/filteranime/episodecount.go
@@ -0,0 +1,18 @@
+package filteranime
+
+import (
+	"github.com/aerogo/aero"
+	"github.com/animenotifier/notify.moe/arn"
+)
+
+// EpisodeCount ...
+func EpisodeCount(ctx aero.Context) error {
+	return editorList(
+		ctx,
+		"Anime without an episode count",
+		func(anime *arn.Anime) bool {
+			return anime.EpisodeCount == 0
+		},
+		nil,
+	)
+}
diff --git a/pages/index/staffroutes/staffroutes.go b/pages/index/staffroutes/staffroutes.go
index baa980a8..0b41c4e5 100644
--- a/pages/index/staffroutes/staffroutes.go
+++ b/pages/index/staffroutes/staffroutes.go
@@ -46,6 +46,7 @@ func Register(app *aero.Application) {
 	editorFilterable("/editor/anime/details/startdate", filteranime.StartDate)
 	editorFilterable("/editor/anime/details/enddate", filteranime.EndDate)
 	editorFilterable("/editor/anime/details/episodelength", filteranime.EpisodeLength)
+	editorFilterable("/editor/anime/details/episodecount", filteranime.EpisodeCount)
 	editorFilterable("/editor/anime/details/source", filteranime.Source)
 	editorFilterable("/editor/anime/details/status", filteranime.Status)