rubytube 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b12cdf8d2dc05d51f957f67a932c9845cf60dfbecac9b183217a521a57bad129
-  data.tar.gz: 6ca47773a0d13cf4ea479e628445835a7fccd7de02ea2e0aa7d7151f2f1806fe
+  metadata.gz: 4cc85afbbd2fe5d5810318f9cf2c7bc16c0194d10e3919673ae393991b28e473
+  data.tar.gz: 8e7c4102505799239962d99f8697cce2457c8d46aef47f46595a65bda35fd6e6
 SHA512:
-  metadata.gz: 657e712bcd49674523a2fafa60f1a026b247d18bd8b5ec4390e2e6c95d4145e652016687d2060033e9fa48738c9e4a4a7e8b46abf6117025f6b33ce8a591f89b
-  data.tar.gz: 6d5ff0d1faf7a6b6951a814430d3b2029f5a72281853d7ef2aaecd70e24e3cab78c7df330dec48e63d73928c8bb9a1d92f16e2ff2bb07cbf515a67cc19dcafd4
+  metadata.gz: 01aab01de46c366279769feefeaa6f9bde56da54eaf1febbba266834dc17772c57707272a239aa7b1ec8d2d23bf805ff9d7f9cae77b2182f35aeff442bd22510
+  data.tar.gz: e7d289d421ad60bff7e84022ba11926a2108a32cd63be2d3d768550f500cf66b9373ecc6c142b3ed61cbbc454096b7b9e4cf019d8267e57aeee359d09bb17ebd

data/README.md

@@ -15,5 +15,5 @@ require 'rubytube'
 video = RubyTube.new('https://www.youtube.com/watch?v=dQw4w9WgXcQ')
 
 # Download video
-video.download(filename: 'my_video.mp4')
+video.streams.first.download
 ```

data/lib/rubytube/cipher.rb

@@ -140,7 +140,7 @@ module RubyTube
       mapper
     end
 
-    def reverse(arr)
+    def reverse(arr, _ = nil)
       # Ruby equivalent of JavaScript's Array.reverse()
       arr.reverse!
     end
@@ -155,6 +155,7 @@ module RubyTube
       temp = arr[0]
       arr[0] = arr[index.to_i % arr.length]
       arr[index.to_i % arr.length] = temp
+      arr
     end
 
     def push(arr, val)

data/lib/rubytube/client.rb

@@ -99,6 +99,14 @@ module RubyTube
       @streams = StreamQuery.new(fmt_streams)
     end
 
+    def thumbnail_url
+      thumbs = vid_info.fetch('videoDetails', {}).fetch('thumbnail', {}).fetch('thumbnails', [])
+
+      return thumbs[-1]['url'] if thumbs.size > 0
+
+      "https://img.youtube.com/vi/#{ideo_id}/maxresdefault.jpg"
+    end
+
     def vid_info
       return @vid_info if @vid_info
 

data/lib/rubytube/stream.rb

@@ -61,6 +61,10 @@ module RubyTube
       monostate.title
     end
 
+    def resolution
+      stream_format.resolution
+    end
+
     private
 
     def get_file_path(filename, output_dir, prefix = '')
@@ -77,5 +81,9 @@ module RubyTube
     def default_filename
       "#{monostate.title}.#{subtype}"
     end
+
+    def stream_format
+      @stream_format ||= StreamFormat.new(itag)
+    end
   end
 end

data/lib/rubytube/stream_format.rb

@@ -0,0 +1,152 @@
+module RubyTube
+  class StreamFormat
+    PROGRESSIVE_VIDEO = {
+      5 => ["240p", "64kbps"],
+      6 => ["270p", "64kbps"],
+      13 => ["144p", nil],
+      17 => ["144p", "24kbps"],
+      18 => ["360p", "96kbps"],
+      22 => ["720p", "192kbps"],
+      34 => ["360p", "128kbps"],
+      35 => ["480p", "128kbps"],
+      36 => ["240p", nil],
+      37 => ["1080p", "192kbps"],
+      38 => ["3072p", "192kbps"],
+      43 => ["360p", "128kbps"],
+      44 => ["480p", "128kbps"],
+      45 => ["720p", "192kbps"],
+      46 => ["1080p", "192kbps"],
+      59 => ["480p", "128kbps"],
+      78 => ["480p", "128kbps"],
+      82 => ["360p", "128kbps"],
+      83 => ["480p", "128kbps"],
+      84 => ["720p", "192kbps"],
+      85 => ["1080p", "192kbps"],
+      91 => ["144p", "48kbps"],
+      92 => ["240p", "48kbps"],
+      93 => ["360p", "128kbps"],
+      94 => ["480p", "128kbps"],
+      95 => ["720p", "256kbps"],
+      96 => ["1080p", "256kbps"],
+      100 => ["360p", "128kbps"],
+      101 => ["480p", "192kbps"],
+      102 => ["720p", "192kbps"],
+      132 => ["240p", "48kbps"],
+      151 => ["720p", "24kbps"],
+      300 => ["720p", "128kbps"],
+      301 => ["1080p", "128kbps"],
+    }
+
+    DASH_VIDEO = {
+      133 => ["240p", nil],
+      134 => ["360p", nil],
+      135 => ["480p", nil],
+      136 => ["720p", nil],
+      137 => ["1080p", nil],
+      138 => ["2160p", nil],
+      160 => ["144p", nil],
+      167 => ["360p", nil],
+      168 => ["480p", nil],
+      169 => ["720p", nil],
+      170 => ["1080p", nil],
+      212 => ["480p", nil],
+      218 => ["480p", nil],
+      219 => ["480p", nil],
+      242 => ["240p", nil],
+      243 => ["360p", nil],
+      244 => ["480p", nil],
+      245 => ["480p", nil],
+      246 => ["480p", nil],
+      247 => ["720p", nil],
+      248 => ["1080p", nil],
+      264 => ["1440p", nil],
+      266 => ["2160p", nil],
+      271 => ["1440p", nil],
+      272 => ["4320p", nil],
+      278 => ["144p", nil],
+      298 => ["720p", nil],
+      299 => ["1080p", nil],
+      302 => ["720p", nil],
+      303 => ["1080p", nil],
+      308 => ["1440p", nil],
+      313 => ["2160p", nil],
+      315 => ["2160p", nil],
+      330 => ["144p", nil],
+      331 => ["240p", nil],
+      332 => ["360p", nil],
+      333 => ["480p", nil],
+      334 => ["720p", nil],
+      335 => ["1080p", nil],
+      336 => ["1440p", nil],
+      337 => ["2160p", nil],
+      394 => ["144p", nil],
+      395 => ["240p", nil],
+      396 => ["360p", nil],
+      397 => ["480p", nil],
+      398 => ["720p", nil],
+      399 => ["1080p", nil],
+      400 => ["1440p", nil],
+      401 => ["2160p", nil],
+      402 => ["4320p", nil],
+      571 => ["4320p", nil],
+      694 => ["144p", nil],
+      695 => ["240p", nil],
+      696 => ["360p", nil],
+      697 => ["480p", nil],
+      698 => ["720p", nil],
+      699 => ["1080p", nil],
+      700 => ["1440p", nil],
+      701 => ["2160p", nil],
+      702 => ["4320p", nil]
+    }
+
+    DASH_AUDIO = {
+      139 => [nil, "48kbps"],
+      140 => [nil, "128kbps"],
+      141 => [nil, "256kbps"],
+      171 => [nil, "128kbps"],
+      172 => [nil, "256kbps"],
+      249 => [nil, "50kbps"],
+      250 => [nil, "70kbps"],
+      251 => [nil, "160kbps"],
+      256 => [nil, "192kbps"],
+      258 => [nil, "384kbps"],
+      325 => [nil, nil],
+      328 => [nil, nil],
+    }
+
+    ITAGS = {
+      **PROGRESSIVE_VIDEO,
+      **DASH_VIDEO,
+      **DASH_AUDIO
+    }
+
+    HDR = [330, 331, 332, 333, 334, 335, 336, 337]
+    FORMAT_3D = [82, 83, 84, 85, 100, 101, 102]
+    LIVE = [91, 92, 93, 94, 95, 96, 132, 151]
+
+    attr_reader :itag, :resolution, :abr
+
+    def initialize(_itag)
+      @itag = _itag
+
+      @resolution, @abr = ITAGS.fetch(itag)
+    end
+
+    def is_live?
+      !!LIVE[itag]
+    end
+
+    def is_3d?
+      !!FORMAT_3D[itag]
+    end
+
+    def is_hdr?
+      !!HDR[itag]
+    end
+
+    def is_dash?
+      !!DASH_AUDIO[itag] || !!DASH_VIDEO[itag]
+    end
+  end
+end

data/lib/rubytube/stream_query.rb

@@ -7,12 +7,13 @@ module RubyTube
       @streams = fmt_streams
     end
 
-    def filter(file_extension: nil, only_audio: false, only_video: false)
+    def filter(file_extension: nil, only_audio: false, only_video: false, resolution: nil)
       filters = []
 
       filters << ->(stream) { stream.subtype == file_extension } if file_extension
       filters << ->(stream) { stream.is_audio? } if only_audio
       filters << ->(stream) { stream.is_video? } if only_video
+      filters << ->(stream) { stream.resolution == resolution } if resolution
 
       r = streams
       filters.each do |f|
@@ -29,5 +30,9 @@ module RubyTube
     def get_by_itag(itag)
       streams.find { |s| s.itag == itag }
     end
+
+    def get_by_resolution(resolution)
+      streams.find { |s| s.resolution == resolution }
+    end
   end
 end

data/lib/rubytube/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyTube
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/rubytube.rb

@@ -12,6 +12,7 @@ require_relative 'rubytube/innertube'
 require_relative 'rubytube/monostate'
 require_relative 'rubytube/parser'
 require_relative 'rubytube/request'
+require_relative 'rubytube/stream_format'
 require_relative 'rubytube/stream_query'
 require_relative 'rubytube/stream'
 require_relative 'rubytube/utils'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rubytube
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - nightswinger
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-09-20 00:00:00.000000000 Z
+date: 2023-09-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -62,6 +62,7 @@ files:
 - lib/rubytube/parser.rb
 - lib/rubytube/request.rb
 - lib/rubytube/stream.rb
+- lib/rubytube/stream_format.rb
 - lib/rubytube/stream_query.rb
 - lib/rubytube/utils.rb
 - lib/rubytube/version.rb
@@ -81,7 +82,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.6.0
+      version: 2.7.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="