package fr.ifremer.quadrige2.core.service;
   
   /*-
    * #%L
    * Quadrige2 Core :: Quadrige2 Client Core
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero 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 Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  import com.google.gson.Gson;
  import fr.ifremer.quadrige2.core.dao.technical.gson.Gsons;
  import fr.ifremer.quadrige2.core.dao.technical.http.HttpHelper;
  import fr.ifremer.quadrige2.core.exception.Quadrige2TechnicalException;
  import fr.ifremer.quadrige2.core.config.Quadrige2Configuration;
  import fr.ifremer.quadrige2.core.security.AuthenticationInfo;
  import org.apache.http.auth.AuthScope;
  import org.apache.http.auth.UsernamePasswordCredentials;
  import org.apache.http.client.CredentialsProvider;
  import org.apache.http.client.config.RequestConfig;
  import org.apache.http.entity.mime.content.FileBody;
  import org.apache.http.impl.client.BasicCredentialsProvider;
  import org.apache.http.util.Args;
  import org.nuiton.jaxx.application.type.ApplicationProgressionModel;
  
  import java.io.*;
  import java.net.URI;
  import java.net.URISyntaxException;
  import java.net.URL;
  
  import static org.nuiton.i18n.I18n.t;
  
  /**
   * Support for REST services
   * <p/>
   * Created by Ludovic on 08/02/2016.
   */
  public class RestServiceSupport {
  
  	protected final Quadrige2Configuration config;
  
  	private Gson gson = null;
  
  	private RequestConfig requestConfig = null;
  
  	/**
  	 * <p>
  	 * Constructor for RestServiceSupport.
  	 * </p>
  	 * 
  	 * @param config
  	 *            a {@link fr.ifremer.quadrige2.core.config.Quadrige2Configuration} object.
  	 */
  	public RestServiceSupport(Quadrige2Configuration config) {
  		this.config = config;
  	}
  
  	/**
  	 * <p>
  	 * Getter for the field <code>gson</code>.
  	 * </p>
  	 * 
  	 * @return a {@link com.google.gson.Gson} object.
  	 */
  	protected Gson getGson() {
  		if (gson == null) {
  			gson = Gsons
  					.newBuilder(Quadrige2Configuration.getInstance().getDbTimezone())
  					.create();
  		}
  		return gson;
  	}
  
  	/**
  	 * <p>
  	 * Getter for the field <code>requestConfig</code>.
  	 * </p>
  	 * 
  	 * @return a {@link org.apache.http.client.config.RequestConfig} object.
  	 */
  	protected RequestConfig getRequestConfig() {
  		if (requestConfig == null) {
  			Integer timeOut = config.getSynchronizationSiteTimeout();
  
 			// build request config for timeout
 			requestConfig = RequestConfig.custom().setSocketTimeout(timeOut).setConnectTimeout(timeOut).build();
 		}
 
 		return requestConfig;
 	}
 
 	/**
 	 * <p>
 	 * getCredentialsProvider.
 	 * </p>
 	 * 
 	 * @param authenticationInfo
 	 *            a {@link fr.ifremer.quadrige2.core.security.AuthenticationInfo} object.
 	 * @return a {@link org.apache.http.client.CredentialsProvider} object.
 	 */
 	protected CredentialsProvider getCredentialsProvider(AuthenticationInfo authenticationInfo) {
 		// For anonymous connection
 		if (authenticationInfo == null) {
 			return null;
 		}
 
 		URL url = getRemoteUrl();
 		// build credentials information
 		CredentialsProvider result = new BasicCredentialsProvider();
 		result.setCredentials(new AuthScope(url.getHost(), url.getPort()),
 				new UsernamePasswordCredentials(authenticationInfo.getLogin(), authenticationInfo.getPassword()));
 		return result;
 	}
 
 	/**
 	 * <p>
 	 * getRemoteUrl.
 	 * </p>
 	 * 
 	 * @return a {@link java.net.URL} object.
 	 */
 	protected URL getRemoteUrl() {
 		return config.getSynchronizationSiteUrl();
 	}
 
 	/**
 	 * <p>
 	 * getPathUri.
 	 * </p>
 	 * 
 	 * @param subPath
 	 *            a {@link java.lang.String} object.
 	 * @return a {@link java.net.URI} object.
 	 */
 	protected URI getPathUri(String subPath) {
 		try {
 			return HttpHelper.getAppendedPath(config.getSynchronizationSiteUrl().toURI(), subPath);
 		} catch (URISyntaxException e) {
 			throw new Quadrige2TechnicalException(t("quadrige2.error.remote.synchronizationSiteUrl", e.getMessage()), e);
 		}
 	}
 
 	protected class ProgressFileBody extends FileBody {
 
 		private final ApplicationProgressionModel progressionModel;
 
 		public ProgressFileBody(File file, ApplicationProgressionModel progressionModel) {
 			super(file);
 			this.progressionModel = progressionModel;
 		}
 
 		@Override
 		public long getContentLength() {
 			long size = super.getContentLength();
 			progressionModel.setTotal((int) size);
 			return size;
 		}
 
 		@Override
 		public void writeTo(OutputStream out) throws IOException {
 			Args.notNull(out, "Output stream");
 			try (InputStream in = new BufferedInputStream(new FileInputStream(getFile()))) {
 				final byte[] tmp = new byte[4096];
 				int l;
 				long count = 0;
 				while ((l = in.read(tmp)) != -1) {
 					out.write(tmp, 0, l);
 					count += l;
 					progressionModel.setCurrent((int) count);
 				}
 				out.flush();
 			}
 		}
 	}
 }