misc: (*EpisodeRange).Split(*Episode) -- Splits a range

[go-anidb.git] / misc / episode.go

diff --git a/misc/episode.go b/misc/episode.go
index 72140d4abb602967f7ba90fe6e3d3c3d34a27469..1dbd1726e899236efb37bea898ad6389d8a7a9dd 100644 (file)
--- a/misc/episode.go
+++ b/misc/episode.go
@@ -134,6 +134,34 @@ func (ep *Episode) FormatLog(max int) string {
        return ep.Format(scale(max))
 }
 
+func (ep *Episode) Inc() {
+       if ep.Parts > 0 && ep.Part == ep.Parts-1 {
+               ep.IncNumber()
+       } else {
+               ep.Part++
+       }
+}
+
+func (ep *Episode) IncNumber() {
+       ep.Part = -1
+       ep.Parts = 0
+       ep.Number++
+}
+
+func (ep *Episode) Dec() {
+       if ep.Part > 0 {
+               ep.Part--
+       } else {
+               ep.DecNumber()
+       }
+}
+
+func (ep *Episode) DecNumber() {
+       ep.Part = -1
+       ep.Parts = 0
+       ep.Number--
+}
+
 // Parses a string in the usual AniDB API episode format and converts into
 // an Episode.
 func ParseEpisode(s string) *Episode {