From 3e19deb8eb6e9303fac01389486f92a8ce388be0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Timoth=C3=A9e=20Mazzucotelli?= <pawamoy@pm.me>
Date: Fri, 27 Mar 2020 19:26:01 +0100
Subject: [PATCH] fix: Re-apply pywal color theme if any

---
 src/aria2p/interface.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/aria2p/interface.py b/src/aria2p/interface.py
index cb06caa..673632e 100644
--- a/src/aria2p/interface.py
+++ b/src/aria2p/interface.py
@@ -17,8 +17,10 @@
 # pylint: disable=invalid-name
 
 import os
+import sys
 import time
 from collections import defaultdict
+from pathlib import Path
 
 from asciimatics.event import KeyboardEvent, MouseEvent
 from asciimatics.screen import ManagedScreen, Screen
@@ -420,10 +422,22 @@ def run(self):
                         time.sleep(self.sleep)
                         self.frame = (self.frame + 1) % self.frames
                     logger.debug("Screen has resized")
+                    self.post_resize()
         except Exception as error:
             logger.exception(error)
             return False
 
+    def post_resize(self):
+        logger.debug(f"Running post-resize function")
+        logger.debug("Trying to re-apply pywal color theme")
+        wal_sequences = Path.home() / ".cache" / "wal" / "sequences"
+        try:
+            with wal_sequences.open("rb") as fd:
+                contents = fd.read()
+                sys.stdout.buffer.write(contents)
+        except Exception:
+            pass
+
     def update_select_sort_rows(self):
         self.select_sort_rows = self.columns_order