diff options
Diffstat (limited to 'src/main/java/Downloader.java')
| -rw-r--r-- | src/main/java/Downloader.java | 220 |
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; + + } +} |