RBToPOPM.java

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.net.URLDecoder;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadPoolExecutor;

/**
 * Created by bozhidar on 11.10.17.
 */
public class RBToPOPM {
    public static HashMap<Integer, Integer> starsToNumber = new HashMap<>();
    static {
        starsToNumber.putIfAbsent(0, 0);
        starsToNumber.putIfAbsent(1, 1);
        starsToNumber.putIfAbsent(2, 64);
        starsToNumber.putIfAbsent(3, 128);
        starsToNumber.putIfAbsent(4, 196);
        starsToNumber.putIfAbsent(5, 255);
    }
    
    public static final String ANSI_GREEN = "\u001B[32m";
    public static final String ANSI_YELLOW = "\u001B[33m";
    public static final String ANSI_RESET = "\u001B[0m";
    private static int total = 0;

    public static void main(String... args) throws ParserConfigurationException, IOException, SAXException, URISyntaxException, InterruptedException, ExecutionException {
        args = Files.readAllLines(Paths.get(args[0])).get(0).split(",;");
        HashMap<String, File> files = new HashMap<>();
        //Read all the files that are passed to the program

        for (String arg : args) {
            files.put(arg, new File(arg));
        }

        String database = System.getProperty("user.home") + "/.local/share/rhythmbox/rhythmdb.xml";
        int threads = Runtime.getRuntime().availableProcessors();
        ThreadPoolExecutor executor = (ThreadPoolExecutor) Executors.newFixedThreadPool(threads);
        System.out.println("Running program on " + threads + " threads");

        File db = new File(database);
        if (db.canRead()) {
            //Read the RB database
            DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
            DocumentBuilder builder = dbf.newDocumentBuilder();
            Document doc = builder.parse(db);

            doc.getDocumentElement().normalize();

            NodeList songs = doc.getElementsByTagName("entry");
            //For all songs
            for (int i = 0; i < songs.getLength(); i++) {
                Node song = songs.item(i);
                Element songElement = (Element) song;
                Node location = songElement.getElementsByTagName("location").item(0);
                //If the location is not null
                if (location != null) {
                    String val = location.getTextContent();
                    val = val.replace("file://", "");
                    val = URLDecoder.decode(val, "UTF-8");
                    //Get the filename and if it is one of the requested files
                    if (files.containsKey(val)) {
                        File toAddPOPM = files.get(val);

                        NodeList temp = songElement.getElementsByTagName("rating");
                        if (temp != null) {
                            //Increment the number of total processed files
                            total ++;
                            //Get the rating of RhythmBox
                            int RBRating;
                            Node stars = temp.item(0);
                            if (stars == null) {
                                RBRating = 0;
                            } else {
                                RBRating = Integer.parseInt(temp.item(0).getTextContent());
                            }
                            int WMPRating = starsToNumber.get(RBRating);

                            executor.execute(() -> {
                                try {
                                    addNewRating(toAddPOPM, RBRating, WMPRating);
                                } catch (IOException | InterruptedException e) {
                                    e.printStackTrace();
                                }
                            });
                        }
                    }
                }
            }
            executor.shutdown();
        }

        //System.out.println("\n" + ANSI_GREEN + "Processed " + total + " songs successfully!" + ANSI_RESET + "");
    }

    public static void addNewRating(File toAddPOPM, int RBRating, int WMPRating) throws IOException, InterruptedException {
        System.out.println(ANSI_YELLOW + "[Deleting obsolete POPM]\t" + ANSI_RESET + toAddPOPM.getAbsolutePath());
        ProcessBuilder pb = new ProcessBuilder("eyeD3", "--remove-frame=POPM", toAddPOPM.getAbsolutePath());
        Process p = pb.start();
        p.waitFor();

        //Add the RB rating
        System.out.println(ANSI_GREEN + "[Adding POPM] [Rating=" + RBRating + "]\t" + ANSI_RESET + toAddPOPM.getAbsolutePath());

        pb = new ProcessBuilder("eyeD3", "--add-popularity=RhythmBox:" + RBRating + ":0", toAddPOPM.getAbsolutePath());
        Process addRhythmBox = pb.start();
        addRhythmBox.waitFor();

        pb = new ProcessBuilder("eyeD3", "--add-popularity=Windows Media Player 9 Series:" + WMPRating + ":0", toAddPOPM.getAbsolutePath());
        Process addWMP = pb.start();
        addWMP.waitFor();
    }
}