vidload 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e25ffa5a4bdbfdadd4ee261da31fab7f78e15f6865196bec9ed6824d827918cc
-  data.tar.gz: 85cbad777cffda334cf26c8992fe42b8a209eaed2b6210cea733e1f54f7cadcb
+  metadata.gz: d7f2ad6f46978b5cad7216f00d0c9a74bb2cd7ec537ddab8145d9f8bd0914cde
+  data.tar.gz: 0146cabc4cb8f7f39c2b3adaa76293eefa86acf4233890a5116f4988ef590fdd
 SHA512:
-  metadata.gz: 898294dbb222d1d84f31cde9b4b30f7689de6cd18bacf2dd452496346be24f83245d24f52c1125291256020533e5b759091f4c67a7d7a9b47ed7457416a29633
-  data.tar.gz: 6bfae7fb0047df91c7aecfc6cc0f6302f895b6872e52dedad7751a2fbb39f97ee29966a437a5f1ba8b1dfd8324c34704e846cd930e1768324e8a2f0d64343bde
+  metadata.gz: c195fbe2098e82a3da3317f61cacf1c10f667a6195c0b4a0ebd9d38eee8cec222703cf00aeb1cba48ff14b9d0a98ffc6334ef5a14f4b74aad259683db471ff4c
+  data.tar.gz: dd4a308b63da84aa3dcfc8796556f8f48fc53eeabb73c92d39ed2689181bc72b1bb99c776e67cd352db415d405b759975d40c3c85ae877bf898c0644bd95dadf

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,130 @@
+# 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)
+          @max_lines = IO.console.winsize[0]
+          @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)
+          @lines = [''] * @max_lines
+          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.0'
 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.0
 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: []