Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2023 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.downloader.lib.impl;

import java.io.File;

import java.io.IOException;

import java.net.ConnectException;

import java.net.URL;

import java.net.UnknownHostException;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Objects;

import org.apache.commons.lang3.StringUtils;

import org.apache.http.entity.ContentType;

import org.gvsig.compat.net.ICancellable;

import org.gvsig.compat.se.net.downloader.se.SEDownloader;

import org.gvsig.downloader.DownloaderAuthenticationConfig;

import org.gvsig.downloader.DownloaderAuthenticationFactory;

import org.gvsig.downloader.DownloaderCredentials;

import org.gvsig.downloader.DownloaderManager;

import static org.gvsig.downloader.DownloaderManager.urlStartsWith;

import org.gvsig.json.Json;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("UseSpecificCatch")

public class DownloaderManagerImpl 

        extends SEDownloader

        implements DownloaderManager {

    private static final Logger LOG = LoggerFactory.getLogger(DownloaderManagerImpl.class);

    private final List<DownloaderCredentials> credentials;

    private final Map<String,DownloaderAuthenticationFactory> authenticationTypes;

    private final List<DownloaderAuthenticationConfig> authenticationConfigurationServices;

    public DownloaderManagerImpl() {

        super();

        this.credentials = new ArrayList<>();

        this.authenticationTypes = new HashMap<>();

        this.authenticationConfigurationServices = new ArrayList<>();

    }

    @Override

    protected Runnable createDownloaderTask(SEDownloader downloader, URL url, String data, File target, Object groupID, int maxbytes) {

        SEAuthDownloaderTask t = new SEAuthDownloaderTask(

            downloader, 

            url, 

            data, 

            target, 

            groupID);

        t.setMaxbytes(maxbytes);

        return t;

    }

    @Override

    protected Runnable createDownloaderTask(SEDownloader downloader, URL url, String method, ContentType contenttype, String data, File target, Object groupID, int maxbytes) {

        SEAuthDownloaderTask t = new SEAuthDownloaderTask(

            downloader, 

            url, 

            method,

            contenttype,

            data, 

            target, 

            groupID);

        t.setMaxbytes(maxbytes);

        return t;

    }

    @Override

    public void addOrReplaceCredentials(DownloaderCredentials credentials) {

        if( credentials == null ) {

            LOG.warn("Invalid credentials (null).");

            return;

        }

        for (int i = 0; i < this.credentials.size(); i++) {

            DownloaderCredentials x = this.credentials.get(i);

            if(DownloaderManager.areSameURLs(credentials.getServiceUrl(), x.getServiceUrl())){

                this.credentials.set(i, credentials);

                return;

            }

        }

        this.credentials.add(credentials);

    }

    @Override

    public void removeCredentials(DownloaderCredentials credentials) {

        if( credentials == null ) {

            return;

        }

        Iterator<DownloaderCredentials> it = this.credentials.iterator();

        while (it.hasNext()) {

            DownloaderCredentials next = it.next();

            if(DownloaderManager.areSameURLs(credentials.getServiceUrl(), next.getServiceUrl())){

                it.remove();

            }

        }

    }

    @Override

    public void removeCredentials(String serviceUrl) {

        if( StringUtils.isBlank(serviceUrl) ) {

            return;

        }

        Iterator<DownloaderCredentials> it = this.credentials.iterator();

        while (it.hasNext()) {

            DownloaderCredentials next = it.next();

            if(DownloaderManager.areSameURLs(serviceUrl, next.getServiceUrl())){

                it.remove();

            }

        }

    }

    @Override

    public DownloaderCredentials getCredentials(URL url) {

        return this.getCredentials(url.toExternalForm());

    }

    @Override

    public DownloaderCredentials getCredentials(String url) {

        if (StringUtils.isBlank(url)) {

            return null;

        }

        for (DownloaderCredentials theCredentials : this.credentials) {

            if(urlStartsWith(url, theCredentials.getServiceUrl())){

                return theCredentials;

            }

        }

        DownloaderAuthenticationConfig config = this.getAuthenticationConfigurationService(url);

        if(config == null){

            return null;

        }

        for (DownloaderCredentials theCredentials : this.credentials) {

            DownloaderCredentials x = theCredentials.createCredentials(config);

            if(x != null) {

                this.addOrReplaceCredentials(x);

                return x;

            }

        }

        return null;

    }

    @Override

    public void registerAuthenticationConfigurationService(DownloaderAuthenticationConfig config ) {

        if( config == null ) {

            LOG.warn("Invalid config (null).");

            return;

        }

        for (int i = 0; i < this.authenticationConfigurationServices.size(); i++) {

            DownloaderAuthenticationConfig x = this.authenticationConfigurationServices.get(i);

            if(DownloaderManager.areSameURLs(config.getServiceUrl(), x.getServiceUrl())){

                this.authenticationConfigurationServices.set(i, config);

                return;

            }

        }

        this.authenticationConfigurationServices.add(config);

    }

    @Override

    public void removeAuthenticationConfigurationService(DownloaderAuthenticationConfig config) {

        if( config == null ) {

            return;

        }

        Iterator<DownloaderAuthenticationConfig> it = this.authenticationConfigurationServices.iterator();

        while (it.hasNext()) {

            DownloaderAuthenticationConfig next = it.next();

            if(DownloaderManager.areSameURLs(config.getServiceUrl(), next.getServiceUrl())){

                it.remove();

            }

        }

    }

    @Override

    public void registerAuthenticationType(DownloaderAuthenticationFactory factory) {

        this.authenticationTypes.put(factory.getProviderName().toLowerCase(), factory);

    }

    @Override

    public DownloaderAuthenticationConfig getAuthenticationConfigurationService(String baseUrl) {

        if (StringUtils.isBlank(baseUrl)) {

            return null;

        }

        for (DownloaderAuthenticationConfig theConfig : this.authenticationConfigurationServices) {

            if (urlStartsWith(baseUrl, theConfig.getServiceUrl())) {

                return theConfig;

            }

        }

        return null;

    }

    @Override

    public Collection<DownloaderAuthenticationFactory> getAuthenticationTypes() {

        return this.authenticationTypes.values();

    }

    @Override

    public Collection<DownloaderAuthenticationConfig> getAuthenticationConfigurationServices() {

        return this.authenticationConfigurationServices;

    }

    @Override

    public Collection<DownloaderCredentials> getCredentials() {

        return this.credentials; //.values();

    }

    public DownloaderAuthenticationConfig requestAutenticationConfig(URL url) { // TODO: falta por implementar

        return null;

    }

    @Override

    public DownloaderAuthenticationFactory getDownloaderAuthenticationFactory(String providerName) {

        return this.authenticationTypes.get(providerName.toLowerCase());

    }

    @Override

    public void registerAuthenticationConfigurationService(String config) {

        if(StringUtils.isNotBlank(config)) {

            DownloaderAuthenticationConfig downloaderConfig = (DownloaderAuthenticationConfig) Json.toObject(config);

            if(downloaderConfig != null){

                this.registerAuthenticationConfigurationService(downloaderConfig);

            }

        }

    }

    @Override

    public String getAuthenticationConfigurationServiceAsString(String baseUrl) {

        DownloaderAuthenticationConfig downloaderConfig = this.getAuthenticationConfigurationService(baseUrl);

        if(downloaderConfig != null){

            return downloaderConfig.toJson().toString();

        }

        return null;

    }

    @Override

    public synchronized File downloadFile(URL url, String method, ContentType contenttype, String data, String name, ICancellable cancel, int maxbytes) throws IOException, ConnectException, UnknownHostException {

        return downloadFile(url, method, contenttype, data, name, cancel, maxbytes, true);

    }

    @Override

    public synchronized File downloadFile(URL url, String method, ContentType contenttype, String data, String name, ICancellable cancel, int maxbytes, boolean useCache) throws IOException, ConnectException, UnknownHostException {

        LOG.info("Download file " + Objects.toString(url));

        if (useCache) {

            File target = getPreviousDownloadedURL(url, data);

            if (target != null) {

                LOG.info("CACHED "+url.toString() + " at '" + target.getAbsolutePath() + "'");

                return target;

            }

        }

        File target = new File(name);

        if (!target.isAbsolute()){

            target = getUniqueTemporaryFile(name);

        }

        if(cancel == null) {

            cancel = ICancellable.DUMB;

        }

        Runnable task = this.createDownloaderTask(this, url, method, contenttype, data, target, cancel.getID(), maxbytes);

        task.run();

        return target;

    }

    public void propagateCredentials(String sourceServiceUrl, Collection<String> targetServicesUrls) {

        if(StringUtils.isBlank(sourceServiceUrl)){

            return;

        }

        DownloaderAuthenticationConfig sourceConfig = this.getAuthenticationConfigurationService(sourceServiceUrl);

        if(sourceConfig == null){

            return;

        }

        DownloaderCredentials sourceCredentials = this.getCredentials(sourceServiceUrl);

        for (String targetServiceUrl : targetServicesUrls) {

            if(StringUtils.isBlank(targetServiceUrl)){

                continue;

            }

            DownloaderAuthenticationConfig targetConfig = this.getAuthenticationConfigurationService(targetServiceUrl);

            if(targetConfig == null) {

                targetConfig = sourceConfig.createAuthenticationConfig(targetServiceUrl);

                if(targetConfig != null) {

                    this.registerAuthenticationConfigurationService(targetConfig);

                }

            }

            if(sourceCredentials != null) {

                DownloaderCredentials targetCredentials = this.getCredentials(targetServiceUrl);

                if(targetCredentials == null){

                    targetCredentials = sourceCredentials.createCredentials(targetServiceUrl);

                    if(targetCredentials != null) {

                        this.addOrReplaceCredentials(targetCredentials);

                    }

                }

            }

        }

    }

}