From b1ab30ddde6dd93ce786f61b362cd58df9eed04e Mon Sep 17 00:00:00 2001
From: Lukas Winkler <git@lw1.at>
Date: Fri, 9 Apr 2021 16:26:58 +0200
Subject: [PATCH] add useragent and redis socket

---
 config.py |  1 +
 server.py | 12 ++++++++----
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/config.py b/config.py
index cd6945f..e534569 100644
--- a/config.py
+++ b/config.py
@@ -1,2 +1,3 @@
 user_agent = "HNClient (in development)"
 debug = True
+# redis_socket = "/run/redis-hnreader/redis-server.sock"
diff --git a/server.py b/server.py
index 08e3674..343e0cb 100644
--- a/server.py
+++ b/server.py
@@ -5,13 +5,18 @@ from starlette.requests import Request
 from starlette.responses import JSONResponse, Response
 from starlette.routing import Route
 
-from config import debug
+from config import debug, user_agent, redis_socket
 from hnapi import HNClient
 from reader import Reader
 
 conn = aiohttp.TCPConnector(ttl_dns_cache=60 * 10)
-session = aiohttp.ClientSession(connector=conn)
-r = Redis()
+session = aiohttp.ClientSession(connector=conn, headers={
+    "User-Agent": user_agent
+})
+if redis_socket:
+    r = Redis(unix_socket_path=redis_socket)
+else:
+    r = Redis()
 reader = Reader()
 
 api = HNClient(session, r)
@@ -51,4 +56,3 @@ app = Starlette(debug=debug, routes=[
     Route('/api/read/{item_id:int}', read),
     Route('/api/topstories', topstories),
 ])
-