root/platform/tdiary/plugin/search-yahoo.rb

#
# search-yahoo.rb - site search plugin sample using Yahoo! Search BOSS API.
#
# Copyright (C) 2008, TADA Tadashi <sho@spc.gr.jp>
# You can redistribute it and/or modify it under GPL.
#
# Needed these options below:
#
# @options['search-yahoo.appid'] : Your BOSS APPID
# @options['search-yahoo.result_filter'] : your dialy's URL format of DAY mode into Regexp.
#

require 'open-uri'
require 'timeout'
require 'rexml/document'

def search_title
        '全文検索 by Yahoo! Search BOSS'
end

def search_input_form( q )
        r = <<-HTML
                <div>
                        <form method="GET" action="#{@conf.index}">
                        検索キーワード: 
                        <input name="q" value="#{h q}">
                        <input type="submit" value="OK">
                        </form>
                </div>
        HTML
end

def search_boss_api( q, start = 0 )
        url = 'http://boss.yahooapis.com/ysearch/web/v1/'
        appid = @conf['search-yahoo.appid']

        url << "#{q}?appid=#{appid}&format=xml&count=50&start=#{start}"

        proxy = @conf['proxy']
        proxy = 'http://' + proxy if proxy
        timeout( 10 ) do
                open( url, :proxy => proxy ) {|f| f.read }
        end
end

def search_result
        query = CGI::unescape( @cgi.params['q'][0] )
        start = CGI::unescape( @cgi.params['start'][0] || '0' ).to_i

        begin
                uri = URI::parse( @conf.base_url )
                q = "#{query} site:#{uri.host}"
                q << " inurl:#{uri.path}" unless uri.path == '/'
                xml = search_boss_api( u( q.untaint ), start )
                doc = REXML::Document::new( xml ).root
                res = doc.elements.to_a( '/ysearchresponse' )[0]
                unless res.attribute( 'responsecode' ).value == '200' then
                        return '<p class="message">ERROR</p>'
                end
        rescue OpenURI::HTTPError
                return %Q|<p class="message">#$!</p>|
        end

        r = search_input_form( query )
        r << '<dl class="search-result">'
        doc.elements.to_a( '*/result' ).each do |elem|
                url = elem.elements.to_a( 'url' )[0].text
                next unless url =~ @conf['search-yahoo.result_filter']
                title = elem.elements.to_a( 'title' )[0].text
                abstract = elem.elements.to_a( 'abstract' )[0].text
                r << %Q|<dt><a href="#{h url}">#{title}</a></dt>|
                r << %Q|<dd>#{abstract}</dd>|
        end
        r << '</dl>'

        r << '<div class="search-navi">'
        doc.elements.to_a( '/ysearchresponse/prevpage' ).each do |p|
                if /start=\d+/ =~ p.text then
                        r << %Q|<a href="#{@conf.index}?q=#{u query}&#$&">&lt;前の50件</a>&nbsp;|
                end
        end

        doc.elements.to_a( '/ysearchresponse/nextpage' ).each do |n|
                if /start=\d+/ =~ n.text then
                        r << %Q|<a href="#{@conf.index}?q=#{u query}&#$&">次の50件&gt;</a>|
                end
        end
        r << '</div>'

        r
end