Add black/whitelist feature to browser-based login

1 files changed, 36 insertions, 0 deletions

diff --git a/src/global.cpp b/src/global.cpp
index fd6e321..4efccad 100644
--- a/src/global.cpp
+++ b/src/global.cpp
@@ -80,3 +80,39 @@ QImage Global::getConfigGradient()
     }
     return img;
 }
+
+QStringList loadUrlList(const QString &file)
+{
+	QStringList stringList;
+	QFile textFile(file);
+	if (!textFile.open(QFile::ReadOnly)) {
+		QTextStream(stdout) << "Cannot open URL list\n";
+		return QStringList();
+	}
+	QTextStream textStream(&textFile);
+	while (true)
+	{
+		QString line = textStream.readLine();
+		if (line.isNull())
+			break;
+		else
+			stringList.append(line);
+	}
+	return stringList;
+}
+
+QStringList Global::urlBlacklist()
+{
+    auto path = Settings::urlBlacklistFile();
+    if (!QFile::exists(path))
+        return QStringList();
+    return loadUrlList(path);
+}
+
+QStringList Global::urlWhitelist()
+{
+    auto path = Settings::urlWhitelistFile();
+    if (!QFile::exists(path))
+        return QStringList();
+    return loadUrlList(path);
+}