зеркало из https://github.com/mozilla/gecko-dev.git
(Not part of the build yet). This file used to be mailnews\mime\html\nsMimeEmitter.h/.cpp if you need to look up a
log for the file.
This commit is contained in:
mscott%netscape.com
Родитель
03974ead3b
Коммит
27129ca7d9
|
|
@ -0,0 +1,461 @@
|
|||
/* -*- 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.0 (the "NPL"); you may not use this file except in
|
||||
* compliance with the NPL. You may obtain a copy of the NPL at
|
||||
* http://www.mozilla.org/NPL/
|
||||
*
|
||||
* Software distributed under the NPL is distributed on an "AS IS" basis,
|
||||
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
|
||||
* for the specific language governing rights and limitations under the
|
||||
* NPL.
|
||||
*
|
||||
* The Initial Developer of this code under the NPL is Netscape
|
||||
* Communications Corporation. Portions created by Netscape are
|
||||
* Copyright (C) 1998 Netscape Communications Corporation. All Rights
|
||||
* Reserved.
|
||||
*/
|
||||
#include "stdio.h"
|
||||
#include "nsMimeRebuffer.h"
|
||||
#include "nsMimeHtmlEmitter.h"
|
||||
#include "plstr.h"
|
||||
#include "nsEmitterUtils.h"
|
||||
#include "nsMailHeaders.h"
|
||||
#include "nscore.h"
|
||||
|
||||
|
||||
nsresult NS_NewMimeHtmlEmitter(const nsIID& iid, void **result)
|
||||
{
|
||||
nsMimeHtmlEmitter *obj = new nsMimeHtmlEmitter();
|
||||
if (obj)
|
||||
return obj->QueryInterface(iid, result);
|
||||
else
|
||||
return NS_ERROR_OUT_OF_MEMORY;
|
||||
}
|
||||
|
||||
/*
|
||||
* The following macros actually implement addref, release and
|
||||
* query interface for our component.
|
||||
*/
|
||||
NS_IMPL_ADDREF(nsMimeHtmlEmitter)
|
||||
NS_IMPL_RELEASE(nsMimeHtmlEmitter)
|
||||
NS_IMPL_QUERY_INTERFACE(nsMimeHtmlEmitter, nsIMimeEmitter::GetIID()); /* we need to pass in the interface ID of this interface */
|
||||
|
||||
/*
|
||||
* nsIMimeEmitter definitions....
|
||||
*/
|
||||
nsMimeHtmlEmitter::nsMimeHtmlEmitter()
|
||||
{
|
||||
/* the following macro is used to initialize the ref counting data */
|
||||
NS_INIT_REFCNT();
|
||||
|
||||
mOutStream = NULL;
|
||||
mBufferMgr = NULL;
|
||||
mTotalWritten = 0;
|
||||
mTotalRead = 0;
|
||||
mDocHeader = PR_FALSE;
|
||||
mAttachContentType = NULL;
|
||||
|
||||
#ifdef DEBUG_rhp
|
||||
mLogFile = NULL; /* Temp file to put generated HTML into. */
|
||||
mReallyOutput = PR_FALSE;
|
||||
#endif
|
||||
}
|
||||
|
||||
nsMimeHtmlEmitter::~nsMimeHtmlEmitter(void)
|
||||
{
|
||||
if (mBufferMgr)
|
||||
delete mBufferMgr;
|
||||
}
|
||||
|
||||
// Set the output stream for processed data.
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::SetOutputStream(nsINetOStream *outStream)
|
||||
{
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
// Note - these is setup only...you should not write
|
||||
// anything to the stream since these may be image data
|
||||
// output streams, etc...
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::Initialize(nsINetOStream *outStream)
|
||||
{
|
||||
mOutStream = outStream;
|
||||
|
||||
// Create rebuffering object
|
||||
mBufferMgr = new MimeRebuffer();
|
||||
|
||||
// Counters for output stream
|
||||
mTotalWritten = 0;
|
||||
mTotalRead = 0;
|
||||
|
||||
#ifdef DEBUG_rhp
|
||||
PR_Delete("C:\\email.html");
|
||||
mLogFile = PR_Open("C:\\email.html", PR_RDWR | PR_CREATE_FILE | PR_TRUNCATE, 493);
|
||||
#endif /* DEBUG */
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
// Note - this is teardown only...you should not write
|
||||
// anything to the stream since these may be image data
|
||||
// output streams, etc...
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::Complete()
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_FALSE;
|
||||
#endif
|
||||
|
||||
// If we are here and still have data to write, we should try
|
||||
// to flush it...if we try and fail, we should probably return
|
||||
// an error!
|
||||
PRUint32 written;
|
||||
if (mBufferMgr->GetSize() > 0)
|
||||
Write("", 0, &written);
|
||||
|
||||
printf("TOTAL WRITTEN = %d\n", mTotalWritten);
|
||||
printf("LEFTOVERS = %d\n", mBufferMgr->GetSize());
|
||||
|
||||
#ifdef DEBUG_rhp
|
||||
if (mLogFile)
|
||||
PR_Close(mLogFile);
|
||||
#endif
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
// Header handling routines.
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::StartHeader(PRBool rootMailHeader, PRBool headerOnly, const char *msgID,
|
||||
const char *outCharset)
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
mDocHeader = rootMailHeader;
|
||||
|
||||
if (mDocHeader)
|
||||
{
|
||||
if ( (!headerOnly) && (outCharset) && (*outCharset) )
|
||||
{
|
||||
UtilityWrite("<META HTTP-EQUIV=\"Content-Type\" CONTENT=\"text/html; charset=");
|
||||
UtilityWrite(outCharset);
|
||||
UtilityWrite("\">");
|
||||
}
|
||||
UtilityWrite("<BLOCKQUOTE><table BORDER=0>");
|
||||
}
|
||||
else
|
||||
UtilityWrite("<BLOCKQUOTE><table BORDER=0 BGCOLOR=\"#CCCCCC\" >");
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::AddHeaderField(const char *field, const char *value)
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
if ( (!field) || (!value) )
|
||||
return NS_OK;
|
||||
|
||||
char *newValue = nsEscapeHTML(value);
|
||||
if (!newValue)
|
||||
return NS_OK;
|
||||
|
||||
UtilityWrite("<TR>");
|
||||
|
||||
UtilityWrite("<td>");
|
||||
UtilityWrite("<div align=right>");
|
||||
UtilityWrite("<B>");
|
||||
UtilityWrite(field);
|
||||
UtilityWrite(":");
|
||||
UtilityWrite("</B>");
|
||||
UtilityWrite("</div>");
|
||||
UtilityWrite("</td>");
|
||||
|
||||
UtilityWrite("<td>");
|
||||
UtilityWrite(newValue);
|
||||
UtilityWrite("</td>");
|
||||
|
||||
UtilityWrite("</TR>");
|
||||
|
||||
PR_FREEIF(newValue);
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::EndHeader()
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
UtilityWrite("</TABLE></BLOCKQUOTE>");
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::ProcessContentType(const char *ct)
|
||||
{
|
||||
if (mAttachContentType)
|
||||
{
|
||||
PR_FREEIF(mAttachContentType);
|
||||
mAttachContentType = NULL;
|
||||
}
|
||||
|
||||
if ( (!ct) || (!*ct) )
|
||||
return NS_OK;
|
||||
|
||||
char *slash = PL_strchr(ct, '/');
|
||||
if (!slash)
|
||||
mAttachContentType = PL_strdup(ct);
|
||||
else
|
||||
{
|
||||
PRInt32 size = (PL_strlen(ct) + 4 + 1);
|
||||
mAttachContentType = (char *)PR_MALLOC( size );
|
||||
if (!mAttachContentType)
|
||||
return NS_ERROR_OUT_OF_MEMORY;
|
||||
|
||||
memset(mAttachContentType, 0, size);
|
||||
PL_strcpy(mAttachContentType, ct);
|
||||
|
||||
char *newSlash = PL_strchr(mAttachContentType, '/');
|
||||
*newSlash = '\0';
|
||||
PL_strcat(mAttachContentType, "%2F");
|
||||
PL_strcat(mAttachContentType, (slash + 1));
|
||||
}
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
// Attachment handling routines
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::StartAttachment(const char *name, const char *contentType, const char *url)
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
PR_FREEIF(mAttachContentType);
|
||||
mAttachContentType = NULL;
|
||||
ProcessContentType(contentType);
|
||||
UtilityWrite("<CENTER>");
|
||||
UtilityWrite("<table BORDER CELLSPACING=0>");
|
||||
UtilityWrite("<tr>");
|
||||
UtilityWrite("<td>");
|
||||
|
||||
if (mAttachContentType)
|
||||
{
|
||||
UtilityWrite("<a href=\"");
|
||||
UtilityWrite(url);
|
||||
UtilityWrite("&outformat=");
|
||||
UtilityWrite(mAttachContentType);
|
||||
UtilityWrite("\" target=new>");
|
||||
}
|
||||
|
||||
UtilityWrite("<img SRC=\"resource:/res/network/gopher-unknown.gif\" BORDER=0 ALIGN=ABSCENTER>");
|
||||
UtilityWrite(name);
|
||||
|
||||
if (mAttachContentType)
|
||||
UtilityWrite("</a>");
|
||||
|
||||
UtilityWrite("</td>");
|
||||
UtilityWrite("<td>");
|
||||
UtilityWrite("<table BORDER=0 BGCOLOR=\"#FFFFCC\">");
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::AddAttachmentField(const char *field, const char *value)
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
// Don't let bad things happen
|
||||
if ( (!value) || (!*value) )
|
||||
return NS_OK;
|
||||
|
||||
char *newValue = nsEscapeHTML(value);
|
||||
PRBool linkIt = (!PL_strcmp(HEADER_X_MOZILLA_PART_URL, field));
|
||||
|
||||
//
|
||||
// For now, let's not output the long URL field, but when prefs are
|
||||
// working this will change.
|
||||
//
|
||||
if (linkIt)
|
||||
return NS_OK;
|
||||
|
||||
UtilityWrite("<TR>");
|
||||
|
||||
UtilityWrite("<td>");
|
||||
UtilityWrite("<div align=right>");
|
||||
UtilityWrite("<B>");
|
||||
UtilityWrite(field);
|
||||
UtilityWrite(":");
|
||||
UtilityWrite("</B>");
|
||||
UtilityWrite("</div>");
|
||||
UtilityWrite("</td>");
|
||||
UtilityWrite("<td>");
|
||||
|
||||
if (linkIt)
|
||||
{
|
||||
UtilityWrite("<a href=\"");
|
||||
UtilityWrite(value);
|
||||
if (mAttachContentType)
|
||||
{
|
||||
UtilityWrite("&outformat=");
|
||||
UtilityWrite(mAttachContentType);
|
||||
}
|
||||
UtilityWrite("\" target=new>");
|
||||
}
|
||||
|
||||
UtilityWrite(newValue);
|
||||
|
||||
if (linkIt)
|
||||
UtilityWrite("</a>");
|
||||
|
||||
UtilityWrite("</td>");
|
||||
UtilityWrite("</TR>");
|
||||
|
||||
PR_FREEIF(newValue);
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::EndAttachment()
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
PR_FREEIF(mAttachContentType);
|
||||
UtilityWrite("</TABLE>");
|
||||
UtilityWrite("</td>");
|
||||
UtilityWrite("</tr>");
|
||||
|
||||
UtilityWrite("</TABLE>");
|
||||
UtilityWrite("</CENTER>");
|
||||
UtilityWrite("<BR>");
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
// Attachment handling routines
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::StartBody(PRBool bodyOnly, const char *msgID, const char *outCharset)
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
if ((bodyOnly) && (outCharset) && (*outCharset))
|
||||
{
|
||||
UtilityWrite("<META HTTP-EQUIV=\"Content-Type\" CONTENT=\"text/html; charset=");
|
||||
UtilityWrite(outCharset);
|
||||
UtilityWrite("\">");
|
||||
}
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::WriteBody(const char *buf, PRUint32 size, PRUint32 *amountWritten)
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_TRUE;
|
||||
#endif
|
||||
|
||||
Write(buf, size, amountWritten);
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::EndBody()
|
||||
{
|
||||
#ifdef DEBUG_rhp
|
||||
mReallyOutput = PR_FALSE;
|
||||
#endif
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
// These are routines necessary for the C based routines in libmime
|
||||
// to access the new world streams.
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::Write(const char *buf, PRUint32 size, PRUint32 *amountWritten)
|
||||
{
|
||||
unsigned int written = 0;
|
||||
PRUint32 rc, aReadyCount = 0;
|
||||
|
||||
#ifdef DEBUG_rhp
|
||||
if ((mLogFile) && (mReallyOutput))
|
||||
PR_Write(mLogFile, buf, size);
|
||||
#endif
|
||||
|
||||
//
|
||||
// Make sure that the buffer we are "pushing" into has enough room
|
||||
// for the write operation. If not, we have to buffer, return, and get
|
||||
// it on the next time through
|
||||
//
|
||||
*amountWritten = 0;
|
||||
rc = mOutStream->WriteReady(&aReadyCount);
|
||||
|
||||
// First, handle any old buffer data...
|
||||
if (mBufferMgr->GetSize() > 0)
|
||||
{
|
||||
if (aReadyCount >= mBufferMgr->GetSize())
|
||||
{
|
||||
rc += mOutStream->Write(mBufferMgr->GetBuffer(),
|
||||
mBufferMgr->GetSize(), &written);
|
||||
mTotalWritten += written;
|
||||
mBufferMgr->ReduceBuffer(written);
|
||||
}
|
||||
else
|
||||
{
|
||||
rc += mOutStream->Write(mBufferMgr->GetBuffer(),
|
||||
aReadyCount, &written);
|
||||
mTotalWritten += written;
|
||||
mBufferMgr->ReduceBuffer(written);
|
||||
mBufferMgr->IncreaseBuffer(buf, size);
|
||||
return rc;
|
||||
}
|
||||
}
|
||||
|
||||
// Now, deal with the new data the best way possible...
|
||||
rc = mOutStream->WriteReady(&aReadyCount);
|
||||
if (aReadyCount >= size)
|
||||
{
|
||||
rc += mOutStream->Write(buf, size, &written);
|
||||
mTotalWritten += written;
|
||||
*amountWritten = written;
|
||||
return rc;
|
||||
}
|
||||
else
|
||||
{
|
||||
rc += mOutStream->Write(buf, aReadyCount, &written);
|
||||
mTotalWritten += written;
|
||||
mBufferMgr->IncreaseBuffer(buf+written, (size-written));
|
||||
*amountWritten = written;
|
||||
return rc;
|
||||
}
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsMimeHtmlEmitter::UtilityWrite(const char *buf)
|
||||
{
|
||||
PRInt32 tmpLen = PL_strlen(buf);
|
||||
PRUint32 written;
|
||||
|
||||
Write(buf, tmpLen, &written);
|
||||
#ifdef DEBUG
|
||||
// Write("\r\n", 2, &written);
|
||||
#endif
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
|
@ -0,0 +1,90 @@
|
|||
/* -*- 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.0 (the "NPL"); you may not use this file except in
|
||||
* compliance with the NPL. You may obtain a copy of the NPL at
|
||||
* http://www.mozilla.org/NPL/
|
||||
*
|
||||
* Software distributed under the NPL is distributed on an "AS IS" basis,
|
||||
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
|
||||
* for the specific language governing rights and limitations under the
|
||||
* NPL.
|
||||
*
|
||||
* The Initial Developer of this code under the NPL is Netscape
|
||||
* Communications Corporation. Portions created by Netscape are
|
||||
* Copyright (C) 1998 Netscape Communications Corporation. All Rights
|
||||
* Reserved.
|
||||
*/
|
||||
#ifndef _nsMimeHtmlEmitter_h_
|
||||
#define _nsMimeHtmlEmitter_h_
|
||||
|
||||
#include "prtypes.h"
|
||||
#include "prio.h"
|
||||
#include "nsMimeRebuffer.h"
|
||||
#include "nsINetOStream.h"
|
||||
#include "nsIMimeEmitter.h"
|
||||
|
||||
class nsMimeHtmlEmitter : public nsIMimeEmitter {
|
||||
public:
|
||||
nsMimeHtmlEmitter ();
|
||||
virtual ~nsMimeHtmlEmitter (void);
|
||||
|
||||
/* this macro defines QueryInterface, AddRef and Release for this class */
|
||||
NS_DECL_ISUPPORTS
|
||||
|
||||
// These will be called to start and stop the total operation
|
||||
NS_IMETHOD Initialize(nsINetOStream *outStream);
|
||||
NS_IMETHOD Complete();
|
||||
|
||||
// Set the output stream for processed data.
|
||||
NS_IMETHOD SetOutputStream(nsINetOStream *outStream);
|
||||
|
||||
// Header handling routines.
|
||||
NS_IMETHOD StartHeader(PRBool rootMailHeader, PRBool headerOnly, const char *msgID,
|
||||
const char *outCharset);
|
||||
NS_IMETHOD AddHeaderField(const char *field, const char *value);
|
||||
NS_IMETHOD EndHeader();
|
||||
|
||||
// Attachment handling routines
|
||||
NS_IMETHOD StartAttachment(const char *name, const char *contentType, const char *url);
|
||||
NS_IMETHOD AddAttachmentField(const char *field, const char *value);
|
||||
NS_IMETHOD EndAttachment();
|
||||
|
||||
// Body handling routines
|
||||
NS_IMETHOD StartBody(PRBool bodyOnly, const char *msgID, const char *outCharset);
|
||||
NS_IMETHOD WriteBody(const char *buf, PRUint32 size, PRUint32 *amountWritten);
|
||||
NS_IMETHOD EndBody();
|
||||
|
||||
// Generic write routine. This is necessary for output that
|
||||
// libmime needs to pass through without any particular parsing
|
||||
// involved (i.e. decoded images, HTML Body Text, etc...
|
||||
NS_IMETHOD Write(const char *buf, PRUint32 size, PRUint32 *amountWritten);
|
||||
NS_IMETHOD UtilityWrite(const char *buf);
|
||||
|
||||
NS_IMETHOD ProcessContentType(const char *ct);
|
||||
|
||||
protected:
|
||||
// For buffer management on output
|
||||
MimeRebuffer *mBufferMgr;
|
||||
|
||||
// For the output stream
|
||||
nsINetOStream *mOutStream;
|
||||
PRUint32 mTotalWritten;
|
||||
PRUint32 mTotalRead;
|
||||
|
||||
// For header determination...
|
||||
PRBool mDocHeader;
|
||||
|
||||
// For content type...
|
||||
char *mAttachContentType;
|
||||
|
||||
#ifdef DEBUG_rhp
|
||||
PRBool mReallyOutput;
|
||||
PRFileDesc *mLogFile; /* Temp file to put generated HTML into. */
|
||||
#endif
|
||||
};
|
||||
|
||||
/* this function will be used by the factory to generate an class access object....*/
|
||||
extern nsresult NS_NewMimeHtmlEmitter(const nsIID& iid, void **result);
|
||||
|
||||
#endif /* _nsMimeHtmlEmitter_h_ */
|
||||
Загрузка…
Ссылка в новой задаче