63588a79da2a43d1d960bed39393d6f908f00e5e
[asciidoctor_to_rss.git] / src / main / java / net / wpitchoune / asciidoctor / HTML.java
1 package net.wpitchoune.asciidoctor;
2
3 import java.io.File;
4 import java.io.IOException;
5 import java.nio.charset.StandardCharsets;
6 import java.nio.file.Files;
7 import java.text.SimpleDateFormat;
8 import java.util.ArrayList;
9 import java.util.Collection;
10 import java.util.Collections;
11 import java.util.Comparator;
12 import java.util.List;
13 import java.util.logging.Logger;
14
15 import com.rometools.rome.feed.synd.SyndEntry;
16
17 /*
18  * Copyright (C) 2016 jeanfi@gmail.com
19  *
20  * This program is free software; you can redistribute it and/or
21  * modify it under the terms of the GNU General Public License as
22  * published by the Free Software Foundation; either version 2 of the
23  * License, or (at your option) any later version.
24  *
25  * This program is distributed in the hope that it will be useful, but
26  * WITHOUT ANY WARRANTY; without even the implied warranty of
27  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
28  * General Public License for more details.
29  *
30  * You should have received a copy of the GNU General Public License
31  * along with this program; if not, write to the Free Software
32  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
33  * 02110-1301 USA
34  */
35
36 public final class HTML {
37         private static final Logger LOG = Logger.getLogger(Main.class.getSimpleName()); 
38         private static final SimpleDateFormat DATE_FORMATTER = new SimpleDateFormat("yyyy-dd-MM");        
39         private final Configuration config; 
40         
41         public HTML(Configuration config) {
42                 this.config = config;
43         }
44         
45         public static void appendHTMLHead(StringBuffer sb, String title, Configuration config)
46                         throws IOException {
47                 File f;
48                 
49                 f = config.getHTMLHeaderFile();
50                 if (f == null) {
51                         LOG.info("There is no declared HTML header file.");
52                         return ;
53                 }
54
55                 sb.append("<!DOCTYPE html>\n");
56                 sb.append("<html>\n");                          
57                 sb.append("<head>\n");
58                 sb.append(new String(Files.readAllBytes(f.toPath()),
59                                      StandardCharsets.UTF_8));
60                 sb.append("<title>");
61                 sb.append(title);
62                 sb.append("</title>\n");
63                 sb.append("</head>\n");
64         }
65
66         private void appendHTMLHead(StringBuffer sb, String title) throws IOException {
67                 appendHTMLHead(sb, title, config);
68         }
69         
70         public static void appendHTMLContentHeader(StringBuffer sb, String title) {
71                 sb.append("<div id='header'>\n");
72                 sb.append("<h1>");
73                 sb.append(title);
74                 sb.append("</h1>\n");
75                 sb.append("</div>");
76         }   
77         
78         
79         public static void appendHTMLFooter(StringBuffer sb) {
80                 sb.append("</body>\n");
81                 sb.append("</html>");           
82         }                     
83         
84         public String toHTML(SyndEntry entry) throws IOException {
85                 StringBuffer buf;
86                 
87                 buf = new StringBuffer();
88                 
89                 appendHTMLHead(buf, entry.getTitle());
90                 
91                 buf.append("<body>\n");
92                 
93                 appendHTMLContentHeader(buf, entry.getTitle());
94                 
95                 buf.append("<div id='content'>\n");
96                 if (entry.getPublishedDate() != null) {
97                         buf.append("<div class='date'>");
98                         buf.append(DATE_FORMATTER.format(entry.getPublishedDate()));
99                         buf.append("</div>");
100                 }
101                 buf.append(entry.getDescription().getValue());
102                 buf.append("</div>");
103                 
104                 appendHTMLFooter(buf);
105                 
106                 return buf.toString();  
107         }
108         
109         public String toHTML(Collection<SyndEntry> entries) throws IOException {
110                 StringBuffer buf;
111                 List<SyndEntry> sortedEntries;
112                 Comparator<SyndEntry> cmp;
113                 
114                 buf = new StringBuffer();
115                 
116                 appendHTMLHead(buf, config.getFeedTitle());
117                 
118                 buf.append("<body>\n");
119                 
120                 appendHTMLContentHeader(buf, config.getFeedTitle());
121                 
122                 buf.append("<div id='content'>\n");
123                 
124                 cmp = new Comparator<SyndEntry>() {
125                         @Override
126                         public int compare(SyndEntry o1, SyndEntry o2) {
127                                 return o2.getPublishedDate().compareTo(o1.getPublishedDate());
128                         }                       
129                 };
130                 
131                 sortedEntries = new ArrayList<SyndEntry>(entries);
132                 Collections.sort(sortedEntries, cmp);
133                 
134                 for(SyndEntry e: sortedEntries) {
135                         buf.append("\n<div>");
136                         buf.append("<h2>");
137                         buf.append("<a href='" + e.getUri() + "'>");
138                         buf.append(e.getTitle());
139                         buf.append("</a></h2>");
140                         if (e.getPublishedDate() != null) {
141                                 buf.append("<div class='date'>");
142                                 buf.append(DATE_FORMATTER.format(e.getPublishedDate()));
143                                 buf.append("</div>");
144                         }
145                         buf.append(e.getDescription().getValue());
146                         buf.append("</div>\n");   
147                 }
148                                 
149                 buf.append("</div>\n");
150                 
151                 HTML.appendHTMLFooter(buf);
152                 
153                 return buf.toString();
154         }
155 }