gecko-dev/lib/libmsg/prsembst.h

/* -*- 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