1 files changed, 16 insertions, 2 deletions

diff --git a/Digitigrade/Policy/BlockInstance.php b/Digitigrade/Policy/BlockInstance.php
index 5f902f2..053418c 100644
--- a/Digitigrade/Policy/BlockInstance.php
+++ b/Digitigrade/Policy/BlockInstance.php
@@ -2,6 +2,7 @@
 namespace Digitigrade\Policy;
 
 use Digitigrade\Model\Actor;
+use Digitigrade\Model\Instance;
 use Digitigrade\Model\Interaction;
 use Digitigrade\Model\Note;
 use Digitigrade\Model\PushableModel;
@@ -15,8 +16,11 @@ class BlockInstance extends Policy {
         $this->blockedHosts = $hosts;
     }
 
-    private function isBlocked(PushableModel $obj): bool {
-        $objHost = hostname_from_uri($obj->uri);
+    private function isBlocked(PushableModel|Instance|null $obj): bool {
+        if ($obj == null) {
+            return false;
+        }
+        $objHost = hostname_from_uri($obj instanceof Instance ? $obj->domain : $obj->uri);
         foreach ($this->blockedHosts as $host) {
             if (str_ends_with($objHost, $host)) {
                 return true;
@@ -34,4 +38,14 @@ class BlockInstance extends Policy {
     public function checkNote(Note $note): bool {
         return !$this->isBlocked($note);
     }
+
+    public function checkActorFederation(Actor $actor, ?Instance $instance): bool {
+        return !$this->isBlocked($instance);
+    }
+    public function checkInteractionFederation(Interaction $interaction, ?Instance $instance): bool {
+        return !$this->isBlocked($instance);
+    }
+    public function checkNoteFederation(Note $note, ?Instance $instance): bool {
+        return !$this->isBlocked($instance);
+    }
 }
\ No newline at end of file