From cd15c6473139c74bb3be20c5c80948a7b822b099 Mon Sep 17 00:00:00 2001
From: Son NK <>
Date: Thu, 15 Oct 2020 16:01:44 +0200
Subject: [PATCH] Add PREMIUM_ALIAS_DOMAINS config

---
 app/config.py | 9 +++++++++
 example.env   | 3 +++
 2 files changed, 12 insertions(+)

diff --git a/app/config.py b/app/config.py
index c794ed60..f25b4f8f 100644
--- a/app/config.py
+++ b/app/config.py
@@ -100,6 +100,15 @@ else:
 
 ALIAS_DOMAINS = [d.lower().strip() for d in ALIAS_DOMAINS]
 
+if "PREMIUM_ALIAS_DOMAINS" in os.environ:
+    PREMIUM_ALIAS_DOMAINS = eval(
+        os.environ["PREMIUM_ALIAS_DOMAINS"]
+    )  # ["domain1.com", "domain2.com"]
+else:
+    PREMIUM_ALIAS_DOMAINS = []
+
+PREMIUM_ALIAS_DOMAINS = [d.lower().strip() for d in PREMIUM_ALIAS_DOMAINS]
+
 # the alias domain used when creating the first alias for user
 FIRST_ALIAS_DOMAIN = os.environ.get("FIRST_ALIAS_DOMAIN") or EMAIL_DOMAIN
 
diff --git a/example.env b/example.env
index bbc7e1c4..a32277e3 100644
--- a/example.env
+++ b/example.env
@@ -27,6 +27,9 @@ OTHER_ALIAS_DOMAINS=["domain1.com", "domain2.com"]
 # domains that can be used to create aliases. If set, override OTHER_ALIAS_DOMAINS
 ALIAS_DOMAINS=["domain1.com", "domain2.com"]
 
+# (optional) domains that are only available to premium accounts
+PREMIUM_ALIAS_DOMAINS=["premium.com"]
+
 # the alias domain used when creating the first alias for user, default to EMAIL_DOMAIN if not set
 # FIRST_ALIAS_DOMAIN = another-domain.com