package org.schabi.newpipe.extractor.services.youtube;

import org.json.JSONException;
import org.json.JSONObject;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.schabi.newpipe.extractor.AbstractStreamInfo;
import org.schabi.newpipe.extractor.Downloader;
import org.schabi.newpipe.extractor.NewPipe;
import org.schabi.newpipe.extractor.Parser;
import org.schabi.newpipe.extractor.UrlIdHandler;
import org.schabi.newpipe.extractor.exceptions.ExtractionException;
import org.schabi.newpipe.extractor.exceptions.ParsingException;
import org.schabi.newpipe.extractor.playlist.PlayListExtractor;
import org.schabi.newpipe.extractor.stream_info.StreamInfoItemCollector;
import org.schabi.newpipe.extractor.stream_info.StreamInfoItemExtractor;

import java.io.IOException;

public class YoutubePlayListExtractor extends PlayListExtractor {

    private String TAG = YoutubePlayListExtractor.class.toString();

    private Document doc = null;

    private boolean isAjaxPage = false;
    private static String name = "";
    private static String feedUrl = "";
    private static String avatarUrl = "";
    private static String bannerUrl = "";
    private static String nextPageUrl = "";

    public YoutubePlayListExtractor(UrlIdHandler urlIdHandler,
                                    String url, int page, int serviceId) throws IOException, ExtractionException {
        super(urlIdHandler, url, page, serviceId);
        Downloader downloader = NewPipe.getDownloader();
        url = urlIdHandler.cleanUrl(url);
        if(page == 0) {
            String channelPageContent = downloader.download(url);
            doc = Jsoup.parse(channelPageContent, url);
            nextPageUrl = getNextPageUrl(doc);
            isAjaxPage = false;
        } else {
            String ajaxDataRaw = downloader.download(nextPageUrl);
            JSONObject ajaxData;
            try {
                ajaxData = new JSONObject(ajaxDataRaw);
                final String htmlDataRaw = "<table><tbody id=\"pl-load-more-destination\">" + ajaxData.getString("content_html") + "</tbody></table>";
                doc = Jsoup.parse(htmlDataRaw, nextPageUrl);
                final String nextPageHtmlDataRaw = ajaxData.getString("load_more_widget_html");
                if(!nextPageHtmlDataRaw.isEmpty()) {
                    final Document nextPageData = Jsoup.parse(nextPageHtmlDataRaw, nextPageUrl);
                    nextPageUrl = getNextPageUrl(nextPageData);
                } else {
                    nextPageUrl = "";
                }
            } catch (JSONException e) {
                throw new ParsingException("Could not parse json data for next page", e);
            }
            isAjaxPage = true;
        }
    }

    @Override
    public String getName() throws ParsingException {
        try {
            if (!isAjaxPage) {
                name = doc.select("span[class=\"qualified-channel-title-text\"]").first()
                        .select("a").first().text() + " - " +
                        doc.select("meta[name=title]").first().attr("content");
            }
            return name;
        } catch (Exception e) {
            throw new ParsingException("Could not get playlist name");
        }
    }

    @Override
    public String getAvatarUrl() throws ParsingException {
        try {
            if(!isAjaxPage) {
                avatarUrl = doc.select("div[id=gh-banner] img[class=channel-header-profile-image]").first().attr("src");
                if(avatarUrl.startsWith("//")) {
                    avatarUrl = "https:" + avatarUrl;
                }
            }
            return avatarUrl;
        } catch(Exception e) {
            throw new ParsingException("Could not get playlist Avatar");
        }
    }

    @Override
    public String getBannerUrl() throws ParsingException {
        try {
            if(!isAjaxPage) {
                Element el = doc.select("div[id=\"gh-banner\"] style").first();
                String cssContent = el.html();
                String url = "https:" + Parser.matchGroup1("url\\((.*)\\)", cssContent);
                if (url.contains("s.ytimg.com")) {
                    bannerUrl = null;
                } else {
                    bannerUrl = url.substring(0, url.indexOf(");"));
                }
            }
            return bannerUrl;
        } catch(Exception e) {
            throw new ParsingException("Could not get playlist Banner");
        }
    }

    @Override
    public StreamInfoItemCollector getStreams() throws ParsingException {
        StreamInfoItemCollector collector = getStreamPreviewInfoCollector();
        Element tbody = doc.select("tbody[id=\"pl-load-more-destination\"]").first();
        final YoutubeStreamUrlIdHandler youtubeStreamUrlIdHandler = YoutubeStreamUrlIdHandler.getInstance();
        for(final Element li : tbody.children()) {
            collector.commit(new StreamInfoItemExtractor() {
                @Override
                public AbstractStreamInfo.StreamType getStreamType() throws ParsingException {
                    return AbstractStreamInfo.StreamType.VIDEO_STREAM;
                }

                @Override
                public String getWebPageUrl() throws ParsingException {
                    try {
                        return youtubeStreamUrlIdHandler.getUrl(li.attr("data-video-id"));
                    } catch (Exception e) {
                        throw new ParsingException("Could not get web page url for the video", e);
                    }
                }

                @Override
                public String getTitle() throws ParsingException {
                    try {
                        return li.attr("data-title");
                    } catch (Exception e) {
                        throw new ParsingException("Could not get title", e);
                    }
                }

                @Override
                public int getDuration() throws ParsingException {
                    try {
                        return YoutubeParsingHelper.parseDurationString(
                                li.select("div[class=\"timestamp\"] span").first().text().trim());
                    } catch(Exception e) {
                        if(isLiveStream(li)) {
                            // -1 for no duration
                            return -1;
                        } else {
                            throw new ParsingException("Could not get Duration: " + getTitle(), e);
                        }
                    }
                }

                @Override
                public String getUploader() throws ParsingException {
                    return li.select("div[class=pl-video-owner] a").text();
                }

                @Override
                public String getUploadDate() throws ParsingException {
                    return "";
                }

                @Override
                public long getViewCount() throws ParsingException {
                    return -1;
                }

                @Override
                public String getThumbnailUrl() throws ParsingException {
                    try {
                        return "https://i.ytimg.com/vi/" + youtubeStreamUrlIdHandler.getId(getWebPageUrl()) + "/hqdefault.jpg";
                    } catch (Exception e) {
                        throw new ParsingException("Could not get thumbnail url", e);
                    }
                }

                @Override
                public boolean isAd() throws ParsingException {
                    return false;
                }

                private boolean isLiveStream(Element item) {
                    Element bla = item.select("span[class*=\"yt-badge-live\"]").first();

                    if(bla == null) {
                        // sometimes livestreams dont have badges but sill are live streams
                        // if video time is not available we most likly have an offline livestream
                        if(item.select("span[class*=\"video-time\"]").first() == null) {
                            return true;
                        }
                    }
                    return bla != null;
                }
            });
        }

        return collector;
    }

    @Override
    public boolean hasNextPage() throws ParsingException {
        return nextPageUrl != null && !nextPageUrl.isEmpty();
    }

    private String getNextPageUrl(Document d) throws ParsingException {
        try {
            Element button = d.select("button[class*=\"yt-uix-load-more\"]").first();
            if(button != null) {
                return "https://www.youtube.com" + button.attr("data-uix-load-more-href");
            } else {
                // sometimes channels are simply so small, they don't have a second/next4q page
                return "";
            }
        } catch(Exception e) {
            throw new ParsingException("could not load next page url", e);
        }
    }
}