Add support for configuring a suffix string + simplify logic (#5)

* Add support for configuring a suffix string + simplify logic

* Document suffix config

* Update VERSION

README.md

@@ -10,6 +10,7 @@ This element is **not** all that useful as a card (it can be used for that to),
 | entity | string | **Required** | The entity_id of the entity you want to show.
 | attribute | string | **Required** | The attribute of the entity you want to show.
 | prefix | string | **Optional** | A string you want to show in front of the attribute, ex. "My prefix string: "
+| suffix | string | **Optional** | A string you want to show after the attribute, ex. "My suffix string: "
 | show_empty | boolean | **Optional** | Give you the possibility to show `unavailable` if the state is empty, default `false`
 
 ## Installation

VERSION

@@ -1 +1 @@
-0.0.4
+0.1.0

state-attribute-element.js

@@ -1,23 +1,14 @@
 class StateAttributeElement extends HTMLElement {
   set hass(hass) {
     const entityId = this.config.entity;
-    const prefix_string = this.config.prefix
+    const prefix_string = this.config.prefix || ''
+    const suffix_string = this.config.suffix || ''
     const show_empty = this.config.show_empty
     const attr = this.config.attribute;
     const state = hass.states[entityId].attributes[attr];
     const card = document.createElement('state-attribute-element');
-    if (state.length != 0) {
-      if (prefix_string) {
-        this.innerHTML = prefix_string + state;
-      } else {
-        this.innerHTML = state;
-      }
-    } else if (show_empty == true) {
-      if (prefix_string) {
-        this.innerHTML = prefix_string + state;
-      } else {
-        this.innerHTML = state;
-      }
+    if (state.length != 0 || show_empty === true) {
+      this.innerHTML = `${prefix_string}${state}${suffix_string}`
     }
   }
   setConfig(config) {
@@ -37,4 +28,4 @@ class StateAttributeElement extends HTMLElement {
     return 1;
   }
 }
-customElements.define('state-attribute-element', StateAttributeElement);
+customElements.define('state-attribute-element', StateAttributeElement);