/*
 * LinShare is an open source filesharing software, part of the LinPKI software
 * suite, developed by Linagora.
 * 
 * Copyright (C) 2016 LINAGORA
 * 
 * This program is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Affero General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option) any
 * later version, provided you comply with the Additional Terms applicable for
 * LinShare software by Linagora pursuant to Section 7 of the GNU Affero General
 * Public License, subsections (b), (c), and (e), pursuant to which you must
 * notably (i) retain the display of the “LinShare™” trademark/logo at the top
 * of the interface window, the display of the “You are using the Open Source
 * and free version of LinShare™, powered by Linagora © 2009–2015. Contribute to
 * Linshare R&D by subscribing to an Enterprise offer!” infobox and in the
 * e-mails sent with the Program, (ii) retain all hypertext links between
 * LinShare and linshare.org, between linagora.com and Linagora, and (iii)
 * refrain from infringing Linagora intellectual property rights over its
 * trademarks and commercial brands. Other Additional Terms apply, see
 * <http://www.linagora.com/licenses/> for more details.
 * 
 * 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 and
 * its applicable Additional Terms for LinShare along with this program. If not,
 * see <http://www.gnu.org/licenses/> for the GNU Affero General Public License
 * version 3 and <http://www.linagora.com/licenses/> for the Additional Terms
 * applicable to LinShare software.
 */
package org.linagora.linshare.core.batches.impl;

import java.util.List;

import org.linagora.linshare.core.domain.constants.LinShareConstants;
import org.linagora.linshare.core.domain.entities.AbstractDomain;
import org.linagora.linshare.core.domain.entities.Account;
import org.linagora.linshare.core.domain.entities.DomainQuota;
import org.linagora.linshare.core.exception.BatchBusinessException;
import org.linagora.linshare.core.exception.BusinessException;
import org.linagora.linshare.core.job.quartz.Context;
import org.linagora.linshare.core.job.quartz.DomainBatchResultContext;
import org.linagora.linshare.core.repository.AbstractDomainRepository;
import org.linagora.linshare.core.repository.AccountRepository;
import org.linagora.linshare.core.repository.DomainQuotaRepository;

public class ComputeSubDomainQuotaBatchImpl extends GenericBatchImpl {

	private final AbstractDomainRepository domainRepository;
	
	private final DomainQuotaRepository domainQuotaRepository;

	public ComputeSubDomainQuotaBatchImpl(
			AccountRepository<Account> accountRepository,
			AbstractDomainRepository domainRepository, DomainQuotaRepository domainQuotaRepository) {
		super(accountRepository);
		this.domainRepository = domainRepository;
		this.domainQuotaRepository = domainQuotaRepository;
	}

	@Override
	public List<String> getAll() {
		logger.info(getClass().toString() + " job starting ...");
		// First all top domains (aka root domain children.
		List<String> entries = domainRepository.getAllSubDomainIdentifiers(LinShareConstants.rootDomainIdentifier);
		// Then the root domain.
		entries.add(LinShareConstants.rootDomainIdentifier);
		logger.info(entries.size() + " domain(s) have been found.");
		return entries;
	}

	@Override
	public Context execute(String identifier, long total, long position)
			throws BatchBusinessException, BusinessException {
		AbstractDomain domain = domainRepository.findById(identifier);
		Context context = new DomainBatchResultContext(domain);
		try {
			logDebug(total, position, "processing domain : " + domain.toString());
			Long subdomains = domainQuotaRepository.sumOfCurrentValueForSubdomains(domain);
			DomainQuota quota = domainQuotaRepository.find(domain);
			quota.setCurrentValueForSubdomains(subdomains);
			domainQuotaRepository.update(quota);
		} catch (BusinessException businessException) {
			logError(total, position,
					"Error while trying to update domain: " + domain.toString());
			logger.info("Error occured while processing quota domain :", businessException);
			BatchBusinessException exception = new BatchBusinessException(
					context, "Error while trying to updating domain.");
			exception.setBusinessException(businessException);
			throw exception;
		}
		
		return context;
	}

	@Override
	public void notify(Context context, long total, long position) {
		DomainBatchResultContext domainContext = (DomainBatchResultContext) context;
		logDebug(total, position, "Subdomains used space updated for domain : " + domainContext.getResource().toString());
	}

	@Override
	public void notifyError(BatchBusinessException exception, String identifier, long total, long position) {
		DomainBatchResultContext domainContext = (DomainBatchResultContext) exception.getContext();
		String ressource = domainContext.getResource().toString();
		logError(total, position, "Subdomains used space update failed for domain : " + ressource);
		logger.error(
				"Error occured while updating Subdomains used space for domain "+ ressource + ". BatchBusinessException ",
				exception);
	}

	@Override
	public void terminate(List<String> all, long errors, long unhandled_errors, long total, long processed) {
		long success = total - errors - unhandled_errors;
		logger.info(success + " domain(s) have been updated.");
		if (errors > 0) {
			logger.error(errors + " domain(s) failed to be updated.");
		}
		if (unhandled_errors > 0) {
			logger.error(unhandled_errors
					+ " domain(s) failed to be updated (unhandled error).");
		}
		logger.info(getClass().toString() + " job terminated.");
	}

}