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                         DateTime cached_time = new DateTime ();
  60                         if (last_write_time_cache.Contains (file.FullName))
  61                                 cached_time = (DateTime) last_write_time_cache [file.FullName];
  62
  63                         last_write_time_cache [file.FullName] = file.LastWriteTimeUtc;
  64
  65                         return cached_time < file.LastWriteTimeUtc;
  66                 }
  67
  68                 private void OnInotifyEvent (Inotify.Watch watch,
  69                                              string path,
  70                                              string subitem,
  71                                              string srcpath,
  72                                              Inotify.EventType type)
  73                 {
  74                         if (subitem == "")
  75                                 return;
  76
  77                         string full_path = Path.Combine (path, subitem);
  78
  79                         if ((type & Inotify.EventType.Create) != 0 && (type & Inotify.EventType.IsDirectory) != 0) {
  80                                 Watch (full_path);
  81                                 return;
  82                         }
  83
  84                         if ((type & Inotify.EventType.Delete) != 0 && (type & Inotify.EventType.IsDirectory) != 0) {
  85                                 watch.Unsubscribe ();
  86                                 return;
  87                         }
  88
  89                         if ((type & Inotify.EventType.MovedTo) != 0) {
  90                                 if (subitem == "summary") {
  91                                         // IMAP summary
  92                                         Logger.Log.Info ("Reindexing updated IMAP summary: {0}", full_path);
  93                                         if (SummaryAddedEvent != null)
  94                                                 SummaryAddedEvent (new FileInfo (full_path));
  95                                 } else if (Path.GetExtension (full_path) == ".ev-summary") {
  96                                         // mbox summary
  97                                         string mbox_file = Path.ChangeExtension (full_path, null);
  98                                         Logger.Log.Info ("Reindexing updated mbox: {0}", mbox_file);
  99                                         if (MboxAddedEvent != null)
 100                                                 MboxAddedEvent (new FileInfo (mbox_file));
 101                                 }
 102                         }
 103                 }
 104
 105                 private void Watch (string root)
 106                 {
 107                         Queue pending = new Queue ();
 108
 109                         pending.Enqueue (root);
 110
 111                         while (pending.Count > 0) {
 112
 113                                 string dir = (string) pending.Dequeue ();
 114
 115                                 foreach (string subdir in DirectoryWalker.GetDirectories (dir)) {
 116                                         if (Shutdown.ShutdownRequested)
 117                                                 return;
 118
 119                                         if (Inotify.Enabled) {
 120                                                 Inotify.Subscribe (dir, OnInotifyEvent,
 121                                                                    Inotify.EventType.Create
 122                                                                    | Inotify.EventType.Delete
 123                                                                    | Inotify.EventType.MovedTo);
 124                                         }
 125
 126                                         pending.Enqueue (subdir);
 127                                 }
 128
 129                                 Stopwatch watch = new Stopwatch ();
 130                                 if (Debug)
 131                                         Logger.Log.Debug ("Starting watch on {0}", dir);
 132                                 watch.Start ();
 133                                 foreach (string path in DirectoryWalker.GetItems (dir, new DirectoryWalker.FileFilter (IsSummary))) {
 134                                         if (Shutdown.ShutdownRequested)
 135                                                 return;
 136
 137                                         FileInfo file = new FileInfo (path);
 138
 139                                         if (file.Name == "summary") {
 140                                                 if (SummaryAddedEvent != null && FileIsInteresting (file))
 141                                                         SummaryAddedEvent (file);
 142                                         } else if (file.Extension == ".ev-summary") {
 143                                                 string mbox_name = Path.Combine (file.DirectoryName,
 144                                                                                  Path.GetFileNameWithoutExtension (file.Name));
 145                                                 FileInfo mbox_file = new FileInfo (mbox_name);
 146                                                 if (MboxAddedEvent != null && FileIsInteresting (mbox_file))
 147                                                         MboxAddedEvent (mbox_file);
 148                                         }
 149                                 }
 150                                 watch.Stop ();
 151                                 if (Debug)
 152                                         Logger.Log.Debug ("Crawled {0} in {1}", dir, watch);
 153                         }
 154                 }
 155
 156                 private bool IsSummary (string path, string name)
 157                 {
 158                         if (name == "summary")
 159                                 return true;
 160
 161                         if (Path.GetExtension (name) == ".ev-summary")
 162                                 return true;
 163
 164                         return false;
 165                 }
 166
 167                 public void Crawl ()
 168                 {
 169                         foreach (string root in roots) {
 170                                 if (Shutdown.ShutdownRequested)
 171                                         return;
 172
 173                                 Watch (root);
 174                         }
 175                 }
 176         }
 177 }