beagled/EvolutionMailDriver/MailCrawler.cs

   1
   2 //
   3 // MailCrawler.cs
   4 //
   5 // Copyright (C) 2004 Novell, Inc.
   6 //
   7 //
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining a
  10 // copy of this software and associated documentation files (the "Software"),
  11 // to deal in the Software without restriction, including without limitation
  12 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
  13 // and/or sell copies of the Software, and to permit persons to whom the
  14 // Software is furnished to do so, subject to the following conditions:
  15 //
  16 // The above copyright notice and this permission notice shall be included in
  17 // all copies or substantial portions of the Software.
  18 //
  19 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  22 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  24 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  25 // DEALINGS IN THE SOFTWARE.
  26 //
  27
  28 using System;
  29 using System.Collections;
  30 using System.IO;
  31
  32 using Beagle.Util;
  33 using Beagle.Daemon;
  34
  35 namespace Beagle.Daemon.EvolutionMailDriver {
  36
  37         class MailCrawler {
  38                 public delegate void ItemAddedHandler (FileInfo file);
  39
  40                 private static bool Debug = false;
  41
  42                 ArrayList roots = new ArrayList ();
  43
  44                 Hashtable last_write_time_cache = new Hashtable ();
  45
  46                 public ItemAddedHandler MboxAddedEvent;
  47                 public ItemAddedHandler SummaryAddedEvent;
  48
  49                 public MailCrawler (params string[] paths)
  50                 {
  51                         foreach (string p in paths) {
  52                                 if (Directory.Exists (p))
  53                                         roots.Add (p);
  54                         }
  55                 }
  56
  57                 private bool FileIsInteresting (FileInfo file)
  58                 {
  59                         if (! file.Exists)
  60                                 return false;
  61
  62                         DateTime cached_time = new DateTime ();
  63                         if (last_write_time_cache.Contains (file.FullName))
  64                                 cached_time = (DateTime) last_write_time_cache [file.FullName];
  65
  66                         last_write_time_cache [file.FullName] = file.LastWriteTimeUtc;
  67
  68                         return cached_time < file.LastWriteTimeUtc;
  69                 }
  70
  71                 private void OnInotifyEvent (Inotify.Watch watch,
  72                                              string path,
  73                                              string subitem,
  74                                              string srcpath,
  75                                              Inotify.EventType type)
  76                 {
  77                         if (subitem == "")
  78                                 return;
  79
  80                         string full_path = Path.Combine (path, subitem);
  81
  82                         if ((type & Inotify.EventType.Create) != 0 && (type & Inotify.EventType.IsDirectory) != 0) {
  83                                 Watch (full_path);
  84                                 return;
  85                         }
  86
  87                         if ((type & Inotify.EventType.Delete) != 0 && (type & Inotify.EventType.IsDirectory) != 0) {
  88                                 watch.Unsubscribe ();
  89                                 return;
  90                         }
  91
  92                         if ((type & Inotify.EventType.MovedTo) != 0) {
  93                                 if (subitem == "summary") {
  94                                         // IMAP summary
  95                                         if (SummaryAddedEvent != null && File.Exists (full_path)) {
  96                                                 Logger.Log.Info ("Reindexing updated IMAP summary: {0}", full_path);
  97                                                 SummaryAddedEvent (new FileInfo (full_path));
  98                                         }
  99                                 } else if (Path.GetExtension (full_path) == ".ev-summary") {
 100                                         // mbox summary
 101                                         string mbox_file = Path.ChangeExtension (full_path, null);
 102                                         if (MboxAddedEvent != null && File.Exists (mbox_file)) {
 103                                                 Logger.Log.Info ("Reindexing updated mbox: {0}", mbox_file);
 104                                                 MboxAddedEvent (new FileInfo (mbox_file));
 105                                         }
 106                                 }
 107                         }
 108                 }
 109
 110                 private void Watch (string root)
 111                 {
 112                         Queue pending = new Queue ();
 113
 114                         pending.Enqueue (root);
 115
 116                         while (pending.Count > 0) {
 117
 118                                 string dir = (string) pending.Dequeue ();
 119
 120                                 foreach (string subdir in DirectoryWalker.GetDirectories (dir)) {
 121                                         if (Shutdown.ShutdownRequested)
 122                                                 return;
 123
 124                                         if (Inotify.Enabled) {
 125                                                 Inotify.Subscribe (subdir, OnInotifyEvent,
 126                                                                    Inotify.EventType.Create
 127                                                                    | Inotify.EventType.Delete
 128                                                                    | Inotify.EventType.MovedTo);
 129                                         }
 130
 131                                         pending.Enqueue (subdir);
 132                                 }
 133
 134                                 Stopwatch watch = new Stopwatch ();
 135                                 if (Debug)
 136                                         Logger.Log.Debug ("Starting watch on {0}", dir);
 137                                 watch.Start ();
 138                                 foreach (string path in DirectoryWalker.GetItems (dir, new DirectoryWalker.FileFilter (IsSummary))) {
 139                                         if (Shutdown.ShutdownRequested)
 140                                                 return;
 141
 142                                         FileInfo file = new FileInfo (path);
 143
 144                                         if (file.Name == "summary") {
 145                                                 if (SummaryAddedEvent != null && FileIsInteresting (file))
 146                                                         SummaryAddedEvent (file);
 147                                         } else if (file.Extension == ".ev-summary") {
 148                                                 string mbox_name = Path.Combine (file.DirectoryName,
 149                                                                                  Path.GetFileNameWithoutExtension (file.Name));
 150                                                 FileInfo mbox_file = new FileInfo (mbox_name);
 151                                                 if (MboxAddedEvent != null && FileIsInteresting (mbox_file))
 152                                                         MboxAddedEvent (mbox_file);
 153                                         }
 154                                 }
 155                                 watch.Stop ();
 156                                 if (Debug)
 157                                         Logger.Log.Debug ("Crawled {0} in {1}", dir, watch);
 158                         }
 159                 }
 160
 161                 private bool IsSummary (string path, string name)
 162                 {
 163                         if (name == "summary")
 164                                 return true;
 165
 166                         if (Path.GetExtension (name) == ".ev-summary")
 167                                 return true;
 168
 169                         return false;
 170                 }
 171
 172                 public void Crawl ()
 173                 {
 174                         foreach (string root in roots) {
 175                                 if (Shutdown.ShutdownRequested)
 176                                         return;
 177
 178                                 Watch (root);
 179                         }
 180                 }
 181         }
 182 }