vidload 0.2.1 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e25ffa5a4bdbfdadd4ee261da31fab7f78e15f6865196bec9ed6824d827918cc
-  data.tar.gz: 85cbad777cffda334cf26c8992fe42b8a209eaed2b6210cea733e1f54f7cadcb
+  metadata.gz: b09d979bfbf863a31dc59989acb918041c1132e0001975383f4baf3d51f34681
+  data.tar.gz: 524ba3271e3c7e79505d7185d547c3c9413b0097e1be173ca8100f994fa36eaa
 SHA512:
-  metadata.gz: 898294dbb222d1d84f31cde9b4b30f7689de6cd18bacf2dd452496346be24f83245d24f52c1125291256020533e5b759091f4c67a7d7a9b47ed7457416a29633
-  data.tar.gz: 6bfae7fb0047df91c7aecfc6cc0f6302f895b6872e52dedad7751a2fbb39f97ee29966a437a5f1ba8b1dfd8324c34704e846cd930e1768324e8a2f0d64343bde
+  metadata.gz: 9549bd83c953a50dac61429be0118eab75b1a63feef3cfeb727efd4d45f2022a5b34bddcc5998eb29872a1269537dbe04a14c718e7bea1849dd53ca818e590f2
+  data.tar.gz: c0b17cc85d56bb53d5cd36a9f947c217ce38dd7e7e7c15aabd5f31b4cc6dbf6f1c6d41ec5a3fced8a7872667d8e88a69aff694c3326968dfeb228784461fdea5

data/lib/vidload/cli.rb

@@ -23,10 +23,27 @@ module Vidload
       process_mp2t(params)
     end
 
+    desc 'mp4 VIDEO_URL', 'download a mp4 video'
+    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
+    def mp4(video_url)
+      params = {
+        video_url: video_url,
+        **options
+      }
+
+      process_mp4(params)
+    end
+
     private
 
     def process_mp2t(params)
       raise NotImplementedError
     end
+
+    def process_mp4(params)
+      raise NotImplementedError
+    end
   end
 end

data/lib/vidload/mp4/api.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+require 'playwright'
+require 'open3'
+require 'io/console'
+
+module Vidload
+  module Mp4
+    module Api
+      VIDEO_DOWNLOADED_EVENT_QUEUE = Queue.new
+
+      class DownloaderBuilder
+        REQUIRED_ARGS = %i[video_url video_hub_url playwright_cli_path].freeze
+
+        def initialize
+          @kwargs = {}
+        end
+
+        def with_kwargs(**kwargs)
+          @kwargs = kwargs
+          self
+        end
+
+        def with_video_name(video_name)
+          @kwargs[:video_name] = video_name
+          self
+        end
+
+        def with_video_url(video_url)
+          @kwargs[:video_url] = video_url
+          self
+        end
+
+        def with_video_hub_url(video_hub_url)
+          @kwargs[:video_hub_url] = video_hub_url
+          self
+        end
+
+        def with_playwright_cli_path(playwright_cli_path)
+          @kwargs[:playwright_cli_path] = playwright_cli_path
+          self
+        end
+
+        def build
+          REQUIRED_ARGS.each do |required_arg|
+            raise ArgumentError, "#{required_arg} must be provided" unless @kwargs[required_arg]
+          end
+
+          @kwargs[:video_name] = @kwargs[:video_url].split('/').last unless @kwargs[:video_name]
+
+          Downloader.new(**@kwargs)
+        end
+      end
+
+      class Downloader
+        def initialize(**kwargs)
+          @kwargs = kwargs
+        end
+
+        def self.builder
+          DownloaderBuilder.new
+        end
+
+        def self.from_hash(hash)
+          builder.with_kwargs(**hash).build
+        end
+
+        # 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
+            page = browser.new_page
+
+            manage_video_download(page)
+            wait_until_video_downloaded
+
+            browser.close
+          end
+        end
+
+        def display_calling_args
+          puts 'Called with:'
+          @kwargs.each do |key, value|
+            puts "\t#{key}=#{value}"
+          end
+        end
+
+        def self.display_with_spinner(loading_msg = 'Loading...')
+          spinner = TTY::Spinner.new("[:spinner] #{loading_msg}")
+          spinner.auto_spin
+          yield
+          spinner.success('(done)')
+        end
+
+        private
+
+        def manage_video_download(page)
+          page.on('response', ->(resp) { listen_to_video_starts(page, resp) })
+          navigate_to_url(@kwargs[:video_url], page)
+        end
+
+        def wait_until_video_downloaded
+          Downloader.display_with_spinner('Downloading mp4 video...') do
+            VIDEO_DOWNLOADED_EVENT_QUEUE.pop
+          end
+        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
+
+          body = response.text
+          File.open("video-#{@kwargs[:video_name]}.mp4", 'wb') do |f|
+            f.write(body)
+          end
+          VIDEO_DOWNLOADED_EVENT_QUEUE << true
+        end
+
+        def navigate_to_url(url, page)
+          Downloader.display_with_spinner("Page #{url} loading...") do
+            page.goto(url)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vidload/mp4.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Vidload
+  module Mp4
+    require_relative 'mp4/api'
+  end
+end

data/lib/vidload/version.rb

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

data/lib/vidload.rb

@@ -3,5 +3,6 @@
 module Vidload
   require_relative 'vidload/version'
   require_relative 'vidload/mp2t'
+  require_relative 'vidload/mp4'
   require_relative 'vidload/cli'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vidload
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.1
 platform: ruby
 authors:
 - Patacode <pata.codegineer@gmail.com>
@@ -105,6 +105,8 @@ files:
 - lib/vidload/mp2t.rb
 - lib/vidload/mp2t/api.rb
 - lib/vidload/mp2t/remuxer.sh
+- lib/vidload/mp4.rb
+- lib/vidload/mp4/api.rb
 - lib/vidload/version.rb
 homepage:
 licenses: []