Add CustomRoleObj method for loading by GroupType

With this we can easily go from the output of ->Roles to an RT::CustomRole
object.

1 files changed, 26 insertions, 0 deletions

diff --git a/lib/RT/Record/Role/Roles.pm b/lib/RT/Record/Role/Roles.pm
index ffcfa5240d..25179a8fc6 100644
--- a/lib/RT/Record/Role/Roles.pm
+++ b/lib/RT/Record/Role/Roles.pm
@@ -851,4 +851,30 @@ you see faster create times.
 
 =cut
 
+=head2 CustomRoleObj
+
+Returns the L<RT::CustomRole> object for this role if and only if it's
+backed by a custom role. If it's a core role (e.g. Ticket Requestors),
+returns C<undef>.
+
+=cut
+
+sub CustomRoleObj {
+    my $self = shift;
+    my $name = shift;
+
+    if (my ($id) = $name =~ /^RT::CustomRole-(\d+)$/) {
+        my $role = RT::CustomRole->new($self->CurrentUser);
+        my ( $ret, $msg ) = $role->Load($id);
+        if ( $ret ) {
+            return $role;
+        }
+        else {
+            RT->Logger->warning("Couldn't load custom role #$id: $msg");
+        }
+    }
+
+    return undef;
+}
+
 1;