summaryrefslogtreecommitdiff
path: root/include/htmlrenderer.h
blob: c63bf7629ce96e4ed16d7be85da9ddf19cc6ebc9 (plain) (blame)
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
#ifndef NEWSBEUTER_HTMLRENDERER__H
#define NEWSBEUTER_HTMLRENDERER__H

#include <textformatter.h>
#include <vector>
#include <string>
#include <istream>
#include <map>

namespace newsbeuter {

enum class link_type { HREF, IMG, EMBED };
enum class htmltag {
	A = 1, EMBED, BR, PRE, ITUNESHACK, IMG, BLOCKQUOTE, H1, H2, H3, H4, H5, H6,
	P, OL, UL, LI, DT, DD, DL, SUP, SUB, HR, STRONG, UNDERLINE, QUOTATION,
	SCRIPT, STYLE, TABLE, TH, TR, TD
};

typedef std::pair<std::string,link_type> linkpair;

class htmlrenderer {
	public:
		explicit htmlrenderer(bool raw = false);
		void render(const std::string& source,
				std::vector<std::pair<LineType, std::string>>& lines,
				std::vector<linkpair>& links,
				const std::string& url);
		void render(std::istream & input,
				std::vector<std::pair<LineType, std::string>>& lines,
				std::vector<linkpair>& links,
				const std::string& url);
		static std::string render_hr(const unsigned int width);
		// only public for unit testing purposes:
		std::string format_ol_count(unsigned int count, char type);

		struct TableCell {
			explicit TableCell(size_t s) : span(s) { }
			size_t span;
			std::vector<std::string> text; // multiline cell text
		};

		struct TableRow {
			TableRow() : inside(false) { }

			void add_text(const std::string& str);
			void start_cell(size_t span);
			void complete_cell();

			bool inside;    // inside a cell
			std::vector<TableCell> cells;
		};

		struct Table {
			explicit Table(bool b) : inside(false), border(b) { }

			void add_text(const std::string& str);
			void start_row();
			void complete_row();
			void start_cell(size_t span);
			void complete_cell();

			bool inside;    // inside a row
			bool border;
			std::vector<TableRow> rows;
		};

	private:
		void prepare_new_line(std::string& line, int indent_level);
		bool line_is_nonempty(const std::string& line);
		unsigned int add_link(std::vector<linkpair>& links, const std::string& link, link_type type);
		std::string quote_for_stfl(std::string str);
		std::string absolute_url(const std::string& url, const std::string& link);
		std::string type2str(link_type type);
		std::map<std::string, htmltag> tags;
		void render_table(
				const Table& table,
				std::vector<std::pair<LineType, std::string>>& lines);
		void add_nonempty_line(
				const std::string& curline,
				std::vector<Table>& tables,
				std::vector<std::pair<LineType, std::string>>& lines);
		void add_line(
				const std::string& curline,
				std::vector<Table>& tables,
				std::vector<std::pair<LineType, std::string>>& lines);
		void add_line_softwrappable(
				const std::string& line,
				std::vector<std::pair<LineType, std::string>>& lines);
		void add_line_nonwrappable(
				const std::string& line,
				std::vector<std::pair<LineType, std::string>>& lines);
		void add_hr(std::vector<std::pair<LineType, std::string>>& lines);
		std::string get_char_numbering(unsigned int count);
		std::string get_roman_numbering(unsigned int count);
		bool raw_;
};

}

#endif