Add Hatena::Bookmark

lib/feedly.rb

@@ -3,8 +3,9 @@
 require 'net/http'
 require 'uri'
 require 'json'
+require_relative 'social_service'
 
-class Feedly < FeedService
+class Feedly < SocialService
   def subscribers
     api_uri = 'http://cloud.feedly.com/v3/feeds/feed%2F' + URI.escape(@blog_rss, ':/')
     JSON.parse(Net::HTTP.get(URI.parse(api_uri)))['subscribers']

lib/growth_forecast.rb

@@ -19,6 +19,10 @@ def post_livedoor_reader number
     post 'livedoor Reader', number
   end
 
+  def post_hatena_bookmark number
+    post 'Hatena::Bookmark', number
+  end
+
   private
 
   def post graph_name, number

lib/hatena_bookmark.rb

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+
+require 'net/http'
+require 'rexml/document'
+require_relative 'social_service'
+
+class HatenaBookmark < SocialService
+  def count
+    # http://developer.hatena.ne.jp/ja/documents/bookmark/apis/getcount#total
+    response = Net::HTTP.new('b.hatena.ne.jp').start do |http|
+      request = <<EOS
+<?xml version="1.0"?>
+<methodCall>
+  <methodName>bookmark.getTotalCount</methodName>
+  <params>
+    <param>
+      <value><string>#{@blog_url}</string></value>
+    </param>
+  </params>
+</methodCall>
+EOS
+      header = {'Content-Type' => 'text/xml; charset=utf-8', 'Content-Length' => request.bytesize.to_s}
+      http.request_post '/xmlrpc', request, header
+    end
+
+    doc = REXML::Document.new response.body
+    doc.elements['/methodResponse/params/param/value/int'].text.to_i
+  end
+end

lib/livedoor_reader.rb

@@ -2,8 +2,9 @@
 
 require 'net/http'
 require 'uri'
+require_relative 'social_service'
 
-class LivedoorReader < FeedService
+class LivedoorReader < SocialService
   def subscribers
     api_uri = 'http://rpc.reader.livedoor.com/count?feedlink=' + @blog_rss
     Net::HTTP.get(URI.parse api_uri).to_i

lib/feed_service.rb → lib/social_service.rb

@@ -1,7 +1,8 @@
 #!/usr/bin/env ruby
 
-class FeedService
+class SocialService
   def initialize settings
+    @blog_url = settings['blog']['url']
     @blog_rss = settings['blog']['rss']
   end
 end

main.rb

@@ -9,3 +9,4 @@
 gf = GrowthForecast.new settings
 gf.post_feedly Feedly.new(settings).subscribers
 gf.post_livedoor_reader LivedoorReader.new(settings).subscribers
+gf.post_hatena_bookmark HatenaBookmark.new(settings).count

settings.example.yml

@@ -6,5 +6,7 @@ growth_forecast:
   section_name:
     any_section_name
 blog:
+  url:
+    http://blog.example.com/
   rss:
     http://blog.example.com/feed.rss