viddl 0.0.2

data/lib/viddl/video/clip/crop.rb

@@ -0,0 +1,59 @@
+module Viddl
+
+  module Video
+
+    class Clip
+
+      module Crop
+
+        extend self
+
+        # Crop options formatted for ffmpeg
+        # @param [Hash] options
+        # @option options [Hash] :crop The desired crop parameters (:x, :y, :width, :height)
+        # @return [Hash]
+        def options_formatted(options = {})
+          result = {}
+          unless options[:crop].nil?
+            crop = {}
+            [:x, :y, :width, :height].each do |property|
+              if options[:crop][property].nil?
+                raise "Crop is missing required #{property} property"
+              else
+                crop[property] = options[:crop][property].to_i
+              end
+            end
+            result[:crop] = crop
+          end
+          result
+        end
+
+        # Command line options for cropping
+        # @param [Hash] options
+        # @option options [Hash] :crop The desired crop parameters (:x, :y, :width, :height)
+        # @return [String]
+        def filter_args(options = {})
+          unless options[:crop].nil?
+            crop = options[:crop]
+            "crop=#{crop[:width]}:#{crop[:height]}:#{crop[:x]}:#{crop[:y]}"
+          end
+        end
+
+        # Token added to clip filename for crop args
+        # @param [Hash] options
+        # @option options [Hash] :crop The desired crop parameters (:x, :y, :width, :height)
+        # @return [String, nil]
+        def filename_token(options = {})
+          unless options[:crop].nil?
+            crop = options[:crop]
+            "cx#{crop[:x]}cy#{crop[:y]}cw#{crop[:width]}ch#{crop[:height]}"
+          end
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/lib/viddl/video/clip/cut.rb

@@ -0,0 +1,83 @@
+module Viddl
+
+  module Video
+
+    class Clip
+
+      module Cut
+
+        extend self
+
+        # Cut options formatted for ffmpeg
+        # @param [Hash] options
+        # @option options [Numeric] :start Time in the source file where the clip starts
+        # @option options [Numeric] :duration Duration of the clip
+        # @option options [Numeric] :end Time in the source file where the clip ends
+        # @return [Hash]
+        def options_formatted(options = {})
+          result = {}
+          result[:start] = options[:start]
+          result[:duration] = duration(options)
+          result
+        end
+
+        # Command line options for the given cut constraints
+        # @param [Hash] options
+        # @option options [Numeric] :start Time in the source file where the clip starts
+        # @option options [Numeric] :duration Duration of the clip
+        # @return [String, nil]
+        def args(options = {})
+          args = []
+          unless options[:start].nil?
+            args << "-ss #{options[:start]}"
+          end
+          unless options[:duration].nil?
+            args << "-t #{options[:duration]}"
+          end
+          args.join(" ") unless args.empty?
+        end
+
+        # Token added to clip filename for cut args
+        # @param [Hash] options
+        # @option options [Numeric] :start Time in the source file where the clip starts
+        # @option options [Numeric] :duration Duration of the clip
+        # @return [String, nil]
+        def filename_token(options = {})
+          if !options[:start].nil? || !options[:duration].nil?
+            args = ""
+            if !options[:start].nil?
+              args += "s#{options[:start]}"
+            end
+            if !options[:duration].nil?
+              args += "d#{options[:duration]}"
+            end
+            args
+          end
+        end
+
+        private
+
+        # Numeric duration for the given options
+        # @param [Hash] options
+        # @option options [Numeric] :duration Duration of the clip
+        # @option options [Numeric] :end Time in the source file where the clip ends
+        # @return [Numeric]
+        def duration(options = {})
+          duration = nil
+          if !options[:duration].nil? && !options[:end].nil?
+            raise "Can not use both end time and duration"
+          elsif !options[:duration].nil? && options[:end].nil?
+            duration = options[:duration]
+          elsif options[:duration].nil? && !options[:end].nil?
+            duration = options[:end] - (options[:start] || 0)
+          end
+          duration
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/lib/viddl/video/clip/resize.rb

@@ -0,0 +1,66 @@
+module Viddl
+
+  module Video
+
+    class Clip
+
+      module Resize
+
+        extend self
+
+        # Resize options formatted for ffmpeg
+        # @param [Hash] options
+        # @option options [Integer, String] :width The desired width to resize to
+        # @option options [Integer, String] :height The desired height to resize to
+        # @return [Hash]
+        def options_formatted(options = {})
+          result = {}
+          [:width, :height].each do |property|
+            result[property] = options[property].to_i unless options[property].nil?
+          end
+          result
+        end
+
+        # Command line options for resize
+        # @param [Hash] options
+        # @option options [Integer] :width The desired width to resize to
+        # @option options [Integer] :height The desired height to resize to
+        # @return [String, nil]
+        def filter_args(options = {})
+          scale = if options[:width].nil? && !options[:height].nil?
+            "-1:#{options[:height]}"
+          elsif !options[:width].nil? && options[:height].nil?
+            "#{options[:width]}:-1"
+          elsif !options[:width].nil? && !options[:height].nil?
+            "#{options[:width]}:#{options[:height]}"
+          end
+          unless scale.nil?
+            "scale=#{scale}"
+          end
+        end
+
+        # Token added to clip filename for resize args
+        # @param [Hash] options
+        # @option options [Integer] :width The desired width to resize to
+        # @option options [Integer] :height The desired height to resize to
+        # @return [String, nil]
+        def filename_token(options = {})
+          if !options[:width].nil? || !options[:height].nil?
+            args = ""
+            if !options[:width].nil?
+              args += "w#{options[:width]}"
+            end
+            if !options[:height].nil?
+              args += "h#{options[:height]}"
+            end
+            args
+          end
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/lib/viddl/video/download.rb

@@ -0,0 +1,49 @@
+module Viddl
+
+  module Video
+
+    class Download
+
+      # download is stored to /tmp before processing
+      TEMPDIR = "/tmp"
+      # download format is forced to mp4 to optimize for quickness
+      FORMAT_ARG = "-f 'best[ext=mp4]'"
+
+      # Download the given video
+      # @param [Video::Instance] video
+      # @param [Hash] options
+      # @return [Download]
+      def self.process(video, options = {})
+        download = new(video)
+        download.process(options)
+        download
+      end
+
+      # @param [Video::Instance] video
+      def initialize(video)
+        @video = video
+        @video.download = self if @video.download.nil?
+      end
+
+      # Download the video file
+      # @param [Hash] options
+      # @return [Boolean]
+      def process(options = {})
+        result = Kernel.system(command_line)
+        raise(result.to_s) unless result
+        true
+      end
+
+      private
+
+      # Command line to download the video file
+      # @return [String]
+      def command_line
+        "youtube-dl #{@video.source_url} #{FORMAT_ARG} -o '#{TEMPDIR}/#{@video.id}s.%(ext)s'"
+      end
+
+    end
+
+  end
+
+end

data/lib/viddl/video/instance.rb

@@ -0,0 +1,62 @@
+module Viddl
+
+  module Video
+
+    class Instance
+
+      attr_accessor :download
+      attr_reader :id, :source_url
+
+      # @param [String] url The url of the video source
+      def initialize(url)
+        @source_url = url
+
+        populate_id
+      end
+
+      # Cut the video source using the given options
+      # @param [Hash] options
+      # @option options [Boolean] :audio Whether to include audio
+      # @option options [Numeric] :start Time in the source file where the clip starts
+      # @option options [Numeric] :duration Duration of the clip
+      # @option options [Numeric] :end Time in the source file where the clip ends
+      # @option options [Integer, String] :width The desired width to resize to
+      # @option options [Integer, String] :height The desired height to resize to
+      # @option options [Hash] :crop The desired crop parameters (:x, :y, :width, :height)
+      # @return [Array<Clip>]
+      def create_clip(options = {})
+        source_filenames.map do |filename|
+          Clip.process(filename, options)
+        end
+      end
+
+      # Download the video source
+      # @param [Hash] options
+      # @return [Download]
+      def process_download(options = {})
+        @download = Download.process(self, options)
+      end
+
+      # The downloaded source filenames
+      # @return [Array<String>]
+      def source_filenames
+        if @download.nil?
+          raise "File must be downloaded"
+        else
+          @source_filenames = Dir["#{Download::TEMPDIR}/#{@id}*"]
+        end
+      end
+
+      private
+
+      # The video instance id
+      # @return [String]
+      def populate_id
+        @id = @source_url.scan(/youtube.com\/watch\?v\=(\S*)&?/).flatten.first
+      end
+
+    end
+
+  end
+
+end

data/spec/helper.rb

@@ -0,0 +1,8 @@
+$:.unshift(File.join("..", "lib"))
+
+require "rspec"
+require "viddl"
+
+#RSpec.configure do |c|
+#  c.mock_with :rspec
+#end

data/spec/system_spec.rb

@@ -0,0 +1,77 @@
+require "helper"
+
+describe Viddl::System do
+
+  context ".validate" do
+
+    context "passes" do
+
+      it "execs command line" do
+        expect(Kernel).to(receive(:system).twice.and_return(true))
+        @result = Viddl::System.send(:validate)
+      end
+
+    end
+
+    context "fails" do
+
+      it "raises" do
+        expect(Kernel).to(receive(:system).and_return(nil))
+        expect {
+          Viddl::System.send(:validate)
+        }.to(raise_error(RuntimeError))
+      end
+
+    end
+
+  end
+
+  context ".validate_ffmpeg" do
+
+    context "passes" do
+
+      it "execs command line" do
+        expect(Kernel).to(receive(:system).and_return(true))
+        @result = Viddl::System.send(:validate_ffmpeg)
+      end
+
+    end
+
+    context "fails" do
+
+      it "raises" do
+        expect(Kernel).to(receive(:system).and_return(nil))
+        expect {
+          Viddl::System.send(:validate_ffmpeg)
+        }.to(raise_error(RuntimeError))
+      end
+
+    end
+
+  end
+
+  context ".validate_youtube_dl" do
+
+    context "passes" do
+
+      it "execs command line" do
+        expect(Kernel).to(receive(:system).and_return(true))
+        @result = Viddl::System.send(:validate_youtube_dl)
+      end
+
+    end
+
+    context "fails" do
+
+      it "raises" do
+        expect(Kernel).to(receive(:system).and_return(nil))
+        expect {
+          Viddl::System.send(:validate_youtube_dl)
+        }.to(raise_error(RuntimeError))
+      end
+
+    end
+
+  end
+
+end

data/spec/video/clip/audio_spec.rb

@@ -0,0 +1,81 @@
+require "helper"
+
+describe Viddl::Video::Clip::Audio do
+
+  context ".options_formatted" do
+
+    context "with no options" do
+
+      it "returns audio = true" do
+        options = {}
+        opts = Viddl::Video::Clip::Audio.send(:options_formatted, options)
+        expect(opts).to(include(audio: true))
+      end
+
+    end
+
+    context "with audio = false" do
+
+      it "returns audio = false" do
+        options = {
+          audio: false
+        }
+        opts = Viddl::Video::Clip::Audio.send(:options_formatted, options)
+        expect(opts).to(include(audio: false))
+      end
+
+    end
+
+    context "with audio = true" do
+
+      it "returns audio = true" do
+        options = {
+          audio: true
+        }
+        opts = Viddl::Video::Clip::Audio.send(:options_formatted, options)
+        expect(opts).to(include(audio: true))
+      end
+
+    end
+
+  end
+
+  context ".args" do
+
+    context "with no options" do
+
+      it "return blank string" do
+        options = {}
+        args = Viddl::Video::Clip::Audio.send(:args, options)
+        expect(args).to(be_nil)
+      end
+
+    end
+
+    context "with audio=false" do
+
+      it "turns off audio" do
+        options = {
+          audio: false
+        }
+        args = Viddl::Video::Clip::Audio.send(:args, options)
+        expect(args).to(eq("-an"))
+      end
+
+    end
+
+    context "with audio=true" do
+
+      it "returns blank string" do
+        options = {
+          audio: true
+        }
+        args = Viddl::Video::Clip::Audio.send(:args, options)
+        expect(args).to(be_nil)
+      end
+
+    end
+
+  end
+
+end