package fr.ifremer.quadrige3.core.service.http;
   
   /*-
    * #%L
    * Quadrige3 Core :: Client API
    * %%
    * Copyright (C) 2017 - 2019 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.quadrige3.core.ProgressionCoreModel;
  import fr.ifremer.quadrige3.core.dao.technical.Files;
  import org.apache.http.entity.mime.content.FileBody;
  import org.apache.http.util.Args;
  
  import java.io.*;
  
  /**
   * @author peck7 on 03/04/2019.
   */
  public class ProgressFileBody extends FileBody {
  
      private final ProgressionCoreModel progressionModel;
  
      public ProgressFileBody(File file, ProgressionCoreModel progressionModel) {
          super(file);
          this.progressionModel = progressionModel;
      }
  
      @Override
      public long getContentLength() {
          long size = super.getContentLength();
          progressionModel.setTotal(size);
          return size;
      }
  
      @Override
      public void writeTo(OutputStream out) throws IOException {
          Args.notNull(out, "Output stream");
          try (InputStream in = new BufferedInputStream(new FileInputStream(getFile()))) {
              Files.copyStream(in, out, progressionModel);
              out.flush();
          }
      }
  }