colesico.framework.translation.codegen.model.DictionaryElement Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of colesico-translation Show documentation
Colesico framework translation (localization) service.
There is a newer version: 5.3.0
/*
 * Copyright 20014-2019 Vladlen V. Larionov and others as noted.
 *
 * 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 colesico.framework.translation.codegen.model;

import colesico.framework.assist.codegen.model.AnnotationElement;
import colesico.framework.assist.codegen.model.ClassElement;
import colesico.framework.assist.codegen.model.MethodElement;
import colesico.framework.translation.Dictionary;
import org.apache.commons.lang3.StringUtils;

import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;

/**
 * Represents dictionary bean
 */
public class DictionaryElement {

    private final ClassElement originBean;
    private final String basePath;
    private final Map bundlesByLocale = new HashMap<>();
    private final Set keyMethods = new LinkedHashSet<>();

    public DictionaryElement(ClassElement dictionaryInterface) {

        if (dictionaryInterface == null) {
            throw new RuntimeException("Dictionary element is null");
        }
        this.originBean = dictionaryInterface;

        AnnotationElement beanAnn = dictionaryInterface.getAnnotation(Dictionary.class);
        String bPath;
        if (StringUtils.isNoneBlank(beanAnn.unwrap().basePath())) {
            bPath = beanAnn.unwrap().basePath();
        } else {
            bPath = dictionaryInterface.getPackageName() + '.' + dictionaryInterface.getSimpleName();
            bPath = bPath.replace('.', '/');
        }
        this.basePath = bPath;

    }

    public void addTranslationMethod(MethodElement keyMethod) {
        keyMethods.add(keyMethod);
    }

    public void addTranslation(MethodElement keyMethod, String localeKey, String translation) {
        BundleElement translationsBundleElement = bundlesByLocale.get(localeKey);
        if (translationsBundleElement == null) {
            translationsBundleElement = new BundleElement(this, localeKey);
            bundlesByLocale.put(localeKey, translationsBundleElement);
        }
        translationsBundleElement.addTranslation(keyMethod, translation);
    }

    public String getBasePath() {
        return basePath;
    }

    public ClassElement getOriginBean() {
        return originBean;
    }

    public Set getKeyMethods() {
        return keyMethods;
    }

    public String getImplClassSimpleName() {
        return getOriginBean().getSimpleName() + "Impl";
    }

    public Map getBundlesByLocale() {
        return bundlesByLocale;
    }

    @Override
    public String toString() {
        return "DictionaryBeanElement{" +
                "originBean=" + originBean.asType().toString() +
                ", basePath='" + basePath + '\'' +
                '}';
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        DictionaryElement that = (DictionaryElement) o;

        return basePath.equals(that.basePath);
    }

    @Override
    public int hashCode() {
        return basePath.hashCode();
    }
}