Sha256: e229fc44a8bab0913a56755288d124919d0010575d97be35ce0879196a602ea3

Contents?: true

Size: 1.77 KB

Versions: 6

Compression:

Stored size: 1.77 KB

Contents

module Zanders
  class Base

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

      Net::FTP.open(Zanders.config.ftp_host, options[:username], options[:password]) do |ftp|
        ftp.passive = true

        yield ftp
      end
    rescue Net::FTPPermError
      raise Zanders::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|
        yield ftp
      end
    end

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

          ftp.chdir(Zanders.config.ftp_directory)
          ftp.getbinaryfile(filename, tempfile.path)

          tempfile.close

          tempfile
        ensure
          ftp.close
        end
      end
    end

    def content_for(xml_doc, field)
      node = xml_doc.css(field).first

      if node.nil?
        nil
      else
        if node.css("DATA").present?
          node.css("DATA").text.strip
        else
          node.content.strip
        end
      end
    end

  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
zanders-2.2.5 lib/zanders/base.rb
zanders-2.2.4 lib/zanders/base.rb
zanders-2.2.3 lib/zanders/base.rb
zanders-2.2.2 lib/zanders/base.rb
zanders-2.2.1 lib/zanders/base.rb
zanders-2.2.0 lib/zanders/base.rb