CCM NG: Model for categories tree in /ccm/admin

git-svn-id: https://svn.libreccm.org/ccm/ccm_ng@4030 8810af33-2d31-482b-a856-94f89814c4df

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 <a href="mailto:jens.pelzetter@googlemail.com">Jens Pelzetter</a>
+ */
+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<CategoryTreeNode> {
+
+        private final Iterator<Category> subCategoriesIterator;
+
+        public SubCategoryNodesIterator(final List<Category> subCategories) {
+            subCategoriesIterator = subCategories.iterator();
+        }
+
+        @Override
+        public boolean hasNext() {
+            return subCategoriesIterator.hasNext();
+        }
+
+        @Override
+        public CategoryTreeNode next() {
+            return new CategoryTreeNode(subCategoriesIterator.next());
+        }
+
+    }
+
+}