gvSIG projects pool

package org.gvsig.online.lib.impl;

import org.gvsig.online.lib.api.OnlineDownloader;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.net.URI;

import java.net.URL;

import java.util.ArrayList;

import java.util.List;

import java.util.Map;

import java.util.Objects;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.apache.http.Header;

import org.apache.http.HttpEntityEnclosingRequest;

import org.apache.http.HttpRequest;

import org.apache.http.HttpResponse;

import org.apache.http.NameValuePair;

import org.apache.http.auth.AuthenticationException;

import org.apache.http.client.ClientProtocolException;

import org.apache.http.client.ResponseHandler;

import org.apache.http.client.entity.UrlEncodedFormEntity;

import org.apache.http.client.methods.HttpEntityEnclosingRequestBase;

import org.apache.http.client.methods.HttpGet;

import org.apache.http.client.methods.HttpPost;

import org.apache.http.client.methods.HttpPut;

import org.apache.http.client.methods.HttpUriRequest;

import org.apache.http.entity.ContentType;

import org.apache.http.entity.StringEntity;

import org.apache.http.impl.client.CloseableHttpClient;

import org.apache.http.impl.client.HttpClientBuilder;

import org.apache.http.message.BasicNameValuePair;

import org.gvsig.online.lib.api.OnlineSite;

import org.gvsig.online.lib.api.OnlineUserIdentificationRequester;

import org.gvsig.tools.ToolsLocator;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class OnlineDownloaderImpl implements OnlineDownloader {

//    private OnlineAuthorizationRequester authorizationRequester;

    private final OnlineSite site;

    private static class DownloaderResponseHandler implements ResponseHandler<Object> {

        private int status;

        private File f;

        private HttpResponse response;

        public DownloaderResponseHandler(String urlpath) {

            this.status = 200;

            String fname = FilenameUtils.getBaseName(StringUtils.removeEnd(urlpath,"/"));

            this.f = ToolsLocator.getFoldersManager().getUniqueTemporaryFile(fname);

        }

        @Override

        public Object handleResponse(HttpResponse response) throws ClientProtocolException, IOException {

            this.status=response.getStatusLine().getStatusCode();

            this.response = response;

            FileOutputStream fs = null;

            try {

                fs = new FileOutputStream(f);

                InputStream content = response.getEntity().getContent();

                IOUtils.copy(content, fs);

            } catch (IOException ex) {

                f = null;

                throw ex;

            } catch (Exception ex) {

                f = null;

                throw new IOException(ex);

            } finally {

                IOUtils.closeQuietly(fs);

            }

            return null;

        }

        private int getStatus() {

            return this.status;

        }

        private File getFile() {

            return this.f;

        }

    }

    public OnlineDownloaderImpl(OnlineSite site) {

        this.site = site;

    }

    private DownloaderResponseHandler executeRequest(HttpUriRequest request) throws IOException {

        CloseableHttpClient httpClient = HttpClientBuilder.create()

                .build();

        String authorization = this.site.getCurrentAuthorizationToken();

        if( StringUtils.isNotBlank(authorization) ) {

            request.addHeader("Authorization", authorization);

        }

        DownloaderResponseHandler responseHandler = new DownloaderResponseHandler(request.getURI().toString());

        System.out.println("REQUEST "+request.getMethod()+" "+request.getURI().toString());

        for (Header header : request.getAllHeaders()) {

            System.out.println(header.getName()+": "+header.getValue());

        }

        if( request instanceof HttpEntityEnclosingRequestBase ) {

            InputStream content = ((HttpEntityEnclosingRequestBase) request).getEntity().getContent();

            System.out.println("BODY data "+(StringUtils.join(IOUtils.readLines(content,"utf-8"),"\n")));                

        }

        httpClient.execute(request,responseHandler);

        System.out.println("RESPONSE CODE "+responseHandler.getStatus());

        return responseHandler;

    }

    @SuppressWarnings("Convert2Lambda")

    @Override

    public File get(URL url) throws IOException {

        try {

            int numretries = 3;

            for (int retries = 0; retries < numretries; retries++) {            

                HttpGet request = new HttpGet(url.toURI());

                request.setHeader("User-Agent","gvSIG-desktop");

                request.setHeader("Referer","http://www.gvsig.com");

                DownloaderResponseHandler responseHandler = executeRequest(request);

                switch(responseHandler.getStatus()) {

                    case 200: 

                        return responseHandler.getFile();

                    case 401: // 401 Authorization Required

                        if( retries < numretries-1 ) {

                            authorize();

                        }

                        break;

                    case 500:

                        throw new IOException("Can't download '"+Objects.toString(url)+"' 500 Server error");

                }

            }

            throw new IOException("Can't download '"+Objects.toString(url)+"'");

        } catch (IOException ex) {

            throw ex;

        } catch (Exception ex) {

            throw new IOException("Can't download '"+Objects.toString(url)+"'", ex);

        }

    }

    @Override

    public File post_json(URL url, String json) throws IOExceptionWithStatus {

        return post_json(url, json, "POST");

    }

    @Override

    public File put_json(URL url, String json) throws IOExceptionWithStatus {

        return post_json(url, json, "PUT");

    }

    @SuppressWarnings("Convert2Lambda")

    private File post_json(URL url, String json, String method) throws IOExceptionWithStatus {

        int status = 500;

        try {

            int numretries = 3;

            for (int retries = 0; retries < numretries; retries++) {            

                HttpEntityEnclosingRequestBase request;

                switch (method.toUpperCase()) {

                    case "PUT":

                        request = new HttpPut(url.toURI());

                        break;

                    case "POST":

                    default:

                        request = new HttpPost(url.toURI());

                        break;

                }

                request.setEntity(new StringEntity(json, ContentType.APPLICATION_JSON));

                request.setHeader("Content-type","application/json");                

                DownloaderResponseHandler responseHandler = executeRequest(request);

                switch(status = responseHandler.getStatus()) {

                    case 200: 

                        return responseHandler.getFile();

                    case 401: // 401 Authorization Required

                        if( retries < numretries-1 ) {

                            authorize();

                        }

                        break;

                    case 409: // 409 Version Conflict

                        retries = numretries;

                        break;

                    case 500:

                        throw new IOExceptionWithStatus(500, "Can't download '"+Objects.toString(url)+"' 500 Server error");

                }

            }

            throw new IOExceptionWithStatus(status, "Can't download '"+Objects.toString(url)+"' too many retries");

        } catch (IOExceptionWithStatus ex) {

            throw ex;

        } catch (Exception ex) {

            throw new IOExceptionWithStatus(status,"Can't download '"+Objects.toString(url)+"'", ex);

        }

    }

    @Override

    public File post_form(URL url, Map<String,String>data) throws IOException {

        try {

            URI uri = url.toURI();

            if( !uri.toString().endsWith("/") ) {

                uri = new URI(url.toString()+"/");

            }

            int numretries = 3;

            for (int retries = 0; retries < numretries; retries++) {            

                HttpPost request = new HttpPost(uri);

                final List<NameValuePair> params = new ArrayList<>();

                for (Map.Entry<String, String> entry : data.entrySet()) {

                    String key = entry.getKey();

                    String value = entry.getValue();

                    params.add(new BasicNameValuePair(key, value));

                }

                UrlEncodedFormEntity formdata = new UrlEncodedFormEntity(params);

                request.setEntity(formdata);

                request.setHeader("Content-type","application/x-www-form-urlencoded");

                DownloaderResponseHandler responseHandler = executeRequest(request);

                switch(responseHandler.getStatus()) {

                    case 200: 

                        return responseHandler.getFile();

                    case 401: // 401 Authorization Required

                        if( retries < numretries-1 ) {

                            authorize();

                        }

                        break;

                    case 500:

                        throw new IOException("Can't download '"+Objects.toString(url)+"' 500 Server error");

                }

            }

            throw new IOException("Can't download '"+Objects.toString(url)+"' too many retries");

        } catch (IOException ex) {

            throw ex;

        } catch (Exception ex) {

            throw new IOException("Can't download '"+Objects.toString(url)+"'", ex);

        }

    }

    public void authorize() throws Exception {

        try {

            System.out.println("AUTHORIZATION REQUIRED");

            OnlineUserIdentificationRequester identificationRequester = this.site.getUserIdentificationRequester();

            if( identificationRequester == null ) {

                throw new AuthenticationException("Not identification requester configured");

            }

            if( !identificationRequester.requestIdentification() ) {

                return;

            }

            String authorization = identificationRequester.getAuthorization();

            this.site.setCurrentUserCode(identificationRequester.getUserId());

            this.site.setCurrentAuthorizationToken(authorization);

            System.out.println("AUTH TOKEN: "+authorization);

        } catch (Exception ex) {

//            LOGGER.warn("Can't authorize", ex);

        }

    }

//    

//    public static void main0(String[] args) throws Exception {

//        new DefaultLibrariesInitializer().fullInitialize();

//        

//        OnlineDownloaderImpl downloader = new OnlineDownloaderImpl();

//        File f = downloader.get(new URL("https://devel.gvsigonline.com/gvsigonline/api/v1/projects"));

//        System.out.println(f);

//    }

//

//    public static void main1(String[] args) throws Exception {

//        new DefaultLibrariesInitializer().fullInitialize();

//        

//        Properties props = new Properties();

//        props.load(FileUtils.openInputStream(FileUtils.getFile(System.getProperty("user.home"),"onlineclient.properties")));

//        OnlineManager manager = OnlineLocator.getOnlineManager();

//        manager.setUserIdentificationRequester(new AbstractOnlineUserIdentificationRequester(

//                props.getProperty("online_user"),

//                props.getProperty("online_password"),

//                null

//            ) {

//            @Override

//            public boolean requestIdentification(String urlbase) {

//                return true;

//            }

//        });

//        File f;

//        String s;

//        OnlineSiteImpl site = (OnlineSiteImpl) manager.connectSite(new URL("https://devel.gvsigonline.com/gvsigonline"));

//        OnlineDownloader downloader = site.getDownloader();

////        f = downloader.get(new URL("https://devel.gvsigonline.com/gvsigonline/api/v1/projects"));

////        System.out.println(f);

////        s = FileUtils.readFileToString(f, StandardCharsets.UTF_8);

////        System.out.println(Json.createObject(s).toString());

//        f = downloader.get(new URL("https://devel.gvsigonline.com/gvsigonline/api/v1/projects/83/layers/"));

//        System.out.println(f);

//        s = FileUtils.readFileToString(f, StandardCharsets.UTF_8);

//        System.out.println(Json.createObject(s).toString());

//        System.out.println("Fin");

//    }

//    

//    public static void main(String[] args) throws Exception {

//        new DefaultLibrariesInitializer().fullInitialize();

//        

//        OnlineDownloaderImpl downloader = new OnlineDownloaderImpl();

//        File f = downloader.get(new URL("https://tile.openstreetmap.org/4/8/8.png"));

//        System.out.println(f);

//    }

}