Medias.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2022 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * 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 Affero 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/>.
 */
 
package org.turro.upload;
 
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.Reader;
import java.nio.charset.Charset;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.turro.formatter.BytesFormatter;
import org.apache.commons.io.input.ReaderInputStream;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.util.Toasts;
import org.turro.file.Document;
import org.turro.file.Folder;
import org.turro.i18n.I_;
import org.turro.util.ImageUtil;
import org.turro.zkoss.text.RepositoryContent;
import org.zkoss.util.media.Media;
 
public class Medias {
  
  public static final long MAX_FILE_SIZE = BytesFormatter.parseBytes("16MB");
 
  public static void toFolder(Media media, Folder folder) {
    toFolder(media, folder, standard());
  }
  
  public static void toFolder(Media media, Folder folder, Consumer<File> consumer) {
    toFolder(media, folder.folder(), consumer);
  }
  
  public static void toFolder(Media media, File folder) {
    toFolder(media, folder, standard());
  }
 
  public static void toFolder(Media media, File folder, Consumer<File> consumer) {
    toFile(media, Folder.from(folder).ensure().path().resolve(Document.correctName(media.getName())).toFile(), consumer);
  }
  
  public static void toFile(Media media, File file) {
    toFile(media, file, standard());
  }
 
  public static void toFile(Media media, File file, Consumer<File> consumer) {
    try(FileOutputStream fos = new FileOutputStream(file)) {
      if(media.inMemory()) {
        fos.write(media.isBinary() ? media.getByteData() : media.getStringData().getBytes());
      } else {
        byte[] buffer = new byte[102400];
        try(InputStream is = toStream(media)) {
          int r;
          while((r = is.read(buffer)) != -1) {
            fos.write(buffer, 0, r);
          }
        }
      }
    } catch (IOException ex) {
      Logger.getLogger(RepositoryContent.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
    if(file.exists() && consumer != null) {
      consumer.accept(file);
    }
  }
  
  public static Reader toReader(Media media) {
    if(!media.isBinary()) {
      return media.getReaderData();
    }
    return null;
  }
 
  public static InputStream toStream(Media media) {
    return media.isBinary() ? media.getStreamData() : new ReaderInputStream(media.getReaderData(), Charset.defaultCharset());
  }
  
  public static Consumer<File> scaling(double scale) {
    return (file) -> {
      try {
        ImageUtil.scale(file, file, scale);
      } catch (IOException ex) {
        Toasts.message(I_.get("Bad format image")).show();
        file.delete();
        Logger.getLogger(Medias.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      }
    };
  }
  
  public static Consumer<File> smart(double width, double height) {
    return (file) -> {
      try {
        ImageUtil.smart(file, file, width, height);
      } catch (IOException ex) {
        Toasts.message(I_.get("Bad format image")).show();
        file.delete();
        Logger.getLogger(Medias.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      }
    };
  }
 
  public static Consumer<File> standard() {
    return (file) -> {
      if(Document.from(file).isImage()) {
        try {
          ImageUtil.scale(file, file, 1200);
        } catch (IOException ex) {
          Toasts.message(I_.get("Bad format image")).show();
          file.delete();
          Logger.getLogger(Medias.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
        }
      } else {
        smaller(MAX_FILE_SIZE).apply(file);
      }
    };
  }
  
  public static Function<File, Boolean> smaller(long size) {
    return (file) -> {
      if(file.length() > size) {
        Toasts.message(I_.get("File is too large")).show();
        file.delete();
        return false;
      }
      return true;
    };
  }
  
  private Medias() {}
  
}