org.finos.legend.sdlc.generation.TextGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of legend-sdlc-generation-shared Show documentation
There is a newer version: 0.177.6
Show newest version
// Copyright 2023 Goldman Sachs
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package org.finos.legend.sdlc.generation;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.function.Supplier;
import java.util.regex.Pattern;

public abstract class TextGenerator
{
    private static final Pattern LINEBREAK_PATTERN = Pattern.compile("\\R");

    private final Map parameters = new HashMap<>();
    private final GeneratorTemplate template;
    private String lineBreak;

    protected TextGenerator(GeneratorTemplate template, String defaultLineBreak)
    {
        this.template = Objects.requireNonNull(template);
        setLineBreak(defaultLineBreak);
    }

    protected TextGenerator(GeneratorTemplate template)
    {
        this(template, null);
    }

    public T generate()
    {
        String text = this.template.generateText(Collections.unmodifiableMap(this.parameters));
        return newGeneratedText(processLineBreaks(text));
    }

    protected void setLineBreak(String lineBreak)
    {
        if ((lineBreak != null) && !LINEBREAK_PATTERN.matcher(lineBreak).matches())
        {
            throw new IllegalArgumentException("Invalid line break: \"" + lineBreak + "\"");
        }
        this.lineBreak = lineBreak;
    }

    @SuppressWarnings("unchecked")
    protected  V getParameter(String key)
    {
        return (V) this.parameters.get(key);
    }

    @SuppressWarnings("unchecked")
    protected  V getOrComputeParameter(String key, Supplier supplier)
    {
        return (V) this.parameters.computeIfAbsent(key, k -> supplier.get());
    }

    protected void setParameter(String key, Object value)
    {
        if (value == null)
        {
            unsetParameter(key);
        }
        else
        {
            this.parameters.put(key, value);
        }
    }

    protected void addToParameter(String key, Object value)
    {
        addAllToParameter(key, Collections.singletonList(value));
    }

    protected void addAllToParameter(String key, Object... values)
    {
        addAllToParameter(key, Arrays.asList(values));
    }

    protected void addAllToParameter(String key, Collection values)
    {
        Collection