BasePropertySectionDataModelExtractor.java

// ============================================================================
//
// Copyright (C) 2006-2016 Talend Inc. - www.talend.com
//
// This source code is available under agreement available at
// %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
//
// You should have received a copy of the agreement
// along with this program; if not, write to Talend SA
// 9 rue Pages 92150 Suresnes, France
//
// ============================================================================
package com.amalto.workbench.detailtabs.sections.util;

import org.eclipse.xsd.XSDSchema;

import com.amalto.workbench.detailtabs.sections.BasePropertySection;
import com.amalto.workbench.models.TreeObject;
import com.amalto.workbench.models.infoextractor.IAllDataModelHolder;
import com.amalto.workbench.utils.Util;

public class BasePropertySectionDataModelExtractor extends BasePropertySectionGlobalInfoExtractor implements IAllDataModelHolder {

    protected String defaultDataModel = "";//$NON-NLS-1$

    protected String defaultEntity = "";//$NON-NLS-1$

    public BasePropertySectionDataModelExtractor(BasePropertySection propSection) {
        this(propSection, "");//$NON-NLS-1$
    }

    public BasePropertySectionDataModelExtractor(BasePropertySection propSection, String defaultDataModel) {
        super(propSection, TreeObject.DATA_MODEL);

        this.defaultDataModel = defaultDataModel;
    }

    public String[] getAllDataModelNames() {
        return getGlobalInfos();
    }

    public boolean hasDataModel() {
        return getAllDataModelNames().length > 0;
    }

    public XSDSchema getDataModel(String dataModelName) {
        XSDSchema xsd = Util.getXSDSchemaOfDirtyEditor(dataModelName);
        if (xsd == null) {
            xsd = MDMRepositoryViewExtensionService.getDataModelXsd(propSection.getTreeObject(), "", dataModelName);
        }
        return xsd;
    }

    public String getDefaultDataModel() {
        return defaultDataModel;
    }

    public void setDefaultDataModel(String defaultDataModel) {
        this.defaultDataModel = defaultDataModel;
    }

    public String getDefaultEntity() {
        return defaultEntity;
    }

    public void setDefaultEntity(String defaultEntity) {
        this.defaultEntity = defaultEntity;
    }

}