Enhance FullPaginationDecorator to handle custom css-tags defined through tag-attributes.

Author: thamacher

src/main/java/org/thymeleaf/dialect/springdata/decorator/FullPaginationDecorator.java

@@ -19,6 +19,10 @@ public final class FullPaginationDecorator implements PaginationDecorator {
     private static final String CSS_ATTR_PREFIX = SpringDataDialect.PREFIX + ":" + DEFAULT_CLASS + "-";
 
     // define custom CSS tags
+    private String cssLaquo = "page-item";
+    private String cssRaquo = "page-item";
+    private String cssPrevious = "page-item";
+    private String cssNext = "page-item";
     private String cssPageItem = "page-item";
     private String cssPageLink = "page-link";
     private String cssDisabled = "disabled";
@@ -104,19 +108,19 @@ private String createPageLinks(final Page<?> page, final ITemplateContext contex
     }
 
     private String getLaquo(Locale locale) {
-        return Messages.getMessage(BUNDLE_NAME, "laquo", locale, cssPageItem, cssPageLink, cssDisabled);
+        return Messages.getMessage(BUNDLE_NAME, "laquo", locale, cssLaquo, cssPageLink, cssDisabled);
     }
 
     private String getLaquo(String firstPage, Locale locale) {
-        return Messages.getMessage(BUNDLE_NAME, "laquo.link", locale, firstPage, cssPageItem, cssPageLink);
+        return Messages.getMessage(BUNDLE_NAME, "laquo.link", locale, firstPage, cssLaquo, cssPageLink);
     }
 
     private String getRaquo(Locale locale) {
-        return Messages.getMessage(BUNDLE_NAME, "raquo", locale, cssPageItem, cssPageLink, cssDisabled);
+        return Messages.getMessage(BUNDLE_NAME, "raquo", locale, cssRaquo, cssPageLink, cssDisabled);
     }
 
     private String getRaquo(String lastPage, Locale locale) {
-        return Messages.getMessage(BUNDLE_NAME, "raquo.link", locale, lastPage, cssPageItem, cssPageLink);
+        return Messages.getMessage(BUNDLE_NAME, "raquo.link", locale, lastPage, cssRaquo, cssPageLink);
     }
 
     private String getLink(int pageNumber, Locale locale) {
@@ -133,7 +137,7 @@ private String getPreviousPageLink(Page<?> page, final ITemplateContext context)
         int previousPage = page.getNumber()-1;
         String link = PageUtils.createPageUrl(context, previousPage);
 
-        return Messages.getMessage(BUNDLE_NAME, msgKey, locale, link, cssPageItem, cssPageLink, cssDisabled);
+        return Messages.getMessage(BUNDLE_NAME, msgKey, locale, link, cssPrevious, cssPageLink, cssDisabled);
     }
 
     private String getNextPageLink(Page<?> page, final ITemplateContext context) {
@@ -142,11 +146,27 @@ private String getNextPageLink(Page<?> page, final ITemplateContext context) {
         int nextPage = page.getNumber() + 1;
         String link = PageUtils.createPageUrl(context, nextPage);
 
-        return Messages.getMessage(BUNDLE_NAME, msgKey, locale, link, cssPageItem, cssPageLink, cssDisabled);
+        return Messages.getMessage(BUNDLE_NAME, msgKey, locale, link, cssNext, cssPageLink, cssDisabled);
     }
 
     private void configureCss(Map<String,String> attributeMap) {
 
+        if (attributeMap.get(CSS_ATTR_PREFIX + "laquo") != null) {
+            this.cssLaquo = attributeMap.get(CSS_ATTR_PREFIX + "laquo");
+        }
+
+        if (attributeMap.get(CSS_ATTR_PREFIX + "raquo") != null) {
+            this.cssRaquo = attributeMap.get(CSS_ATTR_PREFIX + "raquo");
+        }
+
+        if (attributeMap.get(CSS_ATTR_PREFIX + "previous") != null) {
+            this.cssPrevious = attributeMap.get(CSS_ATTR_PREFIX + "previous");
+        }
+
+        if (attributeMap.get(CSS_ATTR_PREFIX + "next") != null) {
+            this.cssNext = attributeMap.get(CSS_ATTR_PREFIX + "next");
+        }
+
         if (attributeMap.get(CSS_ATTR_PREFIX + "page-item") != null) {
             this.cssPageItem = attributeMap.get(CSS_ATTR_PREFIX + "page-item");
         }