tech.kronicle.utils.BaseFileCache Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kronicle-utils Show documentation

Kronicle utils library that contains utility classes for use by Kronicle and its plugins

There is a newer version: 0.1.331

package tech.kronicle.utils;

import lombok.SneakyThrows;

import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Optional;

import static java.nio.file.StandardOpenOption.CREATE_NEW;
import static tech.kronicle.common.StringEscapeUtils.escapeString;

public class BaseFileCache {

    private final FileUtils fileUtils;
    private final Path cacheDir;

    @SneakyThrows
    public BaseFileCache(FileUtils fileUtils, Path cacheDir) {
        this.fileUtils = fileUtils;
        this.cacheDir = cacheDir;
        Files.createDirectories(this.cacheDir);
    }

    protected Optional getFileContent(String key) {
        return readFileContent(getAbsoluteFile(key));
    }

    protected void putFileContent(String key, String content) {
        writeFileContent(getAbsoluteFile(key), content);
    }

    private void writeFileContent(Path file, String content) {
        fileUtils.writeFileContent(file, content, CREATE_NEW);
    }

    private Optional readFileContent(Path file) {
        if (!Files.exists(file)) {
            return Optional.empty();
        }

        return Optional.of(fileUtils.readFileContent(file));
    }

    private Path getAbsoluteFile(String key) {
        Path file = cacheDir.resolve(convertKeyToFileName(key));

        if (!file.startsWith(cacheDir)) {
            throw new RuntimeException(String.format(
                    "File \"%s\" is outside of cache dir \"%s\"",
                    escapeString(file.toString()),
                    escapeString(cacheDir.toString())
            ));
        }

        return file;
    }

    private String convertKeyToFileName(String key) {
        try {
            return URLEncoder.encode(key, StandardCharsets.UTF_8.name());
        } catch (UnsupportedEncodingException e) {
            throw new RuntimeException(e);
        }
    }
}