eu.ciechanowiec.sling.rocket.asset.FileWithExtension Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sling.rocket.commons Show documentation

Common utilities used by Sling Rocket

There is a newer version: 13.32.0

package eu.ciechanowiec.sling.rocket.asset;

import eu.ciechanowiec.conditional.Conditional;
import lombok.extern.slf4j.Slf4j;
import org.apache.tika.mime.MimeType;
import org.apache.tika.mime.MimeTypeException;
import org.apache.tika.mime.MimeTypes;

import java.io.File;
import java.util.Optional;

@Slf4j
class FileWithExtension {

    private final File file;

    FileWithExtension(File file) {
        this.file = file;
    }

    File rename(String newBasicName, String mimeTypeName) {
        log.trace("Renaming '{}' to have the extension for MIME type '{}'", file, mimeTypeName);
        MimeTypes defaultMimeTypes = MimeTypes.getDefaultMimeTypes();
        try {
            MimeType mimeType = defaultMimeTypes.forName(mimeTypeName);
            String extension = mimeType.getExtension();
            return Conditional.conditional(extension.isBlank())
                    .onTrue(() -> {
                        log.warn("Unable to rename '{}' to have the extension for MIME type '{}'", file, mimeTypeName);
                        return file;
                    })
                    .onFalse(() -> renameForResolvedExtension(newBasicName, extension))
                    .get(File.class);
        } catch (MimeTypeException exception) {
            String message = String.format(
                    "Unable to rename '%s' to have the extension for MIME type '%s'",
                    file, mimeTypeName
            );
            log.warn(message, exception);
            return file;
        }
    }

    private File renameForResolvedExtension(String newBasicName, String extension) {
        String newFileName = String.format("%s%s", newBasicName, extension);
        File newFile = Optional.ofNullable(file.getParent())
                               .map(parent -> new File(parent, newFileName))
                               .orElse(new File(newFileName));
        log.trace("New name for '{}' is '{}'", file, newFileName);
        newFile.deleteOnExit();
        boolean wasRenamed = file.renameTo(newFile);
        log.trace("Was '{}' renamed to '{}'? Answer: {}", file, newFile, wasRenamed);
        return Conditional.conditional(wasRenamed)
                .onTrue(() -> newFile)
                .onFalse(() -> file)
                .get(File.class);
    }
}