ES6 class for Padding

Author: dhilt

src/modules/padding.js

@@ -49,28 +49,36 @@ Object.getOwnPropertyNames(CacheProto.prototype).forEach(methodName =>
   Cache.prototype[methodName] = CacheProto.prototype[methodName]
 );
 
-export default function Padding(template) {
-  let result;
-
+function generateElement(template) {
   if(template.nodeType !== Node.ELEMENT_NODE) {
     throw new Error('ui-scroll directive requires an Element node for templating the view');
   }
-
+  let element;
   switch (template.tagName.toLowerCase()) {
     case 'dl':
       throw new Error(`ui-scroll directive does not support <${template.tagName}> as a repeating tag: ${template.outerHTML}`);
     case 'tr':
       let table = angular.element('<table><tr><td><div></div></td></tr></table>');
-      result = table.find('tr');
+      element = table.find('tr');
       break;
     case 'li':
-      result = angular.element('<li></li>');
+      element = angular.element('<li></li>');
       break;
     default:
-      result = angular.element('<div></div>');
+      element = angular.element('<div></div>');
   }
+  return element;
+}
 
-  result.cache = new Cache();
+class Padding {
+  constructor(template) {
+    this.element = generateElement(template);
+    this.cache = new Cache();
+  }
+
+  height() {
+    return this.element.height.apply(this.element, arguments);
+  }
+}
 
-  return result;
-}
+export default Padding;

src/modules/viewport.js

@@ -25,8 +25,8 @@ export default function Viewport(elementRoutines, buffer, element, viewportContr
     createPaddingElements(template) {
       topPadding = new Padding(template);
       bottomPadding = new Padding(template);
-      element.before(topPadding);
-      element.after(bottomPadding);
+      element.before(topPadding.element);
+      element.after(bottomPadding.element);
     },
 
     applyContainerStyle() {
@@ -54,11 +54,11 @@ export default function Viewport(elementRoutines, buffer, element, viewportContr
     },
 
     insertElement(e, sibling) {
-      return elementRoutines.insertElement(e, sibling || topPadding);
+      return elementRoutines.insertElement(e, sibling || topPadding.element);
     },
 
     insertElementAnimated(e, sibling) {
-      return elementRoutines.insertElementAnimated(e, sibling || topPadding);
+      return elementRoutines.insertElementAnimated(e, sibling || topPadding.element);
     },
 
     shouldLoadBottom() {