1 files changed, 107 insertions, 0 deletions
diff --git a/mediagoblin/plugins/indexedsearch/__init__.py b/mediagoblin/plugins/indexedsearch/__init__.py
new file mode 100644
index 00000000..058fb28b
--- /dev/null
+++ b/mediagoblin/plugins/indexedsearch/__init__.py
@@ -0,0 +1,107 @@
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+# from pkg_resources import resource_filename
+
+import os
+import logging
+import importlib
+
+from sqlalchemy import event
+
+from mediagoblin.db.models import (
+    MediaEntry, Comment
+)
+from mediagoblin.tools import pluginapi
+
+_log = logging.getLogger(__name__)
+PLUGIN_DIR = os.path.dirname(__file__)
+
+
+def setup_plugin():
+    """Setup plugin by adding routes and templates to mediagoblin"""
+
+    _log.info('Setting up routes and templates')
+    config = pluginapi.get_config('mediagoblin.plugins.indexedsearch')
+
+    if config.get('USERS_ONLY'):
+        view = 'user_search_results_view'
+    else:
+        view = 'search_results_view'
+
+    routes = [
+        ('mediagoblin.plugins.indexedsearch',
+         '/search/',
+         'mediagoblin.plugins.indexedsearch.views:' + view)]
+
+    pluginapi.register_routes(routes)
+    pluginapi.register_template_path(os.path.join(PLUGIN_DIR, 'templates'))
+
+    search_link_style = config.get('SEARCH_LINK_STYLE')
+    _log.debug("Search link style was specified as: %r", search_link_style)
+
+    if search_link_style in ['button', 'link', 'none', 'form']:
+        header_template = ('indexedsearch/search_link_%s.html' %
+                           search_link_style)
+    else:
+        header_template = 'indexedsearch/search_link_form.html'
+
+    pluginapi.register_template_hooks(
+        {'header_extra': header_template})
+
+
+def setup_engine(mediagoblin_app):
+    """Setup engine by updating the index and adding database hook."""
+    _log.info('Setting up engine')
+    get_engine().update_index()
+    add_event_hooks()
+    return mediagoblin_app
+
+
+hooks = {
+    'setup': setup_plugin,
+    'wrap_wsgi': setup_engine
+}
+
+
+def get_engine():
+    config = pluginapi.get_config('mediagoblin.plugins.indexedsearch')
+    backend_module = importlib.import_module(config.get('BACKEND'))
+    return backend_module.Engine(**config)
+
+
+def comment_change(mapper, connection, comment):
+    """If a comment on a media entry has been removed, reindex the entry."""
+    if isinstance(comment.target(), MediaEntry):
+        get_engine().add_media_entry(comment.target())
+
+
+def media_entry_updated(mapper, connection, media_entry):
+    """If a comment on a media entry has been removed, reindex the entry."""
+    get_engine().add_media_entry(media_entry)
+
+
+def media_entry_deleted(mapper, connection, media_entry):
+    """Delete a media entry"""
+    get_engine().remove_media_entry(media_entry.id)
+
+
+def add_event_hooks():
+    for event_type in 'after_delete', 'after_update', 'after_insert':
+        event.listen(Comment, event_type, comment_change)
+
+    event.listen(MediaEntry, 'after_delete', media_entry_deleted)
+    event.listen(MediaEntry, 'after_update', media_entry_updated)
+    event.listen(MediaEntry, 'after_insert', media_entry_updated)