kopano-core/common/HtmlToTextParser.h

/*
 * Copyright 2005 - 2016 Zarafa and its licensors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#pragma once

#include <kopano/zcdefs.h>
#include <string>
#include <map>
#include <stack>

namespace KC {

class _kc_export CHtmlToTextParser _kc_final {
public:
	CHtmlToTextParser(void);
	bool Parse(const WCHAR *lpwHTML);
	std::wstring& GetText();

protected:
	_kc_hidden void Init(void);
	_kc_hidden void parseTag(const wchar_t *&);
	_kc_hidden bool parseEntity(const wchar_t *&);
	_kc_hidden void parseAttributes(const wchar_t *&);
	_kc_hidden void addChar(wchar_t);
	_kc_hidden void addNewLine(bool force_line);
	_kc_hidden bool addURLAttribute(const wchar_t *attr, bool spaces = false);
	_kc_hidden void addSpace(bool force);

	//Parse tags
	_kc_hidden void parseTagP(void);
	_kc_hidden void parseTagBP(void);
	_kc_hidden void parseTagBR(void);
	_kc_hidden void parseTagTR(void);
	_kc_hidden void parseTagBTR(void);
	_kc_hidden void parseTagTDTH(void);
	_kc_hidden void parseTagIMG(void);
	_kc_hidden void parseTagA(void);
	_kc_hidden void parseTagBA(void);
	_kc_hidden void parseTagSCRIPT(void);
	_kc_hidden void parseTagBSCRIPT(void);
	_kc_hidden void parseTagSTYLE(void);
	_kc_hidden void parseTagBSTYLE(void);
	_kc_hidden void parseTagHEAD(void);
	_kc_hidden void parseTagBHEAD(void);
	_kc_hidden void parseTagNewLine(void);
	_kc_hidden void parseTagHR(void);
	_kc_hidden void parseTagHeading(void);
	_kc_hidden void parseTagPRE(void);
	_kc_hidden void parseTagBPRE(void);
	_kc_hidden void parseTagOL(void);
	_kc_hidden void parseTagUL(void);
	_kc_hidden void parseTagLI(void);
	_kc_hidden void parseTagPopList(void);
	_kc_hidden void parseTagDL(void);
	_kc_hidden void parseTagDT(void);
	_kc_hidden void parseTagDD(void);

	std::wstring strText;
	bool fScriptMode = false;
	bool fHeadMode = false;
	short cNewlines = 0;
	bool fStyleMode = false;
	bool fTDTHMode = false;
	bool fPreMode = false;
	bool fTextMode = false;
	bool fAddSpace = false;

	typedef void ( CHtmlToTextParser::*ParseMethodType )( void );

	struct _kc_hidden tagParser {
		tagParser(void) = default;
		tagParser(bool bParseAttrs, ParseMethodType parserMethod){
			this->bParseAttrs = bParseAttrs;
			this->parserMethod = parserMethod;
		};
		bool bParseAttrs = false;
		ParseMethodType parserMethod = nullptr;
	};

	struct _TableRow {
		bool bFirstCol;
	};

	enum eListMode { lmDefinition, lmOrdered, lmUnordered };
	struct ListInfo {
		eListMode mode = lmDefinition;
		unsigned int count = 0;
	};

	typedef std::map<std::wstring, tagParser>		MapParser;
	typedef std::map<std::wstring, std::wstring>	MapAttrs;

	typedef std::stack<MapAttrs>	StackMapAttrs;
	typedef std::stack<_TableRow>	StackTableRow;
	typedef std::stack<ListInfo>	ListInfoStack;
	
	StackTableRow	stackTableRow;
	MapParser		tagMap;
	StackMapAttrs	stackAttrs;
	ListInfo 		listInfo;
	ListInfoStack	listInfoStack;
};

} /* namespace */