feat: Add extended API endpoints support

Author: JohanDevl

cmd/export_trakt/main.go

@@ -15,7 +15,7 @@ import (
 func main() {
 	// Parse command line flags
 	configPath := flag.String("config", "config/config.toml", "Path to configuration file")
-	exportType := flag.String("export", "watched", "Type of export (watched, collection, shows, all)")
+	exportType := flag.String("export", "watched", "Type of export (watched, collection, shows, ratings, watchlist, all)")
 	flag.Parse()
 
 	// Initialize logger
@@ -65,13 +65,19 @@ func main() {
 		exportCollection(traktClient, letterboxdExporter, log)
 	case "shows":
 		exportShows(traktClient, letterboxdExporter, log)
+	case "ratings":
+		exportRatings(traktClient, letterboxdExporter, log)
+	case "watchlist":
+		exportWatchlist(traktClient, letterboxdExporter, log)
 	case "all":
 		exportWatchedMovies(traktClient, letterboxdExporter, log)
 		exportCollection(traktClient, letterboxdExporter, log)
 		exportShows(traktClient, letterboxdExporter, log)
+		exportRatings(traktClient, letterboxdExporter, log)
+		exportWatchlist(traktClient, letterboxdExporter, log)
 	default:
 		log.Error("errors.invalid_export_type", map[string]interface{}{"type": *exportType})
-		fmt.Printf("Invalid export type: %s. Valid types are 'watched', 'collection', 'shows', or 'all'\n", *exportType)
+		fmt.Printf("Invalid export type: %s. Valid types are 'watched', 'collection', 'shows', 'ratings', 'watchlist', or 'all'\n", *exportType)
 		os.Exit(1)
 	}
 
@@ -144,4 +150,42 @@ func exportShows(client *api.Client, exporter *export.LetterboxdExporter, log lo
 		log.Error("export.export_failed", map[string]interface{}{"error": err.Error()})
 		os.Exit(1)
 	}
+}
+
+func exportRatings(client *api.Client, exporter *export.LetterboxdExporter, log logger.Logger) {
+	// Get ratings
+	log.Info("export.retrieving_ratings", nil)
+	ratings, err := client.GetRatings()
+	if err != nil {
+		log.Error("errors.api_request_failed", map[string]interface{}{"error": err.Error()})
+		os.Exit(1)
+	}
+
+	log.Info("export.ratings_retrieved", map[string]interface{}{"count": len(ratings)})
+
+	// Export ratings
+	log.Info("export.exporting_ratings", nil)
+	if err := exporter.ExportRatings(ratings); err != nil {
+		log.Error("export.export_failed", map[string]interface{}{"error": err.Error()})
+		os.Exit(1)
+	}
+}
+
+func exportWatchlist(client *api.Client, exporter *export.LetterboxdExporter, log logger.Logger) {
+	// Get watchlist
+	log.Info("export.retrieving_watchlist", nil)
+	watchlist, err := client.GetWatchlist()
+	if err != nil {
+		log.Error("errors.api_request_failed", map[string]interface{}{"error": err.Error()})
+		os.Exit(1)
+	}
+
+	log.Info("export.watchlist_retrieved", map[string]interface{}{"count": len(watchlist)})
+
+	// Export watchlist
+	log.Info("export.exporting_watchlist", nil)
+	if err := exporter.ExportWatchlist(watchlist); err != nil {
+		log.Error("export.export_failed", map[string]interface{}{"error": err.Error()})
+		os.Exit(1)
+	}
 } 

config/config.toml

@@ -4,6 +4,7 @@ client_id = ""
 client_secret = ""
 access_token = ""
 api_base_url = "https://api.trakt.tv"
+extended_info = "full"
 
 # Letterboxd Export Configuration
 [letterboxd]

locales/en.json

@@ -23,7 +23,15 @@
     "retrieving_watched_shows": "Retrieving watched shows from Trakt.tv",
     "shows_retrieved": "Retrieved {{.shows}} shows with {{.episodes}} episodes from Trakt.tv",
     "exporting_shows": "Exporting TV shows to CSV format",
-    "shows_export_complete": "Successfully exported {{.shows}} shows with {{.episodes}} episodes to {{.path}}"
+    "shows_export_complete": "Successfully exported {{.shows}} shows with {{.episodes}} episodes to {{.path}}",
+    "retrieving_ratings": "Retrieving movie ratings from Trakt.tv",
+    "ratings_retrieved": "Retrieved {{.count}} movie ratings from Trakt.tv",
+    "exporting_ratings": "Exporting movie ratings to Letterboxd format",
+    "ratings_export_complete": "Successfully exported {{.count}} movie ratings to {{.path}}",
+    "retrieving_watchlist": "Retrieving movie watchlist from Trakt.tv",
+    "watchlist_retrieved": "Retrieved {{.count}} movies from your Trakt.tv watchlist",
+    "exporting_watchlist": "Exporting movie watchlist to Letterboxd format",
+    "watchlist_export_complete": "Successfully exported {{.count}} watchlist movies to {{.path}}"
   },
   "errors": {
     "config_load_failed": "Failed to load configuration: {{.error}}",
@@ -32,6 +40,8 @@
     "file_create_failed": "Failed to create file: {{.error}}",
     "log_file_failed": "Failed to set log file: {{.error}}",
     "translator_failed": "Failed to initialize translator: {{.error}}",
-    "invalid_export_type": "Invalid export type: {{.type}}"
+    "invalid_export_type": "Invalid export type: {{.type}}",
+    "api_url_parse_error": "Failed to parse API URL: {{.error}}",
+    "api_response_parse_failed": "Failed to parse API response: {{.error}}"
   }
 }

locales/fr.json

@@ -23,7 +23,15 @@
     "retrieving_watched_shows": "Récupération des séries TV vues depuis Trakt.tv",
     "shows_retrieved": "{{.shows}} séries avec {{.episodes}} épisodes récupérés depuis Trakt.tv",
     "exporting_shows": "Exportation des séries TV au format CSV",
-    "shows_export_complete": "{{.shows}} séries avec {{.episodes}} épisodes exportés avec succès vers {{.path}}"
+    "shows_export_complete": "{{.shows}} séries avec {{.episodes}} épisodes exportés avec succès vers {{.path}}",
+    "retrieving_ratings": "Récupération des évaluations de films depuis Trakt.tv",
+    "ratings_retrieved": "{{.count}} évaluations de films récupérées depuis Trakt.tv",
+    "exporting_ratings": "Exportation des évaluations de films au format Letterboxd",
+    "ratings_export_complete": "{{.count}} évaluations de films exportées avec succès vers {{.path}}",
+    "retrieving_watchlist": "Récupération de la liste de surveillance depuis Trakt.tv",
+    "watchlist_retrieved": "{{.count}} films récupérés depuis votre liste de surveillance Trakt.tv",
+    "exporting_watchlist": "Exportation de la liste de surveillance au format Letterboxd",
+    "watchlist_export_complete": "{{.count}} films de la liste de surveillance exportés avec succès vers {{.path}}"
   },
   "errors": {
     "config_load_failed": "Échec du chargement de la configuration : {{.error}}",
@@ -32,6 +40,8 @@
     "file_create_failed": "Échec de la création du fichier : {{.error}}",
     "log_file_failed": "Échec de la définition du fichier journal : {{.error}}",
     "translator_failed": "Échec de l'initialisation du traducteur : {{.error}}",
-    "invalid_export_type": "Type d'exportation invalide : {{.type}}"
+    "invalid_export_type": "Type d'exportation invalide : {{.type}}",
+    "api_url_parse_error": "Échec de l'analyse de l'URL de l'API : {{.error}}",
+    "api_response_parse_failed": "Échec de l'analyse de la réponse de l'API : {{.error}}"
   }
 }