// 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() != 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());
  }
}
