gecko-dev/lib/libmsg/prsembst.h

322 строки
10 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
*
* The contents of this file are subject to the Netscape 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/NPL/
*
* 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 mozilla.org code.
*
* The Initial Developer of the Original Code is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All
* Rights Reserved.
*
* Contributor(s):
*/
#ifndef ParseMboxSt_H
#define ParseMboxSt_H
#include "idarray.h"
#include "ptrarray.h"
/* Be sure to include msg.h before including this file. */
class MSG_Master;
class MailDB;
class MSG_Pane;
class MSG_FolderPane;
class MailMessageHdr;
struct MSG_FilterList;
struct MSG_Filter;
struct MSG_Rule;
class MailMessageHdr;
class MSG_UrlQueue;
class TImapFlagAndUidState;
class MSG_FolderInfoContainer;
class MSG_FolderInfoMail;
class MSG_IMAPHost;
typedef enum
{
MBOX_PARSE_ENVELOPE,
MBOX_PARSE_HEADERS,
MBOX_PARSE_BODY
} MBOX_PARSE_STATE;
// This object maintains the parse state for a single mail message.
class ParseMailMessageState
{
public:
ParseMailMessageState();
virtual ~ParseMailMessageState();
void Init(uint32 fileposition);
virtual void Clear();
virtual void FinishHeader();
virtual int32 ParseFolderLine(const char *line, uint32 lineLength);
virtual int StartNewEnvelope(const char *line, uint32 lineLength);
int ParseHeaders();
int GrowHeaders(uint32 desired_size);
int GrowEnvelope(uint32 desired_size);
int FinalizeHeaders();
int ParseEnvelope (const char *line, uint32 line_size);
int InternSubject (struct message_header *header);
int InternRfc822 (struct message_header *header,
char **ret_name);
void SetMailDB(MailDB *mailDB);
MailDB *GetMailDB() {return m_mailDB;}
MailMessageHdr *m_newMsgHdr; /* current message header we're building */
MailDB *m_mailDB;
MBOX_PARSE_STATE m_state;
uint32 m_position;
uint32 m_envelope_pos;
uint32 m_headerstartpos;
char *m_headers;
uint32 m_headers_fp;
uint32 m_headers_size;
char *m_envelope;
uint32 m_envelope_fp;
uint32 m_envelope_size;
struct message_header m_message_id;
struct message_header m_references;
struct message_header m_date;
struct message_header m_from;
struct message_header m_sender;
struct message_header m_newsgroups;
struct message_header m_subject;
struct message_header m_status;
struct message_header m_mozstatus;
struct message_header m_mozstatus2;
// Support for having multiple To or CC header lines in a message
XPPtrArray m_toList;
XPPtrArray m_ccList;
struct message_header *GetNextHeaderInAggregate (XPPtrArray &list);
void GetAggregateHeader (XPPtrArray &list, struct message_header *);
void ClearAggregateHeader (XPPtrArray &list);
struct message_header m_envelope_from;
struct message_header m_envelope_date;
struct message_header m_priority;
// Mdn support
struct message_header m_mdn_original_recipient;
struct message_header m_return_path;
struct message_header m_mdn_dnt; /* MDN Disposition-Notification-To: header */
uint16 m_body_lines;
XP_Bool m_IgnoreXMozillaStatus;
};
// This class should ultimately be part of a mailbox parsing
// state machine - either by inheritance or delegation.
// Currently, a folder pane owns one and libnet message socket
// related messages get passed to this object.
// This class has a few things in common with ListNewsGroupState.
// We might want to consider a common base class that stores the
// view and db and url and master objects, though there will
// be casting involved if we do that so it may not be worthwhile.
// All of the actual work functions are different.
class ParseMailboxState
{
public:
ParseMailboxState(const char *mailboxName);
virtual ~ParseMailboxState();
MSG_Master *GetMaster() {return m_mailMaster;}
void SetMaster(MSG_Master *master)
{m_mailMaster = master;}
void SetView(MessageDBView *view) {m_msgDBView = view;}
void SetPane(MSG_Pane *pane) {m_pane = pane;}
void SetContext(MWContext *context) {m_context = context; }
MWContext* GetContext() { return m_context; }
void SetMailMessageParseState(ParseMailMessageState *mailMessageState) ;
void SetDB (MailDB *mailDB) {m_mailDB = mailDB; if (m_parseMsgState) m_parseMsgState->SetMailDB(mailDB);}
MailDB *GetDB() {return m_mailDB;}
char *GetMailboxName() {return m_mailboxName;}
void SetFolder(MSG_FolderInfo *folder) {m_folder = folder;}
void SetIncrementalUpdate(XP_Bool update) {m_updateAsWeGo = update;}
void SetIgnoreNonMailFolder(XP_Bool ignoreNonMailFolder) {m_ignoreNonMailFolder = ignoreNonMailFolder;}
MailMessageHdr *GetCurrentMsg() {return m_parseMsgState->m_newMsgHdr;}
XP_Bool GetIsRealMailFolder() {return m_isRealMailFolder;}
// message socket libnet callbacks, which come through folder pane
virtual int BeginOpenFolderSock(const char *folder_name,
const char *message_id, int32 msgnum,
void **folder_ptr);
virtual int ParseMoreFolderSock(const char* folder_name,
const char* message_id,
int32 msgnum, void** folder_ptr);
virtual void CloseFolderSock(const char* folder_name, const char* message_id,
int32 msgnum, void* folder_ptr);
virtual int BeginParsingFolder(int32 startPos);
virtual void DoneParsingFolder();
virtual void AbortNewHeader();
virtual int ParseBlock(const char *block, int32 length);
// msg_LineBuffer callback.
static int32 LineBufferCallback (char *line, uint32 line_length,
void *closure);
// which in turn calls this.
int32 ParseFolderLine(const char *line, uint32 lineLength);
void UpdateDBFolderInfo();
void UpdateDBFolderInfo(MailDB *mailDB, const char *mailboxName);
void UpdateStatusText ();
// Update the progress bar based on what we know.
virtual void UpdateProgressPercent ();
ParseMailMessageState *GetMsgState();
protected:
virtual int32 PublishMsgHeader();
virtual void FolderTypeSpecificTweakMsgHeader(MailMessageHdr *tweakMe);
void FreeBuffers();
// data
ParseMailMessageState *m_parseMsgState;
MailDB *m_mailDB;
MessageDBView *m_msgDBView; // open view on current download, if any
MSG_FolderInfo *m_folder; // folder, if any, we are reparsing.
char *m_mailboxName;
MSG_Master *m_mailMaster;
MSG_Pane *m_pane;
int32 m_obuffer_size;
char *m_obuffer;
uint32 m_ibuffer_fp;
char *m_ibuffer;
uint32 m_ibuffer_size;
XP_File m_file;
int32 m_graph_progress_total;
int32 m_graph_progress_received;
MWContext *m_context;
XP_Bool m_updateAsWeGo;
XP_Bool m_parsingDone;
XP_Bool m_ignoreNonMailFolder;
XP_Bool m_isRealMailFolder;
};
class ParseNewMailState : public ParseMailboxState
{
public:
ParseNewMailState(MSG_Master *master, MSG_FolderInfoMail *folder);
virtual ~ParseNewMailState();
virtual void DoneParsingFolder();
virtual void SetUsingTempDB(XP_Bool usingTempDB, char *tmpDBName);
void DisableFilters() {m_disableFilters = TRUE;}
// from jsmsg.cpp
friend void JSMailFilter_MoveMessage(ParseNewMailState *state,
MailMessageHdr *msgHdr,
MailDB *mailDB,
const char *folder,
MSG_Filter *filter,
XP_Bool *pMoved);
protected:
virtual int32 PublishMsgHeader();
virtual void ApplyFilters(XP_Bool *pMoved);
virtual MSG_FolderInfoMail *GetTrashFolder();
virtual int MoveIncorporatedMessage(MailMessageHdr *mailHdr,
MailDB *sourceDB,
char *destFolder,
MSG_Filter *filter);
virtual int MarkFilteredMessageRead(MailMessageHdr *msgHdr);
void LogRuleHit(MSG_Filter *filter, MailMessageHdr *msgHdr);
MSG_FilterList *m_filterList;
XP_File m_logFile;
char *m_tmpdbName; // Temporary filename of new database
XP_Bool m_usingTempDB;
XP_Bool m_disableFilters;
};
class ParseIMAPMailboxState : public ParseNewMailState
{
public:
ParseIMAPMailboxState(MSG_Master *master, MSG_IMAPHost *host, MSG_FolderInfoMail *folder, MSG_UrlQueue *urlQueue, TImapFlagAndUidState *flagStateAdopted);
virtual ~ParseIMAPMailboxState();
// close the db
virtual void DoneParsingFolder();
virtual void SetPublishUID(int32 uid);
virtual void SetPublishByteLength(uint32 byteLength);
virtual void SetNextSequenceNum(int32 seqNum);
const IDArray &GetBodyKeys() { return fFetchBodyKeys; }
MSG_UrlQueue *GetFilterUrlQueue() {return fUrlQueue;}
protected:
virtual int32 PublishMsgHeader();
virtual void FolderTypeSpecificTweakMsgHeader(MailMessageHdr *tweakMe);
virtual void ApplyFilters(XP_Bool *pMoved);
virtual MSG_FolderInfoMail *GetTrashFolder();
virtual int MoveIncorporatedMessage(MailMessageHdr *mailHdr,
MailDB *sourceDB,
char *destFolder,
MSG_Filter *filter);
virtual int MarkFilteredMessageRead(MailMessageHdr *msgHdr);
private:
XP_Bool fParsingInbox;
XP_Bool fB2HaveWarnedUserOfOfflineFiltertarget;
int32 fNextMessagePublishUID;
uint32 fNextMessageByteLength;
MSG_UrlQueue *fUrlQueue;
TImapFlagAndUidState *fFlagState;
IDArray fFetchBodyKeys;
int32 fNextSequenceNum;
MSG_FolderInfoContainer *m_imapContainer;
MSG_IMAPHost *m_host;
};
// If m_out_file is set, this parser will also write out the message,
// adding a mozilla status line if one is not present.
class ParseOutgoingMessage : public ParseMailMessageState
{
public:
ParseOutgoingMessage();
virtual ~ParseOutgoingMessage();
void SetOutFile(XP_File out_file) {m_out_file = out_file;}
XP_File GetOutFile() {return m_out_file;}
virtual int StartNewEnvelope(const char *line, uint32 lineLength);
virtual void FinishHeader();
virtual int32 ParseFolderLine(const char *line, uint32 lineLength);
virtual int32 ParseBlock(const char *block, uint32 lineLength);
virtual void Clear();
void AdvanceOutPosition(uint32 amountToAdvance);
void SetWriteToOutFile(XP_Bool writeToOutFile) {m_writeToOutFile = writeToOutFile;}
void FlushOutputBuffer();
uint32 m_bytes_written;
protected:
static int32 LineBufferCallback(char *line, uint32 lineLength, void *closure);
XP_Bool m_wroteXMozillaStatus;
XP_Bool m_writeToOutFile;
XP_Bool m_lastBodyLineEmpty;
XP_File m_out_file;
uint32 m_ouputBufferSize;
char *m_outputBuffer;
uint32 m_outputBufferIndex;
};
#endif