package cn.orionsec.ops.utils;

import cn.orionsec.kit.lang.utils.Exceptions;
import cn.orionsec.kit.lang.utils.Strings;
import cn.orionsec.kit.lang.utils.collect.Maps;
import cn.orionsec.kit.lang.utils.io.Streams;
import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.io.InputStream;
import java.util.Map;
import java.util.Objects;

@Slf4j
public class ResourceLoader {

    /**
     * key: resource
     * value: content
     */
    private static final Map<String, String> RESOURCE = Maps.newMap();

    /**
     * 获取资源
     *
     * @param resource resource
     * @param loader   加载类
     * @return content
     */
    public static String get(String resource, Class<?> loader) {
        try (InputStream stream = loader.getResourceAsStream(resource)) {
            String content = Streams.toString(Objects.requireNonNull(stream));
            RESOURCE.put(resource, content);
            return content;
        } catch (IOException e) {
            throw Exceptions.notFound(Strings.format("resource not found {} by {}", resource, loader.getName()), e);
        }
    }

}
