diff --git a/ccm-core/src/main/java/com/arsdigita/ui/admin/categories/CategoriesTreeModel.java b/ccm-core/src/main/java/com/arsdigita/ui/admin/categories/CategoriesTreeModel.java new file mode 100644 index 000000000..c6b676ed3 --- /dev/null +++ b/ccm-core/src/main/java/com/arsdigita/ui/admin/categories/CategoriesTreeModel.java @@ -0,0 +1,109 @@ +/* + * Copyright (C) 2016 LibreCCM Foundation. + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, + * MA 02110-1301 USA + */ +package com.arsdigita.ui.admin.categories; + +import com.arsdigita.bebop.PageState; +import com.arsdigita.bebop.tree.TreeModel; +import com.arsdigita.bebop.tree.TreeNode; + +import org.libreccm.categorization.Category; +import org.libreccm.categorization.Domain; + +import java.util.Iterator; +import java.util.List; + +/** + * + * @author Jens Pelzetter + */ +public class CategoriesTreeModel implements TreeModel { + + private final Domain domain; + + public CategoriesTreeModel(final Domain domain) { + this.domain = domain; + } + + @Override + public TreeNode getRoot(final PageState data) { + return new CategoryTreeNode(domain.getRoot()); + } + + @Override + public boolean hasChildren(final TreeNode node, + final PageState state) { + final Category category = ((CategoryTreeNode) node).getCategory(); + + return (category.getSubCategories() != null + && !category.getSubCategories().isEmpty()); + } + + @Override + public Iterator getChildren(final TreeNode node, + final PageState state) { + return new SubCategoryNodesIterator(((CategoryTreeNode) node) + .getCategory().getSubCategories()); + } + + private class CategoryTreeNode implements TreeNode { + + private final Category category; + + public CategoryTreeNode(final Category category) { + this.category = category; + } + + public Category getCategory() { + return category; + } + + @Override + public Object getKey() { + return category.getObjectId(); + } + + @Override + public Object getElement() { + return category.getName(); + } + + } + + private class SubCategoryNodesIterator + implements Iterator { + + private final Iterator subCategoriesIterator; + + public SubCategoryNodesIterator(final List subCategories) { + subCategoriesIterator = subCategories.iterator(); + } + + @Override + public boolean hasNext() { + return subCategoriesIterator.hasNext(); + } + + @Override + public CategoryTreeNode next() { + return new CategoryTreeNode(subCategoriesIterator.next()); + } + + } + +}