Contents

module BillHicks
  class Base

    def self.connect(options = {})
      requires!(options, :username, :password)

      Net::FTP.open(BillHicks.config.ftp_host, options[:username], options[:password]) do |ftp|
        begin
          ftp.debug_mode = BillHicks.config.debug_mode
          ftp.passive = true
          yield ftp
        ensure
          ftp.close
        end
      end
    rescue Net::FTPPermError
      raise BillHicks::NotAuthenticated
    end

    protected

    # Wrapper to `self.requires!` that can be used as an instance method.
    def requires!(*args)
      self.class.requires!(*args)
    end

    def self.requires!(hash, *params)
      params.each do |param|
        if param.is_a?(Array)
          raise ArgumentError.new("Missing required parameter: #{param.first}") unless hash.has_key?(param.first)

          valid_options = param[1..-1]
          raise ArgumentError.new("Parameter: #{param.first} must be one of: #{valid_options.join(', ')}") unless valid_options.include?(hash[param.first])
        else
          raise ArgumentError.new("Missing required parameter: #{param}") unless hash.has_key?(param)
        end
      end
    end

    # Instance methods become class methods through inheritance
    def connect(options)
      self.class.connect(options) do |ftp|
        begin
          yield ftp
        ensure
          ftp.close
        end
      end
    end

    def get_file(filename)
      connect(@options) do |ftp|
        begin
          tempfile = Tempfile.new

          ftp.chdir(BillHicks.config.top_level_dir)
          ftp.getbinaryfile(filename, tempfile.path)

          tempfile
        ensure
          ftp.close
        end
      end
    end

  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
bill_hicks-5.0.2	lib/bill_hicks/base.rb
bill_hicks-5.0.1	lib/bill_hicks/base.rb
bill_hicks-5.0.0	lib/bill_hicks/base.rb
bill_hicks-4.0.0	lib/bill_hicks/base.rb
bill_hicks-3.0.0	lib/bill_hicks/base.rb