Add support for custom headers in CLI arguments

Author: abdlhay

README.md

@@ -28,11 +28,13 @@ To download a video using AbyssVideoDownloader:
 1. Run the following command:
 
    ```bash
-   java -jar abyss-dl.jar [-o <output_file_path>]
+   java -jar mycli.jar [-H <header>] [--header <header>] [-o <output_file_path>]
    ```
 
-    - The `-o <output_file_path>` flag is optional. If not specified, the file will be saved in the Downloads directory with a default name.
-    - The `output_file_path` can either be the file name or the full path (including the file name) where you want the video to be saved.
+   - The `-H <header>` or `--header <header>` flag allows you to add HTTP headers in the format `Header-Name: Header-Value`.
+   - You can specify multiple headers by repeating the `-H` or `--header` flag.
+   - The `-o <output_file_path>` flag is optional. If not specified, the file will be saved in the Downloads directory with a default name.
+   - The `output_file_path` can either be the file name or the full path (including the file name) where you want the video to be saved.
 
 2. After running the command, the application will prompt you to input the video URL or the ID from Abyss.
 
@@ -50,7 +52,13 @@ To download a video using AbyssVideoDownloader:
    java -jar abyss-dl.jar
    ```
 
-3. After running the command, you will be prompted for the video URL:
+3. **Make an HTTP request with custom headers to download a video:**
+
+   ```bash
+   java -jar mycli.jar -H "Ref: Bearer TOKEN" --header "Referer: https://example.com" -o my_video.mp4
+   ```
+
+4. After running the command, you will be prompted for the video URL:
 
    ```
    enter the video URL or ID: https://abysscdn.com/?v=K8R6OOjS7

src/main/kotlin/CliArguments.kt

@@ -0,0 +1,41 @@
+package com.abmo
+
+class CliArguments(private val args: Array<String>) {
+
+    fun getHeaders(): Map<String, String> {
+        val headers = mutableMapOf<String, String>()
+        var i = 0
+
+        while (i < args.size) {
+            when (args[i]) {
+                "--header", "-H" -> {
+                    if (i + 1 < args.size) {
+                        val header = args[i + 1]
+                        val parts = header.split(":", limit = 2)
+                        if (parts.size == 2) {
+                            val key = parts[0].trim()
+                            val value = parts[1].trim()
+                            headers[key] = value
+                        } else {
+                            println("Invalid header format. Use 'Header-Name: Header-Value'")
+                        }
+                        i += 1
+                    }
+                }
+            }
+            i += 1
+        }
+
+        return headers
+    }
+
+    fun getOutputFileName(args: Array<String>): String? {
+        for (i in args.indices) {
+            if (args[i] == "-o" && i + 1 < args.size) {
+                return args[i + 1]
+            }
+        }
+        return null
+    }
+
+}

src/main/kotlin/Main.kt

@@ -14,13 +14,12 @@ suspend fun main(args: Array<String>) {
     val javaScriptRunner = JavaScriptRunner()
     val cryptoHelper = CryptoHelper(javaScriptRunner)
     val scraperService = ScraperService(cryptoHelper)
+    val cliArguments = CliArguments(args)
 
 
-    val outputFileName = if (args.isNotEmpty() && args[0] == "-o") {
-        args.getOrNull(1)
-    } else {
-        null
-    }
+    val outputFileName = cliArguments.getOutputFileName(args)
+    val headers = cliArguments.getHeaders()
+
 
     if (outputFileName != null && !isValidPath(outputFileName)) {
         exitProcess(0)
@@ -34,7 +33,7 @@ suspend fun main(args: Array<String>) {
         val videoID = scanner.nextLine().getVideoID()
 
         val url = "https://abysscdn.com/?v=$videoID"
-        val videoSources = scraperService.getVideoMetaData(url)?.sources
+        val videoSources = scraperService.getVideoMetaData(url, headers)?.sources
             ?.sortedBy { it?.label?.filter { char -> char.isDigit() }?.toInt() }
 
         if (videoSources == null) {
@@ -56,9 +55,9 @@ suspend fun main(args: Array<String>) {
 
             val config = if (outputFileName == null) {
                 println("\nOutput file not specified. The video will be saved in the 'Downloads' folder as '${url.getVideoID()}_$resolution.mp4'.")
-                Config(url, resolution)
+                Config(url, resolution, header = headers)
             } else {
-                Config(url, resolution, File(outputFileName))
+                Config(url, resolution, File(outputFileName), header = headers)
             }
             println("\nvideo with id $videoID and resolution $resolution being processed....")
             scraperService.downloadVideo(config)

src/main/kotlin/ScraperService.kt

@@ -22,7 +22,7 @@ class ScraperService(
     suspend fun downloadVideo(config: Config) {
         var totalBytesDownloaded = 0L
         val startTime = System.currentTimeMillis()
-        val video = getVideoMetaData(config.url)
+        val video = getVideoMetaData(config.url, config.header)
         val simpleVideo = video?.toSimpleVideo(config.resolution)
         val segmentBodies = generateSegmentsBody(simpleVideo)
         val segmentUrl = getSegmentUrl(video)
@@ -46,8 +46,10 @@ class ScraperService(
     }
 
 
-    fun getVideoMetaData(url: String): Video? {
-        val document = Unirest.get(url).asString().body
+    fun getVideoMetaData(url: String, headers: Map<String, String>?): Video? {
+        val document = Unirest.get(url)
+            .headers(headers)
+            .asString().body
         val encryptedVideoData = extractEncryptedVideoMetaData(document)
         return cryptoHelper.decodeEncryptedString(encryptedVideoData)
     }

src/main/kotlin/model/Config.kt

@@ -7,5 +7,6 @@ import java.io.File
 data class Config(
     val url: String,
     val resolution: String,
-    var outputFile: File? = File(systemDownloadFolder(), "${url.getVideoID()}_$resolution.mp4")
+    var outputFile: File? = File(systemDownloadFolder(), "${url.getVideoID()}_$resolution.mp4"),
+    val header: Map<String, String>? = null
 )