possibility to use CSS prefix for menu items

[svjatoslav_commons.git] / src / main / java / eu / svjatoslav / commons / network / navigation / Navigation.java

diff --git a/src/main/java/eu/svjatoslav/commons/network/navigation/Navigation.java b/src/main/java/eu/svjatoslav/commons/network/navigation/Navigation.java
index 408372a..3a2877f 100755 (executable)
--- a/src/main/java/eu/svjatoslav/commons/network/navigation/Navigation.java
+++ b/src/main/java/eu/svjatoslav/commons/network/navigation/Navigation.java
@@ -1,3 +1,12 @@
+/*
+ * Svjatoslav Commons - shared library of common functionality.
+ * Copyright ©2012-2014, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of version 3 of the GNU Lesser General Public License
+ * or later as published by the Free Software Foundation.
+ */
+
 package eu.svjatoslav.commons.network.navigation;
 
 import java.net.MalformedURLException;
@@ -15,6 +24,13 @@ public class Navigation {
                return defaultNavigationItem;
        }
 
+       private String getGetNotNullString(final String string) {
+               if (string != null)
+                       return string;
+
+               return "";
+       }
+
        public NavigationItem getRootNavigationItem() {
                return rootNavigationItem;
        }
@@ -43,17 +59,22 @@ public class Navigation {
                final NavigationItem selectedItem = getSelectedItem(request);
                final StringBuffer result = new StringBuffer();
 
-               result.append("<div class=\"navigationMenu\">");
+               result.append("<div class=\""
+                               + getGetNotNullString(rootNavigationItem.getCssPrefix())
+                               + "navigationMenu\">");
 
                for (final NavigationItem item : getRootNavigationItem()
                                .getSubElements()) {
 
                        if (item == selectedItem)
-                               result.append("<a class=\"menuItemSelected\" href=\""
-                                               + item.getLinkUrl() + "\">");
+                               result.append("<a class=\""
+                                               + getGetNotNullString(item.getCssPrefix())
+                                               + "menuItemSelected\" href=\"" + item.getLinkUrl()
+                                               + "\">");
                        else
-                               result.append("<a class=\"menuItem\" href=\""
-                                               + item.getLinkUrl() + "\">");
+                               result.append("<a class=\""
+                                               + getGetNotNullString(item.getCssPrefix())
+                                               + "menuItem\" href=\"" + item.getLinkUrl() + "\">");
 
                        result.append("<div>" + item.getTitle() + "</div>");
                        result.append("</a>");
@@ -64,13 +85,16 @@ public class Navigation {
                return result.toString();
        }
 
-       public void setDefaultNavigationItem(
+       public NavigationItem setDefaultNavigationItem(
                        final NavigationItem defaultNavigationItem) {
                this.defaultNavigationItem = defaultNavigationItem;
+               return defaultNavigationItem;
        }
 
-       public void setRootNavigationItem(final NavigationItem rootNavigationItem) {
+       public NavigationItem setRootNavigationItem(
+                       final NavigationItem rootNavigationItem) {
                this.rootNavigationItem = rootNavigationItem;
+               return rootNavigationItem;
        }
 
 }