masuda 0.5.0

data/lib/masuda.rb

@@ -0,0 +1,342 @@
+# 
+# = masuda.rb
+# 
+# Copyright (c) 2007 SUGAWARA Genki <sgwr_dts@yahoo.co.jp>
+# 
+# == Example
+# 
+#     require 'masuda'
+#     
+#     diary = Masuda::Diary.new
+#     diary.entries.each {|entry| puts entry.content }
+#     entry = diary.entry('20070712231804')
+#     puts <<EOS
+#       #{entry.title}
+#       #{entry.content}
+#     EOS
+#     entry.trackbacks.each {|trackback| puts trackback.snippet }
+#     
+#     diary.login('my_id', 'my_pass')
+#     diary.my_entries.each {|entry| puts entry.content }
+#     diary.post('Ruby is ...', <<EOS)
+#     A dynamic, open source
+#     programming language with a
+#     ...
+#     EOS
+#     
+#     session[:diary] = diary.raw
+#     ...
+#     diary = Masuda::Diary.restore(session[:diary])
+# 
+
+require 'base64'
+require 'cgi'
+require 'digest/md5'
+require 'net/http'
+require 'stringio'
+require 'time'
+
+module Masuda
+  class Diary
+    @@host = 'anond.hatelabo.jp'  
+    @@login_host = 'www.hatelabo.jp'
+    @@user_agent = "RubyMasudaLibrary/#{VERSION}"
+
+    def initialize
+      @cookies = {}
+    end
+
+    def self.restore(raw_diary)
+      diary = Diary.new
+      diary.update(raw_diary)
+      diary
+    end
+
+    def login(user, pass)
+      logout if logined?
+      res = request('/login', {:mode => 'enter', :key => user, :password => pass, :autologin => 1}, @@login_host)
+      cookies = res.get_fields('set-cookie')
+      set_cookie(cookies) if cookies
+      @user = user if (retval = logined?)
+      retval
+    end
+
+    def logout
+      @cookies.clear
+      @user = nil
+    end
+
+    def logined?
+      %w(rk b).all? {|k| @cookies.has_key?(k) and @cookies[k].valid? }
+    end
+
+    def entry(id)
+      res = request("/#{id}/")
+      return nil unless ('200' == res.code)
+      source = res.body.to_a
+      et= to_entry(source)
+      load_trackbacks(source, self, et)
+      et
    end
+
+    def trackbacks(id)
+      entry(id).trackbacks
+    end
+
+    def entries(page = nil)
+      path = page ? "/?page=#{page}" : '/'
+      res = request(path)
+      ets = nil
+
+      if '200' == res.code
+        ets, pages = to_entries(res.body)
+        ets.instance_variable_set(:@pages, pages)
+        def ets.pages; @pages; end
+      end
+
+      ets
+    end
+
+    def my_entries(page = nil)
+      return nil unless logined?
+      path = page ? "/#{@user}/?page=#{page}" : "/#{@user}/"
+      res = request(path)
+      ets = nil
+
+      if '200' == res.code
+        ets, pages = to_entries(res.body)
+        ets.instance_variable_set(:@pages, pages)
+        def ets.pages; @pages; end
+      end
+
+      ets
+    end
+
+    def post(title, content)
+      return false unless logined?
+      res = request("/#{@user}/edit", {:mode => 'confirm',  :rkm => rkm, :id => '', :title => title, :body => content, :edit => "\343\201\223\343\201\256\345\206\205\345\256\271\343\202\222\347\231\273\351\214\262\343\201\231\343\202\213"})
+       ('302' == res.code)
+    end
+
+    def raw
+      Marshal.dump('user' => @user, 'cookies' => @cookies.values.map {|cookie| cookie.raw })
+    end
+
+    def update(raw_diary)
+      attrs = Marshal.restore(raw_diary)
+      @user = attrs['user']
+      set_cookie(attrs['cookies'])
+    end
+
+    private
+    def request(path, params = {}, host = @@host)
+      Net::HTTP.version_1_2
+
+      Net::HTTP.start(host, 80) {|http|
+        req = Net::HTTP::Post.new(path)
+        req['Host'] = host
+        req['User-Agent'] = @@user_agent
+        req['Cookie'] = @cookies.values.select {|cookie| cookie.apply?(host, path) }.map {|cookie| cookie.header_string }.join('; ')
+
+        req.body = params.map {|k, v| "#{CGI::escape(k.to_s)}=#{CGI::escape(v.to_s)}" }.join('&')
+        http.request(req)
+      }
+    end
+
+    def set_cookie(cookies)
+      cookies.each {|raw_cookie|
+        cookie = Cookie.parse(raw_cookie)
+        @cookies[cookie.name] = cookie
+      }
+    end
+
+    def rkm
+      return nil unless @cookies.has_key?('rk')
+      rk = @cookies['rk'].value
+      Base64.encode64(Digest::MD5.digest(rk)).strip.sub(/=+\Z/o, '')
+    end
+
+    def to_entry(source, date = '')
+      source = source.to_a if source.kind_of?(String)
+
+      while line = source.shift
+        yield(line) if block_given?
+        if line['<span class="date">']
+          date.replace(%r|<span class="date">(.+)</span>|o.match(line)[1])
+        elsif line['<div class="section">']
+          header = source.shift
+          line = source.shift # skip
+          content = StringIO.new
+
+          until line['<p class="sectionfooter">']
+            content << line unless /<a href="[^"]+" class="edit">/o =~ line
+            line = source.shift
+          end
+
+          footer = line  
+
+          id, title = %r|<h3><a href="/([0-9]+)"><span class="sanchor">.+</span></a>(.*)?\Z|o.match(header).captures
+          title.gsub!('</h3>', '')
+          tb_num = %r|トラックバック\((\d+)\)|uo.match(footer)[1].to_i
+          time = %r|(\d{2}:\d{2})|.match(footer)[1]
+
+          return Entry.new(self, id, title, content.string, Time.parse("#{date} #{time}"), tb_num)
+        end
+      end
+
+      nil
+    end
+
+    def to_entries(source)
+      source = source.to_a if source.kind_of?(String)
+      ets = []
+      date = ''
+      pages = nil
+
+      until source.empty?
+        et = to_entry(source, date) {|line|
+          if line['<div class="pager-r">'] and line['</div>']
+            pages = %r|<div class="pager-r">(\d+)[^<]+</div>|o.match(line)[1]
+          end
+        }
+        ets << et if et
+      end
+
+      [ets, pages]
+    end
+
+    def load_trackbacks(source, diary, parent)
+      source = source.to_a if source.kind_of?(String)
+      tbs = []
+
+      while (line = source.shift) and not line['</ul>']
+        if line['<li>']
+          header = source.shift
+          line = source.shift until source.shift['<div class="box-curve">']
+          line = source.shift # skip
+          snippet = StringIO.new
+
+          until line['<span class="curve-bottom">']
+            snippet << line
+            line = source.shift
+          end
+
+          m = %r|<a href="http://anond.hatelabo.jp/([0-9]+)"|o.match(header)
+          m = %r|<a href="http://d.hatena.ne.jp/([^/]+)/.*"|o.match(header) unless m
+          id = m ? m[1] : nil # TODO
+          tbs << (tb = Trackback.new(diary, parent, id, snippet.string))
+
+          until line['</li>']
+            load_trackbacks(source, diary, tb) if line['<ul>']
+            line = source.shift
+          end
+        end
+      end
+
+      parent.trackbacks = tbs
+    end
+  end # Diary
+
+  class Entry
+    attr_reader :diary, :id, :title, :content, :time
+    attr_writer :trackbacks
+    attr_accessor :tb_num
+
+    def initialize(diary, id, title, content, time, tb_num)
+      @diary = diary
+      @id = id
+      @title = title
+      @content = content
+      @time = time
+      @tb_num = tb_num
+    end
+
+    def trackbacks
+      @trackbacks or @trackbacks = diary.trackbacks(@id)
+    end
+  end # Entry
+
+  class Trackback
+    attr_reader :diary, :parent, :id, :snippet
+    attr_accessor :trackbacks
+
+    def initialize(diary, parent, id, snippet)
+      @diary = diary
+      @parent = parent
+      @id = id
+      @snippet = snippet
+    end
+
+    def entry
+      @entry or @entry = @diary.entry(@id)
+    end
+    
+    def tb_num
+      @trackbacks ? @trackbacks.length : 0
+    end
+  end # Trackback
+
+  class Cookie
+    attr_reader :name, :value, :expires, :domain, :path, :secure
+
+    def initialize(source = {})
+      %w(name value expires domain path secure).each {|k|
+        next unless source.has_key?(k)
+        instance_variable_set("@#{k}", source[k])
+      }
+    end
+
+    def valid?(now = Time.now)
+      not @expires or (now <= @expires)
+    end
+
+    def apply?(domain, path, secure = true, now = Time.now)
+      valid?(now) and
+       (domain.slice(-@domain.length, @domain.length) == @domain) and
+       (path.slice(0, @path.length) == @path) and
+       (not @secure or secure)
+    end
+
+    def header_string
+    "#{CGI::escape(@name)}=#{CGI::escape(@value)}"
+    end
+
+    def raw
+      attrs = []
+      attrs << "#{@name}=#{@value}"
+      attrs << "expires=#{@expires.rfc2822}" if @expires
+      attrs << "domain=#{@domain}" if @domain
+      attrs << "path=#{@path}" if @path
+      attrs << "secure=#{@secure}" if @secure
+      attrs.join('; ')
+    end
+
+    def self.parse(raw_cookie)
+      source = {}
+
+      raw_cookie.split(/;\s?/o).each {|pair|
+        key, value = pair.split('=', 2)
+        next unless key and value
+        key = CGI::unescape(key)
+        value = CGI::unescape(value)
+
+        case key
+        when 'expires'
+          source['expires'] = Time.parse(value)
+        when 'domain'
+          source['domain'] = value
+        when 'path'
+          source['path'] = value
+        when 'secure'
+          source['secure'] = ('true' == value.downcase)
+        else
+          unless source.has_key?('name')
+            source['name'] = key
+            source['value'] = value
+          end
+        end
+      }
+
+      Cookie.new(source)
+    end
+  end # Cookie
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.4
+specification_version: 1
+name: masuda
+version: !ruby/object:Gem::Version 
+  version: 0.5.0
+date: 2007-10-18 00:00:00 +09:00
+summary: Hatena AnonymouseDiary Reader/Writer
+require_paths: 
+- lib
+email: sgwr_dts@yahoo.co.jp
+homepage: http://masuda.rubyforge.org
+rubyforge_project: masuda
+description: Hatena AnonymouseDiary Reader/Writer
+autorequire: 
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- winebarrel
+files: 
+- lib/masuda.rb
+test_files: []
+
+rdoc_options: 
+- --main
+- README.txt
+extra_rdoc_files: []
+
+executables: []
+
+extensions: []
+
+requirements: []
+
+dependencies: []
+