[Rt-commit] rt branch, admin_ui, updated. 055f36e32dd118a53ffb99ef00cd179de872c9ad

[sunnavy at bestpractical.com]

The branch, admin_ui has been updated
       via  055f36e32dd118a53ffb99ef00cd179de872c9ad (commit)
      from  019f8763c6a758d60eefe758cb0345918f92765e (commit)

Summary of changes:
 lib/RT/Action/EditUserMemberships.pm |  128 ++++++++++++++++++++++++++++++++++
 1 files changed, 128 insertions(+), 0 deletions(-)
 create mode 100644 lib/RT/Action/EditUserMemberships.pm

- Log -----------------------------------------------------------------
commit 055f36e32dd118a53ffb99ef00cd179de872c9ad
Author: sunnavy <sunnavy at bestpractical.com>
Date:   Wed Dec 23 14:32:51 2009 +0800

    add action EditUserMemberships

diff --git a/lib/RT/Action/EditUserMemberships.pm b/lib/RT/Action/EditUserMemberships.pm
new file mode 100644
index 0000000..5cf0a36
--- /dev/null
+++ b/lib/RT/Action/EditUserMemberships.pm
@@ -0,0 +1,128 @@
+
+use strict;
+use warnings;
+
+package RT::Action::EditUserMemberships;
+use base qw/RT::Action Jifty::Action/;
+use Scalar::Defer;
+
+__PACKAGE__->mk_accessors('object');
+
+sub arguments {
+    my $self = shift;
+    return {} unless $self->object;
+
+    my $args = {};
+    $args->{object_id} = {
+        render_as     => 'hidden',
+        default_value => $self->object->id,
+    };
+    $args->{object_type} = {
+        render_as     => 'hidden',
+        default_value => ref $self->object,
+    };
+
+    $args->{groups} = {
+        render_as        => 'Checkboxes',
+        default_value    => defer { $self->default_value },
+        available_values => defer { $self->available_values },
+    };
+    return $args;
+}
+
+=head2 take_action
+
+=cut
+
+sub take_action {
+    my $self = shift;
+
+    my $object_type = $self->argument_value('object_type');
+    return unless $object_type;
+    if ( $RT::Model::ACE::OBJECT_TYPES{$object_type} ) {
+        my $object    = $object_type->new;
+        my $object_id = $self->argument_value('object_id');
+        $object->load($object_id);
+        unless ( $object->id ) {
+            Jifty->log->error("couldn't load $object_type #$object_id");
+            return;
+        }
+
+        $self->object($object);
+    }
+    else {
+        Jifty->log->error("object type '$object_type' is incorrect");
+        return;
+    }
+
+    my $value = $self->argument_value('groups');
+    my @groups;
+    if ( UNIVERSAL::isa( $value, 'ARRAY' ) ) {
+        @groups = @$value;
+    }
+    else {
+        @groups = $value;
+    }
+    @groups = grep { $_ && /^\d+$/ } @groups;
+
+    my $current_groups = $self->default_value();
+    my %current_groups = map { $_ => 1 } @$current_groups;
+    my %groups         = map { $_ => 1 } @groups;
+
+    my %group_objects;
+    for my $id ( @$current_groups, @groups ) {
+        next if $group_objects{$id};
+        my $group = RT::Model::Group->new;
+        $group->load($id);
+        $group_objects{$id} = $group;
+    }
+
+    for my $group ( keys %current_groups ) {
+        next if $groups{$group};
+
+        my ( $val, $msg ) =
+          $group_objects{$group}->delete_member( $self->object->id );
+        Jifty->log->error($msg) unless $val;
+    }
+
+    for my $group ( keys %groups ) {
+        next if $current_groups{$group};
+        my ( $val, $msg ) =
+          $group_objects{$group}->add_member( $self->object->id );
+        Jifty->log->error($msg) unless $val;
+    }
+
+    $self->report_success;
+    return 1;
+}
+
+=head2 report_success
+
+=cut
+
+sub report_success {
+    my $self = shift;
+
+    # Your success message here
+    $self->result->message('Success');
+}
+
+sub available_values {
+    my $self   = shift;
+    my $object = $self->object;
+    my $groups = RT::Model::GroupCollection->new;
+    $groups->limit_to_user_defined_groups;
+    return [ map { { display => $_->name, value => $_->id } }
+          @{ $groups->items_array_ref } ];
+}
+
+sub default_value {
+    my $self      = shift;
+    my $is_member = RT::Model::GroupCollection->new;
+    $is_member->limit_to_user_defined_groups;
+    $is_member->with_member( principal => $self->object->id );
+    return [ map { $_->id } @{ $is_member->items_array_ref } ];
+}
+
+1;
+

-----------------------------------------------------------------------