vidload 0.1.10 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3416ba597c831a7d6e1460bc0447fee8577e758279e036e820ccd967d93d46ef
-  data.tar.gz: a6d4c4c9e5bc8447cf68335029d60ca156c98ef6fc7bacd066b2335ca0e6966b
+  metadata.gz: 9f2f19f4a3e68894970ee46251031cabb28592bd242f0e1b7e2ecc8f5d40c3ee
+  data.tar.gz: 78b597dba867b0b988021cee4b8d35ea2a42d5df224af9cd052873c92687cada
 SHA512:
-  metadata.gz: '0891fcda0f3d59ea64ce309f35fea44456185bf0c6d56765e3c6d6ca60160819d19428e2b065e55c5c14efd3a7ef5dffa712ae02861e4e098c2d6e8bb8fdef75'
-  data.tar.gz: d5406c2f3e81b23038402e9f249c2359a96dc89edeba437e8274a2c4ada88bf6f1207373ab3205a2c9604bcc88bf0af8d8443e2f483cf93647c5934d780df466
+  metadata.gz: a4261172dc43b40d8f652f96103612d286825816c3d0acc57c2819364545158935c358a0b13b3cb7280d0b9281edc966cb8104c019306ad847548cf7de281120
+  data.tar.gz: ea0934e1a52dee15f4204a16c5e1bd08580742b634ad02f86cf17f776373b5b60216098ef94e07e4770903d341c482fd0c1f15b8b4bf738ab1614649588e90e9

data/lib/vidload/cli.rb

@@ -0,0 +1,36 @@
+require "thor"
+
+class Vidload::Cli < Thor
+  desc "mp2t VIDEO_URL", "download a mp2t containerized video"
+  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 :hls_url, type: :string, required: true
+  method_option :master_playlist_name, type: :string, required: true
+  method_option :playwright_cli_path, type: :string, required: true
+  method_option :video_referer, type: :string, required: true
+  method_option :ts_seg_pattern, type: :string, required: true
+  method_option :hls_index_pattern, type: :string, required: true
+  def mp2t(video_url)
+    params = {
+      video_url: video_url,
+      video_name: options[:video_name],
+      hls_url: options[:hls_url],
+      master_playlist_name: options[:master_playlist_name],
+      playwright_cli_path: options[:playwright_cli_path],
+      video_referer: options[:video_referer],
+      ts_seg_pattern: options[:ts_seg_pattern],
+      hls_index_pattern: options[:hls_index_pattern],
+      author_name: options[:author_name],
+      output_dir: options[:output_dir],
+    }
+
+    process_mp2t(params)
+  end
+
+  private 
+
+  def process_mp2t(params)
+    raise NotImplementedError
+  end
+end

data/lib/vidload/mp2t/api.rb

@@ -19,13 +19,14 @@ module Vidload::Mp2t::Api
     def initialize(
       video_url:,
       video_name:,
+      author_name:,
       hls_url:,
       master_playlist_name:,
       playwright_cli_path:,
       video_referer:,
       ts_seg_pattern:,
       hls_index_pattern:,
-      author_name:
+      output_dir:
     )
       raise ArgumentError, "video_url must be provided" unless video_url
       raise ArgumentError, "hls_url must be provided" unless hls_url
@@ -45,6 +46,7 @@ module Vidload::Mp2t::Api
       @max_lines = IO.console.winsize[0]
       @author_name = author_name
       @video_name = if @author_name then "#{@author_name}_#{video_name}" else nil end
+      @output_dir = output_dir || "./" 
     end
 
     def self.from_argv
@@ -61,6 +63,20 @@ module Vidload::Mp2t::Api
       )
     end
 
+    def self.from_hash(hash)
+      new(
+        video_url: hash[:video_url],
+        author_name: hash[:author_name],
+        video_name: hash[:video_name],
+        hls_url: hash[:hls_url],
+        master_playlist_name: hash[:master_playlist_name],
+        playwright_cli_path: hash[:playwright_cli_path],
+        video_referer: hash[:video_referer],
+        ts_seg_pattern: hash[:ts_seg_pattern],
+        hls_index_pattern: hash[:hls_index_pattern],
+      )
+    end
+
     # main func to be called in your own scripts defined under web/
     def download_video(video_starter_callbacks: [])
       Playwright.create(playwright_cli_executable_path: @playwright_cli_path) do |playwright|
@@ -122,7 +138,7 @@ module Vidload::Mp2t::Api
 
     def trigger_video_download(video_url, seg_qty)
       puts "Video starts. Starting download..."
-      run_cmd(DEMUXER_PATH, video_url, @video_name, @video_referer) do |line|
+      run_cmd(DEMUXER_PATH, video_url, "#{@output_dir}#{@video_name}", @video_referer) do |line|
         if (line.include?("hls @") || line.include?("https @")) && line.match?(/#{@ts_seg_pattern}/i)
           seg_nb = line.match(/#{@ts_seg_pattern}/i)[:seg_nb]
           add_line(line)
@@ -130,7 +146,7 @@ module Vidload::Mp2t::Api
         end
       end
       print "\r\e[2K"
-      puts "✔ Video downloaded successfully! Available in ./#{@video_name}.mp4"
+      puts "✔ Video downloaded successfully! Available in #{@output_dir}#{@video_name}.mp4"
       VIDEO_DOWNLOADED_EVENT_QUEUE << true
     end
 

data/lib/vidload/version.rb

@@ -1,3 +1,3 @@
 module Vidload
-  VERSION = "0.1.10"
+  VERSION = "0.2.0"
 end

data/lib/vidload.rb

@@ -1,4 +1,5 @@
 module Vidload
   require_relative "vidload/version"
   require_relative "vidload/mp2t"
+  require_relative "vidload/cli"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vidload
 version: !ruby/object:Gem::Version
-  version: 0.1.10
+  version: 0.2.0
 platform: ruby
 authors:
 - Patacode <pata.codegineer@gmail.com>
@@ -52,6 +52,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -73,6 +87,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/vidload.rb
+- lib/vidload/cli.rb
 - lib/vidload/mp2t.rb
 - lib/vidload/mp2t/api.rb
 - lib/vidload/mp2t/remuxer.sh