From 9faf40e41aee936d90948595869d27f3b9c4ba3b Mon Sep 17 00:00:00 2001
From: Stefan Kremser <stef.1497@gmail.com>
Date: Wed, 8 May 2019 18:37:34 +0200
Subject: [PATCH] Only include the libraries that are used

---
 esp8266_deauther/DisplayUI.h | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/esp8266_deauther/DisplayUI.h b/esp8266_deauther/DisplayUI.h
index 6023feb..f2f5e17 100644
--- a/esp8266_deauther/DisplayUI.h
+++ b/esp8266_deauther/DisplayUI.h
@@ -9,12 +9,20 @@
 #include "Scan.h"
 #include "Attack.h"
 
-#include <Wire.h>
-#include <SPI.h>
-#include <SH1106Wire.h>
-#include <SSD1306Wire.h>
-#include <SH1106Spi.h>
-#include <SSD1306Spi.h>
+// ===== adjustable ===== //
+#if defined(SSD1306_I2C)
+  #include <Wire.h>
+  #include <SSD1306Wire.h>
+#elif defined(SSD1306_SPI)
+  #include <SPI.h>
+  #include <SSD1306Spi.h>
+#elif defined(SH1106_I2C)
+  #include <Wire.h>
+  #include <SH1106Wire.h>
+#elif defined(SH1106_SPI)
+  #include <SPI.h>
+  #include <SH1106Spi.h>
+#endif /* if defined(SSD1306_I2C) */
 
 #include <SimpleButton.h>