[ruby-list:29397] rnet.rb 〜高レベルネットライブラリ〜

るびきちです。

ネットを扱うクラスでちょっと高レベルなインターフェースが欲しかったので、
書いてみました。
net/* とは違ってURIで指定してあれこれするやつです。
現在はHTTPしか実装してませんが、FTPも仕様が決まれば簡単に
追加できるようになってるはずです。

#!/usr/bin/env ruby
=begin
= RNet
高レベルネットクラス。要uri.rb。
URIで指定してネットワークをいじくる。

== メソッド
--- RNet.new(arg)
      RNetオブジェクトを作る。
      argは文字列かURIオブジェクトでURIを指定。
      URIによりスキームを決定し、それに応じたメソッドが使えるようになる。

=== HTTP
--- RNet.head(header0={})
      HEADリクエストを発行。HTTPResponseオブジェクトを返す。
--- RNet.get(header0={})
      GETリクエストを発行。HTTPResponseオブジェクトを返す。
--- RNet.post(data, header0={})
      POSTリクエストを発行。dataは文字列かハッシュ。
      HTTPResponseオブジェクトを返す。
=end
      

require 'net/http'
require 'uri'
require 'kconv'
require 'cgi'

class RNet
  module Http
    @@HeadHeader = {}
    @@GetHeader = {
      'Accept'=> 'text/*, image/*, audio/*, application/*',
      'Accept-Language'=> 'ja; q=1.0, en; q=0.5',
    }
    @@PostHeader = {
      'Accept'=> 'text/*, image/*, audio/*, application/*',
      'Accept-Language'=> 'ja; q=1.0, en; q=0.5',
      'Content-type'=>'application/x-www-form-urlencoded',
    }
  end
  
  module Ftp
  end
end

class RNet
  def initialize(arg)
    case arg
    when URI
      @uri = arg
    when String
      @uri = URI::create(arg)
    else
      raise TypeError, "RNet.new expected URI or String"
    end
    
    extend self.type.const_get(@uri.scheme.capitalize)
    setup
  end
  
  ################ HTTP
  module Http
    def setup
      @host, @port, @path = @uri.host, @uri.port, @uri.path
      @net = Net::HTTP::new @host, @port
    end
    private :setup
    
    def head(header0={})
      header = @@HeadHeader.dup.update(header0)
      @net.head2 @path, header
    end

    def get(header0={})
      header = @@GetHeader.dup.update(header0)
      @net.get2(@path, header) do |res|
        if res.code =~ /^3[0-9][0-9]$/
          return RNet::new(res['location']).get(header0)
        end
      end
    end
    
    def hash2data(hash)
      data = ''
      hash.each_pair do |name, value|
        case value
        when Array
          value.each do |vv|
            data << "#{name}=#{CGI::escape(Kconv::tosjis vv)}&"
          end
        when NilClass
        else
          data << "#{name}=#{CGI::escape(Kconv::tosjis value)}&"
        end
      end
      data.chop!
      data
    end
    private :hash2data

    def post(data, header0={})
      case data
      when String
        datastr = data
      when Hash
        datastr = hash2data(data)
      else
        raise TypeError, "RNet#post: expected String or Hash!"
      end
      
      header = @@PostHeader.dup.update(header0)
      @net.post2(@path, datastr, header) do |res|
        if res.code =~ /^3[0-9][0-9]$/
          return RNet::new(res['location']).get(header0)
        end
      end
    end
  end
end

if __FILE__ == $0
  get = RNet::new('http://www.ruby-lang.org/en').get
  get.body
  p get.code
  get.each do |h,v|
    p [h,v]
  end
  
end