af_solr_generator 0.0.2

data/af_solr_generator.rb

@@ -0,0 +1,54 @@
+class Stubby
+  def initialize
+    @f = Hash.new
+  end
+  def field(name, type, modifier={})
+    @f[name.to_sym]=type.to_sym
+  end
+  def retrieve
+    @f
+  end
+end
+
+class AfSolrGenerator < Rails::Generator::NamedBase
+  attr_accessor :fields
+  def manifest
+    @fields = Hash.new
+    record do |m|
+      z = name.classify.constantize
+      raise "#{z} is not an ActiveFedora::Base!" unless z.superclass == ActiveFedora::Base
+      z.ds_specs.each do |k,v|
+        stubby = Stubby.new
+        v.last.call(stubby)
+        @fields[k.to_sym] = stubby.retrieve
+      end
+
+
+      m.directory 'app/controllers/'
+      m.directory "app/views/#{name}_search"
+      m.template "controller_template.erb",  "app/controllers/#{name}_search_controller.rb"
+      m.template "search_form.erb",  "app/views/#{name}_search/#{name}_search_form.html.erb"
+      m.template "results.erb",  "app/views/#{name}_search/results.html.erb"
+      m.template "show.erb",  "app/views/#{name}_search/show.html.erb"
+      puts <<-EOS
+        Make sure to add a route for the search interface, if you're using ReST.  The search
+        interface lives at :controller=>#{name}_search_controller, :action=>search
+
+        It's up to you to configure your routes, but the system defaults to work with
+        the default passthrough rails routes:
+
+        map.connect ':controller/:action/:id'
+
+      EOS
+    end
+  end
+
+  protected
+    def banner
+      <<-EOS
+Creates an ActiveFedora search controller based on your ActiveFedora::Base model definitions.
+
+USAGE: #{$0} #{spec.name} <model name>
+EOS
+    end
+end

data/lib/af_solr_generator.rb

@@ -0,0 +1,6 @@
+$:.unshift(File.dirname(__FILE__)) unless
+  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+module AfSolrGenerator
+  VERSION = '0.0.2'
+end

data/templates/controller_template.erb

@@ -0,0 +1,39 @@
+require 'rubygems'
+gem 'ruby-fedora'
+gem 'active-fedora'
+class <%=class_name%>SearchController < ApplicationController
+  before_filter :register_af
+  STREAMS= [<%=fields.keys.map{|x|":#{x}"}.join(', ')%>]
+
+  def index
+    render :action=>'<%=name%>_search_form'
+  end
+
+  def search
+    @hits = ActiveFedora::SolrService.instance.conn.query(transform(params)).to_a
+    render :action=>'results'
+  end
+
+  def transform(params)
+    q= []
+    STREAMS.each do |s|
+      params[s].each{|k,v| q<<"#{k}_field:#{v}" unless v.blank?} 
+    end
+    q.join(' ')
+    
+  end
+
+  #this really doesn't belong here, you should probably create a real controller for your <%=class_name%>s
+  def show
+    @inst = <%=class_name%>.load_instance(params[:id])
+  end
+
+
+  private
+
+  def register_af
+  #change this if your fedora/solr runs somewhere else...
+    ActiveFedora::SolrService.register('http://localhost:8080/solr')
+    Fedora::Repository.register('http://fedoraAdmin:fedoraAdmin@localhost:8080/fedora')
+  end
+end

data/templates/results.erb

@@ -0,0 +1,19 @@
+<h1><%%=@hits.size%> <%=name.humanize%> Found.</h1>
+<table border='1'>
+  <tr>    
+    <th>PID</th>
+    <%%@hits.first.keys.sort.each do |k|%>
+      <th><%%=k.to_s.humanize%></th>
+    <%%end%>
+  </tr>
+
+
+<%%@hits.each  do |f|%>
+  <tr>
+    <td><%%=link_to f['id'], :action=>:show, :id=>f['id']%></td>
+  <%%- f.keys.compact.sort.each do |v| %>
+    <td><%%=f[v] %></td>
+  <%%end%>
+<%%end%>
+</tr>
+</table>

data/templates/search_form.erb

@@ -0,0 +1,12 @@
+<%%-form_tag url_for(:action=>:search, :controller=>:<%=name%>_search) do%>
+  <%-fields.each do |k,v| %>
+    <%% fields_for :<%=k%> do |f|%>
+      <h1><%=k%></h1>
+      <%v.each do |f| %>
+        <label><%=f.first.to_s.humanize%></label>
+        <%%=f.text_field :<%=f.first%>%>
+      <%end%>
+    <%%end%>
+  <%-end%>
+  <%%=submit_tag "Search"%>
+<%%- end %>

data/templates/show.erb

@@ -0,0 +1,15 @@
+<h1><%%=@inst.pid%></h1>
+<table>
+  <tr><th>Key</th><th>Value</th></tr>
+  <%%@inst.datastreams.each do |name, ds| %>
+    <tr ><th colspan="2" style="background:#f0f0f0;">Datastream: <%%=ds.dsid%>(<%%=ds.class%>)</th></tr>
+    <%%ds.fields.each do |k,v|%>
+      <tr><td style="background:#ff4000"><%%=k.to_s%></td><td><%%=v[:values].join("\r")%></td></tr>
+    <%%end%>
+    <%% ds.attributes.each do |k,v| %> 
+      <tr>
+        <td style="background:#707070"><%%=k%></td><td><%%=v%></td>
+      </tr>
+     <%%end%>
+  <%%end%>
+</table>

data/test/test_af_solr.rb

@@ -0,0 +1,11 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestAfSolr < Test::Unit::TestCase
+
+  def setup
+  end
+  
+  def test_truth
+    assert true
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,3 @@
+require 'stringio'
+require 'test/unit'
+require File.dirname(__FILE__) + '/../lib/af_solr'

metadata

@@ -0,0 +1,90 @@
+--- !ruby/object:Gem::Specification 
+name: af_solr_generator
+version: !ruby/object:Gem::Version 
+  version: 0.0.2
+platform: ruby
+authors: 
+- MediaShelf LLC
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-02-26 00:00:00 -06:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: ruby-fedora
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.0.1
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: active-fedora
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.0.1
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.8.0
+    version: 
+description: FIX (describe your package)
+email: 
+- matt.zumwalt@yourmediashelf.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- af_solr_generator.rb
+- lib/af_solr_generator.rb
+- templates/controller_template.erb
+- templates/search_form.erb
+- templates/results.erb
+- templates/show.erb
+has_rdoc: true
+homepage: FIX (url)
+post_install_message: 
+rdoc_options: 
+- --main
+- README.rdoc
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: rubyfedora
+rubygems_version: 1.3.1
+signing_key: 
+specification_version: 2
+summary: FIX (describe your package)
+test_files: 
+- test/test_af_solr.rb
+- test/test_helper.rb