packages/serverless-workflow-diagram-editor/kie-wb-common-stunner/kie-wb-common-stunner-core/kie-wb-common-stunner-api/kie-wb-common-stunner-core-api/src/main/java/org/kie/workbench/common/stunner/core/api/JsDefinitionManager.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.kie.workbench.common.stunner.core.api;

import java.lang.annotation.Annotation;
import java.util.Optional;

import jsinterop.annotations.JsIgnore;
import jsinterop.annotations.JsType;
import jsinterop.base.Js;
import org.kie.workbench.common.stunner.core.definition.adapter.DefinitionId;
import org.kie.workbench.common.stunner.core.definition.jsadapter.JsDefinitionAdapter;
import org.kie.workbench.common.stunner.core.definition.jsadapter.JsDefinitionProperty;
import org.kie.workbench.common.stunner.core.definition.jsadapter.JsDefinitionSetAdapter;
import org.kie.workbench.common.stunner.core.definition.jsadapter.JsPropertyAdapter;
import org.kie.workbench.common.stunner.core.definition.jsadapter.JsRuleAdapter;
import org.kie.workbench.common.stunner.core.definition.property.PropertyMetaTypes;
import org.kie.workbench.common.stunner.core.factory.graph.ElementFactory;
import org.kie.workbench.common.stunner.core.i18n.StunnerTranslationService;
import org.kie.workbench.common.stunner.core.registry.DynamicRegistry;
import org.kie.workbench.common.stunner.core.registry.definition.TypeDefinitionSetRegistry;
import org.kie.workbench.common.stunner.core.rule.RuleSet;

@JsType
public class JsDefinitionManager {

    @JsIgnore
    private TypeDefinitionSetRegistry registry;
    @JsIgnore
    private JsDefinitionSetAdapter definitionSetAdapter;
    @JsIgnore
    private JsDefinitionAdapter definitionAdapter;
    @JsIgnore
    private JsPropertyAdapter propertyAdapter;
    @JsIgnore
    private JsRuleAdapter ruleAdapter;

    public static JsDefinitionManager build(StunnerTranslationService translationService,
                                            TypeDefinitionSetRegistry registry,
                                            JsDefinitionSetAdapter definitionSetAdapter,
                                            JsDefinitionAdapter definitionAdapter,
                                            JsPropertyAdapter propertyAdapter, JsRuleAdapter ruleAdapter) {
        definitionAdapter.setTranslationService(translationService);
        propertyAdapter.setTranslationService(translationService);
        definitionSetAdapter.setTranslationService(translationService);
        return new JsDefinitionManager(registry, definitionSetAdapter, definitionAdapter, propertyAdapter, ruleAdapter);
    }

    @SuppressWarnings("all")
    public JsDefinitionManager(TypeDefinitionSetRegistry registry, JsDefinitionSetAdapter jsDefinitionSetAdapter, JsDefinitionAdapter definitionAdapter, JsPropertyAdapter propertyAdapter, JsRuleAdapter ruleAdapter) {
        this.registry = registry;
        this.definitionSetAdapter = jsDefinitionSetAdapter;
        this.definitionAdapter = definitionAdapter;
        this.propertyAdapter = propertyAdapter;
        this.ruleAdapter = ruleAdapter;
    }

    @SuppressWarnings("all")
    public void initializeDefinitionSet(Object definitionSet) {
        ((DynamicRegistry) registry).register(definitionSet);
    }

    public void initializeDefinitionsField(String definitionsField) {
        definitionSetAdapter.setDefinitionsField(definitionsField);
    }

    public void initializeDomainQualifier(Annotation domainQualifier) {
        definitionSetAdapter.setDomainQualifier(domainQualifier);
    }

    public void initializeCategory(String definitionId, String category) {
        definitionAdapter.setCategory(definitionId, category);
    }

    public void initializeLabels(String definitionId, String[] definitionLabels) {
        definitionAdapter.setLabels(definitionId, definitionLabels);
    }

    public void initializeDefinitionNameField(String definitionId, String nameField) {
        definitionAdapter.setDefinitionNameField(definitionId, nameField);
    }

    public void initializeRules(RuleSet ruleSet) {
        ruleAdapter.setRuleSet(ruleSet);
    }

    public void initializeElementFactory(String category, Class<? extends ElementFactory> factory) {
        definitionAdapter.setElementFactory(category, factory);
    }

    public DefinitionId getId(Object pojo) {
        return definitionAdapter.getId(pojo);
    }

    public String getCategory(Object pojo) {
        return definitionAdapter.getCategory(pojo);
    }

    public String getTitle(Object pojo) {
        return definitionAdapter.getTitle(pojo);
    }

    public String getName(Object pojo) {
        String field = definitionAdapter.getMetaPropertyField(pojo, PropertyMetaTypes.NAME);
        Optional<?> property = definitionAdapter.getProperty(pojo, field);
        if (property.isPresent()) {
            String name = Js.uncheckedCast(propertyAdapter.getValue((JsDefinitionProperty) property.get()));
            return name;
        }
        return null;
    }

    public void setName(Object pojo, String name) {
        String field = definitionAdapter.getMetaPropertyField(pojo, PropertyMetaTypes.NAME);
        Optional<?> property = definitionAdapter.getProperty(pojo, field);
        if (property.isPresent()) {
            propertyAdapter.setValue((JsDefinitionProperty) property.get(), name);
        }
    }

    public String getDescription(Object pojo) {
        return definitionAdapter.getDescription(pojo);
    }

    public String[] getLabels(Object pojo) {
        return definitionAdapter.getLabels(pojo);
    }

    public String[] getPropertyFields(Object pojo) {
        return definitionAdapter.getPropertyFields(pojo);
    }
}