1 files changed, 125 insertions, 0 deletions
diff --git a/src/core/basetypes/MCHTMLCleaner.cpp b/src/core/basetypes/MCHTMLCleaner.cpp
new file mode 100644
index 00000000..81145c9e
--- /dev/null
+++ b/src/core/basetypes/MCHTMLCleaner.cpp
@@ -0,0 +1,125 @@
+//
+//  HTMLCleaner.cpp
+//  mailcore2
+//
+//  Created by DINH Viêt Hoà on 2/3/13.
+//  Copyright (c) 2013 MailCore. All rights reserved.
+//
+
+#include "MCHTMLCleaner.h"
+
+#include "MCString.h"
+#include "MCData.h"
+
+#include <tidy.h>
+#include <buffio.h>
+
+#include "MCUtils.h"
+#include "MCLog.h"
+
+#if __APPLE__
+#include <TargetConditionals.h>
+#endif
+
+using namespace mailcore;
+
+String * HTMLCleaner::cleanHTML(String * input)
+{
+    TidyBuffer output;
+    TidyBuffer errbuf;
+    TidyBuffer docbuf;
+    int rc;
+    
+    TidyDoc tdoc = tidyCreate();
+    tidyBufInit(&output);
+    tidyBufInit(&errbuf);
+    tidyBufInit(&docbuf);
+    
+    Data * data = input->dataUsingEncoding("utf-8");
+    tidyBufAppend(&docbuf, data->bytes(), data->length());
+    
+#if TARGET_IPHONE_SIMULATOR || TARGET_OS_IPHONE
+    // This option is not available on the Mac.
+    tidyOptSetBool(tdoc, TidyDropEmptyElems, no);
+#endif
+    tidyOptSetBool(tdoc, TidyXhtmlOut, yes);
+    tidyOptSetInt(tdoc, TidyDoctypeMode, TidyDoctypeUser);
+    
+    tidyOptSetBool(tdoc, TidyMark, no);
+    tidySetCharEncoding(tdoc, "utf8");
+    tidyOptSetBool(tdoc, TidyForceOutput, yes);
+    //tidyOptSetValue(tdoc, TidyErrFile, "/dev/null");
+    //tidyOptSetValue(tdoc, TidyOutFile, "/dev/null");
+    tidyOptSetBool(tdoc, TidyShowWarnings, no);
+    tidyOptSetInt(tdoc, TidyShowErrors, 0);
+    rc = tidySetErrorBuffer(tdoc, &errbuf);
+    if ((rc > 1) || (rc < 0)) {
+        //fprintf(stderr, "error tidySetErrorBuffer: %i\n", rc);
+        //fprintf(stderr, "1:%s", errbuf.bp);
+        //return NULL;
+    }
+    rc = tidyParseBuffer(tdoc, &docbuf);
+    //MCLog("%s", MCUTF8(input));
+    if ((rc > 1) || (rc < 0)) {
+        //fprintf(stderr, "error tidyParseBuffer: %i\n", rc);
+        //fprintf(stderr, "1:%s", errbuf.bp);
+        //return NULL;
+    }
+    rc = tidyCleanAndRepair(tdoc);
+    if ((rc > 1) || (rc < 0)) {
+        //fprintf(stderr, "error tidyCleanAndRepair: %i\n", rc);
+        //fprintf(stderr, "1:%s", errbuf.bp);
+        //return NULL;
+    }
+    rc = tidySaveBuffer(tdoc, &output);
+    if ((rc > 1) || (rc < 0)) {
+        //fprintf(stderr, "error tidySaveBuffer: %i\n", rc);
+        //fprintf(stderr, "1:%s", errbuf.bp);
+    }
+    
+    String * result = String::stringWithUTF8Characters((const char *) output.bp);
+    
+    tidyBufFree(&docbuf);
+    tidyBufFree(&output);
+    tidyBufFree(&errbuf);
+    tidyRelease(tdoc);
+    
+    return result;
+    
+    /*
+    if ( ok ) {
+        rc = tidySetErrorBuffer( tdoc, &errbuf );      // Capture diagnostics
+    }
+    if ( rc &gt;= 0 ) {
+        rc = tidyParseString( tdoc, input );           // Parse the input
+    }
+    if ( rc &gt;= 0 ) {
+        rc = tidyCleanAndRepair( tdoc );               // Tidy it up!
+    }
+    if ( rc &gt;= 0 ) {
+        rc = tidyRunDiagnostics( tdoc );               // Kvetch
+    }
+    if ( rc &gt; 1 ) {                                    // If error, force output.
+        rc = ( tidyOptSetBool(tdoc, TidyForceOutput, yes) ? rc : -1 );
+    }
+    if ( rc &gt;= 0 ) {
+        rc = tidySaveBuffer( tdoc, &output );          // Pretty Print
+    }
+     */
+    
+    /*
+    if ( rc &gt;= 0 )
+    {
+        if ( rc &gt; 0 )
+            printf( "\\nDiagnostics:\\n\\n\%s", errbuf.bp );
+            printf( "\\nAnd here is the result:\\n\\n\%s", output.bp );
+            }
+    else
+        printf( "A severe error (\%d) occurred.\\n", rc );
+        
+        tidyBufFree( &amp;output );
+        tidyBufFree( &amp;errbuf );
+        tidyRelease( tdoc );
+        return rc;
+     */
+}