// Copyright (C) 2013 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.gerrit.extensions.registration.DynamicMap;
import com.google.gerrit.extensions.restapi.AcceptsCreate;
import com.google.gerrit.extensions.restapi.ChildCollection;
import com.google.gerrit.extensions.restapi.IdString;
import com.google.gerrit.extensions.restapi.ResourceNotFoundException;
import com.google.gerrit.extensions.restapi.RestView;
import com.google.gerrit.server.CurrentUser;
import com.google.gerrit.server.account.AccountResource.Email;
import com.google.inject.Inject;
import com.google.inject.Provider;
import com.google.inject.Singleton;

@Singleton
public class Emails
    implements ChildCollection<AccountResource, AccountResource.Email>,
        AcceptsCreate<AccountResource> {
  private final DynamicMap<RestView<AccountResource.Email>> views;
  private final GetEmails list;
  private final Provider<CurrentUser> self;
  private final CreateEmail.Factory createEmailFactory;

  @Inject
  Emails(
      DynamicMap<RestView<AccountResource.Email>> views,
      GetEmails list,
      Provider<CurrentUser> self,
      CreateEmail.Factory createEmailFactory) {
    this.views = views;
    this.list = list;
    this.self = self;
    this.createEmailFactory = createEmailFactory;
  }

  @Override
  public RestView<AccountResource> list() {
    return list;
  }

  @Override
  public AccountResource.Email parse(AccountResource rsrc, IdString id)
      throws ResourceNotFoundException {
    if (!self.get().hasSameAccountId(rsrc.getUser())
        && !self.get().getCapabilities().canAdministrateServer()) {
      throw new ResourceNotFoundException();
    }

    if ("preferred".equals(id.get())) {
      String email = rsrc.getUser().getAccount().getPreferredEmail();
      if (Strings.isNullOrEmpty(email)) {
        throw new ResourceNotFoundException();
      }
      return new AccountResource.Email(rsrc.getUser(), email);
    } else if (rsrc.getUser().hasEmailAddress(id.get())) {
      return new AccountResource.Email(rsrc.getUser(), id.get());
    } else {
      throw new ResourceNotFoundException();
    }
  }

  @Override
  public DynamicMap<RestView<Email>> views() {
    return views;
  }

  @SuppressWarnings("unchecked")
  @Override
  public CreateEmail create(AccountResource parent, IdString email) {
    return createEmailFactory.create(email.get());
  }
}
