vidload 0.3.4 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a4ec46038fec427696fc8d8e1f86c84f3df2d0ea39663a3a39daf5dfcafcd2c5
-  data.tar.gz: f519d1682b9fb02c9eaf123b9cc68850bf120234b84281cb9fd175dc1c40ba07
+  metadata.gz: d991f1940c3b16c4d30982c802a24b21aa1e467fc0e5f3544bacce4b711d8e8a
+  data.tar.gz: 303da39859fa070f77c47da10b0d8070e4e8bd0edb525141d7e113d021ea81da
 SHA512:
-  metadata.gz: a4f75f282726d7c831bd716c6b7bcdb9fd892832350a8fb8d84b5a0c93801bf347fa6e3aaaccfb6fccc5d5733130083cf82c3a6b90609ff23b28e763751f855d
-  data.tar.gz: aaf65273085bff9ad12b516c6a264c96b07f1246c989b01a479583a402d6a1097d169c51596a324147003a4f0a52b876ac7625ec532bd682699a03d5a5d40d0d
+  metadata.gz: 1654b94190934e9d8d771564e33d601c2e823d01a93cc649574f7124617e1bca33c01a0dde97f285268d8e59be42637c7f43d9da41832c446da81d83f5d16bc8
+  data.tar.gz: 04fb1174a63b6675e290be8c05448b6db8e0bfc402bbc5e8a42b5840379a4c78c245f3b204f5b8420b3de34cd69b4e37157c1db5382b712c318762c581689246

data/lib/vidload/cli.rb

@@ -8,6 +8,8 @@ module Vidload
     method_option :video_name, type: :string, required: false
     method_option :author_name, type: :string, required: false
     method_option :output_dir, type: :string, required: false
+    method_option :non_headless, type: :boolean, default: false
+    method_option :author_dir, type: :boolean, default: false
     method_option :hls_url, type: :string, required: true
     method_option :master_playlist_name, type: :string, required: true
     method_option :playwright_cli_path, type: :string, required: true
@@ -27,6 +29,7 @@ module Vidload
     method_option :video_name, type: :string, required: false
     method_option :video_hub_url, type: :string, required: true
     method_option :playwright_cli_path, type: :string, required: true
+    method_option :non_headless, type: :boolean, default: false
     def mp4(video_url)
       params = {
         video_url: video_url,

data/lib/vidload/mp2t/api.rb

@@ -46,6 +46,21 @@ module Vidload
           self
         end
 
+        def with_output_dir(output_dir)
+          @kwargs[:output_dir] = output_dir
+          self
+        end
+
+        def non_headless?(non_headless)
+          @kwargs[:non_headless] = non_headless
+          self
+        end
+
+        def author_dir?(author_dir)
+          @kwargs[:author_dir] = author_dir
+          self
+        end
+
         def with_hls_url(hls_url)
           @kwargs[:hls_url] = hls_url
           self
@@ -76,17 +91,11 @@ module Vidload
           self
         end
 
-        def with_output_dir(output_dir)
-          @kwargs[:output_dir] = output_dir
-          self
-        end
-
         def build
           REQUIRED_ARGS.each do |required_arg|
             raise ArgumentError, "#{required_arg} must be provided" unless @kwargs[required_arg]
           end
 
-          @kwargs[:output_dir] = './' unless @kwargs[:output_dir]
           @kwargs[:video_name] = "#{@kwargs[:author_name]}_#{@kwargs[:video_name]}" if @kwargs[:author_name]
 
           Downloader.new(**@kwargs)
@@ -110,7 +119,7 @@ module Vidload
         # main func to be called in your own scripts defined under web/
         def download_video(video_starter_callbacks: [])
           Playwright.create(playwright_cli_executable_path: @kwargs[:playwright_cli_path]) do |playwright|
-            browser = playwright.chromium.launch(headless: false)
+            browser = playwright.chromium.launch(headless: !@kwargs[:non_headless])
             page = browser.new_page
 
             manage_video_download(page, *video_starter_callbacks)
@@ -171,8 +180,19 @@ module Vidload
         def trigger_video_download(video_url, seg_qty)
           VIDEO_START_DOWNLOAD_EVENT_QUEUE << true
           puts 'Video starts. Starting download...'
-          FileUtils.mkdir_p(@kwargs[:output_dir], mode: 0o755)
-          run_cmd(DEMUXER_PATH, video_url, "#{@kwargs[:output_dir]}#{@kwargs[:video_name]}",
+          video_parent_dirs = if @kwargs[:author_dir]
+                                if @kwargs[:output_dir]
+                                  "#{@kwargs[:output_dir]}/#{@kwargs[:author_name]}"
+                                else
+                                  @kwargs[:author_name]
+                                end
+                              else
+                                @kwargs[:output_dir]
+                              end
+
+          puts video_parent_dirs
+          FileUtils.mkdir_p(video_parent_dirs, mode: 0o755)
+          run_cmd(DEMUXER_PATH, video_url, "#{video_parent_dirs}/#{@kwargs[:video_name]}",
                   @kwargs[:video_referer]) do |line|
             if (line.include?('hls @') || line.include?('https @')) && line.match?(/#{@kwargs[:ts_seg_pattern]}/i)
               seg_nb = line.match(/#{@kwargs[:ts_seg_pattern]}/i)[:seg_nb]
@@ -181,7 +201,7 @@ module Vidload
             end
           end
           print "\r\e[2K"
-          puts "✔ Video downloaded successfully! Available in #{@kwargs[:output_dir]}#{@kwargs[:video_name]}.mp4"
+          puts "✔ Video downloaded successfully! Available in #{video_parent_dirs}/#{@kwargs[:video_name]}.mp4"
           VIDEO_DOWNLOADED_EVENT_QUEUE << true
         end
 

data/lib/vidload/mp4/api.rb

@@ -22,13 +22,13 @@ module Vidload
           self
         end
 
-        def with_video_name(video_name)
-          @kwargs[:video_name] = video_name
+        def with_video_url(video_url)
+          @kwargs[:video_url] = video_url
           self
         end
 
-        def with_video_url(video_url)
-          @kwargs[:video_url] = video_url
+        def with_video_name(video_name)
+          @kwargs[:video_name] = video_name
           self
         end
 
@@ -42,6 +42,11 @@ module Vidload
           self
         end
 
+        def non_headless?(non_headless)
+          @kwargs[:non_headless] = non_headless
+          self
+        end
+
         def build
           REQUIRED_ARGS.each do |required_arg|
             raise ArgumentError, "#{required_arg} must be provided" unless @kwargs[required_arg]
@@ -69,7 +74,7 @@ module Vidload
         # main func to be called in your own scripts defined under web/
         def download_video
           Playwright.create(playwright_cli_executable_path: @kwargs[:playwright_cli_path]) do |playwright|
-            browser = playwright.chromium.launch
+            browser = playwright.chromium.launch(headless: !@kwargs[:non_headless])
             page = browser.new_page
 
             manage_video_download(page)

data/lib/vidload/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Vidload
-  VERSION = '0.3.4'
+  VERSION = '0.5.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vidload
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  version: 0.5.0
 platform: ruby
 authors:
 - Patacode <pata.codegineer@gmail.com>