/*******************************************************************************
 * Copyright (c) 2014, 2015 Cisco Systems, Inc. and others.  All rights reserved.
 *  
 *  This program and the accompanying materials are made available under the
 *  terms of the Eclipse Public License v1.0 which accompanies this distribution,
 *  and is available at http://www.eclipse.org/legal/epl-v10.html
 *  
 *******************************************************************************/
package com.cisco.yangide.ext.model.editor.patterns.connections;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.graphiti.features.context.IAddConnectionContext;
import org.eclipse.graphiti.features.context.IAddContext;
import org.eclipse.graphiti.features.context.ICreateConnectionContext;
import org.eclipse.graphiti.mm.pictograms.Connection;
import org.eclipse.graphiti.mm.pictograms.PictogramElement;
import org.eclipse.graphiti.pattern.AbstractConnectionPattern;
import org.eclipse.graphiti.pattern.IConnectionPattern;

import com.cisco.yangide.ext.model.editor.util.YangModelUIUtil;

public abstract class DomainConnectionPattern extends AbstractConnectionPattern implements IConnectionPattern {

    protected abstract EObject createEObject(ICreateConnectionContext context);

    @Override
    public Connection create(ICreateConnectionContext context) {
        // create the domain object connection here
        EObject newDomainObjectConnetion = createEObject(context);

        return YangModelUIUtil.drawConnection(newDomainObjectConnetion, context.getSourceAnchor(),
                context.getTargetAnchor(), getFeatureProvider());
    }

    @Override
    public PictogramElement add(IAddContext context) {
        IAddConnectionContext addConContext = (IAddConnectionContext) context;
        return YangModelUIUtil.drawPictogramConnectionElement(addConContext, getFeatureProvider(), getCreateName());
    }

    @Override
    public boolean canAdd(IAddContext context) {
        return context instanceof IAddConnectionContext;
    }

    @Override
    public String getCreateDescription() {
        return "Creates new " + getCreateName() + " connection";
    }
}