From 82462b24619f98afbeb50788bc281751f989eed2 Mon Sep 17 00:00:00 2001
From: Kyle Drake <kyle@kyledrake.net>
Date: Wed, 18 Dec 2019 20:43:04 -0800
Subject: [PATCH] whitelisting for blackbox

---
 migrations/114_site_blackbox_whitelist.rb | 9 +++++++++
 models/site.rb                            | 5 ++++-
 2 files changed, 13 insertions(+), 1 deletion(-)
 create mode 100644 migrations/114_site_blackbox_whitelist.rb

diff --git a/migrations/114_site_blackbox_whitelist.rb b/migrations/114_site_blackbox_whitelist.rb
new file mode 100644
index 00000000..cbce4d7b
--- /dev/null
+++ b/migrations/114_site_blackbox_whitelist.rb
@@ -0,0 +1,9 @@
+Sequel.migration do
+  up {
+    DB.add_column :sites, :blackbox_whitelisted, :boolean, default: false
+  }
+
+  down {
+    DB.drop_column :sites, :blackbox_whitelisted
+  }
+end
diff --git a/models/site.rb b/models/site.rb
index bb636223..181a3691 100644
--- a/models/site.rb
+++ b/models/site.rb
@@ -1684,7 +1684,10 @@ class Site < Sequel::Model
     end
 
     if pathname.extname.match(HTML_REGEX) && defined?(BlackBox)
-      BlackBox.tos_violation_check self, uploaded
+      begin
+        BlackBox.tos_violation_check self, uploaded
+      rescue
+      end
     end
 
     relative_path_dir = Pathname(relative_path).dirname