251 lines
8.0 KiB
JavaScript
251 lines
8.0 KiB
JavaScript
# -*- Mode: Java; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
# ***** BEGIN LICENSE BLOCK *****
|
|
# Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
#
|
|
# The contents of this file are subject to the Mozilla Public License Version
|
|
# 1.1 (the "License"); you may not use this file except in compliance with
|
|
# the License. You may obtain a copy of the License at
|
|
# http://www.mozilla.org/MPL/
|
|
#
|
|
# Software distributed under the License is distributed on an "AS IS" basis,
|
|
# WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
# for the specific language governing rights and limitations under the
|
|
# License.
|
|
#
|
|
# The Original Code is spotlight integration code.
|
|
#
|
|
# The Initial Developer of the Original Code is
|
|
# David Bienvenu <bienvenu@mozilla.com>
|
|
# Portions created by the Initial Developer are Copyright (C) 2007
|
|
# the Initial Developer. All Rights Reserved.
|
|
#
|
|
# Contributor(s):
|
|
# Siddharth Agarwal <sid1337@gmail.com>
|
|
#
|
|
# Alternatively, the contents of this file may be used under the terms of
|
|
# either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
# the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
# in which case the provisions of the GPL or the LGPL are applicable instead
|
|
# of those above. If you wish to allow use of your version of this file only
|
|
# under the terms of either the GPL or the LGPL, and not to allow others to
|
|
# use your version of this file under the terms of the MPL, indicate your
|
|
# decision by deleting the provisions above and replace them with the notice
|
|
# and other provisions required by the GPL or the LGPL. If you do not delete
|
|
# the provisions above, a recipient may use your version of this file under
|
|
# the terms of any one of the MPL, the GPL or the LGPL.
|
|
#
|
|
# ***** END LICENSE BLOCK *****
|
|
|
|
#include content/searchCommon.js
|
|
|
|
const MSG_DB_LARGE_COMMIT = 1;
|
|
const MSG_FLAG_ATTACHMENT = 0x10000000;
|
|
const CRLF="\r\n";
|
|
|
|
// The property of the header that's used to check if a message is indexed
|
|
const gHdrIndexedProperty = "wds_indexed";
|
|
|
|
// The file extension that is used for support files of this component
|
|
const gFileExt = ".wdseml";
|
|
|
|
// The pref base
|
|
const gPrefBase = "mail.winsearch";
|
|
|
|
Components.utils.import("resource://gre/modules/XPCOMUtils.jsm");
|
|
|
|
function InitWinSearchIntegration()
|
|
{
|
|
var enabled;
|
|
try {
|
|
enabled = gPrefBranch.getBoolPref(gPrefBase + ".enable");
|
|
gLastFolderIndexedUri = gPrefBranch.getCharPref(gPrefBase + ".lastFolderIndexedUri");
|
|
} catch (ex) {}
|
|
|
|
if (!enabled)
|
|
return;
|
|
|
|
SIDump("Initializing Windows Search integration\n");
|
|
InitSupportIntegration();
|
|
}
|
|
|
|
// The stream listener to read messages
|
|
var gStreamListener = {
|
|
_buffer: "",
|
|
outputFile: null,
|
|
outputStream: null,
|
|
unicodeConverter: null,
|
|
// subject: null,
|
|
message: null,
|
|
msgHdr: null,
|
|
mimeHdrObj: null,
|
|
mimeHdrParamObj: null,
|
|
|
|
onDoneStreamingCurMessage: function(successful)
|
|
{
|
|
if (this.outputStream)
|
|
this.outputStream.close();
|
|
if (!successful && this.msgHdr)
|
|
{
|
|
var file = GetSupportFileForMsgHdr(this.msgHdr);
|
|
if (file && file.exists())
|
|
file.remove(false);
|
|
}
|
|
// should we try to delete the file on disk in case not successful?
|
|
gMsgHdrsToIndex.shift();
|
|
|
|
if (gMsgHdrsToIndex.length > 0)
|
|
GenerateSupportFile(gMsgHdrsToIndex[0]);
|
|
},
|
|
|
|
QueryInterface: function(aIId, instance) {
|
|
if (aIId.equals(Ci.nsIStreamListener) || aIId.equals(Ci.nsISupports))
|
|
return this;
|
|
|
|
Components.returnCode = Components.results.NS_ERROR_NO_INTERFACE;
|
|
return null;
|
|
},
|
|
|
|
onStartRequest: function(request, context) {
|
|
try
|
|
{
|
|
var outputFileStream = Cc["@mozilla.org/network/file-output-stream;1"]
|
|
.createInstance(Ci.nsIFileOutputStream);
|
|
outputFileStream.init(this.outputFile, -1, -1, 0);
|
|
this.outputStream = Cc["@mozilla.org/intl/converter-output-stream;1"]
|
|
.createInstance(Ci.nsIConverterOutputStream);
|
|
this.outputStream.init(outputFileStream, "UTF-8", 0, 0x0000);
|
|
}
|
|
catch (ex)
|
|
{
|
|
onDoneStreamingCurMessage(false);
|
|
}
|
|
},
|
|
|
|
onStopRequest: function(request, context, status, errorMsg) {
|
|
try
|
|
{
|
|
// If, for some reason, the first four characters are "null", then remove them
|
|
if (this.message.substring(0, 4) == "null")
|
|
this.message = this.message.substring(4);
|
|
|
|
// First get all the headers (everything till a double CRLF)
|
|
var headerLength = this.message.indexOf(CRLF+CRLF)+1;
|
|
|
|
// XXX the below has to be replaced with getMsgTextFromStream
|
|
|
|
// Filter out attachments. First locate the content-type header.
|
|
if (!this.mimeHdrObj)
|
|
this.mimeHdrObj = Cc["@mozilla.org/messenger/mimeheaders;1"].createInstance(Ci.nsIMimeHeaders);
|
|
this.mimeHdrObj.initialize(this.message.substring(0, headerLength), headerLength);
|
|
var contentTypeHeader = this.mimeHdrObj.extractHeader("Content-Type", false);
|
|
|
|
if (!this.mimeHdrParamObj)
|
|
this.mimeHdrParamObj = Cc["@mozilla.org/network/mime-hdrparam;1"].createInstance(Ci.nsIMIMEHeaderParam);
|
|
// If a multipart header, then strip out attachments
|
|
var contentType = this.mimeHdrParamObj.getParameter(contentTypeHeader, null, null, true, {});
|
|
if (contentType.substring(0,10) == "multipart/")
|
|
{
|
|
var boundary = this.mimeHdrParamObj.getParameter(contentTypeHeader, "boundary", null, true, {});
|
|
|
|
// Search for the boundary after the headers. Remove everything from the second occurrence.
|
|
boundary = CRLF + "--" + boundary;
|
|
var getRidFrom = this.message.indexOf(boundary, headerLength);
|
|
getRidFrom = this.message.indexOf(boundary, getRidFrom + boundary.length);
|
|
|
|
// In case we've got more than one boundary so far
|
|
if (getRidFrom != -1) {
|
|
this.message = this.message.substring(0, getRidFrom);
|
|
|
|
// Add RFC required stuff to the end
|
|
this.message += boundary + "--" + CRLF;
|
|
}
|
|
}
|
|
|
|
// We just write whatever we have in |message| out to the file
|
|
this.outputStream.writeString(this.message);
|
|
|
|
this.msgHdr.setUint32Property(gHdrIndexedProperty, 1);
|
|
var msgDB = this.msgHdr.folder.getMsgDatabase(null);
|
|
msgDB.Commit(MSG_DB_LARGE_COMMIT);
|
|
|
|
this.message = "";
|
|
SIDump("Successfully written file\n");
|
|
}
|
|
catch (ex)
|
|
{
|
|
SIDump(ex);
|
|
this.onDoneStreamingCurMessage(false);
|
|
return;
|
|
}
|
|
this.onDoneStreamingCurMessage(true);
|
|
},
|
|
|
|
onDataAvailable: function(request, context, inputStream, offset, count) {
|
|
try
|
|
{
|
|
var inStream = Cc["@mozilla.org/scriptableinputstream;1"].createInstance(Ci.nsIScriptableInputStream);
|
|
inStream.init(inputStream);
|
|
|
|
// It is necessary to read in data from the input stream
|
|
var inData = inStream.read(count);
|
|
|
|
// If we've already reached the attachments, safely ignore.
|
|
if (this.filteredAttachments)
|
|
return 0;
|
|
|
|
// Also ignore stuff after the first 20K or so
|
|
if (this.message && this.message.length > 20000)
|
|
return 0;
|
|
var inStream = Cc["@mozilla.org/scriptableinputstream;1"].
|
|
createInstance(Ci.nsIScriptableInputStream);
|
|
|
|
inStream.init(inputStream);
|
|
|
|
this.message += inData;
|
|
return 0;
|
|
}
|
|
catch (ex)
|
|
{
|
|
SIDump(ex);
|
|
onDoneStreamingCurMessage(false);
|
|
}
|
|
}
|
|
};
|
|
|
|
/* XPCOM boilerplate code */
|
|
function WinSearchIntegration() { }
|
|
WinSearchIntegration.prototype = {
|
|
classDescription: "Windows Search Integration",
|
|
classID: Components.ID("{451a70f0-1b4f-11dd-bd0b-0800200c9a66}"),
|
|
contractID: "@mozilla.org/windows-search-integration;1",
|
|
|
|
_xpcom_categories: [{
|
|
category: "app-startup",
|
|
service: true
|
|
}],
|
|
|
|
QueryInterface: XPCOMUtils.generateQI([Ci.nsIObserver, Ci.nsISupports]),
|
|
|
|
observe : function(aSubject, aTopic, aData)
|
|
{
|
|
switch(aTopic)
|
|
{
|
|
case "app-startup":
|
|
var obsSvc = Cc["@mozilla.org/observer-service;1"].getService(Ci.nsIObserverService);
|
|
obsSvc.addObserver(this, "profile-after-change", false);
|
|
break;
|
|
case "profile-after-change":
|
|
try { InitWinSearchIntegration(); }
|
|
catch(err) { SIDump("Could not initialize winsearch component"); }
|
|
break;
|
|
default:
|
|
throw Components.Exception("Unknown topic: " + aTopic);
|
|
}
|
|
}
|
|
};
|
|
var components = [WinSearchIntegration];
|
|
function NSGetModule(aCompMgr, aFileSpec)
|
|
{
|
|
return XPCOMUtils.generateModule(components);
|
|
}
|