Add youtu.be as a default domain (#251)

aquelemiguel · Jul 12, 2024 · a6c1e88 · a6c1e88
1 parent fcf9338
commit a6c1e88
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/guild/settings.rs b/src/guild/settings.rs
@@ -12,7 +12,7 @@ use std::{
 use crate::errors::ParrotError;
 
 const DEFAULT_SETTINGS_PATH: &str = "data/settings";
-const DEFAULT_ALLOWED_DOMAINS: [&str; 1] = ["youtube.com"];
+const DEFAULT_ALLOWED_DOMAINS: [&str; 2] = ["youtube.com", "youtu.be"];
 
 lazy_static! {
     static ref SETTINGS_PATH: String =
@@ -103,7 +103,11 @@ impl GuildSettings {
         }
 
         if self.allowed_domains.is_empty() && self.banned_domains.is_empty() {
-            self.allowed_domains.insert(String::from("youtube.com"));
+            self.allowed_domains = DEFAULT_ALLOWED_DOMAINS
+                .iter()
+                .map(|d| d.to_string())
+                .collect();
+
             self.banned_domains.clear();
         }
     }