google_suggest 0.0.2

data/lib/google_suggest.rb

@@ -0,0 +1,77 @@
+# encoding: utf-8
+
+require 'uri'
+require 'nokogiri'
+require 'net/http'
+
+class GoogleSuggest
+  class Configure
+    attr_accessor :home_language
+    attr_accessor :proxy
+
+    def initialize
+      @home_language = 'en'
+    end
+  end
+
+  attr_accessor :home_language
+  attr_accessor :proxy
+
+  @@configure = Configure.new
+
+  class << self
+    def configure
+      yield @@configure if block_given?
+      @@configure
+    end
+
+    def suggest_for(keyword)
+      self.new.suggest_for(keyword)
+    end
+  end
+
+  def initialize
+    @home_language = @@configure.home_language
+    @proxy = @@configure.proxy
+  end
+
+  def suggest_for(keyword)
+    query = {:output => 'toolbar',
+             :hl => self.home_language,
+             :q => URI.encode(keyword)}
+    res = http_get('/complete/search', query)
+    xml = Nokogiri::XML(res.body.to_s)
+    suggestions = []
+    xml.css('toplevel CompleteSuggestion').each do |node|
+      suggest = {}
+      node.children.each do |child|
+        suggest[child.name] = (child['data'] || child['int'].to_i)
+      end
+      if suggest['suggestion'] and not suggest['suggestion'].valid_encoding?
+        suggest['suggestion'].force_encoding('Shift_JIS').encode!('UTF-8')
+      end
+      suggestions << suggest
+    end
+    return suggestions
+  end
+
+  private
+  GOOGLE_HOST = 'www.google.com'
+
+  def http
+    if @proxy
+      proxy_url = URI.parse(@proxy) 
+      http_class = Net::HTTP.Proxy(proxy_url.host, proxy_url.port)
+    else
+      http_class = Net::HTTP
+    end
+    http_class.new(GOOGLE_HOST)
+  end
+
+  def http_get(path, query) 
+    path = path + '?' + query.map{|k,v| "#{k}=#{v}"}.join('&')
+    req = Net::HTTP::Get.new(path)
+    http.request(req)
+  end
+
+end

data/spec/google_suggest-spec.rb

@@ -0,0 +1,127 @@
+# encoding: utf-8
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+
+describe GoogleSuggest do
+  context "default settiong" do
+    subject { GoogleSuggest.new }
+
+    its(:home_language) { should be_eql 'en' }
+    its(:proxy) { should be_nil }
+  end
+
+  context "Class Methods" do
+    before do
+      google_suggest = GoogleSuggest.new
+      res = Object.new
+      res.stub!(:body) do
+        doc = nil
+        File.open(File.join(File.dirname(__FILE__), 'sample_us.xml')) do |f|
+          doc = f.read
+        end
+        doc
+      end
+      google_suggest.stub!(:http_get) { res }
+      suggestions = google_suggest.suggest_for('google')
+      GoogleSuggest.stub!(:new).and_return { mock('google_suggest', :suggest_for => suggestions) }
+
+      @suggestions = GoogleSuggest.suggest_for('google')
+    end
+     
+    it do 
+      @suggestions.size.should be 10
+    end
+    it do
+      @suggestions.should be_all do
+      end
+    end
+  end
+
+  describe GoogleSuggest::Configure do
+    context "#configure called without block" do
+      before do
+        @configure = GoogleSuggest.configure
+        @configure.home_language = 'ja'
+        @configure.proxy = 'http://proxy.example.com'
+      end
+      subject { GoogleSuggest.new }
+      its(:home_language) { should be_eql 'ja' }
+      its(:proxy) { should be_eql 'http://proxy.example.com' }
+    end
+    context "#configure called and given block" do
+      before do
+        GoogleSuggest.configure do |c|
+          c.home_language = 'us'
+          c.proxy = 'http://proxy.example.com'
+        end
+      end
+      subject { GoogleSuggest.new }
+      its(:home_language) { should be_eql 'us' }
+      its(:proxy) { should be_eql 'http://proxy.example.com' }
+    end
+  end
+
+  context "#suggest_from method" do
+    before :all do
+      GoogleSuggest.configure do |c|
+        c.home_language = 'us'
+      end
+    end
+
+    before do
+      @google_suggest = GoogleSuggest.new
+      res = Object.new
+      res.stub!(:body) do
+        doc = nil
+        File.open(File.join(File.dirname(__FILE__), 'sample_us.xml')) do |f|
+          doc = f.read
+        end
+        doc
+      end
+      @google_suggest.stub!(:http_get) { res }
+    end
+
+    it do
+      suggestions = @google_suggest.suggest_for 'google'
+      suggestions.size.should == 10
+    end
+
+    it 'all suggestions should have the value of the key \'suggest\' 'do
+      suggestions = @google_suggest.suggest_for 'google'
+      suggestions.should be_all do |suggest|
+        not suggest['suggest'].nil?
+      end
+    end
+
+
+    it 'all suggestions should have the value of the key \'num_queries\' 'do
+      suggestions = @google_suggest.suggest_for 'google'
+      suggestions.should be_all do |suggest|
+        not suggest['num_queries'].nil?
+      end
+    end
+  end
+  describe "#suggest_for with home language 'ja'" do
+    before :all do
+      GoogleSuggest.configure do |c|
+        c.home_language = 'ja'
+      end
+    end
+
+    before do
+      @google_suggest = GoogleSuggest.new
+      res = Object.new
+      res.stub!(:body) do
+        doc = nil
+        File.open(File.join(File.dirname(__FILE__), 'sample_ja.xml')) do |f|
+          doc = f.read
+        end
+        doc
+      end
+      @google_suggest.stub!(:http_get) { res }
+    end
+      subject {@google_suggest.suggest_for('グーグル').shift['suggestion']}
+
+    its(:encoding) { should be Encoding.find("UTF-8") }
+    it { should be_valid_encoding }
+  end
+end

data/spec/sample_ja.xml

@@ -0,0 +1 @@
+<?xml version="1.0"?><toplevel><CompleteSuggestion><suggestion data="グーグル"/><num_queries int="13000000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルマップ"/><num_queries int="293000000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルアース"/><num_queries int="183000000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルクローム"/><num_queries int="782000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルカレンダー"/><num_queries int="9830000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグル翻訳"/><num_queries int="5140000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルショッピング"/><num_queries int="21900000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルスカラー"/><num_queries int="91600"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルメール"/><num_queries int="62700000"/></CompleteSuggestion><CompleteSuggestion><suggestion data="グーグルドキュメント"/><num_queries int="2260000"/></CompleteSuggestion></toplevel>

data/spec/sample_us.xml

@@ -0,0 +1,12 @@
+<?xml version="1.0"?><toplevel>
+<CompleteSuggestion><suggestion data="google ebook store"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google chrome os"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google earth"/><num_queries int="113000000"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google 翻訳"/><num_queries int="5290000"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google chrome"/><num_queries int="69000000"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google apps"/><num_queries int="102000000"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google カレンダー"/><num_queries int="155000000"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google scholar"/><num_queries int="20600000"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="google usa"/><num_queries int="215000000"/></CompleteSuggestion>
+<CompleteSuggestion><suggestion data="googlemap"/><num_queries int="84800000"/></CompleteSuggestion>
+</toplevel>% 

data/spec/spec_helper.rb

@@ -0,0 +1,8 @@
+# encoding: utf-8
+
+require File.expand_path(File.join(File.dirname(__FILE__), '..',
+                           'lib', 'google_suggest'))
+
+RSpec.configure do |c|
+  c.mock_with :rspec
+end

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification 
+name: google_suggest
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 2
+  version: 0.0.2
+platform: ruby
+authors: 
+- Tatsuya Sato
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-17 00:00:00 +09:00
+default_executable: 
+dependencies: []
+
+description: 
+email: satoryu.1981@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/google_suggest.rb
+- spec/sample_ja.xml
+- spec/google_suggest-spec.rb
+- spec/spec_helper.rb
+- spec/sample_us.xml
+has_rdoc: true
+homepage: http://github.com/satoryu/google_suggest/
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 1
+      - 8
+      - 7
+      version: 1.8.7
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: A gem which allows us to retrieve suggest words from Google in your Ruby Code.
+test_files: []
+