package fr.ifremer.quadrige2.core.service.extraction;
   
   /*-
    * #%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 fr.ifremer.quadrige2.core.exception.Quadrige2BusinessException;
  import fr.ifremer.quadrige2.core.exception.Quadrige2TechnicalException;
  import fr.ifremer.quadrige2.core.config.Quadrige2Configuration;
  import fr.ifremer.quadrige2.core.dao.technical.http.HttpHelper;
  import fr.ifremer.quadrige2.core.security.AuthenticationInfo;
  import fr.ifremer.quadrige2.core.service.RestServiceSupport;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.apache.http.client.CredentialsProvider;
  import org.apache.http.client.methods.HttpPost;
  import org.apache.http.entity.ContentType;
  import org.apache.http.entity.mime.MultipartEntityBuilder;
  import org.apache.http.entity.mime.content.StringBody;
  import org.apache.http.impl.client.CloseableHttpClient;
  import org.apache.http.impl.client.HttpClients;
  import org.apache.http.protocol.HTTP;
  import org.nuiton.i18n.I18n;
  import org.nuiton.jaxx.application.type.ApplicationProgressionModel;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.context.annotation.Lazy;
  import org.springframework.stereotype.Service;
  
  import java.io.File;
  import java.io.IOException;
  import java.nio.charset.Charset;
  
  /**
   * Rest client service for extraction
   * 
   * Created by Ludovic on 08/02/2016.
   */
  @Service("extractionRestClientService")
  @Lazy
  public class ExtractionRestClientServiceImpl extends RestServiceSupport implements ExtractionRestClientService {
  
  	private final static Log log = LogFactory.getLog(ExtractionRestClientServiceImpl.class);
  
  	private static final String URL_EXPORT_UPLOAD = "/service/extraction/upload";
  	private static final String URL_EXPORT_UPLOAD_POST_PARAM_FILE = "file";
  	private static final String URL_EXPORT_UPLOAD_POST_PARAM_FILENAME = "filename";
  	private static final String URL_EXPORT_UPLOAD_POST_PARAM_TYPE = "type";
  
  	/**
  	 * <p>
  	 * Constructor for ExtractionRestClientServiceImpl.
  	 * </p>
  	 * 
  	 * @param config
  	 *            a {@link fr.ifremer.quadrige2.core.config.Quadrige2Configuration} object.
  	 */
  	@Autowired
  	public ExtractionRestClientServiceImpl(Quadrige2Configuration config) {
  		super(config);
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	public void uploadExtractionFile(AuthenticationInfo authenticationInfo, File extractionFile, String extractionType,
  			ApplicationProgressionModel progressionModel) throws Quadrige2TechnicalException {
  
  		if (!extractionFile.exists()) {
  			throw new Quadrige2BusinessException(I18n.t("quadrige2.error.file.not.exists", extractionFile.getAbsolutePath()));
  		}
  
  		if (log.isDebugEnabled()) {
  			log.debug(String.format("file to transfer to server: %s", extractionFile.getAbsolutePath()));
  		}
  
  		// get credentials provider
  		CredentialsProvider credsProvider = getCredentialsProvider(authenticationInfo);
  
  		try (CloseableHttpClient httpClient = HttpClients.custom().setDefaultRequestConfig(getRequestConfig())
  				.setDefaultCredentialsProvider(credsProvider)
  				.build()) {
 
 			HttpPost uploadHttpPost = new HttpPost(getPathUri(URL_EXPORT_UPLOAD));
 
 			MultipartEntityBuilder builder = MultipartEntityBuilder.create();
 			// builder.setMode(HttpMultipartMode.BROWSER_COMPATIBLE);
 			// add type
 			builder.addPart(URL_EXPORT_UPLOAD_POST_PARAM_TYPE, new StringBody(extractionType, ContentType.TEXT_PLAIN));
 			// add file
 			builder.addPart(URL_EXPORT_UPLOAD_POST_PARAM_FILE, new ProgressFileBody(extractionFile, progressionModel));
 			// add file name
 			builder.addPart(URL_EXPORT_UPLOAD_POST_PARAM_FILENAME,
 					new StringBody(extractionFile.getName(), ContentType.create("text/plain", Charset.defaultCharset())));
 
 			uploadHttpPost.setEntity(builder.build());
 			uploadHttpPost.addHeader(HTTP.EXPECT_DIRECTIVE, HTTP.EXPECT_CONTINUE);
 
 			// execute request
 			HttpHelper.executeRequest(httpClient, uploadHttpPost);
 
 		} catch (Quadrige2TechnicalException e) {
 			log.error(I18n.t("quadrige2.error.remote.upload.failed", e.getMessage()), e);
 			throw e;
 		} catch (IOException e) {
 			log.error(I18n.t("quadrige2.error.remote.upload.failed", e.getMessage()), e);
 			throw new Quadrige2TechnicalException(e);
 		}
 
 	}
 
 }