java/com/google/gerrit/server/account/AbstractRealm.java

// Copyright (C) 2014 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.account;

import com.google.common.base.Strings;
import com.google.common.collect.Sets;
import com.google.gerrit.extensions.client.AccountFieldName;
import com.google.gerrit.server.IdentifiedUser;
import com.google.gerrit.server.account.externalids.ExternalId;
import com.google.gerrit.server.mail.send.EmailSender;
import com.google.inject.Inject;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

/** Basic implementation of {@link Realm}. */
public abstract class AbstractRealm implements Realm {
  private EmailSender emailSender;

  @Inject(optional = true)
  void setEmailSender(EmailSender emailSender) {
    this.emailSender = emailSender;
  }

  @Override
  public Set<AccountFieldName> getEditableFields() {
    Set<AccountFieldName> fields = new HashSet<>();
    for (AccountFieldName n : AccountFieldName.values()) {
      if (allowsEdit(n)) {
        if (n == AccountFieldName.REGISTER_NEW_EMAIL) {
          if (emailSender != null && emailSender.isEnabled()) {
            fields.add(n);
          }
        } else {
          fields.add(n);
        }
      }
    }
    return fields;
  }

  @Override
  public boolean hasEmailAddress(IdentifiedUser user, String email) {
    for (ExternalId ext : user.state().externalIds()) {
      if (email != null && email.equalsIgnoreCase(ext.email())) {
        return true;
      }
    }
    return false;
  }

  @Override
  public Set<String> getEmailAddresses(IdentifiedUser user) {
    Collection<ExternalId> ids = user.state().externalIds();
    Set<String> emails = Sets.newHashSetWithExpectedSize(ids.size());
    for (ExternalId ext : ids) {
      if (!Strings.isNullOrEmpty(ext.email())) {
        emails.add(ext.email());
      }
    }
    return emails;
  }
}