diff options
Diffstat (limited to 'app/models/instance.rb')
-rw-r--r-- | app/models/instance.rb | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/app/models/instance.rb b/app/models/instance.rb index 0839894d9..556ee9f70 100644 --- a/app/models/instance.rb +++ b/app/models/instance.rb @@ -7,8 +7,9 @@ class Instance def initialize(resource) @domain = resource.domain - @accounts_count = resource.is_a?(DomainBlock) ? nil : resource.accounts_count - @domain_block = resource.is_a?(DomainBlock) ? resource : DomainBlock.find_by(domain: domain) + @accounts_count = resource.respond_to?(:accounts_count) ? resource.accounts_count : nil + @domain_block = resource.is_a?(DomainBlock) ? resource : DomainBlock.rule_for(domain) + @domain_allow = resource.is_a?(DomainAllow) ? resource : DomainAllow.rule_for(domain) @updated_at = resource.is_a?(DomainBlock) ? resource.updated_at : 0 end |