Add instance methods

The Agency class can `grant` and `revoke` roles for a given Agent and User.
Since it is primarily used in context of both an Agent and User, we add
convenience methods so that one can grant or revoke a role simply by passing
the desired role to the instance.

1 files changed, 10 insertions, 0 deletions

diff --git a/app/models/agency.rb b/app/models/agency.rb
index fe81687..03e6e5a 100644
--- a/app/models/agency.rb
+++ b/app/models/agency.rb
@@ -19,4 +19,14 @@ class Agency < ApplicationRecord
       r&.public_send("#{role}=", false) && r&.save
     end
   end
+
+  # Grant role to current user in current agent
+  def grant(role)
+    self.class.grant(agent, user, role)
+  end
+
+  # Revoke role from current user in current agent
+  def revoke(role)
+    self.class.revoke(agent, user, role)
+  end
 end