Class: FileServe

Inherits:

Object

• Object
• FileServe

Defined in:

lib/vendor/xmpp4r/data/doc/xmpp4r/examples/advanced/fileserve.rb

Instance Method Summary

• #cleaner ⇒ Object
• #command(from, cmd, arg) ⇒ Object
• #initialize(conf) ⇒ FileServe constructor

  A new instance of FileServe.

• #presence ⇒ Object
• #register_handlers ⇒ Object
• #say(to, text) ⇒ Object

Constructor Details

#initialize(conf) ⇒ FileServe

Returns a new instance of FileServe.

# File 'lib/vendor/xmpp4r/data/doc/xmpp4r/examples/advanced/fileserve.rb', line 177

def initialize(conf)
  @uploads = 0
  @downloads = 0

  @transfers = []
  @transfers_lock = Mutex.new

  @client = Jabber::Client.new Jabber::JID.new(conf['jabber']['jid'])
  @client.connect
  @client.auth(conf['jabber']['password'])

  @ft = Jabber::FileTransfer::Helper.new(@client)
  Jabber::Version::SimpleResponder.new(@client,
                                     "XMPP4R FileServe example",
                                     Jabber::XMPP4R_VERSION,
                                     IO.popen('uname -sr').readlines.to_s.strip)
  register_handlers

  @directory = conf['directory']
  @directory.gsub!(/\/+$/, '')

  @socksserver = Jabber::Bytestreams::SOCKS5BytestreamsServer.new(conf['socks']['port'])

  conf['socks']['addresses'].each { |addr| @socksserver.add_address(addr) }

  @proxies = []
  conf['socks']['proxies'].collect { |jid|
    puts "Querying proxy #{jid}..."
    begin
      @proxies.push(Jabber::Bytestreams::SOCKS5Bytestreams::query_streamhost(@client, jid))
    rescue
      puts "Error: #{$!}"
    end
  }

  Thread.new { presence }
  Thread.new { cleaner }

  # Panic reboot ;-)
  @client.on_exception { initialize(conf) }
end

Instance Method Details

#cleaner ⇒ Object

# File 'lib/vendor/xmpp4r/data/doc/xmpp4r/examples/advanced/fileserve.rb', line 331

def cleaner
  loop {

    @transfers_lock.synchronize {
      @transfers.delete_if { |t| t.done? }
    }

    sleep 1
  }
end

#command(from, cmd, arg) ⇒ Object

# File 'lib/vendor/xmpp4r/data/doc/xmpp4r/examples/advanced/fileserve.rb', line 283

def command(from, cmd, arg)
  say = lambda { |text|
    say(from, text)
  }
  socksconf = lambda { |stream|
    stream.add_streamhost(@socksserver)
    @proxies.each { |sh|
      stream.add_streamhost(sh)
    }
  }

  case cmd
    when 'get'
      arg.gsub!(/\//, '')
      arg.gsub!(/^\.+/, '')
      transfer = Download.new(@ft, from, "#{@directory}/#{arg}", say, socksconf)
      @downloads += 1
      @transfers_lock.synchronize {
        @transfers.push(transfer)
      }
    when 'ls'
      text = ""
      Dir.foreach(@directory) { |file|
        next if file =~ /^\./
        path = "#{@directory}/#{file}"
        text += "#{file} (#{human_readable File.size(path)})\n" if File.file?(path)
      }
      say.call(text.strip)
    when 'stat'
      @transfers_lock.synchronize {
        text = "#{@transfers.size} transfers:\n"
        @transfers.each { |t|
          text += "#{t.filename} (#{t.peer}): #{(t.bytes * 100) / t.filesize}% (#{human_readable t.stats}/s)\n"
        }
      }
      say.call(text.strip)
    when 'help'
      say.call "Download a file: get <filename>\nList directory contents: ls\nLook who is currently wasting bandwidth: stat\nUpload a file, simply send this file"
    else
      say.call "Unknown command: #{cmd}, try help"
  end
end

#presence ⇒ Object

# File 'lib/vendor/xmpp4r/data/doc/xmpp4r/examples/advanced/fileserve.rb', line 219

def presence
  rate_upload = 0
  rate_download = 0

  old_status = nil
  loop {
    # Update the @rate_* variables
    @transfers_lock.synchronize {
      @transfers.each { |t|
        if t.kind_of?(Upload) and t.stats > rate_upload
          rate_upload = t.stats
        elsif t.kind_of?(Download) and t.stats > rate_download
          rate_download = t.stats
        end
      }
    }
    status = "Attempted #{@downloads} downloads (max. #{human_readable rate_download}/s) and #{@uploads} uploads (max. #{human_readable rate_upload}/s)"

    @client.send(Jabber::Presence.new(:chat, status)) if status != old_status
    old_status = status

    sleep 1

  }
end

#register_handlers ⇒ Object

# File 'lib/vendor/xmpp4r/data/doc/xmpp4r/examples/advanced/fileserve.rb', line 245

def register_handlers
  @client.add_message_callback { |msg|
    if msg.type == :chat and msg.body and msg.from != 'pentabarf@pentabarf.org/rails'
      puts "<#{msg.from}> #{msg.body.strip}"
      cmd, arg = msg.body.split(/ /, 2)

      command(msg.from, cmd, arg)
    end
  }

  @ft.add_incoming_callback { |iq,file|

    say = lambda { |text|
      say(iq.from, text)
    }

    puts "Incoming file transfer from #{iq.from}: #{file.fname} (#{file.size / 1024} KB)"
    filename = file.fname.split(/\//).last
    filename.gsub!(/^\.+/, '')

    puts "Range: #{file.range != nil}"
    transfer = Upload.new(@ft, iq, "#{@directory}/#{filename}", file.size, file.range != nil, say)
    @uploads += 1

    @transfers_lock.synchronize {
      @transfers.push(transfer)
    }

  }

  roster = Jabber::Roster::Helper.new(@client)

  roster.add_subscription_request_callback { |item,presence|
    roster.accept_subscription(presence.from)
  }

end

#say(to, text) ⇒ Object

# File 'lib/vendor/xmpp4r/data/doc/xmpp4r/examples/advanced/fileserve.rb', line 326

def say(to, text)
  puts ">#{to}< #{text.strip}"
  @client.send(Jabber::Message.new(to, text).set_type(:chat))
end