From 078b20eb387084f2d05052c9e18d131c9f4168dc Mon Sep 17 00:00:00 2001
From: mwiegand <mwiegand@seibert-media.net>
Date: Wed, 1 Jun 2022 20:05:49 +0200
Subject: [PATCH] wip

---
 steam_chat_viewer/__init__.py         | 2 ++
 steam_chat_viewer/templates/home.html | 6 +++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/steam_chat_viewer/__init__.py b/steam_chat_viewer/__init__.py
index 0da654f..77b656a 100644
--- a/steam_chat_viewer/__init__.py
+++ b/steam_chat_viewer/__init__.py
@@ -2,6 +2,7 @@
 
 from os import environ
 from flask import Flask, request, render_template
+from urllib.parse import urlparse
 
 from .pg import query, select
 
@@ -49,4 +50,5 @@ def home():
             '''
         ),
         messages=messages,
+        urlparse=urlparse,
     )
diff --git a/steam_chat_viewer/templates/home.html b/steam_chat_viewer/templates/home.html
index 3f274db..9d6077a 100644
--- a/steam_chat_viewer/templates/home.html
+++ b/steam_chat_viewer/templates/home.html
@@ -26,7 +26,11 @@
             at
             <span class=date>{{ message['date'] }}</span>
           </div>
-          <div class=body>{{ message['message'].replace('\n', '<br/>') |safe}}</div>
+          {% if message['message'].startswith('https://steamuserimages-a.akamaihd.net/'): -%}
+            <img src="/steamuserimages/{{ urlparse(message['message']).path.strip('/').replace('/', '_') }}"
+          {% else: -%}
+            <div class=body>{{ message['message'].replace('\n', '<br/>') |safe}}</div>
+          {% endif -%}
         </li>
 {% endfor -%}
       </ul>