/*****************************************************************
* 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.apache.cayenne.modeler.undo;
import javax.swing.undo.CannotRedoException;
import javax.swing.undo.CannotUndoException;
import org.apache.cayenne.access.DataDomain;
import org.apache.cayenne.map.DataMap;
import org.apache.cayenne.map.DbAttribute;
import org.apache.cayenne.map.DbEntity;
import org.apache.cayenne.map.ObjAttribute;
import org.apache.cayenne.map.ObjEntity;
import org.apache.cayenne.modeler.action.CreateAttributeAction;
import org.apache.cayenne.modeler.action.RemoveAttributeAction;
import org.apache.cayenne.modeler.event.EntityDisplayEvent;
public class CreateAttributeUndoableEdit extends CayenneUndoableEdit {
@Override
public boolean canRedo() {
return true;
}
@Override
public String getPresentationName() {
return "Create Attribute";
}
private ObjEntity objEntity;
private ObjAttribute objAttr;
private DataDomain domain;
private DataMap dataMap;
private DbEntity dbEntity;
private DbAttribute dbAttr;
@Override
public void redo() throws CannotRedoException {
CreateAttributeAction action = (CreateAttributeAction) actionManager
.getAction(CreateAttributeAction.getActionName());
if (objEntity != null) {
action.createObjAttribute(domain, dataMap, objEntity, objAttr);
}
if (dbEntity != null) {
action.createDbAttribute(domain, dataMap, dbEntity, dbAttr);
}
}
@Override
public void undo() throws CannotUndoException {
RemoveAttributeAction action = (RemoveAttributeAction) actionManager
.getAction(RemoveAttributeAction.getActionName());
if (objEntity != null) {
action.removeObjAttributes(objEntity,
new ObjAttribute[] { objAttr });
controller.fireObjEntityDisplayEvent(new EntityDisplayEvent(
this,
objEntity,
dataMap,
domain));
}
if (dbEntity != null) {
action.removeDbAttributes(dataMap, dbEntity,
new DbAttribute[] { dbAttr });
controller.fireDbEntityDisplayEvent(new EntityDisplayEvent(
this,
dbEntity,
dataMap,
domain));
}
}
public CreateAttributeUndoableEdit(DataDomain domain, DataMap map,
ObjEntity objEntity, ObjAttribute attr) {
this.domain = domain;
this.dataMap = map;
this.objEntity = objEntity;
this.objAttr = attr;
}
public CreateAttributeUndoableEdit(DataDomain domain, DataMap map,
DbEntity dbEntity, DbAttribute attr) {
this.domain = domain;
this.dataMap = map;
this.dbEntity = dbEntity;
this.dbAttr = attr;
}
}