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.net.URL;

import java.util.Collection;

import java.util.HashMap;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;

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 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 Map<String,DownloaderCredentials> credentials;

    private final Map<String,DownloaderAuthenticationFactory> authenticationTypes;

    private final Map<String,DownloaderAuthenticationConfig> authenticationConfigurationServices;

    public DownloaderManagerImpl() {

        super();

        this.credentials = new HashMap<>();

        this.authenticationTypes = new HashMap<>();

        this.authenticationConfigurationServices = new HashMap<>();

    }

    @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;

    }

    private String getUrlWithoutQuery(URL url) {

        try {

            URL x = new URL(url.getProtocol(), url.getHost(), url.getPath());

            return x.toExternalForm();

        } catch(Exception ex) {

            return null;

        }

    }

    private String getUrlWithoutQuery(String url) {

        try {

            return this.getUrlWithoutQuery(new URL(url));

        } catch(Exception ex) {

            return null;

        }

    }

    @Override

    public void addOrReplaceCredentials(DownloaderCredentials credentials) {

        if( credentials == null ) {

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

            return;

        }

        String url = this.getUrlWithoutQuery(credentials.getServiceUrl());

        if( url == null ) {

            LOG.warn("Invalid URL '"+credentials.getServiceUrl()+"'");

            return;

        }

        this.credentials.put(url, credentials);

    }

    @Override

    public void removeCredentials(DownloaderCredentials credentials) {

        if( credentials == null ) {

            return;

        }

        String x = credentials.getServiceUrl();

        if( x == null ) {

            return;

        }

        this.credentials.remove(x);

    }

    @Override

    public DownloaderCredentials getCredentials(URL url) {

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

    }

    @Override

    public DownloaderCredentials getCredentials(String url) {

        for (DownloaderCredentials theCredentials : this.credentials.values()) {

            if( StringUtils.startsWith(url, theCredentials.getServiceUrl()) ) {

                return theCredentials;

            }

        }

        return null;

    }

    @Override

    public void registerAuthenticationConfigurationService(DownloaderAuthenticationConfig config ) {

        authenticationConfigurationServices.put(config.getKey(), config);

    }

    @Override

    public void removeAuthenticationConfigurationService(DownloaderAuthenticationConfig config) {

        authenticationConfigurationServices.remove(config.getKey());

    }

    @Override

    public void registerAuthenticationType(DownloaderAuthenticationFactory factory) {

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

    }

    @Override

    public DownloaderAuthenticationConfig getAuthenticationConfigurationService(String baseUrl) {

        for (DownloaderAuthenticationConfig config : this.authenticationConfigurationServices.values()) {

            if( StringUtils.startsWith(baseUrl, config.getServiceUrl()) ) {

                return config;

            }

        }

        return null;

    }

    @Override

    public Collection<DownloaderAuthenticationFactory> getAuthenticationTypes() {

        return this.authenticationTypes.values();

    }

    @Override

    public Collection<DownloaderAuthenticationConfig> getAuthenticationConfigurationServices() {

        return this.authenticationConfigurationServices.values();

    }

    @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);

    }

}