com.structurizr.dsl.BrandingParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of structurizr-dsl Show documentation
Structurizr for Java
The newest version!
package com.structurizr.dsl;

import com.structurizr.util.ImageUtils;
import com.structurizr.view.Font;

import java.io.File;

final class BrandingParser extends AbstractParser {

    private static final String LOGO_GRAMMAR = "logo ";
    private static final String FONT_GRAMMAR = "font  [url]";

    private static final int LOGO_FILE_INDEX = 1;

    private static final int FONT_NAME_INDEX = 1;
    private static final int FONT_URL_INDEX = 2;

    void parseLogo(BrandingDslContext context, Tokens tokens, boolean restricted) {
        // logo 

        if (tokens.hasMoreThan(LOGO_FILE_INDEX)) {
            throw new RuntimeException("Too many tokens, expected: " + LOGO_GRAMMAR);
        } else if (tokens.includes(LOGO_FILE_INDEX)) {
            String path = tokens.get(1);

            if (path.startsWith("data:image/") || path.startsWith("https://") || path.startsWith("http://")) {
                if (IconUtils.isSupported(path)) {
                    context.getWorkspace().getViews().getConfiguration().getBranding().setLogo(path);
                } else {
                    throw new IllegalArgumentException("Only PNG and JPG URLs and data URIs are supported: " + path);
                }
            } else {
                if (!restricted) {
                    File file = new File(context.getFile().getParent(), path);
                    if (file.exists() && !file.isDirectory()) {
                        try {
                            String dataUri = ImageUtils.getImageAsDataUri(file);
                            context.getWorkspace().getViews().getConfiguration().getBranding().setLogo(dataUri);
                        } catch (Exception e) {
                            throw new RuntimeException(e);
                        }
                    } else {
                        throw new RuntimeException(path + " does not exist");
                    }
                }
            }
        } else {
            throw new RuntimeException("Expected: " + LOGO_GRAMMAR);
        }
    }

    void parseFont(BrandingDslContext context, Tokens tokens) {
        // font  [url]

        if (tokens.hasMoreThan(FONT_URL_INDEX)) {
            throw new RuntimeException("Too many tokens, expected: " + FONT_GRAMMAR);
        } else if (tokens.includes(FONT_URL_INDEX)) {
            String name = tokens.get(FONT_NAME_INDEX);
            String url = tokens.get(FONT_URL_INDEX);

            context.getWorkspace().getViews().getConfiguration().getBranding().setFont(new Font(name, url));
        } else if (tokens.includes(FONT_NAME_INDEX)) {
            String name = tokens.get(FONT_NAME_INDEX);

            context.getWorkspace().getViews().getConfiguration().getBranding().setFont(new Font(name));
        } else {
            throw new RuntimeException("Expected: " + FONT_GRAMMAR);
        }
    }

}