From ce99b3a3c622914794b1b727720e6b5fe96fd5f6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?O=CC=81scar=20M=2E=20Lage?= <info@oscarmlage.com>
Date: Thu, 10 Oct 2024 14:22:15 +0200
Subject: [PATCH] Added --media argument

You can execute the script as it is to process all the kind of media content
OR you can select the media you want to only process that kind of content:

--media movies
--media series
--media books
--media games
--media music
---
 main.go | 37 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 36 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 813ce86..1efccc5 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"hugo-medialog/internal/books"
 	"hugo-medialog/internal/movies"
@@ -12,6 +13,37 @@ func main() {
 	// Load .env file
 	utils.LoadConfig()
 
+	// Parameter
+	media := flag.String("media", "", "Specify the media type to process: movies, series, books, games, or music")
+	flag.Parse()
+
+	if *media == "" {
+		processAll()
+	} else {
+		switch *media {
+		case "movies":
+			processMovies()
+		case "series":
+			processSeries()
+		case "books":
+			processBooks()
+		// case "games":
+		// 	processGames()
+		// case "music":
+		// 	processMusic()
+		default:
+			fmt.Printf("Invalid media type: %s. Please use movies, series, books, games, or music.\n", *media)
+		}
+	}
+}
+
+func processAll() {
+	processMovies()
+	processSeries()
+	processBooks()
+}
+
+func processMovies() {
 	// Movies
 	moviesList, err := movies.LoadMovies()
 	if err != nil {
@@ -22,7 +54,9 @@ func main() {
 	if err != nil {
 		fmt.Printf("Error processing movies: %v\n", err)
 	}
+}
 
+func processSeries() {
 	// Series
 	seriesList, err := series.LoadSeries()
 	if err != nil {
@@ -33,7 +67,9 @@ func main() {
 	if err != nil {
 		fmt.Printf("Error processing series: %v\n", err)
 	}
+}
 
+func processBooks() {
 	// Books
 	booksList, err := books.LoadBooks()
 	if err != nil {
@@ -44,5 +80,4 @@ func main() {
 	if err != nil {
 		fmt.Printf("Error processing books: %v\n", err)
 	}
-
 }