package nodebox.client;

import java.io.File;
import java.util.HashMap;
import java.util.Map;

public final class ImageFormat {

    public static final ImageFormat PDF = new ImageFormat("PDF", "pdf");
    public static final ImageFormat PNG = new ImageFormat("PNG", "png");

    private static final Map<String, ImageFormat> FORMAT_MAP;

    static {
        FORMAT_MAP = new HashMap<String, ImageFormat>();
        FORMAT_MAP.put("PDF", PDF);
        FORMAT_MAP.put("PNG", PNG);
    }

    public static ImageFormat of(String name) {
        return FORMAT_MAP.get(name.toUpperCase());
    }

    private final String label;
    private final String extension;

    public ImageFormat(String label, String extension) {
        this.label = label;
        this.extension = extension;
    }

    public String getLabel() {
        return label;
    }

    public String getExtension() {
        return extension;
    }

    public File ensureFileExtension(File file) {
        return new File(ensureFileExtension(file.getPath()));
    }

    public String ensureFileExtension(String file) {
        if (file.endsWith("." + getExtension()))
            return file;
        return file + "." + getExtension();
    }

}