diff options
Diffstat (limited to 'java/com/google/gerrit/server/args4j/AccountGroupUUIDHandler.java')
-rw-r--r-- | java/com/google/gerrit/server/args4j/AccountGroupUUIDHandler.java | 92 |
1 files changed, 92 insertions, 0 deletions
diff --git a/java/com/google/gerrit/server/args4j/AccountGroupUUIDHandler.java b/java/com/google/gerrit/server/args4j/AccountGroupUUIDHandler.java new file mode 100644 index 0000000000..2dd0c7ab5b --- /dev/null +++ b/java/com/google/gerrit/server/args4j/AccountGroupUUIDHandler.java @@ -0,0 +1,92 @@ +// Copyright (C) 2009 The Android Open Source Project +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package com.google.gerrit.server.args4j; + +import static com.google.gerrit.util.cli.Localizable.localizable; + +import com.google.gerrit.common.data.GroupDescription; +import com.google.gerrit.common.data.GroupReference; +import com.google.gerrit.reviewdb.client.AccountGroup; +import com.google.gerrit.server.account.GroupBackend; +import com.google.gerrit.server.account.GroupBackends; +import com.google.gerrit.server.account.GroupCache; +import com.google.gerrit.server.group.InternalGroup; +import com.google.gerrit.server.group.InternalGroupDescription; +import com.google.inject.Inject; +import com.google.inject.assistedinject.Assisted; +import java.util.Optional; +import org.kohsuke.args4j.CmdLineException; +import org.kohsuke.args4j.CmdLineParser; +import org.kohsuke.args4j.OptionDef; +import org.kohsuke.args4j.spi.OptionHandler; +import org.kohsuke.args4j.spi.Parameters; +import org.kohsuke.args4j.spi.Setter; + +public class AccountGroupUUIDHandler extends OptionHandler<AccountGroup.UUID> { + private final GroupBackend groupBackend; + private final GroupCache groupCache; + + @Inject + public AccountGroupUUIDHandler( + final GroupBackend groupBackend, + @Assisted final CmdLineParser parser, + @Assisted final OptionDef option, + @Assisted final Setter<AccountGroup.UUID> setter, + GroupCache groupCache) { + super(parser, option, setter); + this.groupBackend = groupBackend; + this.groupCache = groupCache; + } + + @Override + public final int parseArguments(Parameters params) throws CmdLineException { + final String n = params.getParameter(0); + AccountGroup.UUID uuid = new AccountGroup.UUID(n); + if (groupBackend.handles(uuid)) { + GroupDescription.Basic d = groupBackend.get(uuid); + if (d != null) { + setter.addValue(uuid); + return 1; + } + } + + // Might be a numeric AccountGroup.Id. -> Internal group. + if (n.matches("^[1-9][0-9]*$")) { + try { + AccountGroup.Id groupId = AccountGroup.Id.parse(n); + Optional<InternalGroup> groupInternal = groupCache.get(groupId); + if (groupInternal.isPresent()) { + uuid = new InternalGroupDescription(groupInternal.get()).getGroupUUID(); + setter.addValue(uuid); + return 1; + } + } catch (IllegalArgumentException e) { + // Ignored + } + } + + GroupReference group = GroupBackends.findExactSuggestion(groupBackend, n); + if (group == null) { + throw new CmdLineException(owner, localizable("Group \"%s\" does not exist"), n); + } + setter.addValue(group.getUUID()); + return 1; + } + + @Override + public final String getDefaultMetaVariable() { + return "GROUP"; + } +} |