screen-recorder 1.1.0 → 1.2.0

data/appveyor.yml

@@ -0,0 +1,23 @@
+build: off
+cache:
+  - vendor/bundle
+environment:
+  matrix:
+    - RUBY_VERSION: 24
+      RAKE_TASK: spec
+    - RUBY_VERSION: 24
+      RAKE_TASK: rubocop
+    - RUBY_VERSION: 26
+      RAKE_TASK: spec
+install:
+  - choco install ffmpeg
+  - set PATH=C:\Ruby%RUBY_VERSION%\bin;%PATH%
+  - bundle config --local path vendor/bundle
+  - bundle install
+before_test:
+  - ffmpeg -version
+  - ruby -v
+  - gem -v
+  - bundle -v
+test_script:
+  - bundle exec rake %RAKE_TASK%

data/bin/setup

@@ -1,8 +1,8 @@
-#!/usr/bin/env bash
-set -euo pipefail
-IFS=$'\n\t'
-set -vx
-
-bundle install
-
-# Do any other automated setup that you need to do here
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/screen-recorder.rb

@@ -1,54 +1,55 @@
-require 'streamio-ffmpeg'
-require 'os'
-require 'logger'
-
-# @since 1.0.0.beta11
-module ScreenRecorder
-  #
-  # Uses user given FFMPEG binary
-  #
-  # @example
-  #   ScreenRecorder.ffmpeg_binary = 'C:\ffmpeg.exe'
-  #
-  def self.ffmpeg_binary=(bin)
-    FFMPEG.ffmpeg_binary = bin
-  end
-
-  #
-  # Returns path to ffmpeg binary
-  #
-  def self.ffmpeg_binary
-    FFMPEG.ffmpeg_binary
-  end
-
-  #
-  # Set external logger if you want.
-  #
-  def self.logger=(log)
-    @logger = log
-  end
-
-  #
-  # ScreenRecorder.logger
-  #
-  def self.logger
-    return @logger if @logger
-
-    logger           = Logger.new(STDOUT)
-    logger.level     = Logger::ERROR
-    logger.progname  = 'ScreenRecorder'
-    logger.formatter = proc do |severity, time, progname, msg|
-      "#{time.strftime('%F %T')} #{progname} - #{severity} - #{msg}\n"
-    end
-    logger.debug 'Logger initialized.'
-    @logger = logger
-  end
-end
-
-require 'screen-recorder/type_checker'
-require 'screen-recorder/errors'
-require 'screen-recorder/options'
-require 'screen-recorder/titles'
-require 'screen-recorder/common'
-require 'screen-recorder/desktop'
+require 'childprocess'
+require 'logger'
+require 'os'
+require 'streamio-ffmpeg'
+
+# @since 1.0.0.beta11
+module ScreenRecorder
+  #
+  # Uses user given FFMPEG binary
+  #
+  # @example
+  #   ScreenRecorder.ffmpeg_binary = 'C:\ffmpeg.exe'
+  #
+  def self.ffmpeg_binary=(bin)
+    FFMPEG.ffmpeg_binary = bin
+  end
+
+  #
+  # Returns path to ffmpeg binary
+  #
+  def self.ffmpeg_binary
+    FFMPEG.ffmpeg_binary
+  end
+
+  #
+  # Set external logger if you want.
+  #
+  def self.logger=(log)
+    @logger = log
+  end
+
+  #
+  # ScreenRecorder.logger
+  #
+  def self.logger
+    return @logger if @logger
+
+    logger           = Logger.new(STDOUT)
+    logger.level     = Logger::ERROR
+    logger.progname  = 'ScreenRecorder'
+    logger.formatter = proc do |severity, time, progname, msg|
+      "#{time.strftime('%F %T')} #{progname} - #{severity} - #{msg}\n"
+    end
+    logger.debug 'Logger initialized.'
+    @logger = logger
+  end
+end
+
+require 'screen-recorder/type_checker'
+require 'screen-recorder/errors'
+require 'screen-recorder/options'
+require 'screen-recorder/titles'
+require 'screen-recorder/common'
+require 'screen-recorder/desktop'
 require 'screen-recorder/window'

data/lib/screen-recorder/common.rb

@@ -1,147 +1,153 @@
-# @since 1.0.0-beta11
-module ScreenRecorder
-  # @since 1.0.0-beta11
-  class Common
-    attr_reader :options, :video
-
-    def initialize(input:, output:, advanced: {})
-      @options = Options.new(input: input, output: output, advanced: advanced)
-      @video   = nil
-      @process = nil
-    end
-
-    #
-    # Starts the recording
-    #
-    def start
-      @video     = nil # New file
-      start_time = Time.now
-      @process   = start_ffmpeg
-      elapsed    = Time.now - start_time
-      ScreenRecorder.logger.debug "Process started in #{elapsed}s"
-      ScreenRecorder.logger.info 'Recording...'
-      @process
-    end
-
-    #
-    # Stops the recording
-    #
-    def stop
-      ScreenRecorder.logger.debug 'Stopping ffmpeg.exe...'
-      elapsed = stop_ffmpeg
-      ScreenRecorder.logger.debug "Stopped ffmpeg.exe in #{elapsed}s"
-      ScreenRecorder.logger.info 'Recording complete.'
-      @video = FFMPEG::Movie.new(options.output)
-    end
-
-    #
-    # Discards the recorded file. Useful in automated testing
-    # when a test passes and the recorded file is no longer
-    # needed.
-    #
-    def discard
-      FileUtils.rm options.output
-    end
-
-    alias delete discard
-
-    private
-
-    #
-    # Launches the ffmpeg binary using a generated command based on
-    # the given options.
-    #
-    def start_ffmpeg
-      raise Errors::DependencyNotFound, 'ffmpeg binary not found.' unless ffmpeg_exists?
-
-      ScreenRecorder.logger.debug "Command: #{command}"
-      process = IO.popen(command, 'r+')
-      sleep(1.5) # Takes ~1.5s on average to initialize
-      process
-    end
-
-    #
-    # Sends 'q' to the ffmpeg binary to gracefully stop the process.
-    # Forcefully terminates it if it takes more than 10s.
-    #
-    def stop_ffmpeg
-      @process.puts 'q' # Gracefully exit ffmpeg
-      elapsed = wait_for_io_eof(10)
-      @process.close_write # Close IO
-      elapsed
-    rescue Timeout::Error
-      ScreenRecorder.logger.error 'FFmpeg failed to stop. Force killing it...'
-      force_kill_ffmpeg
-      ScreenRecorder.logger.error "Check '#{@options.log}' for more information."
-    rescue Errno::EPIPE
-      # Gets last line from log file
-      err_line = get_lines_from_log(:last, 2)
-      raise FFMPEG::Error, err_line
-    end
-
-    #
-    # Generates the command line arguments based on the given
-    # options.
-    #
-    def command
-      cmd = "#{ScreenRecorder.ffmpeg_binary} -y "
-      cmd << @options.parsed
-    end
-
-    #
-    # Waits for IO#eof? to return true
-    # after 'q' is sent to the ffmpeg process.
-    #
-    def wait_for_io_eof(timeout)
-      start = Time.now
-      Timeout.timeout(timeout) do
-        sleep(0.1) until @process.eof?
-      end
-      ScreenRecorder.logger.debug "IO#eof? #{@process.eof?}"
-      Time.now - start
-    end
-
-    #
-    # Returns true if ffmpeg binary is found.
-    #
-    def ffmpeg_exists?
-      return !`which ffmpeg`.empty? if OS.linux? # "" if not found
-
-      return !`where ffmpeg`.empty? if OS.windows?
-
-      # If the user does not use ScreenRecorder.ffmpeg_binary=() to set the binary path,
-      # ScreenRecorder.ffmpeg_binary returns 'ffmpeg' assuming it must be in ENV. However,
-      # if the above two checks fail, it is not in the ENV either.
-      return false if ScreenRecorder.ffmpeg_binary == 'ffmpeg'
-
-      true
-    end
-
-    #
-    # Returns lines from the log file
-    #
-    def get_lines_from_log(position = :last, count = 2)
-      f     = File.open(options.log)
-      lines = f.readlines
-      lines = lines.last(count) if position == :last
-      lines = lines.first(count) if position == :first
-      f.close
-
-      lines.join(' ')
-    end
-
-    #
-    # Force kills the ffmpeg process.
-    #
-    def force_kill_ffmpeg
-      if OS.windows?
-        pid = `powershell (Get-Process -name 'ffmpeg').Id`.strip.to_i
-        `taskkill /f /pid #{pid}`
-        return
-      end
-
-      `killall -9 ffmpeg` # Linux and macOS
-      nil
-    end
-  end
+# @since 1.0.0-beta11
+#
+# @api private
+module ScreenRecorder
+  # @since 1.0.0-beta11
+  class Common
+    PROCESS_TIMEOUT = 5 # Seconds to wait for ffmpeg to quit
+
+    attr_reader :options, :video
+
+    def initialize(input:, output:, advanced: {})
+      @options = Options.new(input: input, output: output, advanced: advanced)
+      @video   = nil
+      @process = nil
+    end
+
+    #
+    # Starts the recording
+    #
+    def start
+      @video     = nil # New file
+      start_time = Time.now
+      @process   = start_ffmpeg
+      elapsed    = Time.now - start_time
+      ScreenRecorder.logger.debug "Process started in #{elapsed}s"
+      ScreenRecorder.logger.info 'Recording...'
+      @process
+    end
+
+    #
+    # Stops the recording
+    #
+    def stop
+      ScreenRecorder.logger.debug 'Stopping ffmpeg...'
+      stop_ffmpeg
+      ScreenRecorder.logger.debug 'Stopped ffmpeg.'
+      ScreenRecorder.logger.info 'Recording complete.'
+      @video = FFMPEG::Movie.new(options.output)
+    end
+
+    #
+    # Discards the recorded file. Useful in automated testing
+    # when a test passes and the recorded file is no longer
+    # needed.
+    #
+    def discard
+      FileUtils.rm options.output
+    end
+
+    alias delete discard
+
+    private
+
+    #
+    # Launches the ffmpeg binary using a generated command based on
+    # the given options.
+    #
+    def start_ffmpeg
+      raise Errors::DependencyNotFound, 'ffmpeg binary not found.' unless ffmpeg_exists?
+
+      ScreenRecorder.logger.debug "Command: #{command}"
+      process           = build_command
+      @log_file         = File.new(options.log, 'w+')
+      process.io.stdout = process.io.stderr = @log_file
+      @log_file.sync    = true
+      process.duplex    = true
+      process.start
+      sleep(1.5) # Takes ~1.5s on average to initialize
+      # Stopped because of an error
+      raise FFMPEG::Error, "Failed to start ffmpeg. Reason: #{lines_from_log(:last, 2)}" if process.exited?
+
+      process
+    end
+
+    #
+    # Sends 'q' to the ffmpeg binary to gracefully stop the process.
+    # Forcefully terminates it if it takes more than 5s.
+    #
+    def stop_ffmpeg
+      @process.io.stdin.puts 'q' # Gracefully exit ffmpeg
+      @process.io.stdin.close
+      @log_file.close
+      @process.poll_for_exit(PROCESS_TIMEOUT)
+      @process.exit_code
+    rescue ChildProcess::TimeoutError
+      ScreenRecorder.logger.error 'FFmpeg failed to stop. Force killing it...'
+      @process.stop # Tries increasingly harsher methods to kill the process.
+      ScreenRecorder.logger.error "Check '#{@options.log}' for more information."
+    end
+
+    #
+    # Generates the command line arguments based on the given
+    # options.
+    #
+    def command
+      cmd = "#{ScreenRecorder.ffmpeg_binary} -y "
+      cmd << @options.parsed
+    end
+
+    #
+    # Waits for IO#eof? to return true
+    # after 'q' is sent to the ffmpeg process.
+    #
+    def wait_for_io_eof(timeout)
+      start = Time.now
+      Timeout.timeout(timeout) do
+        sleep(0.1) until @process.eof?
+      end
+      ScreenRecorder.logger.debug "IO#eof? #{@process.eof?}"
+      Time.now - start
+    end
+
+    #
+    # Returns true if ffmpeg binary is found.
+    #
+    def ffmpeg_exists?
+      return !`which ffmpeg`.empty? if OS.linux? # "" if not found
+
+      return !`where ffmpeg`.empty? if OS.windows?
+
+      # If the user does not use ScreenRecorder.ffmpeg_binary=() to set the binary path,
+      # ScreenRecorder.ffmpeg_binary returns 'ffmpeg' assuming it must be in ENV. However,
+      # if the above two checks fail, it is not in the ENV either.
+      return false if ScreenRecorder.ffmpeg_binary == 'ffmpeg'
+
+      true
+    end
+
+    #
+    # Returns lines from the log file
+    #
+    def lines_from_log(position = :last, count = 2)
+      f     = File.open(options.log)
+      lines = f.readlines
+      lines = lines.last(count) if position == :last
+      lines = lines.first(count) if position == :first
+      f.close
+
+      lines.join(' ')
+    end
+
+    #
+    # Returns OS specific arguments for Childprocess.build
+    #
+    def build_command
+      if OS.windows?
+        ChildProcess.build('cmd.exe', '/c', command)
+      else
+        ChildProcess.build('sh', '-c', command)
+      end
+    end
+  end
 end