Plex implementation

Author: LumePart

src/main.go

@@ -190,8 +190,15 @@ func (cfg *Config) systemSetup() { // Verifies variables and does setup
 		if cfg.PlaylistDir == "" {
 			log.Fatal("PLAYLIST_DIR variable not set, exiting")
 		}
+	case "plex":
+		if (cfg.Creds.User == "" && cfg.Creds.Password == "") {
+			log.Fatal("USER and/or PASSWORD variable not set, exiting")
+		}
+		cfg.Creds.getPlexAuth()
+		cfg.Creds.PlexHeader() // Adds client headers
+		cfg.getPlexLibrary()
 	default:
-		log.Fatalf("system: %s not known, please use a supported system (jellyfin, mpd or subsonic)", cfg.System)
+		log.Fatalf("system: %s not known, please use a supported system (jellyfin, mpd, subsonic or plex)", cfg.System)
 }
 }
 

src/playlist.go

@@ -40,8 +40,12 @@ func checkTracks(cfg Config, tracks []Track) []Track { // Returns updated slice
 		switch cfg.System {
 		case "subsonic":
 			ID, _ = searchTrack(cfg, track)
+
 		case "jellyfin":
 			ID, _ = getJfSong(cfg, track)
+
+		case "plex":
+			ID, _ = searchPlexSong(cfg, track)
 		}
 		if ID != "" {
 			tracks[i].Present = true
@@ -83,12 +87,29 @@ func createPlaylist(cfg Config, tracks []Track) error {
 			return fmt.Errorf("failed to create playlist: %s", err.Error())
 		}
 		return nil
+
 	case "mpd": 
 
 		if err := createM3U(cfg, cfg.PlaylistName, tracks); err != nil {
 			return fmt.Errorf("failed to create M3U playlist: %s", err.Error())
 		}
 		return nil
+
+	case "plex":
+		if err := refreshPlexLibrary(cfg); err != nil {
+			return fmt.Errorf("createPlaylist(): %s", err.Error())
+		}
+		log.Printf("sleeping for %d minutes, to allow scan to complete..", cfg.Sleep)
+		time.Sleep(time.Duration(cfg.Sleep) * time.Minute)
+		ID, err := getPlexServer(cfg)
+		if err != nil {
+			return fmt.Errorf("createPlaylist(): %s", err.Error())
+		}
+		key, err := createPlexPlaylist(cfg, ID)
+		if err != nil {
+			return fmt.Errorf("createPlaylist(): %s", err.Error())
+		}
+		addToPlexPlaylist(cfg, key, ID, tracks)
 	}
 	return fmt.Errorf("something very strange happened")
 }
@@ -120,6 +141,14 @@ func handlePlaylistDeletion(cfg Config) error {
 			if err := os.Remove(cfg.PlaylistDir+cfg.PlaylistName+".m3u"); err != nil {
 				return err
 			}
+		case "plex":
+			key, err := searchPlexPlaylist(cfg)
+			if err != nil {
+				return err
+			}
+			if err := deletePlexPlaylist(cfg, key); err != nil {
+				return err
+			}
 		}
 	return nil
 }

src/plex.go

@@ -3,6 +3,7 @@ package main
 import (
 	"bytes"
 	"encoding/json"
+	"explo/debug"
 	"fmt"
 	"log"
 	"net/url"
@@ -221,21 +222,21 @@ func searchPlexPlaylist(cfg Config) (string, error) {
 		return "", fmt.Errorf("searchPlexPlaylist(): failed to parse response: %s", err.Error())
 	}
 
-	key, err := getPlexPlaylist(playlists, cfg.PlaylistName)
-	if err != nil {
-		return "", fmt.Errorf("getPlexPlaylist(): %s", err.Error())
+	key := getPlexPlaylist(playlists, cfg.PlaylistName)
+	if key == "" {
+		debug.Debug("no playlist found")
 	}
 	return key, nil
 }
 
-func getPlexPlaylist(playlists PlexPlaylist, playlistName string) (string, error) {
+func getPlexPlaylist(playlists PlexPlaylist, playlistName string) string {
 
 	for _, playlist := range playlists.MediaContainer.Metadata {
 		if playlist.Title == playlistName {
-			return playlist.Key, nil
+			return playlist.Key
 		}
 	}
-	return "", fmt.Errorf("failed to find playlist")
+	return ""
 }
 
 func getPlexServer(cfg Config) (string, error) {
@@ -255,8 +256,8 @@ func getPlexServer(cfg Config) (string, error) {
 	return server.MediaContainer.MachineIdentifier, nil
 }
 
-func createPlexPlaylist(cfg Config, libraryKey, machineID string) (string, error) {
-	params := fmt.Sprintf("/playlists?title=%s&type=audio&smart=0&uri=server://%s/com.plexapp.plugins.library/%s&X-Plex-Token=%s", cfg.PlaylistName, machineID, libraryKey, cfg.Creds.APIKey)
+func createPlexPlaylist(cfg Config, machineID string) (string, error) {
+	params := fmt.Sprintf("/playlists?title=%s&type=audio&smart=0&uri=server://%s/com.plexapp.plugins.library/%s&X-Plex-Token=%s", cfg.PlaylistName, machineID, cfg.Plex.LibraryID, cfg.Creds.APIKey)
 
 	body, err := makeRequest("POST", cfg.URL+params, nil, cfg.Creds.Headers)
 	if err != nil {
@@ -273,16 +274,15 @@ func createPlexPlaylist(cfg Config, libraryKey, machineID string) (string, error
 	return playlist.MediaContainer.Metadata[0].Key, nil
 }
 
-func addToPlexPlaylist(cfg Config, playlistKey, machineID string, tracks []Track) error {
+func addToPlexPlaylist(cfg Config, playlistKey, machineID string, tracks []Track) {
 	for _, track := range tracks {
 		params := fmt.Sprintf("/playlists/%s?uri=server://%s/com.plexapp.plugins.library/%s&X-Plex-Token=%s", playlistKey, machineID, track.ID, cfg.Creds.APIKey)
 
 		_, err := makeRequest("PUT", cfg.URL+params, nil, cfg.Creds.Headers)
 		if err != nil {
-			return fmt.Errorf("addToPlexPlaylist(): failed to add %s to playlist: %s", track.Title, err.Error())
+			log.Printf("addToPlexPlaylist(): failed to add %s to playlist: %s", track.Title, err.Error())
 		}
 	}
-	return nil
 }
 
 func deletePlexPlaylist(cfg Config, playlistKey string) error {