vidload 0.3.2 → 0.3.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a24a8d21e0c6a4d1c29211c2fc730a34d7f27e1629cec415e3014820a7b13a2e
-  data.tar.gz: 64b502cf0d8b0ddb8f3c7e17246df47c8a04e415b91d67c167a941f8bcf98cc8
+  metadata.gz: a4ec46038fec427696fc8d8e1f86c84f3df2d0ea39663a3a39daf5dfcafcd2c5
+  data.tar.gz: f519d1682b9fb02c9eaf123b9cc68850bf120234b84281cb9fd175dc1c40ba07
 SHA512:
-  metadata.gz: 04e6bef3507ec3580f0ed69ddf427d9b6d9405ea7e46ddf35149e6babde4ee2f680670145bba4179282938dbc199e62bdd5124b3e2a06de5824025525c5d9a22
-  data.tar.gz: 36613f4b862a7adeccf1b5581049e40e9f7e0f3cc3e9602d3e0c919e6f39178784989f636f285949facbc731b5d333d0ecef56eea13f2f5cc807b1d6efd19a12
+  metadata.gz: a4f75f282726d7c831bd716c6b7bcdb9fd892832350a8fb8d84b5a0c93801bf347fa6e3aaaccfb6fccc5d5733130083cf82c3a6b90609ff23b28e763751f855d
+  data.tar.gz: aaf65273085bff9ad12b516c6a264c96b07f1246c989b01a479583a402d6a1097d169c51596a324147003a4f0a52b876ac7625ec532bd682699a03d5a5d40d0d

data/lib/vidload/mp2t/api.rb

@@ -5,12 +5,14 @@ require 'tty-spinner'
 require 'open3'
 require 'm3u8'
 require 'io/console'
+require 'fileutils'
 
 module Vidload
   module Mp2t
     module Api
-      DEMUXER_PATH = "#{__dir__}/remuxer.sh"
+      DEMUXER_PATH = "#{__dir__}/remuxer.sh".freeze
       VIDEO_DOWNLOADED_EVENT_QUEUE = Queue.new
+      VIDEO_START_DOWNLOAD_EVENT_QUEUE = Queue.new
       VIDEO_INDEX_EVENT_QUEUE = Queue.new
       ANSI_BOLD_WHITE = "\033[1;97m"
       ANSI_LIGHT_GREY = "\033[37m"
@@ -108,13 +110,19 @@ 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
+            browser = playwright.chromium.launch(headless: false)
             page = browser.new_page
 
             manage_video_download(page, *video_starter_callbacks)
-            wait_until_video_downloaded
 
-            browser.close
+            if wait_until_video_start_downloading(timeout: 10).nil?
+              puts 'Not possible to download video. Restarting new session'
+              browser.close
+              download_video(video_starter_callbacks: video_starter_callbacks)
+            else
+              wait_until_video_downloaded
+              browser.close
+            end
           end
         end
 
@@ -156,8 +164,14 @@ module Vidload
           VIDEO_DOWNLOADED_EVENT_QUEUE.pop
         end
 
+        def wait_until_video_start_downloading(timeout:)
+          VIDEO_START_DOWNLOAD_EVENT_QUEUE.pop(timeout: timeout)
+        end
+
         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]}",
                   @kwargs[:video_referer]) do |line|
             if (line.include?('hls @') || line.include?('https @')) && line.match?(/#{@kwargs[:ts_seg_pattern]}/i)
@@ -209,11 +223,7 @@ module Vidload
 
           _rows, cols = IO.console.winsize
           @lines.each do |line|
-            if line.length > cols
-              puts "#{line.slice(0, cols - 3)}..."
-            else
-              puts line
-            end
+            line.length > cols ? (puts "#{line.slice(0, cols - 3)}...") : (puts line)
           end
         end
 

data/lib/vidload/mp4/api.rb

@@ -107,9 +107,8 @@ module Vidload
         end
 
         def listen_to_video_starts(_page, response)
-          unless response.url.start_with?(@kwargs[:video_hub_url]) && response.headers['content-type']&.include?('video/mp4')
-            return
-          end
+          content_type = response.headers['content-type']
+          return unless response.url.start_with?(@kwargs[:video_hub_url]) && content_type&.include?('video/mp4')
 
           body = response.text
           File.open("video-#{@kwargs[:video_name]}.mp4", 'wb') do |f|

data/lib/vidload/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vidload
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.4
 platform: ruby
 authors:
 - Patacode <pata.codegineer@gmail.com>
@@ -119,7 +119,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: '3.3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="