Contents

require 'net/sftp'
require 'tempfile'

module Service
  class RemoteRevisionError < StandardError; end
  
  class Service
    def initialize(host, username, path)
      @host = host
      @username = username
      @path = path
    end
    
    def uri
      "#{@scheme}://#{@username}@#{@host}/#{@path}"
    end
  end
  
  class SFTP < Service
    def initialize(host, username, password, path)
      super(host, username, path)
      @scheme = 'sftp'
      @sftp = Net::SFTP.start(host, username, :password => password)
    end
    
    def read(file)
      begin
        @sftp.file.open(File.join(@path, file), 'r') do |f|
          f.gets
        end
      rescue Net::SFTP::StatusException => e
        raise unless e.code == 2
        raise RemoteRevisionError
      end
    end
    
    def write(file, data)
      path = File.join(@path, file)
      begin
        dir = File.dirname(path)
        @sftp.stat!(dir)
      rescue Net::SFTP::StatusException => e
        raise unless e.code == 2
        mkdir_p(dir)
      end
      tmp = Tempfile.new(file.gsub('/', '.'))
      tmp << data
      tmp.flush
      @sftp.upload!(tmp.path, path)
      tmp.close
    end
    
    def delete(file)
      begin
        path = File.join(@path, file)
        @sftp.remove!(path)
        cleanup(File.dirname(path))
      rescue Net::SFTP::StatusException => e
        raise unless e.code == 2
      end
    end
    
    private
    
    def cleanup(dir)
      unless File.identical?(dir, @path)
        if empty?(dir)
          @sftp.rmdir!(dir)
          cleanup(File.dirname(dir))
        end
      end
    end
    
    def empty?(dir)
      @sftp.dir.entries(dir).map do |entry|
        entry.name unless entry.name == '.' or entry.name == '..'
      end.compact.empty?
    end
    
    def mkdir_p(dir)
      begin
        @sftp.mkdir!(dir)
      rescue Net::SFTP::StatusException => e
        raise unless e.code == 2
        mkdir_p(File.dirname(dir))
        mkdir_p(dir)
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
dandelion-0.1.4	lib/dandelion/service.rb