| // 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 com.google.gerrit.common.data.GroupDescription; |
| import com.google.gerrit.common.data.GroupReference; |
| import com.google.gerrit.common.errors.NoSuchGroupException; |
| 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.GroupControl; |
| import com.google.inject.Inject; |
| import com.google.inject.assistedinject.Assisted; |
| 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 GroupControl.Factory groupControlFactory; |
| |
| @Inject |
| public AccountGroupUUIDHandler( |
| final GroupBackend groupBackend, |
| final GroupControl.Factory groupControlFactory, |
| @Assisted final CmdLineParser parser, |
| @Assisted final OptionDef option, |
| @Assisted final Setter<AccountGroup.UUID> setter) { |
| super(parser, option, setter); |
| this.groupBackend = groupBackend; |
| this.groupControlFactory = groupControlFactory; |
| } |
| |
| @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 legacy AccountGroup.Id. |
| if (n.matches("^[1-9][0-9]*$")) { |
| try { |
| AccountGroup.Id legacyId = AccountGroup.Id.parse(n); |
| uuid = groupControlFactory.controlFor(legacyId).getGroup().getGroupUUID(); |
| setter.addValue(uuid); |
| return 1; |
| } catch (IllegalArgumentException | NoSuchGroupException e) { |
| // Ignored |
| } |
| } |
| |
| GroupReference group = GroupBackends.findExactSuggestion(groupBackend, n); |
| if (group == null) { |
| throw new CmdLineException(owner, "Group \"" + n + "\" does not exist"); |
| } |
| setter.addValue(group.getUUID()); |
| return 1; |
| } |
| |
| @Override |
| public final String getDefaultMetaVariable() { |
| return "GROUP"; |
| } |
| } |