1 package be
.nikiroo
.gofetch
.support
;
3 import java
.io
.IOException
;
4 import java
.util
.ArrayList
;
7 import org
.jsoup
.helper
.StringUtil
;
8 import org
.jsoup
.nodes
.Element
;
9 import org
.jsoup
.nodes
.Node
;
10 import org
.jsoup
.nodes
.TextNode
;
11 import org
.jsoup
.select
.Elements
;
12 import org
.jsoup
.select
.NodeTraversor
;
13 import org
.jsoup
.select
.NodeVisitor
;
15 import be
.nikiroo
.gofetch
.data
.Story
;
16 import be
.nikiroo
.utils
.Downloader
;
18 public abstract class BasicSupport
{
19 protected static Downloader downloader
= new Downloader("gofetcher");
22 SLASHDOT
, PIPEDOT
, LWN
, LEMONDE
,
26 * Used to process an element into lines.
30 public interface ElementProcessor
{
32 * Detect if this node is a quote and should be trated as such.
36 * @return TRUE if it is
38 public boolean detectQuote(Node node
);
41 * Process text content (will be called on each text element, allowing
42 * you to modify it if needed).
48 public String
processText(String text
);
55 * @return TRUE if it has to be ignored
57 public boolean ignoreNode(Node node
);
60 * Manually process this node (and return the manual processing value)
63 * If the node is manually processed, it and its children will not be
64 * automatically processed.
67 * the node to optionally process
69 * @return NULL if not processed (will thus be automatically processed
70 * as usual), a {@link String} (may be empty) if we process it
71 * manually -- the given {@link String} will be used instead of
72 * the usual automatic processing if not NULL
74 public String
manualProcessing(Node node
);
78 * A default {@link ElementProcessor} (will not detect or process anything
83 public class BasicElementProcessor
implements ElementProcessor
{
85 public boolean detectQuote(Node node
) {
90 public String
processText(String text
) {
95 public boolean ignoreNode(Node node
) {
100 public String
manualProcessing(Node node
) {
105 static private String preselector
;
110 * List all the recent items, but only assure the ID and internal URL to
111 * fetch it later on (until it has been fetched, the rest of the
112 * {@link Story} is not confirmed).
114 * @return the list of new stories
116 * @throws IOException
119 abstract public List
<Story
> list() throws IOException
;
122 * Fetch the full article content as well as all the comments associated to
123 * this {@link Story}, if any (can be empty, but not NULL).
126 * the story to fetch the comments of
128 * @throws IOException
129 * in case of I/O error
131 abstract public void fetch(Story story
) throws IOException
;
133 abstract public String
getDescription();
135 public String
getSelector() {
136 return getSelector(type
);
139 public Type
getType() {
143 protected void setType(Type type
) {
149 * the preselector to set
151 static public void setPreselector(String preselector
) {
152 BasicSupport
.preselector
= preselector
;
156 * Return a {@link BasicSupport} that is compatible with the given
157 * {@link Type} if it exists (or NULL if not).
162 * @return a compatible {@link BasicSupport} if it exists (or NULL if not)
164 static public BasicSupport
getSupport(Type type
) {
165 BasicSupport support
= null;
170 support
= new Slashdot();
173 support
= new Pipedot();
179 support
= new LeMonde();
183 if (support
!= null) {
184 support
.setType(type
);
191 static public String
getSelector(Type type
) {
192 return preselector
+ "/" + type
+ "/";
196 * Get the first {@link Element} of the given class, or an empty span
197 * {@link Element} if none found.
200 * the element to look in
202 * the class to look for
204 * @return the value or an empty span {@link Element}
206 static protected Element
firstOrEmpty(Element element
, String className
) {
207 Elements subElements
= element
.getElementsByClass(className
);
208 if (subElements
.size() > 0) {
209 return subElements
.get(0);
212 return new Element("span");
216 * Get the first {@link Element} of the given tag, or an empty span
217 * {@link Element} if none found.
220 * the element to look in
222 * the tag to look for
224 * @return the value or an empty span {@link Element}
226 static protected Element
firstOrEmptyTag(Element element
, String tagName
) {
227 Elements subElements
= element
.getElementsByTag(tagName
);
228 if (subElements
.size() > 0) {
229 return subElements
.get(0);
232 return new Element("span");
236 * Process the given element into text (each line is a text paragraph and
237 * can be prepended with ">" signs to indicate a quote or sub-quote or
241 * the element to process
242 * @param elementProcessor
243 * the element processor, must not be NULL
245 * @return text lines, each line is a paragraph
247 static protected List
<String
> toLines(Element element
,
248 final ElementProcessor elementProcessor
) {
249 final List
<String
> lines
= new ArrayList
<String
>();
250 final StringBuilder currentLine
= new StringBuilder();
251 final List
<Integer
> quoted
= new ArrayList
<Integer
>();
252 final List
<Node
> ignoredNodes
= new ArrayList
<Node
>();
254 if (element
!= null) {
255 new NodeTraversor(new NodeVisitor() {
257 public void head(Node node
, int depth
) {
258 String manual
= null;
259 boolean ignore
= elementProcessor
.ignoreNode(node
)
260 || ignoredNodes
.contains(node
.parentNode());
262 manual
= elementProcessor
.manualProcessing(node
);
263 if (manual
!= null) {
264 currentLine
.append(manual
);
270 ignoredNodes
.add(node
);
275 for (int i
= 0; i
< quoted
.size(); i
++) {
280 boolean enterQuote
= elementProcessor
.detectQuote(node
);
281 boolean leaveQuote
= quoted
.contains(depth
);
288 quoted
.remove(Integer
.valueOf(depth
));
291 if (enterQuote
|| leaveQuote
) {
292 if (currentLine
.length() > 0) {
293 if (currentLine
.charAt(currentLine
.length() - 1) == '\n') {
294 currentLine
.setLength(currentLine
.length() - 1);
296 for (String l
: currentLine
.toString().split("\n")) {
300 currentLine
.setLength(0);
303 if (node
instanceof Element
) {
304 Element element
= (Element
) node
;
305 boolean block
= element
.isBlock()
306 || element
.tagName().equalsIgnoreCase("br");
307 if (block
&& currentLine
.length() > 0) {
308 currentLine
.append("\n");
310 } else if (node
instanceof TextNode
) {
311 TextNode textNode
= (TextNode
) node
;
312 String line
= StringUtil
.normaliseWhitespace(textNode
315 currentLine
.append(elementProcessor
.processText(line
));
316 currentLine
.append(" ");
321 public void tail(Node node
, int depth
) {
323 }).traverse(element
);
326 if (currentLine
.length() > 0) {
328 for (int i
= 0; i
< quoted
.size(); i
++) {
332 if (currentLine
.length() > 0) {
333 if (currentLine
.charAt(currentLine
.length() - 1) == '\n') {
334 currentLine
.setLength(currentLine
.length() - 1);
336 for (String l
: currentLine
.toString().split("\n")) {
342 for (int i
= 0; i
< lines
.size(); i
++) {
343 lines
.set(i
, lines
.get(i
).replace(" ", " ").trim());