add new

google-assistant-webserver/app/main.py

@@ -0,0 +1,67 @@
+"""Main entrypoint: webserver handling commands to google assistant."""
+import json
+import logging
+import os
+from pathlib import Path
+
+from aiohttp import web
+
+from assistant import GoogleTextAssistant
+from auth import AuthHandler
+
+LOGGER = logging.getLogger()
+
+CLIENT_JSON = Path(os.path.join(os.path.dirname(os.path.abspath(__file__)), "client_secrets.json"))
+CRED_JSON = Path("/data/cred.json")
+if not os.path.isdir("/data"):
+    CRED_JSON = Path(os.path.join(os.path.dirname(os.path.abspath(__file__)), "cred.json"))
+
+routes = web.RouteTableDef()
+
+
+@routes.get("/broadcast")
+async def broadcast_message(request):
+    message = request.query.get("message", default="This is a test!")
+    text_query = "broadcast " + message
+    with GoogleTextAssistant("en-US", "HA_GA", "HA_GA_TEXT_SERVER", CRED_JSON) as assistant:
+        response_text, response_html = assistant.assist(text_query=text_query)
+    return web.Response(text=response_text)
+
+
+@routes.get("/command")
+async def command(request):
+    message = request.query.get("message", default="This is a test!")
+    with GoogleTextAssistant("en-US", "HA_GA", "HA_GA_TEXT_SERVER", CRED_JSON) as assistant:
+        response_text, response_html = assistant.assist(text_query=message)
+    return web.Response(text=response_text)
+
+
+@routes.get("/")
+async def index(request):
+    """Landingpage."""
+    if not CRED_JSON.exists():
+        raise web.HTTPFound("/auth")
+    html_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "index.html")
+    return web.FileResponse(html_file)
+
+
+@routes.get("/auth")
+async def auth(request):
+    """Authenticate with google."""
+    html_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "auth.html")
+    with open(html_file) as _file:
+        html = _file.read()
+    html = html.replace("[[AUTH_URL]]", request.app["auth"].auth_url)
+    return web.Response(text=html, content_type="text/html")
+
+
+app = web.Application()
+app.add_routes(routes)
+
+with CLIENT_JSON.open("r") as data:
+    user_data = json.load(data)["installed"]
+auth = AuthHandler(user_data, CRED_JSON)
+app.router.add_post("/token", auth.token)
+app["auth"] = auth
+
+web.run_app(app, port=5000)