Searching Beast and WordPress from a Rails app
published 2008-05-17
I did some work for ecolect recently, integrating search across their main site, a Beast forum and a WordPress blog. It was pretty straightforward once I had it figured out, but I couldn't find a walkthrough on the net.
So, I decided to write one.
The Search Engine
Ecolect was already using
acts_as_ferret for
their main site search, so it was a no-brainer to keep using it. For a
fun and thorough introduction to acts_as_ferret, see the Rails Envy
Tutorial
Searching the Beast Forum
(Note: the beast Forum search isn't live at the moment as the Forums are not fully functioning yet.)
Beast setup
In Beast, each Forum has many Topics (saved in the topics table), and each Topic has many Posts (saved in the posts table).
The Beast forum was set up by Shanti Braford as described in option #3 of this great article on integrating a beast forum into a Rails app.
In this setup, the Beast tables are added to the main site's database. So, the posts and topics tables that we want to search are already in the main site's database. This made things pretty easy: you can search the Beast forum just like you would any database table.
Searching a non-integrated Beast forum
If you don't have the Beast tables integrated in to your main site's database, you can still search them. You just need to point the Topic and Post models to the correct database. This is a two step process.
First, set up a database entry for your Beast forum in
config/database.yml. Something like this:
beast: adapter: mysql database: beast_forum username: app password: your_password host: localhost
Then, in the bottom of config/environment.rb, add the following lines:
Post.establish_connection "beast_forum" Topic.establish_connection "beast_forum"
I haven't actually tried this, so let me know if you get it working or if you needed to make any changes to what I've written here.
The Topic Model
Even with the integrated setup there were a few wrinkles. First, although the Beast tables are in the database, there are no models associated with them. I wanted to search post body and topic titles, so I created Topic and Post models.
Here's the Topic model. It's only here so that the Post model can search a posts's titles, so there's not much to it.
class Topic < ActiveRecord::Base has_many :posts end
The Post Model
The Post model is a bit more complicated.
class Post < ActiveRecord::Base belongs_to :topic SEARCH_FIELDS = [:scrubbed_body, :topic_title] acts_as_ferret :fields => { :scrubbed_body => {:boost => 0, :store => :yes}, :topic_title => {:boost => 3, :store => :yes}} extend FullTextSearch # remove the html tags from body def scrubbed_body body_html.gsub(/<\/?[^>]*>/, "") end def topic_title topic.title end # Construct the url to the post def url File.join("http://forums.ecolect.net", "forums", topic.forum_id.to_s, "topics", topic.id.to_s + "#post_#{id}") end end
acts_as_ferret
The acts_as_ferret declaration makes the Post model searchable. Notice
that the actual fields being searched are not taken directly from the
database; they are both manipulated in some way. acts_as_ferret
doesn't really care if the stuff it is indexing is coming directly from
the database or from methods you have added to your model.
scrubbing the html tags
The post body is stored with HTML tags in them, so I wanted to search
and show the posts with tags scrubbed out of them. This is done using
the Post#scrubbed_body method, which is just an ugly regexp that takes
out anything between < and > signs.
the url method
I also wanted to link to the posts, so I created a Post#url method
which is used in the view.
Finally, the actual search is done using the FullTextSearch mixin,
which adds a class method Post::full_text_search to Post. The
FullTextSearch mixin is described in more detail below.
Searching the WordPress Blog
The only table from the WordPress Blog that you really care about is the
wp_post table. To get access to it in your Rails app, make a WpPost
model and point it at your WordPress db.
First, create a database entry in config/database.yml that looks like
this:
wordpress: adapter: mysql database: wordpress username: app password: your_password host: localhost
Then, add the following line to the bottom of config/environment.rb
WpPost.establish_connection "wordpress"
Here's the model:
class WpPost < ActiveRecord::Base primary_key = "ID" SEARCH_FIELDS = [:scrubbed_title, :scrubbed_content] acts_as_ferret :fields => { :scrubbed_title => {:boost => 3, :store => :yes}, :scrubbed_content => {:boost => 0, :store => :yes}} extend FullTextSearch def id read_attribute(:ID) end # title and content need to have the html tags removed from them, and should # only be searchable if the post has been assigned a url (the guid) and the post # is actually a post and not an asset. def scrubbed_title post_title.gsub(/<\/?[^>]*>/, "") if post_type == "post" and !guid.empty? end def scrubbed_content post_content.gsub(/<\/?[^>]*>/, "") if post_type == "post" and !guid.empty? end end
There are a few things to note here:
WordPress uses ID, rather than id, as its primary key. The line
primary_key = "ID" lets Rails know about that. You also need to add an
id method that returns ID to get ferret indexing things properly.
You will need to scrub the html tags from the content and title; that's
what the scrubbed_title and scrubbed_content methods do.
Finally, you don't want search results to index assets (which are stored in the wp_post model as well) or any un-published posts.
- Real posts will have a
post_typeof'post'. - An unpublished post won't have its guid set.
This is taken care of by only returning titles or content
if post_type == "post" and !guid.empty?.
The FullTextSearch mixin
This is based on code by Roman Mackovcak's article on full text search in Rails. All I did was extract the method he provides out in to a mixin so I could use it in multiple models.
To use the mixin in a model, the model needs to define SEARCH_FIELDS
and have an acts_as_ferret declaration. SEARCH_FIELDS is an array of
symbols giving the model fields to be searched.
module FullTextSearch ## # FERRET SEARCH METHOD # This method requires that you set the following in the model: # SEARCH_FIELDS: # a list of symbols giving the fields to be searched. # E.g., SEARCH_FIELDS = [:post_title, :post_content] # # The acts_as_ferret declaration. # Use :store => :yes for each field if you want to use highlighting for that field. # E.g., # acts_as_ferret :fields => { :post_title => {:boost => 3, :store => :yes}, # :post_content => {:boost => 0, :store => :yes}} ## DEFAULT_PER_PAGE = 10 def full_text_search(q, options = {}) return nil if q.nil? || q == "" default_options = {:limit => FullTextSearch::DEFAULT_PER_PAGE, :page => 1, :lazy => self.const_get(:SEARCH_FIELDS) } options = default_options.merge options # Get the offset based on what page we're on options[:offset] = options[:limit] * (options.delete(:page).to_i-1) # Now do the query with our options results = self.find_by_contents(q, options) return [results.total_hits, results] end end
You use it like this:
WpPost.full_text_search('test') WpPost.full_text_search('test', :limit => 30, :page => 2)
Re-indexing the non-local models
As Ruben pointed out in the comments, I forgot to mention how I deal with re-indexing the WordPress and Beast database tables. This is necessary as these tables have data that is modified by another application, so your Rails app doesn't know that changes have been made to them.
To deal with this, I wrote a simple Rake task that reindexes the
WpPost, WpArticle and Post models, and then added a line to the
crontab to run it hourly. Here's the rake file, which I put in
RAILS_ROOT/lib/tasks/ferret.rake:
namespace :ferret do desc "rebuild the non-local Ferret indices, as these need to be done manually." task :rebuild_nonlocal_indices => :environment do [WpPost, WpArticle, Post].each do |model| puts "rebuilding #{model} index" model.rebuild_index end end end
and, for posterity, here's the crontab line:
20 * * * * cd <rails_root> && /usr/local/bin/rake ferret:rebuild_nonlocal_indices >> <log_directory>/ferret_reindex.log 2>&1