com.qiniu.process.qai.ImageCensor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qsuits Show documentation
qiniu-suits is a efficient tools for qiniu api implemented by java8.
There is a newer version: 8.4.8
package com.qiniu.process.qai;

import com.google.gson.JsonObject;
import com.qiniu.process.Base;
import com.qiniu.storage.Configuration;
import com.qiniu.util.Auth;
import com.qiniu.util.CloudApiUtils;
import com.qiniu.util.RequestUtils;

import java.io.IOException;
import java.util.Map;

public class ImageCensor extends Base> {

    private String protocol;
    private String domain;
    private String urlIndex;
    private String suffixOrQuery;
    private boolean useQuery;
    private JsonObject paramsJson;
    private Configuration configuration;
    private CensorManager censorManager;

    public ImageCensor(String accessKey, String secretKey, Configuration configuration, String protocol, String domain,
                       String urlIndex, String suffixOrQuery, String[] scenes)
            throws IOException {
        super("imagecensor", accessKey, secretKey, null);
        set(configuration, protocol, domain, urlIndex, suffixOrQuery, scenes);
        Auth auth = Auth.create(accessKey, secretKey);
        CloudApiUtils.checkQiniu(auth);
        censorManager = new CensorManager(auth, configuration.clone());
    }

    public ImageCensor(String accessKey, String secretKey, Configuration configuration, String protocol, String domain,
                       String urlIndex, String suffixOrQuery, String[] scenes, String savePath, int saveIndex) throws IOException {
        super("imagecensor", accessKey, secretKey, null, savePath, saveIndex);
        set(configuration, protocol, domain, urlIndex, suffixOrQuery, scenes);
        Auth auth = Auth.create(accessKey, secretKey);
        CloudApiUtils.checkQiniu(auth);
        censorManager = new CensorManager(auth, configuration.clone());
    }

    public ImageCensor(String accesskey, String secretKey, Configuration configuration, String protocol, String domain,
                       String urlIndex, String suffixOrQuery, String[] scenes, String savePath) throws IOException {
        this(accesskey, secretKey, configuration, protocol, domain, urlIndex, suffixOrQuery, scenes, savePath, 0);
    }

    private void set(Configuration configuration, String protocol, String domain, String urlIndex, String suffixOrQuery,
                     String[] scenes) throws IOException {
        this.configuration = configuration;
        if (urlIndex == null || "".equals(urlIndex)) {
            this.urlIndex = "url";
            if (domain == null || "".equals(domain)) {
                throw new IOException("please set one of domain and url-index.");
            } else {
                this.protocol = protocol == null || !protocol.matches("(http|https)") ? "http" : protocol;
                RequestUtils.lookUpFirstIpFromHost(domain);
                this.domain = domain;
            }
        } else {
            this.urlIndex = urlIndex;
        }
        this.suffixOrQuery = suffixOrQuery == null ? "" : suffixOrQuery;
        useQuery = !"".equals(this.suffixOrQuery);
        this.paramsJson = new JsonObject();
        paramsJson.add("scenes", CensorManager.getScenes(scenes));
    }

    public ImageCensor clone() throws CloneNotSupportedException {
        ImageCensor videoCensor = (ImageCensor)super.clone();
        videoCensor.censorManager = new CensorManager(Auth.create(accessId, secretKey), configuration.clone());
        return videoCensor;
    }

    @Override
    protected String resultInfo(Map line) {
        String key = line.get("key");
        return (key == null ? "\t" : key + "\t") + line.get(urlIndex);
    }

    @Override
    protected String singleResult(Map line) throws Exception {
        String url = line.get(urlIndex);
        String key = line.get("key");
        if (url == null || "".equals(url)) {
            if (key == null) throw new IOException("key is not exists or empty in " + line);
            url = protocol + "://" + domain + "/" + key.replace("\\?", "%3f") + suffixOrQuery;
            line.put(urlIndex, url);
            return key + "\t" + url + "\t" + censorManager.doImageCensor(url, paramsJson);
        } else if (useQuery) {
            url = url + suffixOrQuery;
            line.put(urlIndex, url);
        }
        return (key == null ? "\t" : key + "\t") + url + "\t" + censorManager.doImageCensor(url, paramsJson);
    }

    @Override
    public void closeResource() {
        super.closeResource();
        protocol = null;
        domain = null;
        urlIndex = null;
        paramsJson = null;
        configuration = null;
        censorManager = null;
    }
}