beagled/EvolutionMailDriver/EvolutionMailIndexableGenerator.cs

   1
   2 //
   3 // EvolutionMailIndexableGenerator.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 using System.Runtime.Serialization.Formatters.Binary;
  32 using System.Threading;
  33 using System.Xml;
  34
  35 using Beagle.Util;
  36 using Beagle.Daemon;
  37
  38 using Camel = Beagle.Util.Camel;
  39
  40 namespace Beagle.Daemon.EvolutionMailDriver {
  41
  42         public abstract class EvolutionMailIndexableGenerator : IIndexableGenerator {
  43
  44                 protected static bool Debug = false;
  45
  46                 private static bool gmime_initialized = false;
  47
  48                 private static ArrayList excludes = new ArrayList ();
  49
  50                 static EvolutionMailIndexableGenerator () {
  51                         foreach (ExcludeItem exclude in Conf.Indexing.Excludes)
  52                                 if (exclude.Type == ExcludeType.MailFolder)
  53                                         excludes.Add (exclude);
  54                 }
  55
  56                 protected EvolutionMailQueryable queryable;
  57
  58                 protected string account_name, folder_name;
  59                 protected int count, indexed_count;
  60
  61                 protected EvolutionMailIndexableGenerator (EvolutionMailQueryable queryable)
  62                 {
  63                         this.queryable = queryable;
  64                 }
  65
  66                 protected abstract string GetFolderName (FileSystemInfo info);
  67                 protected abstract bool Setup ();
  68                 protected abstract FileInfo CrawlFile { get; }
  69                 public abstract string GetTarget ();
  70                 public abstract bool HasNextIndexable ();
  71                 public abstract Indexable GetNextIndexable ();
  72                 public abstract void Checkpoint ();
  73
  74                 public void PostFlushHook ()
  75                 {
  76                         Checkpoint ();
  77                 }
  78
  79                 protected static void InitializeGMime ()
  80                 {
  81                         if (!gmime_initialized) {
  82                                 GMime.Global.Init ();
  83                                 gmime_initialized = true;
  84                         }
  85                 }
  86
  87                 protected bool IsSpamFolder (string name)
  88                 {
  89                         if (name.ToLower () == "spam" || name.ToLower () == "junk")
  90                                 return true;
  91                         else
  92                                 return false;
  93                 }
  94
  95                 protected bool IgnoreFolder (string path)
  96                 {
  97                         // FIXME: Use System.IO.Path
  98                         foreach (ExcludeItem exclude in excludes) {
  99                                 if (exclude.IsMatch (path))
 100                                         return true;
 101                         }
 102                         return false;
 103                 }
 104
 105                 protected void CrawlFinished ()
 106                 {
 107                         // FIXME: This is a little sketchy
 108                         this.queryable.FileAttributesStore.AttachLastWriteTime (this.CrawlFile.FullName, DateTime.UtcNow);
 109
 110                         this.queryable.SetGeneratorProgress (this, 100);
 111                 }
 112
 113                 public string StatusName {
 114                         get { return this.CrawlFile.FullName; }
 115                 }
 116
 117                 public override bool Equals (object o)
 118                 {
 119                         EvolutionMailIndexableGenerator generator = o as EvolutionMailIndexableGenerator;
 120
 121                         if (generator == null)
 122                                 return false;
 123
 124                         if (Object.ReferenceEquals (this, generator))
 125                                 return true;
 126
 127                         if (this.CrawlFile.FullName == generator.CrawlFile.FullName)
 128                                 return true;
 129                         else
 130                                 return false;
 131                 }
 132
 133                 public override int GetHashCode ()
 134                 {
 135                         return this.CrawlFile.FullName.GetHashCode ();
 136                 }
 137         }
 138
 139         public class EvolutionMailIndexableGeneratorMbox : EvolutionMailIndexableGenerator {
 140                 private FileInfo mbox_info;
 141                 private int mbox_fd = -1;
 142                 private GMime.StreamFs mbox_stream;
 143                 private GMime.Parser mbox_parser;
 144                 private long file_size;
 145
 146                 public EvolutionMailIndexableGeneratorMbox (EvolutionMailQueryable queryable, FileInfo mbox_info) : base (queryable)
 147                 {
 148                         this.mbox_info = mbox_info;
 149                 }
 150
 151                 protected override string GetFolderName (FileSystemInfo info)
 152                 {
 153                         FileInfo file_info = (FileInfo) info;
 154                         DirectoryInfo di;
 155                         string folder_name = "";
 156
 157                         di = file_info.Directory;
 158                         while (di != null) {
 159                                 // Evo uses ".sbd" as the extension on a folder
 160                                 if (di.Extension == ".sbd")
 161                                         folder_name = Path.Combine (Path.GetFileNameWithoutExtension (di.Name), folder_name);
 162                                 else
 163                                         break;
 164
 165                                 di = di.Parent;
 166                         }
 167
 168                         return Path.Combine (folder_name, file_info.Name);
 169                 }
 170
 171                 protected override bool Setup ()
 172                 {
 173                         this.account_name = "local@local";
 174                         this.folder_name = this.GetFolderName (this.mbox_info);
 175
 176                         if (this.IsSpamFolder (this.folder_name))
 177                                 return false;
 178
 179                         if (this.IgnoreFolder (this.mbox_info.FullName))
 180                                 return false;
 181
 182                         return true;
 183                 }
 184
 185                 private long MboxLastOffset {
 186                         get {
 187                                 string offset_str = this.queryable.ReadDataLine ("offset-" + this.folder_name.Replace ('/', '-'));
 188                                 long offset = Convert.ToInt64 (offset_str);
 189
 190                                 return offset;
 191                         }
 192
 193                         set {
 194                                 this.queryable.WriteDataLine ("offset-" + this.folder_name.Replace ('/', '-'), value.ToString ());
 195                         }
 196                 }
 197
 198                 public override bool HasNextIndexable ()
 199                 {
 200                         if (this.account_name == null) {
 201                                 if (!Setup ()) {
 202                                         this.queryable.RemoveGeneratorProgress (this);
 203                                         return false;
 204                                 }
 205                         }
 206
 207                         if (this.mbox_fd < 0) {
 208                                 Logger.Log.Debug ("Opening mbox {0}", this.mbox_info.Name);
 209
 210                                 try {
 211                                         InitializeGMime ();
 212                                 } catch (Exception e) {
 213                                         Logger.Log.Warn ("Caught exception trying to initalize gmime:");
 214                                         Logger.Log.Warn (e);
 215                                         return false;
 216                                 }
 217
 218                                 this.mbox_fd = Mono.Unix.Native.Syscall.open (this.mbox_info.FullName, Mono.Unix.Native.OpenFlags.O_RDONLY);
 219                                 this.mbox_stream = new GMime.StreamFs (this.mbox_fd);
 220                                 this.mbox_stream.Seek ((int) this.MboxLastOffset);
 221                                 this.mbox_parser = new GMime.Parser (this.mbox_stream);
 222                                 this.mbox_parser.ScanFrom = true;
 223
 224                                 FileInfo info = new FileInfo (this.mbox_info.FullName);
 225                                 this.file_size = info.Length;
 226                         }
 227
 228                         if (this.mbox_parser.Eos ()) {
 229                                 long offset = this.mbox_parser.FromOffset;
 230
 231                                 this.mbox_stream.Close ();
 232
 233                                 this.mbox_fd = -1;
 234                                 this.mbox_stream.Dispose ();
 235                                 this.mbox_stream = null;
 236                                 this.mbox_parser.Dispose ();
 237                                 this.mbox_parser = null;
 238
 239                                 Logger.Log.Debug ("{0}: Finished indexing {1} messages", this.folder_name, this.indexed_count);
 240
 241                                 if (offset >= 0)
 242                                         this.MboxLastOffset = offset;
 243                                 this.CrawlFinished ();
 244
 245                                 return false;
 246                         } else
 247                                 return true;
 248                 }
 249
 250                 public override Indexable GetNextIndexable ()
 251                 {
 252                         using (GMime.Message message = this.mbox_parser.ConstructMessage ()) {
 253                                 // Work around what I think is a bug in GMime: If you
 254                                 // have a zero-byte file or seek to the end of a
 255                                 // file, parser.Eos () will return true until it
 256                                 // actually tries to read something off the wire.
 257                                 // Since parser.ConstructMessage() always returns a
 258                                 // message (which may also be a bug), we'll often get
 259                                 // one empty message which we need to deal with here.
 260                                 //
 261                                 // Check if its empty by seeing if the Headers
 262                                 // property is null or empty.
 263                                 if (message == null || message.Headers == null || message.Headers == "")
 264                                         return null;
 265
 266                                 ++this.count;
 267
 268                                 string x_evolution = message.GetHeader ("X-Evolution");
 269                                 if (x_evolution == null || x_evolution == "") {
 270                                         Logger.Log.Info ("{0}: Message at offset {1} has no X-Evolution header!",
 271                                                          this.folder_name, this.mbox_parser.FromOffset);
 272                                         return null;
 273                                 }
 274
 275                                 // This extracts the UID and flags from the X-Evolution header.
 276                                 // It may also contain user-defined flags and tags, but we don't
 277                                 // support those right now.
 278                                 int separator_idx = x_evolution.IndexOf ('-');
 279
 280                                 string uid_str = x_evolution.Substring (0, separator_idx);
 281                                 string uid = Convert.ToUInt32 (uid_str, 16).ToString (); // ugh.
 282                                 uint flags = Convert.ToUInt32 (x_evolution.Substring (separator_idx + 1, 4), 16);
 283
 284                                 Indexable indexable = this.GMimeMessageToIndexable (uid, message, flags);
 285
 286                                 if (Debug) {
 287                                         Logger.Log.Debug ("Constructed message {0} with uid {1}, flags {2}.  Indexable {3} null",
 288                                                           this.count, uid, flags, indexable == null ? "" : "not");
 289                                 }
 290
 291                                 if (indexable == null)
 292                                         return null;
 293
 294                                 ++this.indexed_count;
 295
 296                                 return indexable;
 297                         }
 298                 }
 299
 300                 private static bool CheckFlags (uint flags, Camel.CamelFlags test)
 301                 {
 302                         return (flags & (uint) test) == (uint) test;
 303                 }
 304
 305                 private Indexable GMimeMessageToIndexable (string uid, GMime.Message message, uint flags)
 306                 {
 307                         // Don't index messages flagged as junk
 308                         if (CheckFlags (flags, Camel.CamelFlags.Junk))
 309                                 return null;
 310
 311                         System.Uri uri = EvolutionMailQueryable.EmailUri (this.account_name, this.folder_name, uid);
 312                         Indexable indexable = new Indexable (uri);
 313
 314                         indexable.Timestamp = message.Date.ToUniversalTime ();
 315                         indexable.HitType = "MailMessage";
 316                         indexable.MimeType = "message/rfc822";
 317                         indexable.CacheContent = false;
 318
 319                         indexable.AddProperty (Property.NewUnsearched ("fixme:client", "evolution"));
 320                         indexable.AddProperty (Property.NewUnsearched ("fixme:account", "Local"));
 321                         indexable.AddProperty (Property.NewUnsearched ("fixme:folder", this.folder_name));
 322
 323                         GMime.InternetAddressList addrs;
 324
 325                         addrs = message.GetRecipients (GMime.Message.RecipientType.To);
 326                         foreach (GMime.InternetAddress ia in addrs) {
 327                                 if (this.folder_name == "Sent" && ia.AddressType != GMime.InternetAddressType.Group)
 328                                         indexable.AddProperty (Property.NewUnsearched ("fixme:sentTo", ia.Addr));
 329                         }
 330                         addrs.Dispose ();
 331
 332                         addrs = message.GetRecipients (GMime.Message.RecipientType.Cc);
 333                         foreach (GMime.InternetAddress ia in addrs) {
 334                                 if (this.folder_name == "Sent"  && ia.AddressType != GMime.InternetAddressType.Group)
 335                                         indexable.AddProperty (Property.NewUnsearched ("fixme:sentTo", ia.Addr));
 336                         }
 337                         addrs.Dispose ();
 338
 339                         addrs = GMime.InternetAddressList.ParseString (GMime.Utils.HeaderDecodePhrase (message.Sender));
 340                         foreach (GMime.InternetAddress ia in addrs) {
 341                                 if (this.folder_name != "Sent"  && ia.AddressType != GMime.InternetAddressType.Group)
 342                                         indexable.AddProperty (Property.NewUnsearched ("fixme:gotFrom", ia.Addr));
 343                         }
 344                         addrs.Dispose ();
 345
 346                         if (this.folder_name == "Sent")
 347                                 indexable.AddProperty (Property.NewFlag ("fixme:isSent"));
 348
 349                         Property flag_prop = Property.NewUnsearched ("fixme:flags", flags);
 350                         flag_prop.IsMutable = true;
 351                         indexable.AddProperty (flag_prop);
 352
 353                         if (CheckFlags (flags, Camel.CamelFlags.Answered))
 354                                 indexable.AddProperty (Property.NewFlag ("fixme:isAnswered"));
 355
 356                         if (CheckFlags (flags, Camel.CamelFlags.Deleted))
 357                                 indexable.AddProperty (Property.NewFlag ("fixme:isDeleted"));
 358
 359                         if (CheckFlags (flags, Camel.CamelFlags.Draft))
 360                                 indexable.AddProperty (Property.NewFlag ("fixme:isDraft"));
 361
 362                         if (CheckFlags (flags, Camel.CamelFlags.Flagged))
 363                                 indexable.AddProperty (Property.NewFlag ("fixme:isFlagged"));
 364
 365                         if (CheckFlags (flags, Camel.CamelFlags.Seen))
 366                                 indexable.AddProperty (Property.NewFlag ("fixme:isSeen"));
 367
 368                         if (CheckFlags (flags, Camel.CamelFlags.AnsweredAll))
 369                                 indexable.AddProperty (Property.NewFlag ("fixme:isAnsweredAll"));
 370
 371                         indexable.SetBinaryStream (message.Stream);
 372
 373                         return indexable;
 374                 }
 375
 376                 public override void Checkpoint ()
 377                 {
 378                         long offset = -1;
 379
 380                         if (this.mbox_parser != null) {
 381                                 offset = this.mbox_parser.FromOffset;
 382
 383                                 if (offset >= 0)
 384                                         this.MboxLastOffset = offset;
 385                         }
 386
 387                         string progress = "";
 388                         if (this.file_size > 0 && offset > 0) {
 389                                 progress = String.Format (" ({0}/{1} bytes {2:###.0}%)",
 390                                                           offset, this.file_size, 100.0 * offset / this.file_size);
 391
 392                                 this.queryable.SetGeneratorProgress (this, (int) (100.0 * offset / this.file_size));
 393                         }
 394
 395                         Logger.Log.Debug ("{0}: indexed {1} messages{2}",
 396                                           this.folder_name, this.indexed_count, progress);
 397                 }
 398
 399                 public override string GetTarget ()
 400                 {
 401                         return "mbox-file:" + mbox_info.FullName;
 402                 }
 403
 404                 protected override FileInfo CrawlFile {
 405                         get { return this.mbox_info; }
 406                 }
 407         }
 408
 409         public class EvolutionMailIndexableGeneratorImap : EvolutionMailIndexableGenerator {
 410                 private enum ImapBackendType {
 411                         Imap,
 412                         Imap4
 413                 };
 414
 415                 private FileInfo summary_info;
 416                 private string imap_name;
 417                 private ImapBackendType backend_type;
 418                 private Camel.Summary summary;
 419                 private IEnumerator summary_enumerator;
 420                 private ICollection accounts;
 421                 private string folder_cache_name;
 422                 private Hashtable mapping;
 423                 private ArrayList deleted_list;
 424
 425                 public EvolutionMailIndexableGeneratorImap (EvolutionMailQueryable queryable, FileInfo summary_info) : base (queryable)
 426                 {
 427                         this.summary_info = summary_info;
 428                 }
 429
 430                 protected override string GetFolderName (FileSystemInfo info)
 431                 {
 432                         DirectoryInfo dir_info = (DirectoryInfo) info;
 433                         string folder_name = "";
 434
 435                         while (dir_info != null) {
 436                                 folder_name = Path.Combine (dir_info.Name, folder_name);
 437
 438                                 dir_info = dir_info.Parent;
 439
 440                                 if (dir_info.Name != "subfolders")
 441                                         break;
 442                                 else
 443                                         dir_info = dir_info.Parent;
 444                         }
 445
 446                         return folder_name;
 447                 }
 448
 449                 protected override bool Setup ()
 450                 {
 451                         string dir_name = summary_info.DirectoryName;
 452                         int imap_start_idx;
 453
 454                         int idx = dir_name.IndexOf (".evolution/mail/imap4/");
 455
 456                         if (idx >= 0) {
 457                                 this.backend_type = ImapBackendType.Imap4;
 458                                 imap_start_idx = idx + 22;
 459                         } else {
 460                                 this.backend_type = ImapBackendType.Imap;
 461                                 imap_start_idx = dir_name.IndexOf (".evolution/mail/imap/") + 21;
 462                         }
 463
 464                         string imap_start = dir_name.Substring (imap_start_idx);
 465                         this.imap_name = imap_start.Substring (0, imap_start.IndexOf ('/'));
 466
 467                         try {
 468                                 this.accounts = (ICollection) GConfThreadHelper.Get ("/apps/evolution/mail/accounts");
 469                         } catch (Exception ex) {
 470                                 Logger.Log.Warn ("Caught exception in Setup(): " + ex.Message);
 471                                 Logger.Log.Warn ("There are no configured evolution accounts, ignoring {0}", this.imap_name);
 472                                 return false;
 473                         }
 474
 475                         // This should only happen if we shut down while waiting for the GConf results to come back.
 476                         if (this.accounts == null)
 477                                 return false;
 478
 479                         foreach (string xml in this.accounts) {
 480                                 XmlDocument xmlDoc = new XmlDocument ();
 481
 482                                 xmlDoc.LoadXml (xml);
 483
 484                                 XmlNode account = xmlDoc.SelectSingleNode ("//account");
 485
 486                                 if (account == null)
 487                                         continue;
 488
 489                                 string uid = null;
 490
 491                                 foreach (XmlAttribute attr in account.Attributes) {
 492                                         if (attr.Name == "uid") {
 493                                                 uid = attr.InnerText;
 494                                                 break;
 495                                         }
 496                                 }
 497
 498                                 if (uid == null)
 499                                         continue;
 500
 501                                 XmlNode imap_url_node = xmlDoc.SelectSingleNode ("//source/url");
 502
 503                                 if (imap_url_node == null)
 504                                         continue;
 505
 506                                 string imap_url = imap_url_node.InnerText;
 507                                 // If there is a semicolon in the username part of the URL, it
 508                                 // indicates that there's an auth scheme there.  We don't care
 509                                 // about that, so remove it.
 510                                 int user_end = imap_url.IndexOf ('@');
 511                                 int semicolon = imap_url.IndexOf (';', 0, user_end + 1);
 512
 513                                 if (semicolon != -1)
 514                                         imap_url = imap_url.Substring (0, semicolon) + imap_url.Substring (user_end);
 515
 516                                 // Escape backslashes, which frequently appear when using IMAP against Exchange servers
 517                                 this.imap_name = this.imap_name.Replace ("\\", "%5c");
 518
 519                                 // Escape out additional @s in the name.  I hate the class libs so much.
 520                                 int lastIdx = this.imap_name.LastIndexOf ('@');
 521                                 if (this.imap_name.IndexOf ('@') != lastIdx) {
 522                                         string toEscape = this.imap_name.Substring (0, lastIdx);
 523                                         this.imap_name = toEscape.Replace ("@", "%40") + this.imap_name.Substring (lastIdx);
 524                                 }
 525
 526                                 string backend_url_prefix;
 527                                 if (this.backend_type == ImapBackendType.Imap)
 528                                         backend_url_prefix = "imap";
 529                                 else
 530                                         backend_url_prefix = "imap4";
 531
 532                                 if (imap_url.StartsWith (backend_url_prefix + "://" + this.imap_name + "/")) {
 533                                         this.account_name = uid;
 534                                         break;
 535                                 }
 536                         }
 537
 538                         if (this.account_name == null) {
 539                                 Logger.Log.Info ("Unable to determine account name for {0}", this.imap_name);
 540                                 return false;
 541                         }
 542
 543                         // Need to check the directory on disk to see if it's a junk/spam folder,
 544                         // since the folder name will be "foo/spam" and not match the check below.
 545                         DirectoryInfo dir_info = new DirectoryInfo (dir_name);
 546                         if (this.IsSpamFolder (dir_info.Name))
 547                                 return false;
 548
 549                         // Check if the folder is listed in the configuration as to be excluded from indexing
 550                         if (this.IgnoreFolder (dir_info.FullName))
 551                                 return false;
 552
 553                         this.folder_name = GetFolderName (new DirectoryInfo (dir_name));
 554
 555                         return true;
 556                 }
 557
 558                 private string FolderCacheName {
 559                         get {
 560                                 if (this.account_name == null || this.folder_name == null)
 561                                         return null;
 562
 563                                 if (this.folder_cache_name == null)
 564                                         this.folder_cache_name = "status-" + this.account_name + "-" + this.folder_name.Replace ('/', '-');
 565
 566                                 return this.folder_cache_name;
 567                         }
 568                 }
 569
 570                 private bool LoadCache ()
 571                 {
 572                         Stream cacheStream;
 573                         BinaryFormatter formatter;
 574
 575                         if (this.FolderCacheName == null) {
 576                                 this.mapping = new Hashtable ();
 577                                 return false;
 578                         }
 579
 580                         try {
 581                                 cacheStream = this.queryable.ReadDataStream (this.FolderCacheName);
 582                                 formatter = new BinaryFormatter ();
 583                                 this.mapping = formatter.Deserialize (cacheStream) as Hashtable;
 584                                 cacheStream.Close ();
 585                                 Logger.Log.Debug ("Successfully loaded previous crawled data from disk: {0}", this.FolderCacheName);
 586
 587                                 return true;
 588                         } catch {
 589                                 this.mapping = new Hashtable ();
 590
 591                                 return false;
 592                         }
 593                 }
 594
 595                 private void SaveCache ()
 596                 {
 597                         Stream cacheStream;
 598                         BinaryFormatter formatter;
 599
 600                         if (this.FolderCacheName == null)
 601                                 return;
 602
 603                         cacheStream = this.queryable.WriteDataStream (this.FolderCacheName);
 604                         formatter = new BinaryFormatter ();
 605                         formatter.Serialize (cacheStream, mapping);
 606                         cacheStream.Close ();
 607                 }
 608
 609                 public override bool HasNextIndexable ()
 610                 {
 611                         if (this.account_name == null) {
 612                                 if (!Setup ()) {
 613                                         this.queryable.RemoveGeneratorProgress (this);
 614                                         return false;
 615                                 }
 616                         }
 617
 618                         if (this.mapping == null) {
 619                                 bool cache_loaded = this.LoadCache ();
 620
 621                                 this.deleted_list = new ArrayList (this.mapping.Keys);
 622                                 this.deleted_list.Sort ();
 623                                 Logger.Log.Debug ("Deleted list starting at {0} for {1}", this.deleted_list.Count, this.folder_name);
 624
 625                                 // Check to see if we even need to bother walking the summary
 626                                 if (cache_loaded && this.queryable.FileAttributesStore.IsUpToDate (this.CrawlFile.FullName)) {
 627                                         Logger.Log.Debug ("{0}: summary has not been updated; crawl unncessary", this.folder_name);
 628                                         this.queryable.RemoveGeneratorProgress (this);
 629                                         return false;
 630                                 }
 631                         }
 632
 633                         if (this.summary == null) {
 634                                 try {
 635                                         if (this.backend_type == ImapBackendType.Imap)
 636                                                 this.summary = Camel.Summary.LoadImapSummary (this.summary_info.FullName);
 637                                         else
 638                                                 this.summary = Camel.Summary.LoadImap4Summary (this.summary_info.FullName);
 639                                 } catch (Exception e) {
 640                                         Logger.Log.Warn ("Unable to index {0}: {1}", this.folder_name,
 641                                                                          e.Message);
 642                                         this.queryable.RemoveGeneratorProgress (this);
 643                                         return false;
 644                                 }
 645                         }
 646
 647                         if (this.summary_enumerator == null)
 648                                 this.summary_enumerator = this.summary.GetEnumerator ();
 649
 650                         if (this.summary_enumerator.MoveNext ())
 651                                 return true;
 652
 653                         Logger.Log.Debug ("Queuing up {0} removals for deleted messages in ", this.deleted_list.Count, this.folder_name);
 654                         foreach (string uid in this.deleted_list) {
 655                                 Uri uri = EvolutionMailQueryable.EmailUri (this.account_name, this.folder_name, uid);
 656
 657                                 // FIXME: This is kind of a hack, but it's the only way
 658                                 // with the IndexableGenerator to handle our removals.
 659                                 Scheduler.Task task = this.queryable.NewRemoveTask (uri);
 660                                 task.Priority = Scheduler.Priority.Immediate;
 661                                 this.queryable.ThisScheduler.Add (task);
 662                         }
 663
 664                         string progress = "";
 665                         if (this.count > 0 && this.summary.header.count > 0) {
 666                                 progress = String.Format (" ({0}/{1} {2:###.0}%)",
 667                                                           this.count,
 668                                                           this.summary.header.count,
 669                                                           100.0 * this.count / this.summary.header.count);
 670                         }
 671
 672                         Logger.Log.Debug ("{0}: Finished indexing {1} messages{2}", this.folder_name, this.indexed_count, progress);
 673
 674                         this.SaveCache ();
 675                         this.CrawlFinished ();
 676
 677                         return false;
 678                 }
 679
 680                 // Kind of nasty, but we need the function.
 681                 [System.Runtime.InteropServices.DllImport("libglib-2.0.so.0")]
 682                 static extern int g_str_hash (string str);
 683
 684                 // Stolen from deep within e-d-s's camel-data-cache.c  Very evil.
 685                 private const int CAMEL_DATA_CACHE_MASK = ((1 << 6) - 1);
 686
 687                 public override Indexable GetNextIndexable ()
 688                 {
 689                         Indexable indexable = null;
 690
 691                         Camel.MessageInfo mi = (Camel.MessageInfo) this.summary_enumerator.Current;
 692
 693                         ++this.count;
 694
 695                         if (Debug) {
 696                                 Logger.Log.Debug ("Constructed message {0} with uid {1}, flags {2}.",
 697                                                   this.count, mi.uid, mi.flags);
 698                         }
 699
 700                         // Try to load the cached message data off disk
 701                         object flags = this.mapping[mi.uid];
 702
 703                         if (flags == null) {
 704                                 // New, previously unseen message
 705                                 string msg_file;
 706
 707                                 if (this.backend_type == ImapBackendType.Imap)
 708                                         msg_file = Path.Combine (summary_info.DirectoryName, mi.uid + ".");
 709                                 else {
 710                                         // This is taken from e-d-s's camel-data-cache.c.  No doubt
 711                                         // NotZed would scream bloody murder if he saw this here.
 712                                         int hash = (g_str_hash (mi.uid) >> 5) & CAMEL_DATA_CACHE_MASK;
 713                                         string cache_path = String.Format ("cache/{0:x}/{1}", hash, mi.uid);
 714                                         msg_file = Path.Combine (summary_info.DirectoryName, cache_path);
 715                                 }
 716
 717                                 indexable = this.CamelMessageToIndexable (mi, msg_file);
 718
 719                                 if (Debug)
 720                                         Logger.Log.Debug ("Unseen message, indexable {0} null", indexable == null ? "" : "not");
 721
 722                                 this.mapping[mi.uid] = mi.flags;
 723
 724                                 if (indexable != null)
 725                                         ++this.indexed_count;
 726                         } else if ((uint) flags != mi.flags) {
 727                                 // Previously seen message, but flags have changed.
 728                                 Uri uri = CamelMessageUri (mi);
 729                                 indexable = new Indexable (uri);
 730                                 indexable.Type = IndexableType.PropertyChange;
 731
 732                                 Property flag_prop = Property.NewUnsearched ("fixme:flags", mi.flags);
 733                                 flag_prop.IsMutable = true;
 734                                 indexable.AddProperty (flag_prop);
 735
 736                                 if (Debug)
 737                                         Logger.Log.Debug ("Previously seen message, flags changed: {0} -> {1}", flags, mi.flags);
 738
 739                                 ++this.indexed_count;
 740                         } else {
 741                                 if (Debug)
 742                                         Logger.Log.Debug ("Previously seen message, unchanged.");
 743                         }
 744
 745                         if (flags != null)
 746                                 this.deleted_list.Remove (mi.uid);
 747
 748                         return indexable;
 749                 }
 750
 751                 private Uri CamelMessageUri (Camel.MessageInfo message_info)
 752                 {
 753                         return EvolutionMailQueryable.EmailUri (this.account_name, this.folder_name, message_info.uid);
 754                 }
 755
 756                 private Indexable CamelMessageToIndexable (Camel.MessageInfo messageInfo, string msg_file)
 757                 {
 758                         // Don't index messages flagged as junk
 759                         if (messageInfo.IsJunk)
 760                                 return null;
 761
 762                         // Many properties will be set by the filter when
 763                         // processing the cached data, if it's there.  So
 764                         // don't set a number of properties in that case.
 765                         bool have_content = File.Exists (msg_file);
 766
 767                         Uri uri = CamelMessageUri (messageInfo);
 768                         Indexable indexable = new Indexable (uri);
 769
 770                         indexable.Timestamp = messageInfo.SentDate;
 771                         indexable.MimeType = "message/rfc822";
 772                         indexable.HitType = "MailMessage";
 773
 774                         indexable.AddProperty (Property.NewUnsearched ("fixme:account",  this.imap_name));
 775                         indexable.AddProperty (Property.NewUnsearched ("fixme:folder",   this.folder_name));
 776                         indexable.AddProperty (Property.NewUnsearched ("fixme:client", "evolution"));
 777
 778                         if (!have_content) {
 779                                 indexable.AddProperty (Property.New ("dc:title", GMime.Utils.HeaderDecodePhrase (messageInfo.subject)));
 780                                 indexable.AddProperty (Property.NewDate ("fixme:date", messageInfo.SentDate));
 781                         }
 782
 783                         GMime.InternetAddressList addrs;
 784                         addrs = GMime.InternetAddressList.ParseString (messageInfo.to);
 785                         foreach (GMime.InternetAddress ia in addrs) {
 786                                 if (!have_content) {
 787                                         indexable.AddProperty (Property.NewUnsearched ("fixme:to", ia.ToString (false)));
 788                                         if (ia.AddressType != GMime.InternetAddressType.Group) {
 789                                                 indexable.AddProperty (Property.New ("fixme:to_address", ia.Addr));
 790                                                 indexable.AddProperty (Property.New ("fixme:to_sanitized", StringFu.SanitizeEmail (ia.Addr)));
 791                                         }
 792                                         indexable.AddProperty (Property.New ("fixme:to_name", ia.Name));
 793                                 }
 794
 795                                 if (this.folder_name == "Sent" && ia.AddressType != GMime.InternetAddressType.Group)
 796                                         indexable.AddProperty (Property.NewUnsearched ("fixme:sentTo", ia.Addr));
 797                         }
 798                         addrs.Dispose ();
 799
 800                         addrs = GMime.InternetAddressList.ParseString (messageInfo.cc);
 801                         foreach (GMime.InternetAddress ia in addrs) {
 802                                 if (!have_content) {
 803                                         indexable.AddProperty (Property.NewUnsearched ("fixme:cc", ia.ToString (false)));
 804                                         if (ia.AddressType != GMime.InternetAddressType.Group) {
 805                                                 indexable.AddProperty (Property.New ("fixme:cc_address", ia.Addr));
 806                                                 indexable.AddProperty (Property.New ("fixme:cc_sanitized", StringFu.SanitizeEmail (ia.Addr)));
 807                                         }
 808                                         indexable.AddProperty (Property.New ("fixme:cc_name", ia.Name));
 809                                 }
 810
 811                                 if (this.folder_name == "Sent" && ia.AddressType != GMime.InternetAddressType.Group)
 812                                         indexable.AddProperty (Property.NewUnsearched ("fixme:sentTo", ia.Addr));
 813                         }
 814                         addrs.Dispose ();
 815
 816                         addrs = GMime.InternetAddressList.ParseString (messageInfo.from);
 817                         foreach (GMime.InternetAddress ia in addrs) {
 818                                 if (!have_content) {
 819                                         indexable.AddProperty (Property.NewUnsearched ("fixme:from", ia.ToString (false)));
 820                                         if (ia.AddressType != GMime.InternetAddressType.Group) {
 821                                                 indexable.AddProperty (Property.New ("fixme:from_address", ia.Addr));
 822                                                 indexable.AddProperty (Property.New ("fixme:from_sanitized", StringFu.SanitizeEmail (ia.Addr)));
 823                                         }
 824                                         indexable.AddProperty (Property.New ("fixme:from_name", ia.Name));
 825                                 }
 826
 827                                 if (this.folder_name != "Sent" && ia.AddressType != GMime.InternetAddressType.Group)
 828                                         indexable.AddProperty (Property.NewUnsearched ("fixme:gotFrom", ia.Addr));
 829                         }
 830                         addrs.Dispose ();
 831
 832                         indexable.AddProperty (Property.NewKeyword ("fixme:mlist", messageInfo.mlist));
 833
 834                         Property flag_prop = Property.NewUnsearched ("fixme:flags", messageInfo.flags);
 835                         flag_prop.IsMutable = true;
 836                         indexable.AddProperty (flag_prop);
 837
 838                         if (this.folder_name == "Sent")
 839                                 indexable.AddProperty (Property.NewFlag ("fixme:isSent"));
 840
 841                         if (messageInfo.IsAnswered)
 842                                 indexable.AddProperty (Property.NewFlag ("fixme:isAnswered"));
 843
 844                         if (messageInfo.IsDeleted)
 845                                 indexable.AddProperty (Property.NewFlag ("fixme:isDeleted"));
 846
 847                         if (messageInfo.IsDraft)
 848                                 indexable.AddProperty (Property.NewFlag ("fixme:isDraft"));
 849
 850                         if (messageInfo.IsFlagged)
 851                                 indexable.AddProperty (Property.NewFlag ("fixme:isFlagged"));
 852
 853                         if (messageInfo.IsSeen)
 854                                 indexable.AddProperty (Property.NewFlag ("fixme:isSeen"));
 855
 856                         if (messageInfo.HasAttachments && !have_content)
 857                                 indexable.AddProperty (Property.NewFlag ("fixme:hasAttachments"));
 858
 859                         if (messageInfo.IsAnsweredAll)
 860                                 indexable.AddProperty (Property.NewFlag ("fixme:isAnsweredAll"));
 861
 862                         if (have_content)
 863                                 indexable.ContentUri = UriFu.PathToFileUri (msg_file);
 864                         else
 865                                 indexable.NoContent = true;
 866
 867                         return indexable;
 868                 }
 869
 870                 public override void Checkpoint ()
 871                 {
 872                         if (this.summary != null) {
 873
 874                                 string progress = "";
 875                                 if (this.count > 0 && this.summary.header.count > 0) {
 876                                         progress = String.Format (" ({0}/{1} {2:###.0}%)",
 877                                                                   this.count,
 878                                                                   this.summary.header.count,
 879                                                                   100.0 * this.count / this.summary.header.count);
 880
 881                                         this.queryable.SetGeneratorProgress (this, (int) (100.0 * this.count / this.summary.header.count));
 882
 883                                 }
 884
 885                                 Logger.Log.Debug ("{0}: indexed {1} messages{2}",
 886                                                   this.folder_name, this.indexed_count, progress);
 887                         }
 888
 889                         this.SaveCache ();
 890                 }
 891
 892                 public override string GetTarget ()
 893                 {
 894                         return "summary-file:" + summary_info.FullName;
 895                 }
 896
 897                 protected override FileInfo CrawlFile {
 898                         get { return this.summary_info; }
 899                 }
 900         }
 901 }