Changeset 851

Timestamp:

04/04/10 03:41:53 (14 years ago)

Author:

nanardon

Message:

• add primary subcommand to user/group()

File:

• LATMOS-Accounts/lib/LATMOS/Accounts/Cli.pm (modified) (2 diffs)

LATMOS-Accounts/lib/LATMOS/Accounts/Cli.pm

@@ -325 +325 @@
     if (lc($otype) eq 'user') {
         $objenv->add_func('group', {
-            help => 'group add|remove goupname',
+            help => 'group add|remove|primary goupname',
             code => sub {
                 my ($env, $action, @groups) = @_;
                 foreach my $obj (@{$env->{_objects}}) {
-                    my %gr;
-                    foreach ($obj->get_attributes('memberOf')) {
-                        $gr{$_} = 1;
+                    if ($action eq 'primary') {
+                        my $gid = $groups[0];
+                        if ($gid !~ /^\d/) {
+                            my $gobj = $env->base->get_object('group', $gid) or
+                            do {
+                                print $OUT "Cannot find group $gid";
+                                return;
+                            };
+                            $gid = $gobj->get_attributes('gidNumber');
+                        }
+                        $obj->set_c_fields('gidNumber', $gid);
+                    } else {
+                        my %gr;
+                        foreach ($obj->get_attributes('memberOf')) {
+                            $gr{$_} = 1;
+                        }
+                        if ($action eq 'add') {
+                            $gr{$_} = 1 foreach(@groups);
+                        } elsif ($action eq 'remove') {
+                            delete($gr{$_}) foreach(@groups);
+                        } else {
+                            print $OUT 'invalid action';
+                            return;
+                        }
+                        defined $obj->set_c_fields('memberOf' => [ keys %gr ]) or do {
+                            print $OUT "cannot set memberOf attributes for " .
+                            $obj->id;
+                            return;
+                        };
                     }
-                    if ($action eq 'add') {
-                        $gr{$_} = 1 foreach(@groups);
-                    } elsif ($action eq 'remove') {
-                        delete($gr{$_}) foreach(@groups);
-                    } else {
-                        print $OUT 'invalid action';
-                        return;
-                    }
-                    defined $obj->set_c_fields('memberOf' => [ keys %gr ]) or do {
-                        print $OUT "cannot set memberOf attributes for " .
-                        $obj->id;
-                        return;
-                    };
                 }
                 $env->base->commit;
@@ -351 +364 @@
             completion => sub {
                 if (!$_[2]) {
-                    return (qw(add remove));
+                    return (qw(add remove primary));
                 } else {
                     return $_[0]->base->search_objects('group');