rubytube 0.1.0

data/lib/rubytube/stream.rb

@@ -0,0 +1,81 @@
+module RubyTube
+  class Stream
+    attr_accessor(
+      :monostate,
+      :url,
+      :itag,
+      :mime_type,
+      :codecs,
+      :type,
+      :subtype,
+      :file_size,
+      :is_otf,
+      :bitrate
+    )
+
+    def initialize(stream, monostate)
+      self.monostate = monostate
+
+      self.url = stream['url']
+      self.itag = stream['itag'].to_i
+
+      self.mime_type, self.codecs = Extractor.mime_type_codec(stream['mimeType'])
+      self.type, self.subtype = mime_type.split('/')
+
+      self.is_otf = stream['is_otf']
+      self.bitrate = stream['bitrate']
+
+      self.file_size = stream.fetch('contentLength', 0).to_i
+    end
+
+    def download(filename: nil, output_dir: nil)
+      file_path = get_file_path(filename, output_dir)
+
+      return file_path if File.exist?(file_path)
+
+      bytes_remaining = file_size
+
+      File.open(file_path, 'wb') do |f|
+        begin
+          Request.stream(url) do |chunk|
+            bytes_remaining -= chunk.bytesize
+            f.write(chunk)
+          end
+        rescue HTTPError => e
+          raise e if e.code != 404
+        end
+      end
+
+      file_path
+    end
+
+    def is_audio?
+      type == 'audio'
+    end
+
+    def is_video?
+      type == 'video'
+    end
+
+    def title
+      monostate.title
+    end
+
+    private
+
+    def get_file_path(filename, output_dir, prefix = '')
+      filename = default_filename unless filename
+
+      if prefix
+        filename = "#{prefix}#{filename}"
+      end
+
+      output_path = Utils.target_directory(output_dir)
+      File.join(output_path, filename)
+    end
+
+    def default_filename
+      "#{monostate.title}.#{subtype}"
+    end
+  end
+end

data/lib/rubytube/stream_query.rb

@@ -0,0 +1,33 @@
+module RubyTube
+  class StreamQuery < Array
+    attr_reader :streams
+
+    def initialize(fmt_streams)
+      super
+      @streams = fmt_streams
+    end
+
+    def filter(file_extension: nil, only_audio: false, only_video: false)
+      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
+
+      r = streams
+      filters.each do |f|
+        r = r.select(&f)
+      end
+
+      r
+    end
+
+    def first
+      streams.first
+    end
+
+    def get_by_itag(itag)
+      streams.find { |s| s.itag == itag }
+    end
+  end
+end

data/lib/rubytube/utils.rb

@@ -0,0 +1,24 @@
+module RubyTube
+  module Utils
+    module_function
+
+    def regex_search(pattern, string, group)
+      match = string.match(pattern)
+      if match
+        return match[group]
+      end
+      nil
+    end
+
+    def target_directory(output_path = nil)
+      if output_path
+        result = File.join(Dir.pwd, output_path) unless File.absolute_path?(output_path)
+      else
+        result = Dir.pwd
+      end
+
+      FileUtils.mkdir_p(result) unless Dir.exist?(result)
+      result
+    end
+  end
+end

data/lib/rubytube/version.rb

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

data/lib/rubytube.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+require 'faraday'
+require 'faraday/follow_redirects'
+
+require_relative 'rubytube/version'
+
+require_relative 'rubytube/cipher'
+require_relative 'rubytube/client'
+require_relative 'rubytube/extractor'
+require_relative 'rubytube/innertube'
+require_relative 'rubytube/monostate'
+require_relative 'rubytube/parser'
+require_relative 'rubytube/request'
+require_relative 'rubytube/stream_query'
+require_relative 'rubytube/stream'
+require_relative 'rubytube/utils'
+
+module RubyTube
+  class Error < StandardError; end
+  class HTMLParseError < StandardError; end
+  class ExtractError < StandardError; end
+  class MaxRetriesExceeded < StandardError; end
+  class VideoUnavailable < StandardError; end
+
+  class RegexMatchError < StandardError
+    def initialize(caller, pattern)
+      super("Regex match error in #{caller} for pattern #{pattern}")
+    end
+  end
+
+  class MembersOnly < StandardError
+    def initialize(video_id)
+      super("Members only video: #{video_id}")
+    end
+  end
+  
+  class RecordingUnavailable < StandardError
+    def initialize(video_id)
+      super("Recording unavailable: #{video_id}")
+    end
+  end
+  
+  class VideoUnavailable < StandardError
+    def initialize(video_id)
+      super("Video unavailable: #{video_id}")
+    end
+  end
+  
+  class VideoPrivate < StandardError
+    def initialize(video_id)
+      super("Video is private: #{video_id}")
+    end
+  end
+  
+  class LiveStreamError < StandardError
+    def initialize(video_id)
+      super("Video is a live stream: #{video_id}")
+    end
+  end
+
+  class << self
+    def new(url)
+      Client.new(url)
+    end
+  end
+end

data/sig/rubytube.rbs

@@ -0,0 +1,4 @@
+module RubyTube
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: rubytube
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- nightswinger
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2023-09-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.0.1
+- !ruby/object:Gem::Dependency
+  name: faraday-follow_redirects
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Write a longer description or delete this line.
+email:
+- stardustkids83@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".standard.yml"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/rubytube.rb
+- lib/rubytube/cipher.rb
+- lib/rubytube/client.rb
+- lib/rubytube/extractor.rb
+- lib/rubytube/innertube.rb
+- lib/rubytube/monostate.rb
+- lib/rubytube/parser.rb
+- lib/rubytube/request.rb
+- lib/rubytube/stream.rb
+- lib/rubytube/stream_query.rb
+- lib/rubytube/utils.rb
+- lib/rubytube/version.rb
+- sig/rubytube.rbs
+homepage: https://github.com/nightswinger/rubytube
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/nightswinger/rubytube
+  source_code_uri: https://github.com/nightswinger/rubytube
+  changelog_uri: https://github.com/nightswinger/rubytube
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.10
+signing_key:
+specification_version: 4
+summary: Write a short summary, because RubyGems requires one.
+test_files: []