package com.rarchives.ripme.ripper.rippers;

import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.log4j.Logger;
import org.json.JSONArray;
import org.json.JSONObject;
import org.jsoup.Connection.Method;
import org.jsoup.Jsoup;

import com.rarchives.ripme.ripper.AlbumRipper;
import com.rarchives.ripme.ui.RipStatusMessage.STATUS;
import com.rarchives.ripme.utils.Utils;

public class ImagestashRipper extends AlbumRipper {

    private static final String DOMAIN = "imagestash.org",
                                HOST   = "imagestash";
    private static final Logger logger = Logger.getLogger(ImagestashRipper.class);

    public ImagestashRipper(URL url) throws IOException {
        super(url);
    }

    public boolean canRip(URL url) {
        System.err.println(url.getHost() +"/"+DOMAIN);
        return url.getHost().equals(DOMAIN);
    }

    public URL sanitizeURL(URL url) throws MalformedURLException {
        return url;
    }

    @Override
    public void rip() throws IOException {
        // Given URL: https://imagestash.org/tag/everydayuncensor
        // GID:       "everydayuncensor"
        // JSON URL:  https://imagestash.org/images?tags=everydayuncensor&page=1
        String baseURL = "https://imagestash.org/images?tags=" + getGID(this.url);
        int page = 0, index = 0;
        while (true) {
            page++;
            String nextURL = baseURL + "&page=" + page;
            logger.info("[ ] Retrieving " + nextURL);
            sendUpdate(STATUS.LOADING_RESOURCE, nextURL);
            String jsonText = Jsoup.connect(nextURL)
                                   .ignoreContentType(true)
                                   .userAgent(USER_AGENT)
                                   .method(Method.GET)
                                   .execute()
                                   .body();
            logger.info(jsonText);
            JSONObject json = new JSONObject(jsonText);
            JSONArray images = json.getJSONArray("images");
            for (int i = 0; i < images.length(); i++) {
                JSONObject image = images.getJSONObject(i);
                String imageURL = image.getString("src");
                if (imageURL.startsWith("/")) {
                    imageURL = "http://imagestash.org" + imageURL;
                }
                index += 1;
                String prefix = "";
                if (Utils.getConfigBoolean("download.save_order", true)) {
                    prefix = String.format("%03d_", index);
                }
                addURLToDownload(new URL(imageURL), prefix);
            }
            // Check if there are more images to fetch
            int count  = json.getInt("count"),
                offset = json.getInt("offset"),
                total  = json.getInt("total");
            if (count + offset >= total || images.length() == 0) {
                break;
            }
            // Wait a bit
            try {
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                logger.error("Interrupted while waiting to load next page", e);
                break;
            }
        }
        waitForThreads();
    }

    @Override
    public String getHost() {
        return HOST;
    }

    @Override
    public String getGID(URL url) throws MalformedURLException {
        Pattern p = Pattern.compile("^.*imagestash.org/tag/([a-zA-Z0-9\\-_]+)$");
        Matcher m = p.matcher(url.toExternalForm());
        if (m.matches()) {
            return m.group(1);
        }
        throw new MalformedURLException(
                "Expected imagestash.org tag formats: "
                        + "imagestash.org/tag/tagname"
                        + " Got: " + url);
    }
}