2018-04-07 01:23:03 +02:00
|
|
|
import voluptuous as vol
|
|
|
|
|
2018-10-07 16:52:14 +02:00
|
|
|
import esphomeyaml.config_validation as cv
|
2018-05-21 15:47:30 +02:00
|
|
|
from esphomeyaml.const import CONF_AP, CONF_CHANNEL, CONF_DNS1, CONF_DNS2, CONF_DOMAIN, \
|
2018-11-19 22:12:24 +01:00
|
|
|
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
|
|
|
|
from esphomeyaml.cpp_types import App, Component, esphomelib_ns, global_ns
|
2018-04-07 01:23:03 +02:00
|
|
|
|
2018-04-18 18:43:13 +02:00
|
|
|
|
|
|
|
def validate_password(value):
|
|
|
|
value = cv.string(value)
|
|
|
|
if not value:
|
|
|
|
return value
|
|
|
|
if len(value) < 8:
|
|
|
|
raise vol.Invalid(u"WPA password must be at least 8 characters long")
|
|
|
|
if len(value) > 63:
|
|
|
|
raise vol.Invalid(u"WPA password must be at most 63 characters long")
|
|
|
|
return value
|
|
|
|
|
|
|
|
|
2018-06-11 10:01:54 +02:00
|
|
|
def validate_channel(value):
|
|
|
|
value = cv.positive_int(value)
|
|
|
|
if value < 1:
|
|
|
|
raise vol.Invalid("Minimum WiFi channel is 1")
|
|
|
|
if value > 14:
|
|
|
|
raise vol.Invalid("Maximum WiFi channel is 14")
|
|
|
|
return value
|
|
|
|
|
|
|
|
|
2018-04-18 18:43:13 +02:00
|
|
|
AP_MANUAL_IP_SCHEMA = vol.Schema({
|
|
|
|
vol.Required(CONF_STATIC_IP): cv.ipv4,
|
|
|
|
vol.Required(CONF_GATEWAY): cv.ipv4,
|
|
|
|
vol.Required(CONF_SUBNET): cv.ipv4,
|
|
|
|
})
|
|
|
|
|
|
|
|
STA_MANUAL_IP_SCHEMA = AP_MANUAL_IP_SCHEMA.extend({
|
|
|
|
vol.Inclusive(CONF_DNS1, 'dns'): cv.ipv4,
|
|
|
|
vol.Inclusive(CONF_DNS2, 'dns'): cv.ipv4,
|
|
|
|
})
|
|
|
|
|
2018-06-11 10:01:54 +02:00
|
|
|
WIFI_NETWORK_BASE = vol.Schema({
|
2018-06-11 15:22:03 +02:00
|
|
|
vol.Required(CONF_SSID): cv.ssid,
|
2018-06-11 10:01:54 +02:00
|
|
|
vol.Optional(CONF_PASSWORD): validate_password,
|
|
|
|
vol.Optional(CONF_CHANNEL): validate_channel,
|
|
|
|
vol.Optional(CONF_MANUAL_IP): AP_MANUAL_IP_SCHEMA,
|
|
|
|
})
|
|
|
|
|
2018-06-11 15:22:03 +02:00
|
|
|
WIFI_NETWORK_AP = WIFI_NETWORK_BASE.extend({
|
2018-06-11 10:01:54 +02:00
|
|
|
vol.Optional(CONF_MANUAL_IP): AP_MANUAL_IP_SCHEMA,
|
2018-06-11 15:22:03 +02:00
|
|
|
})
|
2018-06-11 10:01:54 +02:00
|
|
|
|
2018-06-11 15:22:03 +02:00
|
|
|
WIFI_NETWORK_STA = WIFI_NETWORK_BASE.extend({
|
2018-06-11 10:01:54 +02:00
|
|
|
vol.Optional(CONF_MANUAL_IP): STA_MANUAL_IP_SCHEMA,
|
2018-06-11 15:22:03 +02:00
|
|
|
})
|
2018-06-11 10:01:54 +02:00
|
|
|
|
|
|
|
|
2018-06-11 16:13:46 +02:00
|
|
|
def validate(config):
|
2018-06-11 10:01:54 +02:00
|
|
|
if CONF_PASSWORD in config and CONF_SSID not in config:
|
|
|
|
raise vol.Invalid("Cannot have WiFi password without SSID!")
|
2018-11-24 18:32:18 +01:00
|
|
|
if (CONF_SSID not in config) and (CONF_AP not in config):
|
2018-06-11 16:13:46 +02:00
|
|
|
raise vol.Invalid("Please specify at least an SSID or an Access Point "
|
|
|
|
"to create.")
|
2018-06-11 10:01:54 +02:00
|
|
|
return config
|
|
|
|
|
|
|
|
|
2018-11-12 23:30:31 +01:00
|
|
|
IPAddress = global_ns.class_('IPAddress')
|
|
|
|
ManualIP = esphomelib_ns.struct('ManualIP')
|
|
|
|
WiFiComponent = esphomelib_ns.class_('WiFiComponent', Component)
|
|
|
|
WiFiAp = esphomelib_ns.struct('WiFiAp')
|
2018-06-02 22:22:20 +02:00
|
|
|
|
2018-11-12 23:30:31 +01:00
|
|
|
WiFiPowerSaveMode = esphomelib_ns.enum('WiFiPowerSaveMode')
|
2018-10-07 16:52:14 +02:00
|
|
|
WIFI_POWER_SAVE_MODES = {
|
2018-11-12 23:30:31 +01:00
|
|
|
'NONE': WiFiPowerSaveMode.WIFI_POWER_SAVE_NONE,
|
|
|
|
'LIGHT': WiFiPowerSaveMode.WIFI_POWER_SAVE_LIGHT,
|
|
|
|
'HIGH': WiFiPowerSaveMode.WIFI_POWER_SAVE_HIGH,
|
2018-10-07 16:52:14 +02:00
|
|
|
}
|
|
|
|
|
2018-06-11 10:01:54 +02:00
|
|
|
CONFIG_SCHEMA = vol.All(vol.Schema({
|
2018-06-02 22:22:20 +02:00
|
|
|
cv.GenerateID(): cv.declare_variable_id(WiFiComponent),
|
2018-04-18 18:43:13 +02:00
|
|
|
vol.Optional(CONF_SSID): cv.ssid,
|
|
|
|
vol.Optional(CONF_PASSWORD): validate_password,
|
2018-06-11 15:22:03 +02:00
|
|
|
vol.Optional(CONF_MANUAL_IP): STA_MANUAL_IP_SCHEMA,
|
2018-06-11 10:01:54 +02:00
|
|
|
vol.Optional(CONF_AP): WIFI_NETWORK_AP,
|
2018-04-07 01:23:03 +02:00
|
|
|
vol.Optional(CONF_HOSTNAME): cv.hostname,
|
2018-10-20 15:18:12 +02:00
|
|
|
vol.Optional(CONF_DOMAIN, default='.local'): cv.domain_name,
|
2018-06-12 21:18:04 +02:00
|
|
|
vol.Optional(CONF_REBOOT_TIMEOUT): cv.positive_time_period_milliseconds,
|
2018-11-23 13:51:22 +01:00
|
|
|
vol.Optional(CONF_POWER_SAVE_MODE): cv.one_of(*WIFI_POWER_SAVE_MODES, upper=True),
|
2018-06-11 16:13:46 +02:00
|
|
|
}), validate)
|
2018-04-07 01:23:03 +02:00
|
|
|
|
|
|
|
|
|
|
|
def safe_ip(ip):
|
|
|
|
if ip is None:
|
2018-06-11 10:01:54 +02:00
|
|
|
return IPAddress(0, 0, 0, 0)
|
2018-04-07 01:23:03 +02:00
|
|
|
return IPAddress(*ip.args)
|
|
|
|
|
|
|
|
|
2018-04-18 18:43:13 +02:00
|
|
|
def manual_ip(config):
|
2018-06-11 10:01:54 +02:00
|
|
|
if config is None:
|
|
|
|
return None
|
2018-04-18 18:43:13 +02:00
|
|
|
return StructInitializer(
|
2018-05-20 12:41:52 +02:00
|
|
|
ManualIP,
|
2018-04-18 18:43:13 +02:00
|
|
|
('static_ip', safe_ip(config[CONF_STATIC_IP])),
|
|
|
|
('gateway', safe_ip(config[CONF_GATEWAY])),
|
|
|
|
('subnet', safe_ip(config[CONF_SUBNET])),
|
|
|
|
('dns1', safe_ip(config.get(CONF_DNS1))),
|
|
|
|
('dns2', safe_ip(config.get(CONF_DNS2))),
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2018-06-11 10:01:54 +02:00
|
|
|
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))),
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2018-04-07 01:23:03 +02:00
|
|
|
def to_code(config):
|
2018-06-11 16:13:46 +02:00
|
|
|
rhs = App.init_wifi()
|
2018-06-02 22:22:20 +02:00
|
|
|
wifi = Pvariable(config[CONF_ID], rhs)
|
2018-04-18 18:43:13 +02:00
|
|
|
|
2018-06-11 16:13:46 +02:00
|
|
|
if CONF_SSID in config:
|
|
|
|
add(wifi.set_sta(wifi_network(config)))
|
2018-04-18 18:43:13 +02:00
|
|
|
|
2018-06-11 10:01:54 +02:00
|
|
|
if CONF_AP in config:
|
|
|
|
add(wifi.set_ap(wifi_network(config[CONF_AP])))
|
2018-04-18 18:43:13 +02:00
|
|
|
|
2018-04-07 01:23:03 +02:00
|
|
|
if CONF_HOSTNAME in config:
|
|
|
|
add(wifi.set_hostname(config[CONF_HOSTNAME]))
|
2018-05-14 11:50:56 +02:00
|
|
|
|
2018-06-12 21:18:04 +02:00
|
|
|
if CONF_REBOOT_TIMEOUT in config:
|
|
|
|
add(wifi.set_reboot_timeout(config[CONF_REBOOT_TIMEOUT]))
|
|
|
|
|
2018-10-07 16:52:14 +02:00
|
|
|
if CONF_POWER_SAVE_MODE in config:
|
2018-10-13 21:17:19 +02:00
|
|
|
add(wifi.set_power_save_mode(WIFI_POWER_SAVE_MODES[config[CONF_POWER_SAVE_MODE]]))
|
2018-10-07 16:52:14 +02:00
|
|
|
|
2018-05-14 11:50:56 +02:00
|
|
|
|
|
|
|
def lib_deps(config):
|
2018-11-19 22:12:24 +01:00
|
|
|
if CORE.is_esp8266:
|
2018-05-14 11:50:56 +02:00
|
|
|
return 'ESP8266WiFi'
|
|
|
|
return None
|