1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef _nsMsgFolderCompactor_h
#define _nsMsgFolderCompactor_h
#include "mozilla/Attributes.h"
#include "nsCOMPtr.h"
#include "nsIMsgFolder.h"
#include "nsIStreamListener.h"
#include "nsIMsgFolderCompactor.h"
#include "nsICopyMsgStreamListener.h"
#include "nsMsgKeyArray.h"
#include "nsIMsgWindow.h"
#include "nsIStringBundle.h"
#include "nsIMsgMessageService.h"
#define COMPACTOR_READ_BUFF_SIZE 16384
class nsFolderCompactState : public nsIMsgFolderCompactor,
public nsIStreamListener,
public nsICopyMessageStreamListener,
public nsIUrlListener
{
public:
NS_DECL_ISUPPORTS
NS_DECL_NSIREQUESTOBSERVER
NS_DECL_NSISTREAMLISTENER
NS_DECL_NSICOPYMESSAGESTREAMLISTENER
NS_DECL_NSIURLLISTENER
NS_DECL_NSIMSGFOLDERCOMPACTOR
nsFolderCompactState(void);
protected:
virtual ~nsFolderCompactState(void);
virtual nsresult InitDB(nsIMsgDatabase *db);
virtual nsresult StartCompacting();
virtual nsresult FinishCompact();
void CloseOutputStream();
void CleanupTempFilesAfterError();
nsresult Init(nsIMsgFolder *aFolder, const char* aBaseMsgUri, nsIMsgDatabase *aDb,
nsIFile *aPath, nsIMsgWindow *aMsgWindow);
nsresult GetMessage(nsIMsgDBHdr **message);
nsresult BuildMessageURI(const char *baseURI, nsMsgKey key, nsCString& uri);
nsresult ShowStatusMsg(const nsString& aMsg);
nsresult ReleaseFolderLock();
void ShowCompactingStatusMsg();
void CompactCompleted(nsresult exitCode);
void ShowDoneStatus();
nsresult CompactNextFolder();
nsCString m_baseMessageUri; // base message uri
nsCString m_messageUri; // current message uri being copy
nsCOMPtr<nsIMsgFolder> m_folder; // current folder being compact
nsCOMPtr<nsIMsgDatabase> m_db; // new database for the compact folder
nsCOMPtr <nsIFile> m_file; // new mailbox for the compact folder
nsCOMPtr <nsIOutputStream> m_fileStream; // output file stream for writing
// all message keys that need to be copied over
RefPtr<nsMsgKeyArray> m_keyArray;
uint32_t m_size;
// sum of the sizes of the messages, accumulated as we visit each msg.
uint64_t m_totalMsgSize;
// number of bytes that can be expunged while compacting.
uint64_t m_totalExpungedBytes;
uint32_t m_curIndex; // index of the current copied message key in key array
uint64_t m_startOfNewMsg; // offset in mailbox of new message
char m_dataBuffer[COMPACTOR_READ_BUFF_SIZE + 1]; // temp data buffer for copying message
nsresult m_status; // the status of the copying operation
nsCOMPtr <nsIMsgMessageService> m_messageService; // message service for copying
nsCOMPtr<nsIArray> m_folderArray; // folders we are compacting, if compacting multiple.
nsCOMPtr <nsIMsgWindow> m_window;
nsCOMPtr <nsIMsgDBHdr> m_curSrcHdr;
uint32_t m_folderIndex; // tells which folder to compact in case of compact all
bool m_compactAll; //flag for compact all
bool m_compactOfflineAlso; //whether to compact offline also
bool m_compactingOfflineFolders; // are we in the offline folder compact phase
bool m_parsingFolder; //flag for parsing local folders;
// these members are used to add missing status lines to compacted messages.
bool m_needStatusLine;
bool m_startOfMsg;
int32_t m_statusOffset;
uint32_t m_addedHeaderSize;
nsCOMPtr<nsIArray> m_offlineFolderArray;
nsCOMPtr<nsIUrlListener> m_listener;
bool m_alreadyWarnedDiskSpace;
};
class nsOfflineStoreCompactState : public nsFolderCompactState
{
public:
nsOfflineStoreCompactState(void);
virtual ~nsOfflineStoreCompactState(void);
NS_IMETHOD OnStopRequest(nsIRequest *request, nsISupports *ctxt,
nsresult status) override;
NS_IMETHODIMP OnDataAvailable(nsIRequest *request, nsISupports *ctxt,
nsIInputStream *inStr,
uint64_t sourceOffset, uint32_t count) override;
protected:
nsresult CopyNextMessage(bool &done);
virtual nsresult InitDB(nsIMsgDatabase *db) override;
virtual nsresult StartCompacting() override;
virtual nsresult FinishCompact() override;
uint32_t m_offlineMsgSize;
};
#endif
|
