code cleanup + add inital crossplatform implementationcross-platform-clean

1 files changed, 220 insertions, 0 deletions

diff --git a/src/main/java/Downloader.java b/src/main/java/Downloader.java
new file mode 100644
index 0000000..d6a1f13
--- /dev/null
+++ b/src/main/java/Downloader.java
@@ -0,0 +1,220 @@
+import org.jaudiotagger.audio.AudioFile;
+import org.jaudiotagger.audio.AudioFileIO;
+import org.jaudiotagger.tag.FieldKey;
+import org.jaudiotagger.tag.Tag;
+import org.jaudiotagger.tag.datatype.Artwork;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.text.SimpleDateFormat;
+
+import javax.swing.*;
+import java.io.File;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Date;
+
+public class Downloader {
+    private String outputDirectory;
+    private JTextArea outputArea;
+    private boolean removeNonAlphaNumeric;
+    String formats[] = {"maxresdefault.jpg", "mqdefault.jpg", "hqdefault.jpg"};
+    FileUtility fileUtil = new FileUtility();
+    public Downloader(String outputDirectory, JTextArea outputArea, boolean removeNonAlphaNumeric){
+        this.outputDirectory = outputDirectory;
+        this.outputArea = outputArea;
+        this.removeNonAlphaNumeric = removeNonAlphaNumeric;
+    }
+
+    public Downloader(String outputDirectory, JTextArea outputArea){
+        this.outputDirectory = outputDirectory;
+        this.outputArea = outputArea;
+        this.removeNonAlphaNumeric = false;
+    }
+
+    /**
+     * Tag mp3 with title, uploader, and image
+     * @param uploader Uploader of the video
+     * @param title Title of the video
+     * @param imageUrl URL of to the thumbnail image
+     */
+    public boolean tagMp3InDir(String uploader, String title, String imageUrl, String filePath) {//Tag mp3 file in downloaded directory
+        try {
+            AudioFile f = AudioFileIO.read(fileUtil.findFileWithType(filePath, "mp3"));
+            System.out.println("File found at: " + fileUtil.findFileWithType(filePath, "mp3"));
+            Tag tag = f.getTag();
+            System.out.println("Uploader: " + uploader);
+            System.out.println("Title: " + title);
+            tag.setField(FieldKey.ARTIST, uploader);
+            tag.setField(FieldKey.TITLE, title);
+            fileUtil.downloadImage(imageUrl, "img.jpg", formats);
+            Artwork cover = Artwork.createArtworkFromFile(new File("img.jpg"));
+            tag.addField(cover);
+            f.commit();
+            fileUtil.deleteFile("img.jpg");
+        }
+        catch(Exception e){
+            JOptionPane.showMessageDialog(
+                    null,
+                    "Error occured while tagging mp3. Check your program version",
+                    "ERROR", JOptionPane.ERROR_MESSAGE);
+            e.printStackTrace();
+            return false;
+        }
+        return true;
+    }
+
+
+    /**
+     * Relays the console output from the CMD to the outputArea
+     * @param p The process to relay to the outputArea from
+     */
+    public void relayConsole(Process p) {
+        BufferedReader r = new BufferedReader(new InputStreamReader(p.getInputStream()));
+        String cmd_line;
+        while (true) {
+            try {
+                cmd_line = r.readLine();
+                if (cmd_line == null) {
+                    break;
+                }
+                System.out.println(cmd_line);
+                outputArea.setText(outputArea.getText() + "\n" + cmd_line);
+            }
+            catch (IOException e) {
+                System.out.println("Error while relaying from CMD");
+            }
+        }
+    }
+
+
+    /*
+    Download a part of a video
+     */
+    public boolean download(String url, String stamp){
+        ArrayList<String> times = new ArrayList<>(Arrays.asList(stamp.split("-")));
+        String startTime = times.get(0);
+        String endTime = times.get(1);
+        int startSec = timestampToSeconds(startTime);
+        int endSec = timestampToSeconds(endTime);
+        String ytDlpExecutable = "yt-dlp" + (System.getProperty("os.name").startsWith("Windows") ? ".exe" : "");
+        String ffmpegExecutable = "ffmpeg" + (System.getProperty("os.name").startsWith("Windows") ? ".exe" : "");
+        try {
+            ProcessBuilder builder = new ProcessBuilder(
+                    ytDlpExecutable,
+                    "-vU",
+                    "-f", "(bestaudio)",
+                    "--external-downloader", ffmpegExecutable,
+                    "--external-downloader-args", "ffmpeg_i:-ss " + startSec + " -to " + endSec,
+                    "--output", "download/%(title)s_%(id)s.webm",
+                    "--write-info-json",
+                    url
+            );
+            builder.directory(new File(outputDirectory));
+            builder.redirectErrorStream(true);
+            Process p = builder.start();
+            relayConsole(p);
+        } catch (Exception e) {
+            JOptionPane.showMessageDialog(null, "An error occurred while downloading using yt-dlp: " + e.getMessage(), "Error", JOptionPane.ERROR_MESSAGE);
+            e.printStackTrace();
+        }
+        File downloadedFile = fileUtil.findFileWithType(outputDirectory+"/download", "webm");
+        try{
+            ProcessBuilder ffmpegBuilder = new ProcessBuilder(
+                    ffmpegExecutable,
+                    "-i", downloadedFile.getAbsolutePath(),
+                    "-vn",
+                    "-ab", "128k",
+                    "-ar", "44100",
+                    "-y",
+                    downloadedFile.getAbsolutePath().replace(".webm", ".mp3")
+            );
+            ffmpegBuilder.directory(new File(outputDirectory));
+            ffmpegBuilder.redirectErrorStream(true);
+            Process p = ffmpegBuilder.start();
+            relayConsole(p);
+        }
+        catch(Exception e){
+            JOptionPane.showMessageDialog(null, "An error occurred while converting to mp3: " + e.getMessage(), "Error", JOptionPane.ERROR_MESSAGE);
+            e.printStackTrace();
+        }
+        fileUtil.deleteFile(downloadedFile.getAbsolutePath()); // Delete old webm file
+
+        String info[] = fileUtil.parseInfoJSON(fileUtil.jsonToString(fileUtil.findJsonFile(outputDirectory+"/download")));
+        String uploader = info[1];
+        String title = info[0];
+        String urlID = info[2];
+        String imageUrl = "https://img.youtube.com/vi/" + urlID+"/";
+
+        if (removeNonAlphaNumeric) {
+            String newTitle = fileUtil.removeNonAlphaNumeric(title);
+            String newUploader = fileUtil.removeNonAlphaNumeric(uploader);
+            String newFileName = newTitle + "_" + newUploader + "_"+stamp+".mp3";
+            File oldFile = new File(outputDirectory + "/" + title + "[" + urlID + "].mp3");
+            File newFile = new File(outputDirectory + "/" + newFileName);
+            oldFile.renameTo(newFile);
+        }
+        tagMp3InDir(uploader, title, imageUrl, outputDirectory+"/download");
+        fileUtil.deleteFile(fileUtil.findJsonFile(outputDirectory+"/download"));
+        File downloadedMp3 = fileUtil.findFileWithType(outputDirectory+"/download", "mp3");
+        downloadedMp3.renameTo(new File(outputDirectory+"/"+downloadedMp3.getName()));
+        fileUtil.deleteFile(outputDirectory+"/download");
+        return true;
+    }
+
+    public boolean download(String url){
+        String ytDlpExecutable = "yt-dlp" + (System.getProperty("os.name").startsWith("Windows") ? ".exe" : "");
+        try {
+            String[] command = {ytDlpExecutable, "-f", "ba", "-x", "--audio-format", "mp3", "--write-info-json", url, "-o", "%(title)s[%(id)s].%(ext)s"};
+            ProcessBuilder processBuilder = new ProcessBuilder(command);
+            processBuilder.directory(new File(outputDirectory));
+            Process process = processBuilder.start();
+            relayConsole(process);
+            process.waitFor();
+        } catch(Exception e){
+            JOptionPane.showMessageDialog(
+                    null,
+                    "Error occured while downloading mp3. Check that you have yt-dlp installed",
+                    "ERROR", JOptionPane.ERROR_MESSAGE);
+            return false;
+        }
+        String info[] = fileUtil.parseInfoJSON(fileUtil.jsonToString(fileUtil.findJsonFile(outputDirectory)));
+        String uploader = info[1];
+        String title = info[0];
+        String urlID = info[2];
+        String imageUrl = "https://img.youtube.com/vi/" + urlID + "/";
+
+        if (removeNonAlphaNumeric) {
+            String newTitle = fileUtil.removeNonAlphaNumeric(title);
+            String newUploader = fileUtil.removeNonAlphaNumeric(uploader);
+            String newFileName = newTitle + "_" + newUploader + ".mp3";
+            File oldFile = new File(outputDirectory + "/" + title + "[" + urlID + "].mp3");
+            File newFile = new File(outputDirectory + "/" + newFileName);
+            oldFile.renameTo(newFile);
+        }
+        fileUtil.deleteFile(fileUtil.findJsonFile(outputDirectory));
+        tagMp3InDir(uploader, title, imageUrl, outputDirectory);
+        return true;
+    }
+
+
+    public static int timestampToSeconds(String timestamp){
+        int totalSeconds = 0;
+        try {
+            SimpleDateFormat sdf = new SimpleDateFormat("HH:mm:ss");
+            Date date = sdf.parse(timestamp);
+            int hours = date.getHours();
+            int minutes = date.getMinutes();
+            int seconds = date.getSeconds();
+            totalSeconds = hours * 3600 + minutes * 60 + seconds;
+            System.out.println(totalSeconds);
+        }
+        catch (Exception e){
+            System.out.println("Error converting timestamp to seconds");
+            e.printStackTrace();
+        }
+        return totalSeconds;
+
+    }
+}