a8feb2ac18dde7ad4c5e030f7d63824e0009963a
[WebThing.git] / SearchHandler.cs
1 using System;
2 using System.Collections.Generic;
3 using System.Text.RegularExpressions;
4
5 namespace bytex64.WebThing {
6     public class SearchHandler {
7         WebThing wt;
8         Dictionary<string,ISearchPlugin> Handlers;
9         ISearchPlugin DefaultSearchHandler;
10         
11         public SearchHandler(WebThing wt) {
12             Handlers = new Dictionary<string,ISearchPlugin>();
13             this.wt = wt;
14
15             foreach (SearchHandlerPair p in Config.SearchHandlers) {
16                 AddHandler(p.Shortcut, p.Plugin);
17             }
18             if (Config.DefaultSearchHandler != null) {
19                 SetDefaultHandler(Config.DefaultSearchHandler);
20             }
21         }
22
23         private bool IsValidSearchHandler(WebThingPlugin p) {
24             Type ptype = p.GetType();
25             if (ptype.GetInterface("ISearchPlugin") != null)
26                 return true;
27             return false;
28         }
29
30         public void AddHandler(string key, string plugin) {
31             if (Handlers.ContainsKey(key)) {
32                 Console.WriteLine("Cannot add search handler {0}: shortcut {1} already registered", plugin, key);
33                 return;
34             }
35             if (!wt.Plugins.Plugins.ContainsKey(plugin)) {
36                 Console.WriteLine("Cannot add search handler {0}: Plugin for {0} not loaded", plugin);
37                 return;
38             }
39
40             WebThingPlugin p = wt.Plugins.Plugins[plugin];
41
42             if (IsValidSearchHandler(p)) {
43                 Handlers[key] = (ISearchPlugin) p;
44                 Console.WriteLine("Added handler {0} for key {1}", plugin, key);
45             } else {
46                 Console.WriteLine("Cannot add search handler {0}: {0} Does not implement ISearchPlugin", plugin);
47             }
48         }
49
50         public void SetDefaultHandler(string plugin) {
51             if (wt.Plugins.Plugins.ContainsKey(plugin)) {
52                 WebThingPlugin p = wt.Plugins.Plugins[plugin];
53                 if (IsValidSearchHandler(p)) {
54                     DefaultSearchHandler = (ISearchPlugin) p;
55                 } else {
56                     Console.WriteLine("Could not set default search handler: {0} does not implement ISearchPlugin", plugin);
57                 }
58             } else {
59                 Console.WriteLine("Could not set default search handler: {0} is not loaded", plugin);
60             }
61         }
62
63         // Call a search handler based on the leading word
64         public string Transform(string search) {
65             Regex get_handler_re = new Regex(@"^([\w-]+)\s+");
66             Match m = get_handler_re.Match(search);
67             if (m.Success) {
68                 string key = m.Groups[1].Value;
69                 string query = get_handler_re.Replace(search, "");
70                 if (Handlers.ContainsKey(key)) {
71                     return Handlers[key].SearchTransform(query);
72                 } else {
73                     if (DefaultSearchHandler != null) {
74                         return DefaultSearchHandler.SearchTransform(search);
75                     } else {
76                         Console.WriteLine("Could not search with {0} and no default search handler defined", key);
77                         return null;
78                     }
79                 }
80             } else {
81                 if (DefaultSearchHandler != null) {
82                     return DefaultSearchHandler.SearchTransform(search);
83                 } else {
84                     Console.WriteLine("Could not get search key and no default search handler defined");
85                     return null;
86                 }
87             }
88         }
89     }
90 }