feat(imdb): format title runtime

Jisin0 · Sep 16, 2024 · 1c618ae · 1c618ae
1 parent 0242b9d
commit 1c618ae
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/imdb/getmovie.go b/imdb/getmovie.go
@@ -204,6 +204,10 @@ func (c *ImdbClient) GetMovie(id string) (*Movie, error) {
 		movie.ReleaseYear = htmlquery.InnerText(releaseYearNode)
 	}
 
+	if movie.Runtime != "" {
+		movie.Runtime = parseIMDbDuration(movie.Runtime)
+	}
+
 	// Cache data for next time
 	if !c.disabledCaching {
 		err := c.cache.MovieCache.Save(id, movie)

diff --git a/imdb/helpers.go b/imdb/helpers.go
@@ -0,0 +1,13 @@
+package imdb
+
+import "strings"
+
+// parseIMDbDuration parses the time returned from imdb into human-readable format. for ex: PT2H1M -> 2h 1min
+func parseIMDbDuration(s string) string {
+	s = strings.ReplaceAll(s, "PT", "")
+	s = strings.ReplaceAll(s, "H", "h ")
+	s = strings.ReplaceAll(s, "M", "min ")
+	s = strings.ReplaceAll(s, "S", "s")
+
+	return s
+}