From d06653dbba1f644afa3453207205aa9bc01e40bc Mon Sep 17 00:00:00 2001
From: Otto Winter <otto@otto-winter.com>
Date: Tue, 16 Oct 2018 23:08:24 +0200
Subject: [PATCH] Add template text sensor

---
 .../components/text_sensor/template.py        | 30 +++++++++++++++++++
 tests/test2.yaml                              |  4 +++
 2 files changed, 34 insertions(+)
 create mode 100644 esphomeyaml/components/text_sensor/template.py

diff --git a/esphomeyaml/components/text_sensor/template.py b/esphomeyaml/components/text_sensor/template.py
new file mode 100644
index 0000000000..8e440b3fa5
--- /dev/null
+++ b/esphomeyaml/components/text_sensor/template.py
@@ -0,0 +1,30 @@
+import voluptuous as vol
+
+from esphomeyaml.components import text_sensor
+import esphomeyaml.config_validation as cv
+from esphomeyaml.const import CONF_LAMBDA, CONF_MAKE_ID, CONF_NAME, CONF_UPDATE_INTERVAL
+from esphomeyaml.helpers import App, Application, add, optional, process_lambda, std_string, \
+    variable
+
+MakeTemplateTextSensor = Application.MakeTemplateTextSensor
+
+PLATFORM_SCHEMA = cv.nameable(text_sensor.TEXT_SENSOR_PLATFORM_SCHEMA.extend({
+    cv.GenerateID(CONF_MAKE_ID): cv.declare_variable_id(MakeTemplateTextSensor),
+    vol.Required(CONF_LAMBDA): cv.lambda_,
+    vol.Optional(CONF_UPDATE_INTERVAL): cv.update_interval,
+}))
+
+
+def to_code(config):
+    rhs = App.make_template_text_sensor(config[CONF_NAME], config.get(CONF_UPDATE_INTERVAL))
+    make = variable(config[CONF_MAKE_ID], rhs)
+    text_sensor.setup_text_sensor(make.Ptemplate_, make.Pmqtt, config)
+
+    template_ = None
+    for template_ in process_lambda(config[CONF_LAMBDA], [],
+                                    return_type=optional.template(std_string)):
+        yield
+    add(make.Ptemplate_.set_template(template_))
+
+
+BUILD_FLAGS = '-DUSE_TEMPLATE_TEXT_SENSOR'
diff --git a/tests/test2.yaml b/tests/test2.yaml
index 44677e0b13..b891206871 100644
--- a/tests/test2.yaml
+++ b/tests/test2.yaml
@@ -116,3 +116,7 @@ text_sensor:
     name: "MQTT Subscribe Text"
     topic: "the/topic"
     qos: 2
+  - platform: template
+    name: "Template Text Sensor"
+    lambda: |-
+      return "Hello World";