/**
* Edits the container and all of the children containers.
*
* @pre parameters != null
* @pre canEditContainer( newContainer ) == true
* @post if( checkCanEdit(container) ) { edit(container); }
* @param oldContainer The original container
* @param newContainer The new container values
* @throws IllegalArgumentException if oldContainer == null or newContainer == null or
* newContainer.canEdit == false
*/
public void editContainer(Container oldContainer, Container newContainer)
throws IllegalArgumentException {
if (oldContainer == null || newContainer == null || !canEditContainer(newContainer)) {
throw new IllegalArgumentException();
}
System.out.println(" ------ Editing container --------");
System.out.println("Before: " + oldContainer + "\nAfter:" + newContainer + "");
if (oldContainer instanceof ProductGroup) {
Container parent = oldContainer.getContainer(); // added for bug fix
parent.getProductGroups().remove((ProductGroup) oldContainer);
oldContainer.setName(newContainer.getName());
((ProductGroup) oldContainer)
.setThreeMonthSupply(((ProductGroup) newContainer).getThreeMonthSupply());
parent.getProductGroups().add((ProductGroup) oldContainer);
} else {
storageUnits.remove(oldContainer);
oldContainer.setName(newContainer.getName());
storageUnits.add((StorageUnit) oldContainer); // added for bug fix
}
containerDAO.update(new ContainerDTO(oldContainer));
ChangeObject hint = getHintObject(oldContainer);
setChanged();
notifyObservers(hint);
}
public void load() {
Collection<ContainerDTO> productContainers = containerDAO.readAll();
TreeMap<Integer, Set<ProductGroup>> parentToChild = new TreeMap<Integer, Set<ProductGroup>>();
for (ContainerDTO containerDTO : productContainers) {
if (containerDTO.getContainerId() == -1) {
StorageUnit storageUnit = new StorageUnit().storageUnitConverter(containerDTO);
storageUnits.add(storageUnit);
idToContainer.put(storageUnit.getId(), storageUnit);
} else {
addToMap(containerDTO, parentToChild);
}
}
setProductGroups(parentToChild, null);
}
/**
* Initializes container and adds it to the parent. Also adds container to list of ProductGroup if
* instance of ProductGroup, otherwise parent is null.
*
* @pre container != null
* @param parent
* @param container
*/
private void addContainerToTree(Container parent, Container container) {
assert container != null;
container.setContainer(parent);
// container.setId( uniqueId++ );
if (container instanceof ProductGroup) {
parent.addProductGroup((ProductGroup) container);
} else {
storageUnits.add((StorageUnit) container);
}
ContainerDTO containerDTO = new ContainerDTO(container);
containerDAO.create(containerDTO);
container.setId(containerDTO.getId());
idToContainer.put(container.getId(), container);
ChangeObject hint = getHintObject(container);
setChanged();
notifyObservers(hint);
}
/**
* Deletes the container and all of the children containers.
*
* @pre Container.items() == 0, recursively, see condition below ***(can't assert that items == 0,
* have to trust)***
* @pre container.exists() == true
* @post delete(container), recursively
* @param container Current container to be deleted
* @throws IllegalArgumentException if container.container == null && container instanceof
* ProductGroup
*/
public void deleteContainer(Container container) throws IllegalArgumentException {
Container parent = container.getContainer();
if (container instanceof ProductGroup) {
if (parent == null) {
throw new IllegalArgumentException();
}
Set<ProductGroup> productGroupList = container.getContainer().getProductGroups();
productGroupList.remove(container);
} else {
storageUnits.remove(container);
}
ChangeObject hint;
if (parent == null || parent.getProductGroups().isEmpty()) {
hint = getHintObject(parent);
} else {
hint = getHintObject(parent.getProductGroups().iterator().next());
}
containerDAO.delete(new ContainerDTO(container));
idToContainer.remove(container.getId());
setChanged();
notifyObservers(hint);
}