add m3u8script

2021-11-17 00:00:38 +01:00
parent 2f941f6eeb
commit b11b35cde2
9 changed files with 198 additions and 0 deletions
--- a/m3u8script/m3u8script.java
+++ b/m3u8script/m3u8script.java
@@ -0,0 +1,137 @@
+package de.eneller.scripts;
+
+import java.io.*;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.util.Arrays;
+import java.util.Scanner;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+public class m3u8script {
+    public static Logger logger = Logger.getLogger(Logger.GLOBAL_LOGGER_NAME);
+
+    public static void main(String[] args) {
+        logger.setLevel(Level.ALL);
+        if (args.length < 2) {
+            printHelp();
+            System.exit(1);
+        }
+        System.out.println(Arrays.toString(args));
+        String url = "";
+        String inputFile = "";
+        String prefix = "media_";
+        String output = "";
+        boolean m = false;
+        for (int i = 0; i < args.length - 1; i++) {
+            switch (args[i]) {
+                case "-single":
+                    url = args[i + 1].trim();
+                    break;
+                case "-multi":
+                    inputFile = args[i + 1].trim();
+                    m = true;
+                    break;
+                case "-prefix":
+                    prefix = args[i + 1].trim();
+                    break;
+                case "-output":
+                    output = args[i + 1].trim();
+                    break;
+            }
+        }
+        if (m == false) {
+            if (output.matches(".*\\.m3u8$") == false) {
+                output = output + "/";
+            }
+            parse(url, output, prefix);
+        } else {
+            String[] pathArr = output.split("/");
+
+            if (output.matches(".*/$") == false) {
+                output = output + "/";
+            }
+
+            try (Scanner scn = new Scanner(new File(inputFile));) {
+
+                String data;
+                int c = 0;
+                int a = 0;
+                while (scn.hasNextLine()) {
+                    a++;
+                    data = scn.nextLine();
+                    if (parse(data, output, prefix)) {
+                        c++;
+                    }
+                }
+                logger.log(Level.INFO, c + " out of " + a + " operations successfully completed. In case of errors, please check the log.");
+            } catch (FileNotFoundException e) {
+                logger.log(Level.SEVERE, "Couldnt find input file");
+            }
+        }
+
+    }
+
+    public static boolean parse(String m3u8WebPath, String outputPath, String matchPrefix) {
+        boolean b = true;
+        try {
+            URL url = new URL(m3u8WebPath);
+            Scanner scn = new Scanner((new BufferedReader(new InputStreamReader(url.openStream()))));
+            String[] splitWebPath = m3u8WebPath.split("/");
+            String m3u8WebPrefix = "";
+            for (int i = 0; i < splitWebPath.length - 1; i++) {
+                m3u8WebPrefix = m3u8WebPrefix + splitWebPath[i] + "/";
+            }
+            if (outputPath.matches(".*/$")) {
+                String generatedFileName = splitWebPath[splitWebPath.length - 2];
+                generatedFileName = generatedFileName.replaceAll("%20", "_");
+                generatedFileName = generatedFileName.replaceAll("\\.smil", ".m3u8");
+                outputPath = outputPath + generatedFileName;
+            }
+
+            File outm3u8 = new File(outputPath);
+
+            if (outm3u8.createNewFile()) {
+                logger.log(Level.INFO, "File created at " + outputPath);
+            } else {
+                logger.log(Level.WARNING, "File already exists, please specify a different path " + outputPath);
+                return false;
+            }
+
+            FileWriter outWriter = new FileWriter(outm3u8);
+            String data = "";
+            while (scn.hasNextLine()) {
+                data = scn.nextLine();
+                if (data.startsWith(matchPrefix)) {
+                    data = m3u8WebPrefix + data;
+                }
+                outWriter.write(data + "\n");
+            }
+            outWriter.close();
+            scn.close();
+        } catch (FileNotFoundException e) {
+            logger.log(Level.SEVERE, "Couldnt find .m3u8 file at specified path");
+        } catch (MalformedURLException e) {
+            logger.log(Level.SEVERE, "Couldnt read file at URL");
+        } catch (IOException e) {
+            logger.log(Level.SEVERE, "An error occured while trying to create the output file, please check that your path exists");
+        }
+
+
+        return true;
+    }
+
+    public static String getInput(String s) {
+        Scanner scn = new Scanner(System.in);
+        System.out.println(s);
+        return scn.nextLine();
+    }
+
+    public static void printHelp() {
+        System.out.println("Seems like you tried to run the script without specifying necessary parameters." + System.lineSeparator() +
+                "The following commands are available:" + System.lineSeparator() + System.lineSeparator() +
+                "Use -multi yourfile.txt to specify the absolute path to an input file containing a list of URLs" + System.lineSeparator() +
+                "Or use -single urlto.m3u8 in combination with -output /path/to/output/folderorfile" + System.lineSeparator() + System.lineSeparator() +
+                "Please go to https://github.com/eneller/m3u8script for additional help");
+    }
+}