Add native ESPHome API (#265)

* Esphomeapi

* Updates

* Remove MQTT from wizard

* Add protobuf to requirements

* Fix

* API Client updates

* Dump config on API connect

* Old WiFi config migration

* Home Assistant state import

* Lint

esphomeyaml/components/wifi.py

@@ -3,12 +3,25 @@ import voluptuous as vol
 import esphomeyaml.config_validation as cv
 from esphomeyaml.const import CONF_AP, CONF_CHANNEL, CONF_DNS1, CONF_DNS2, CONF_DOMAIN, \
     CONF_GATEWAY, CONF_HOSTNAME, CONF_ID, CONF_MANUAL_IP, CONF_PASSWORD, CONF_POWER_SAVE_MODE, \
-    CONF_REBOOT_TIMEOUT, CONF_SSID, CONF_STATIC_IP, CONF_SUBNET
-from esphomeyaml.core import CORE
-from esphomeyaml.cpp_generator import Pvariable, StructInitializer, add
+    CONF_REBOOT_TIMEOUT, CONF_SSID, CONF_STATIC_IP, CONF_SUBNET, CONF_NETWORKS, CONF_BSSID
+from esphomeyaml.core import CORE, HexInt
+from esphomeyaml.cpp_generator import Pvariable, StructInitializer, add, variable, ArrayInitializer
 from esphomeyaml.cpp_types import App, Component, esphomelib_ns, global_ns
 
 
+IPAddress = global_ns.class_('IPAddress')
+ManualIP = esphomelib_ns.struct('ManualIP')
+WiFiComponent = esphomelib_ns.class_('WiFiComponent', Component)
+WiFiAP = esphomelib_ns.struct('WiFiAP')
+
+WiFiPowerSaveMode = esphomelib_ns.enum('WiFiPowerSaveMode')
+WIFI_POWER_SAVE_MODES = {
+    'NONE': WiFiPowerSaveMode.WIFI_POWER_SAVE_NONE,
+    'LIGHT': WiFiPowerSaveMode.WIFI_POWER_SAVE_LIGHT,
+    'HIGH': WiFiPowerSaveMode.WIFI_POWER_SAVE_HIGH,
+}
+
+
 def validate_password(value):
     value = cv.string(value)
     if not value:
@@ -41,10 +54,11 @@ STA_MANUAL_IP_SCHEMA = AP_MANUAL_IP_SCHEMA.extend({
 })
 
 WIFI_NETWORK_BASE = vol.Schema({
-    vol.Required(CONF_SSID): cv.ssid,
+    cv.GenerateID(): cv.declare_variable_id(WiFiAP),
+    vol.Optional(CONF_SSID): cv.ssid,
     vol.Optional(CONF_PASSWORD): validate_password,
     vol.Optional(CONF_CHANNEL): validate_channel,
-    vol.Optional(CONF_MANUAL_IP): AP_MANUAL_IP_SCHEMA,
+    vol.Optional(CONF_MANUAL_IP): STA_MANUAL_IP_SCHEMA,
 })
 
 WIFI_NETWORK_AP = WIFI_NETWORK_BASE.extend({
@@ -53,35 +67,39 @@ WIFI_NETWORK_AP = WIFI_NETWORK_BASE.extend({
 
 WIFI_NETWORK_STA = WIFI_NETWORK_BASE.extend({
     vol.Optional(CONF_MANUAL_IP): STA_MANUAL_IP_SCHEMA,
+    vol.Optional(CONF_BSSID): cv.mac_address,
 })
 
 
 def validate(config):
     if CONF_PASSWORD in config and CONF_SSID not in config:
         raise vol.Invalid("Cannot have WiFi password without SSID!")
-    if (CONF_SSID not in config) and (CONF_AP not in config):
+
+    if CONF_SSID in config:
+        network = {CONF_SSID: config.pop(CONF_SSID)}
+        if CONF_PASSWORD in config:
+            network[CONF_PASSWORD] = config.pop(CONF_PASSWORD)
+        if CONF_MANUAL_IP in config:
+            network[CONF_MANUAL_IP] = config.pop(CONF_MANUAL_IP)
+        if CONF_NETWORKS in config:
+            raise vol.Invalid("You cannot use the 'ssid:' option together with 'networks:'. Please "
+                              "copy your network into the 'networks:' key")
+        config[CONF_NETWORKS] = cv.ensure_list(WIFI_NETWORK_STA)(network)
+
+    if (CONF_NETWORKS not in config) and (CONF_AP not in config):
         raise vol.Invalid("Please specify at least an SSID or an Access Point "
                           "to create.")
     return config
 
 
-IPAddress = global_ns.class_('IPAddress')
-ManualIP = esphomelib_ns.struct('ManualIP')
-WiFiComponent = esphomelib_ns.class_('WiFiComponent', Component)
-WiFiAp = esphomelib_ns.struct('WiFiAp')
-
-WiFiPowerSaveMode = esphomelib_ns.enum('WiFiPowerSaveMode')
-WIFI_POWER_SAVE_MODES = {
-    'NONE': WiFiPowerSaveMode.WIFI_POWER_SAVE_NONE,
-    'LIGHT': WiFiPowerSaveMode.WIFI_POWER_SAVE_LIGHT,
-    'HIGH': WiFiPowerSaveMode.WIFI_POWER_SAVE_HIGH,
-}
-
 CONFIG_SCHEMA = vol.All(vol.Schema({
     cv.GenerateID(): cv.declare_variable_id(WiFiComponent),
+    vol.Optional(CONF_NETWORKS): cv.ensure_list(WIFI_NETWORK_STA),
+
     vol.Optional(CONF_SSID): cv.ssid,
     vol.Optional(CONF_PASSWORD): validate_password,
     vol.Optional(CONF_MANUAL_IP): STA_MANUAL_IP_SCHEMA,
+
     vol.Optional(CONF_AP): WIFI_NETWORK_AP,
     vol.Optional(CONF_HOSTNAME): cv.hostname,
     vol.Optional(CONF_DOMAIN, default='.local'): cv.domain_name,
@@ -110,21 +128,28 @@ def manual_ip(config):
 
 
 def wifi_network(config):
-    return StructInitializer(
-        WiFiAp,
-        ('ssid', config.get(CONF_SSID, "")),
-        ('password', config.get(CONF_PASSWORD, "")),
-        ('channel', config.get(CONF_CHANNEL, -1)),
-        ('manual_ip', manual_ip(config.get(CONF_MANUAL_IP))),
-    )
+    ap = variable(config[CONF_ID], WiFiAP())
+    if CONF_SSID in config:
+        add(ap.set_ssid(config[CONF_SSID]))
+    if CONF_PASSWORD in config:
+        add(ap.set_password(config[CONF_PASSWORD]))
+    if CONF_BSSID in config:
+        bssid = [HexInt(i) for i in config[CONF_BSSID].parts]
+        add(ap.set_bssid(ArrayInitializer(*bssid, multiline=False)))
+    if CONF_CHANNEL in config:
+        add(ap.set_channel(config[CONF_CHANNEL]))
+    if CONF_MANUAL_IP in config:
+        add(ap.set_manual_ip(manual_ip(config[CONF_MANUAL_IP])))
+
+    return ap
 
 
 def to_code(config):
     rhs = App.init_wifi()
     wifi = Pvariable(config[CONF_ID], rhs)
 
-    if CONF_SSID in config:
-        add(wifi.set_sta(wifi_network(config)))
+    for network in config.get(CONF_NETWORKS, []):
+        add(wifi.add_sta(wifi_network(network)))
 
     if CONF_AP in config:
         add(wifi.set_ap(wifi_network(config[CONF_AP])))