openeggbert/youtubedl-frontend
1
0
Fork 0
mirror of https://github.com/openeggbert/youtubedl-frontend.git synced 2025-03-14 21:23:27 +01:00
Code Projects Releases Activity

Refactoring

Browse Source
This commit is contained in:
Robert Vokac 2024-07-22 18:47:48 +02:00
parent 7e65b4363e
commit 76291ea038
No known key found for this signature in database
GPG Key ID: C459E1E4B4A986BB
6 changed files with 208 additions and 122 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/main/java/org/nanoboot/youtubedlfrontend/Arg.java Normal file
View File

@ -0,0 +1,34 @@
///////////////////////////////////////////////////////////////////////////////////////////////
// youtubedl-frontend: Tool generating html pages for Archive Box.
// Copyright (C) 2024 the original author or authors.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; version 2
// of the License only.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
///////////////////////////////////////////////////////////////////////////////////////////////
package org.nanoboot.youtubedlfrontend;
import lombok.AllArgsConstructor;
import lombok.Data;
/**
*
* @author robertvokac
*/
@Data
@AllArgsConstructor
public class Arg {
private ArgType argType;
private String value;
}

44
src/main/java/org/nanoboot/youtubedlfrontend/ArgType.java Normal file
View File

@ -0,0 +1,44 @@
///////////////////////////////////////////////////////////////////////////////////////////////
// youtubedl-frontend: Tool generating html pages for Archive Box.
// Copyright (C) 2024 the original author or authors.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; version 2
// of the License only.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
///////////////////////////////////////////////////////////////////////////////////////////////
package org.nanoboot.youtubedlfrontend;
import lombok.Getter;
/**
*
* @author robertvokac
*/
public enum ArgType {
VIDEO("video", null),
CHANNEL("channel", null),
VIDEOS_PER_ROW("videos-per-row", "4"),
ALWAYS_GENERATE_METADATA("always-generate-metadata", "true"),
ALWAYS_GENERATE_HTML_FILES("always-generate-html-files", "true"),
THUMBNAIL_AS_BASE64("thumbnail-as-base64", "false"),
THUMBNAIL_LINKS_TO_YOUTUBE("thumbnail-links-to-youtube", "false");
@Getter
private String name;
@Getter
private String defaultValue;
ArgType(String name, String defaultValue) {
this.name = name;
this.defaultValue = defaultValue;
}
}

89
src/main/java/org/nanoboot/youtubedlfrontend/Args.java Normal file
View File

@ -0,0 +1,89 @@
///////////////////////////////////////////////////////////////////////////////////////////////
// youtubedl-frontend: Tool generating html pages for Archive Box.
// Copyright (C) 2024 the original author or authors.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; version 2
// of the License only.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
///////////////////////////////////////////////////////////////////////////////////////////////
package org.nanoboot.youtubedlfrontend;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Stream;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.ToString;
/**
*
* @author robertvokac
*/
@Data
@AllArgsConstructor
@ToString
public class Args {
public static final String TWO_DASHES = "--";
private final Map<ArgType, Arg> map = new HashMap<>();
public Args(String[] args) {
if (args.length > 0) {
for (int i = 0; i < args.length; i++) {
String arg = args[i];
if (i == 0 && !arg.startsWith(TWO_DASHES)) {
continue;
}
Optional<ArgType> argType = Stream.of((ArgType.values()))
.filter(a -> arg.equals(TWO_DASHES + a.getName())).findFirst();
if (argType.isPresent()) {
i++;
final ArgType argTypeGet = argType.get();
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for " + TWO_DASHES + argTypeGet.getName());
}
String value = args[i];
if (argTypeGet == ArgType.VIDEOS_PER_ROW) {
int argVideosPerRow = Integer.parseInt(args[i]);
if (argVideosPerRow < 2) {
value = "0";
}
}
map.put(argTypeGet, new Arg(argTypeGet, value));
}
}
}
}
public Optional<String> getString(ArgType argType) {
if (!map.containsKey(argType)) {
return (argType.getDefaultValue() == null || argType.getDefaultValue().isEmpty()) ? Optional.empty() : Optional.of(argType.getDefaultValue());
}
return Optional.of(map.get(argType).getValue());
}
public Optional<Boolean> getBoolean(ArgType argType) {
Optional<String> o = getString(argType);
return o.isPresent() ? Optional.of(Utils.convertStringToBoolean(o.get())) : Optional.empty();
}
public Optional<Integer> getInteger(ArgType argType) {
Optional<String> o = getString(argType);
return o.isPresent() ? Optional.of(Integer.valueOf(o.get())) : Optional.empty();
}
}

132
src/main/java/org/nanoboot/youtubedlfrontend/Main.java
View File

@ -33,6 +33,7 @@ import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import static org.nanoboot.youtubedlfrontend.Args.TWO_DASHES;
/**
* @author <a href="mailto:robertvokac@nanoboot.org">Robert Vokac</a>
@ -42,118 +43,23 @@ public class Main {
private static int iii = 0;
private static int internalStaticVariableVideoNumberPerRow = 0;
public static boolean argAlwaysGenerateMetadata = true;
public static boolean argAlwaysGenerateHtmlFiles = true;
public static boolean thumbnailAsBase64 = false;
public static boolean thumbnailLinksToYoutube = false;
public static int argVideosPerRow = 4;
public static int THUMBNAIL_WIDTH = 250;
public static String argVideo;
public static String argChannel;
public static void main(String[] args) throws IOException, InterruptedException {
System.out.println("youtubedlfrontend - HTML generator\n");
//args = "/rv/databig/youtube --_video UDpsz1yIwiw --always-generate-metadata 1 --always-generate-html-files 1 --videos-per-row 4".split(" ");
if (args.length < 1) {
//System.err.println("At least one argument is expected, but the count of arguments is: " + args.length + ".");
String argsS = "/rv/blupi/archivebox --_video UDpsz1yIwiw --always-generate-metadata 0 "
+ " --always-generate-html-files 1 --videos-per-row 4 --thumbnail-links-to-youtube 1"
+ " --always-generate-html-files 0 --videos-per-row 4 --thumbnail-links-to-youtube 1"
+ " --thumbnail-as-base64 1";
args = argsS.split(" ");
//System.exit(1);
}
argVideo = "";
argChannel = "";
if (args.length > 0) {
for (int i = 0; i < args.length; i++) {
String arg = args[i];
if (i == 0 && !arg.startsWith(TWO_DASHES)) {
continue;
}
if (arg.equals("--video")) {
i++;
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for --video");
}
argVideo = args[i];
}
if (arg.equals("--channel")) {
i++;
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for --channel");
}
argChannel = args[i];
}
if (arg.equals("--videos-per-row")) {
i++;
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for --videos-per-row");
}
argVideosPerRow = Integer.parseInt(args[i]);
if (argVideosPerRow < 2) {
argVideosPerRow = 0;
}
}
if (arg.equals("--always-generate-metadata")) {
i++;
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for --always-generate-metadata");
}
String s = args[i];
try {
argAlwaysGenerateMetadata = Utils.convertStringToBoolean(s);
} catch (Exception e) {
throw new YoutubedlFrontendException("Invalid value for --always-generate-metadata");
}
}
if (arg.equals("--always-generate-html-files")) {
i++;
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for --always-generate-html-files");
}
String s = args[i];
try {argAlwaysGenerateHtmlFiles = Utils.convertStringToBoolean(s);} catch(Exception e) {
throw new YoutubedlFrontendException("Invalid value for --always-generate-html-files");
}
}
if (arg.equals("--thumbnail-as-base64")) {
i++;
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for --thumbnail-as-base64");
}
String s = args[i];
try {thumbnailAsBase64 = Utils.convertStringToBoolean(s);} catch(Exception e) {
throw new YoutubedlFrontendException("Invalid value for --thumbnail-as-base64");
}
}
if (arg.equals("--thumbnail-links-to-youtube")) {
i++;
if (i >= args.length) {
throw new YoutubedlFrontendException("Fatal error: missing value for --thumbnail-links-to-youtube");
}
String s = args[i];
try {
thumbnailLinksToYoutube = Utils.convertStringToBoolean(s);
} catch (Exception e) {
throw new YoutubedlFrontendException("Invalid value for --thumbnail-links-to-youtube");
}
}
}
}
Args argsInstance = new Args(args);
System.out.println(argsInstance.toString());
String workingDirectory = args.length > 0 && !args[0].startsWith(TWO_DASHES) ? args[0] : new File(".").getAbsolutePath();
File archiveBoxRootDirectory = new File(workingDirectory);
@ -167,16 +73,14 @@ public class Main {
//nothing to do
continue;
}
YoutubeVideo youtubeVideo = new YoutubeVideo(mediaDirectory);
if (!Main.argVideo.isBlank() && !youtubeVideo.getId().equals(Main.argVideo)) {
YoutubeVideo youtubeVideo = new YoutubeVideo(mediaDirectory, argsInstance.getBoolean(ArgType.ALWAYS_GENERATE_METADATA).get(), argsInstance.getString(ArgType.VIDEO).orElse(""));
if (argsInstance.getString(ArgType.VIDEO).isPresent() && !argsInstance.getString(ArgType.VIDEO).equals(youtubeVideo.getId())) {
continue;
}
if (!argVideo.isBlank() && !youtubeVideo.getId().equals(argVideo)) {
continue;
}
if (!argChannel.isBlank() && !youtubeVideo.getChannelId().equals(argChannel)) {
if (argsInstance.getString(ArgType.CHANNEL).isPresent() && !argsInstance.getString(ArgType.CHANNEL).equals(youtubeVideo.getChannelId())) {
continue;
}
i++;
System.out.println("\n\nFound video #" + i);
@ -244,6 +148,9 @@ public class Main {
<head>
<link rel="icon" type="image/x-icon" href="favicon.ico" sizes="16x16">
<title>Youtube videos</title>
<!-- Generated by: https://code.nanoboot.org/nanoboot/youtubedl-frontend -->
<style>
body {padding:20px;}
* {
@ -264,7 +171,7 @@ public class Main {
NumberFormat formatter = new DecimalFormat("#0.00");
channels.forEach(c -> {
sb.append("<h1>").append(c).append("</h1>\n");
sb.append("<div style=\"max-width:").append((Main.THUMBNAIL_WIDTH + 20) * Main.argVideosPerRow).append("px\"><a href =\"").append(channelUrls.get(c)).append("\">").append(channelUrls.get(c)).append("</a><div class=\"videos\">");
sb.append("<div style=\"max-width:").append((Main.THUMBNAIL_WIDTH + 20) * argsInstance.getInteger(ArgType.VIDEOS_PER_ROW).get()).append("px\"><a href =\"").append(channelUrls.get(c)).append("\">").append(channelUrls.get(c)).append("</a><div class=\"videos\">");
iii = 0;
internalStaticVariableVideoNumberPerRow = 0;
sb.append("<table>\n");
@ -277,7 +184,7 @@ public class Main {
sb.append("<td><div class=\"box\"><table style=\"margin:5px;max-width:")
.append(THUMBNAIL_WIDTH)
.append("px;\">\n<tr><td><a href=\"");
if (thumbnailLinksToYoutube) {
if (argsInstance.getBoolean(ArgType.THUMBNAIL_LINKS_TO_YOUTUBE).get()) {
sb.append("https://www.youtube.com/watch?v=").append(youtubeVideo.getId());
} else {
sb.append("videos/" + youtubeVideo.getId() + ".html");
@ -288,7 +195,7 @@ public class Main {
.append(youtubeVideo.getSnapshot())
.append("/media/mini-thumbnail.")
.append(youtubeVideo.getMiniThumbnailFormat()).toString();
if (thumbnailAsBase64) {
if (argsInstance.getBoolean(ArgType.THUMBNAIL_AS_BASE64).get()) {
try {
byte[] bytes = Files.readAllBytes(new File(archiveBoxRootDirectory + "/" + thumbnailPath).toPath());
System.out.println("###=" + archiveBoxRootDirectory + "/" + thumbnailPath);
@ -319,12 +226,12 @@ public class Main {
.append("</td></tr>\n");
youtubeVideo.setNumber(iii);
sb.append("</table></div></td>\n");
if (internalStaticVariableVideoNumberPerRow == argVideosPerRow) {
if (internalStaticVariableVideoNumberPerRow == argsInstance.getInteger(ArgType.VIDEOS_PER_ROW).get()) {
sb.append("<tr>");
internalStaticVariableVideoNumberPerRow = 0;
}
File videoHtmlFile = new File(videosDirectory, youtubeVideo.getId() + ".html");
if (!videoHtmlFile.exists() || argAlwaysGenerateHtmlFiles) {
if (!videoHtmlFile.exists() || argsInstance.getBoolean(ArgType.ALWAYS_GENERATE_HTML_FILES).get()) {
{
StringBuilder videoHtml = new StringBuilder("""
@ -484,7 +391,7 @@ public class Main {
}
}
});
if (internalStaticVariableVideoNumberPerRow < argVideosPerRow) {
if (internalStaticVariableVideoNumberPerRow < argsInstance.getInteger(ArgType.VIDEOS_PER_ROW).get()) {
sb.append("<tr>");
}
sb.append("</table>\n");
@ -501,6 +408,5 @@ public class Main {
System.out.println("[Warning] Snapshots without videos:");
YoutubeVideo.missingYoutubeVideos.forEach(s -> System.out.println(s));
}
private static final String TWO_DASHES = "--";
}

22
src/main/java/org/nanoboot/youtubedlfrontend/YoutubeComment.java
View File

@ -1,7 +1,21 @@
/*
* Click nbfs://nbhost/SystemFileSystem/Templates/Licenses/license-default.txt to change this license
* Click nbfs://nbhost/SystemFileSystem/Templates/Classes/Class.java to edit this template
*/
///////////////////////////////////////////////////////////////////////////////////////////////
// youtubedl-frontend: Tool generating html pages for Archive Box.
// Copyright (C) 2024 the original author or authors.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; version 2
// of the License only.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
///////////////////////////////////////////////////////////////////////////////////////////////
package org.nanoboot.youtubedlfrontend;
import java.util.ArrayList;

9
src/main/java/org/nanoboot/youtubedlfrontend/YoutubeVideo.java
View File

@ -23,7 +23,6 @@ import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import java.util.Properties;
import javax.imageio.ImageIO;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
@ -62,9 +61,9 @@ public class YoutubeVideo implements Comparable<YoutubeVideo> {
//
public static final List<String> missingYoutubeVideos = new ArrayList<>();
public YoutubeVideo(File mediaDirectory) throws InterruptedException, IOException {
public YoutubeVideo(File mediaDirectory, boolean argAlwaysGenerateMetadata, String argVideo) throws InterruptedException, IOException {
File metadataFile = new File(mediaDirectory, "metadata");
if (!Main.argAlwaysGenerateMetadata && metadataFile.exists()) {
if (argAlwaysGenerateMetadata && metadataFile.exists()) {
YoutubeVideo yv = new YoutubeVideo();
//new ObjectMapper().readValue(Utils.readTextFromFile(metadataFile), YoutubeVideo.class);
@ -75,7 +74,7 @@ public class YoutubeVideo implements Comparable<YoutubeVideo> {
id = properties.getProperty("id");
if (!Main.argVideo.isBlank() && !id.equals(Main.argVideo)) {
if (!argVideo.isBlank() && !id.equals(argVideo)) {
return;
}
snapshot = properties.getProperty("snapshot");
@ -212,7 +211,7 @@ public class YoutubeVideo implements Comparable<YoutubeVideo> {
.filter(f
-> (f.getName().endsWith("." + ext))
|| (f.getName().endsWith(".mp4"))
|| (f.getName().endsWith(".mkv"))
|| (f.getName().endsWith(".mkv"))|| (f.getName().endsWith(".webm"))
)
// .filter(
// f -> !f.getName().endsWith(".description")