User management (#15420)

* User management

* Lint

* Fix dict

* Reuse data instance

* OrderedDict all the way

homeassistant/components/config/auth.py

@@ -0,0 +1,113 @@
+"""Offer API to configure Home Assistant auth."""
+import voluptuous as vol
+
+from homeassistant.core import callback
+from homeassistant.components import websocket_api
+
+
+WS_TYPE_LIST = 'config/auth/list'
+SCHEMA_WS_LIST = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
+    vol.Required('type'): WS_TYPE_LIST,
+})
+
+WS_TYPE_DELETE = 'config/auth/delete'
+SCHEMA_WS_DELETE = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
+    vol.Required('type'): WS_TYPE_DELETE,
+    vol.Required('user_id'): str,
+})
+
+WS_TYPE_CREATE = 'config/auth/create'
+SCHEMA_WS_CREATE = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
+    vol.Required('type'): WS_TYPE_CREATE,
+    vol.Required('name'): str,
+})
+
+
+async def async_setup(hass):
+    """Enable the Home Assistant views."""
+    hass.components.websocket_api.async_register_command(
+        WS_TYPE_LIST, websocket_list,
+        SCHEMA_WS_LIST
+    )
+    hass.components.websocket_api.async_register_command(
+        WS_TYPE_DELETE, websocket_delete,
+        SCHEMA_WS_DELETE
+    )
+    hass.components.websocket_api.async_register_command(
+        WS_TYPE_CREATE, websocket_create,
+        SCHEMA_WS_CREATE
+    )
+    return True
+
+
+@callback
+@websocket_api.require_owner
+def websocket_list(hass, connection, msg):
+    """Return a list of users."""
+    async def send_users():
+        """Send users."""
+        result = [_user_info(u) for u in await hass.auth.async_get_users()]
+
+        connection.send_message_outside(
+            websocket_api.result_message(msg['id'], result))
+
+    hass.async_add_job(send_users())
+
+
+@callback
+@websocket_api.require_owner
+def websocket_delete(hass, connection, msg):
+    """Delete a user."""
+    async def delete_user():
+        """Delete user."""
+        if msg['user_id'] == connection.request.get('hass_user').id:
+            connection.send_message_outside(websocket_api.error_message(
+                msg['id'], 'no_delete_self',
+                'Unable to delete your own account'))
+            return
+
+        user = await hass.auth.async_get_user(msg['user_id'])
+
+        if not user:
+            connection.send_message_outside(websocket_api.error_message(
+                msg['id'], 'not_found', 'User not found'))
+            return
+
+        await hass.auth.async_remove_user(user)
+
+        connection.send_message_outside(
+            websocket_api.result_message(msg['id']))
+
+    hass.async_add_job(delete_user())
+
+
+@callback
+@websocket_api.require_owner
+def websocket_create(hass, connection, msg):
+    """Create a user."""
+    async def create_user():
+        """Create a user."""
+        user = await hass.auth.async_create_user(msg['name'])
+
+        connection.send_message_outside(
+            websocket_api.result_message(msg['id'], {
+                'user': _user_info(user)
+            }))
+
+    hass.async_add_job(create_user())
+
+
+def _user_info(user):
+    """Format a user."""
+    return {
+        'id': user.id,
+        'name': user.name,
+        'is_owner': user.is_owner,
+        'is_active': user.is_active,
+        'system_generated': user.system_generated,
+        'credentials': [
+            {
+                'type': c.auth_provider_type,
+            } for c in user.credentials
+        ]
+    }