tt-rss-android/org.fox.ttrss/src/main/java/org/fox/ttrss/types/Article.java

package org.fox.ttrss.types;

import android.os.Parcel;
import android.os.Parcelable;
import android.util.Log;

import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;

import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

// TODO: serialize Labels
public class Article implements Parcelable {
	public int id;
	public boolean unread; 
	public boolean marked; 
	public boolean published; 
	public int score;
	public int updated; 
	public boolean is_updated; 
	public String title; 
	public String link; 
	public int feed_id; 
	public List<String> tags;
	public List<Attachment> attachments;
	public String content;
    public String excerpt;
	public List<List<String>> labels;
	public String feed_title;	
	public int comments_count;
	public String comments_link;
	public boolean always_display_attachments;
	public String author;
	public String note;
    public boolean selected;

	/* not serialized */
	public Document articleDoc;
	public Element flavorImage;

	public String flavorImageUri;
	public String flavorStreamUri;
	public String youtubeVid;

	//public int flavorImageCount;

	public List<Element> mediaList = new ArrayList<>();

	public Article(Parcel in) {
		readFromParcel(in);
	}
	
	public Article() {
		
	}

	public void collectMediaInfo() {
		articleDoc = Jsoup.parse(content);

		if (articleDoc != null) {
			mediaList = articleDoc.select("img,video,iframe[src*=youtube.com/embed/]");

			for (Element e : mediaList) {
				if ("iframe".equals(e.tagName().toLowerCase())) {
					flavorImage = e;
					break;
				} else if ("video".equals(e.tagName().toLowerCase())) {
					flavorImage = e;
					break;
				}
			}

			if (flavorImage == null) {
				for (Element e : mediaList) {
					flavorImage = e;
					break;
				}
			}

			if (flavorImage != null) {

				try {

					if ("video".equals(flavorImage.tagName().toLowerCase())) {
						Element source = flavorImage.select("source").first();
						flavorStreamUri = source.attr("src");

						flavorImageUri = flavorImage.attr("poster");
					} else if ("iframe".equals(flavorImage.tagName().toLowerCase())) {

						String srcEmbed = flavorImage.attr("src");

						if (srcEmbed.length() > 0) {
							Pattern pattern = Pattern.compile("/embed/([\\w-]+)");
							Matcher matcher = pattern.matcher(srcEmbed);

							if (matcher.find()) {
								youtubeVid = matcher.group(1);

								flavorImageUri = "https://img.youtube.com/vi/" + youtubeVid + "/mqdefault.jpg";
								flavorStreamUri = "https://youtu.be/" + youtubeVid;
							}
						}
					} else {
						flavorImageUri = flavorImage.attr("src");
						flavorStreamUri = null;
					}
				} catch (Exception e) {
					e.printStackTrace();

					flavorImage = null;
					flavorImageUri = null;
					flavorStreamUri = null;
				}
			}
		}

		Log.d("Article", "collectMediaInfo: " + flavorImage);
	}

	public Article(int id) {
		this.id = id;
		this.title = "";
		this.link = "";
		this.tags = new ArrayList<String>();
	}

	@Override
	public int describeContents() {
		return 0;
	}
	
	@Override
	public void writeToParcel(Parcel out, int flags) {
		out.writeInt(id);
		out.writeInt(unread ? 1 : 0);
		out.writeInt(marked ? 1 : 0);
		out.writeInt(published ? 1 : 0);
		out.writeInt(score);
		out.writeInt(updated);
		out.writeInt(is_updated ? 1 : 0);
		out.writeString(title);
		out.writeString(link);
		out.writeInt(feed_id);
		out.writeStringList(tags);
		out.writeString(content);
        out.writeString(excerpt);
		out.writeList(attachments);
		out.writeString(feed_title);
		out.writeInt(comments_count);
		out.writeString(comments_link);
		out.writeInt(always_display_attachments ? 1 : 0);
		out.writeString(author);
		out.writeString(note);
        out.writeInt(selected ? 1 : 0);
	}
	
	public void readFromParcel(Parcel in) {
		id = in.readInt();
		unread = in.readInt() == 1;
		marked = in.readInt() == 1;
		published = in.readInt() == 1;
		score = in.readInt();
		updated = in.readInt();
		is_updated = in.readInt() == 1;
		title = in.readString();
		link = in.readString();
		feed_id = in.readInt();
		
		if (tags == null) tags = new ArrayList<String>();
		in.readStringList(tags);
		
		content = in.readString();
        excerpt = in.readString();
		
		attachments = new ArrayList<Attachment>();
		in.readList(attachments, Attachment.class.getClassLoader());
		
		feed_title = in.readString();
		
		comments_count = in.readInt();
		comments_link = in.readString();
		always_display_attachments = in.readInt() == 1;
		author = in.readString();
		note = in.readString();
        selected = in.readInt() == 1;
	}
	
	@SuppressWarnings("rawtypes")
	public static final Parcelable.Creator CREATOR =
    	new Parcelable.Creator() {
            public Article createFromParcel(Parcel in) {
                return new Article(in);
            }
 
            public Article[] newArray(int size) {
                return new Article[size];
            }
        };
}