text: /**
   *    Copyright (C) 2016 MongoDB Inc.
   *
   *    This program is free software: you can redistribute it and/or  modify
   *    it under the terms of the GNU Affero General Public License, version 3,
   *    as published by the Free Software Foundation.
   *
   *    This program is distributed in the hope that it will be useful,
   *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   *    GNU Affero General Public License for more details.
   *
   *    You should have received a copy of the GNU Affero General Public License
   *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
   *
   *    As a special exception, the copyright holders give permission to link the
   *    code of portions of this program with the OpenSSL library under certain
   *    conditions as described in each individual source file and distribute
   *    linked combinations including the program with the OpenSSL library. You
   *    must comply with the GNU Affero General Public License in all respects for
   *    all of the code used other than as permitted herein. If you modify file(s)
   *    with this exception, you may extend this exception to your version of the
   *    file(s), but you are not obligated to do so. If you do not wish to do so,
   *    delete this exception statement from your version. If you delete this
   *    exception statement from all source files in the program, then also delete
   *    it in the license file.
   */
  
  #define MONGO_LOG_DEFAULT_COMPONENT ::mongo::logger::LogComponent::kCommand
  
  #include "mongo/platform/basic.h"
  
  #include "mongo/db/auth/authorization_session.h"
  #include "mongo/db/commands.h"
  #include "mongo/db/write_concern_options.h"
  #include "mongo/s/catalog/type_shard.h"
  #include "mongo/s/client/shard_registry.h"
  #include "mongo/s/grid.h"
  #include "mongo/s/request_types/add_shard_to_zone_request_type.h"
  #include "mongo/util/log.h"
  
  namespace mongo {
  namespace {
  
  const ReadPreferenceSetting kPrimaryOnlyReadPreference{ReadPreference::PrimaryOnly};
  const WriteConcernOptions kMajorityWriteConcern(WriteConcernOptions::kMajority,
                                                  // Note: Even though we're setting UNSET here,
                                                  // kMajority implies JOURNAL if journaling is
                                                  // supported by mongod and
                                                  // writeConcernMajorityJournalDefault is set to true
                                                  // in the ReplSetConfig.
                                                  WriteConcernOptions::SyncMode::UNSET,
                                                  Seconds(15));
  
  /**
   * {
   *   addShardToZone: <string shardName>,
   *   zone: <string zoneName>
   * }
   */
  class AddShardToZoneCmd : public BasicCommand {
  public:
      AddShardToZoneCmd() : BasicCommand("addShardToZone", "addshardtozone") {}
  
      AllowedOnSecondary secondaryAllowed() const override {
          return AllowedOnSecondary::kAlways;
      }
  
      bool adminOnly() const override {
          return true;
      }
  
      bool supportsWriteConcern(const BSONObj& cmd) const override {
          return false;
      }
  
      std::string help() const override {
          return "adds a shard to zone";
      }
  
      Status checkAuthForCommand(Client* client,
                                 const std::string& dbname,
                                 const BSONObj& cmdObj) const override {
          if (!AuthorizationSession::get(client)->isAuthorizedForActionsOnResource(
                  ResourcePattern::forExactNamespace(ShardType::ConfigNS), ActionType::update)) {
              return Status(ErrorCodes::Unauthorized, "Unauthorized");
          }
          return Status::OK();
      }
  
      bool run(OperationContext* opCtx,
               const std::string& dbname,
               const BSONObj& cmdObj,
               BSONObjBuilder& result) override {
          auto parsedRequest = uassertStatusOK(AddShardToZoneRequest::parseFromMongosCommand(cmdObj));
  
          BSONObjBuilder cmdBuilder;
          parsedRequest.appendAsConfigCommand(&cmdBuilder);
          cmdBuilder.append("writeConcern", kMajorityWriteConcern.toBSON());
  
          auto configShard = Grid::get(opCtx)->shardRegistry()->getConfigShard();
          auto cmdResponseStatus = uassertStatusOK(
              configShard->runCommandWithFixedRetryAttempts(opCtx,
                                                            kPrimaryOnlyReadPreference,
                                                            "admin",
                                                            cmdBuilder.obj(),
                                                            Shard::RetryPolicy::kIdempotent));
          uassertStatusOK(cmdResponseStatus.commandStatus);
          return true;
      }
  
  } addShardToZoneCmd;
  
  }  // namespace
  }  // namespace mongo
  