/*
 * Copyright 2013 - 2014 Felix Müller
 *
 * This file is part of CodeQ Invest.
 *
 * CodeQ Invest is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * CodeQ Invest 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with CodeQ Invest.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.codeqinvest.quality.analysis;

import com.google.common.collect.Sets;
import lombok.extern.slf4j.Slf4j;
import org.codeqinvest.codechanges.scm.factory.ScmAvailabilityCheckerServiceFactory;
import org.codeqinvest.quality.Project;
import org.codeqinvest.quality.repository.ProjectRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Service;

import java.util.Set;

/**
 * This component handles all quality analyzer runs. Quality analyzer can
 * be scheduled for later execution triggered by cron expression or
 * they can be executed immediately. For scheduling analyzers this
 * implementation uses a {@link ThreadPoolTaskScheduler}.
 *
 * @author fmueller
 */
@Slf4j
@Service
public class DefaultQualityAnalyzerScheduler implements QualityAnalyzerScheduler {

  // TODO make this configurable
  private static final int DEFAULT_POOL_SIZE = 10;

  private final ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();

  private final ProjectRepository projectRepository;
  private final ViolationsCalculatorService violationsCalculatorService;
  private final ScmAvailabilityCheckerServiceFactory scmAvailabilityCheckerServiceFactory;
  private final CodeChangeProbabilityCalculatorFactory codeChangeProbabilityCalculatorFactory;
  private final SecureChangeProbabilityCalculator secureChangeProbabilityCalculator;
  private final QualityViolationCostsCalculator costsCalculator;
  private final QualityAnalysisRepository qualityAnalysisRepository;

  private final Set<Project> alreadyScheduledProjects = Sets.newCopyOnWriteArraySet();

  @Autowired
  public DefaultQualityAnalyzerScheduler(ProjectRepository projectRepository,
                                         ViolationsCalculatorService violationsCalculatorService,
                                         ScmAvailabilityCheckerServiceFactory scmAvailabilityCheckerServiceFactory,
                                         CodeChangeProbabilityCalculatorFactory codeChangeProbabilityCalculatorFactory,
                                         SecureChangeProbabilityCalculator secureChangeProbabilityCalculator, QualityViolationCostsCalculator costsCalculator,
                                         QualityAnalysisRepository qualityAnalysisRepository) {
    this.projectRepository = projectRepository;
    this.violationsCalculatorService = violationsCalculatorService;
    this.scmAvailabilityCheckerServiceFactory = scmAvailabilityCheckerServiceFactory;
    this.codeChangeProbabilityCalculatorFactory = codeChangeProbabilityCalculatorFactory;
    this.secureChangeProbabilityCalculator = secureChangeProbabilityCalculator;
    this.costsCalculator = costsCalculator;
    this.qualityAnalysisRepository = qualityAnalysisRepository;

    scheduler.setPoolSize(DEFAULT_POOL_SIZE);
    scheduler.initialize();
  }

  /**
   * {@inheritDoc}
   */
  public void executeAnalyzer(Project project) {
    project.setHadAnalysis(true);
    projectRepository.save(project);

    QualityAnalyzerService qualityAnalyzerService = createDefaultQualityAnalyzer();
    scheduler.execute(new AnalyzerRunnable(project, projectRepository, qualityAnalyzerService));
    log.info("Executing analyzer job for project {}", project.getName());
  }

  /**
   * {@inheritDoc}
   */
  public boolean scheduleAnalyzer(Project project) {
    if (alreadyScheduledProjects.contains(project)) {
      log.info("Project {} is already scheduled!", project.getName());
      return false;
    }
    alreadyScheduledProjects.add(project);

    QualityAnalyzerService qualityAnalyzerService = createDefaultQualityAnalyzer();
    scheduler.schedule(new AnalyzerRunnable(project, projectRepository, qualityAnalyzerService), new CronTrigger(project.getCronExpression()));
    log.info("Scheduled analyzer job for project {} with cron expression {}", project.getName(), project.getCronExpression());
    return true;
  }

  private DefaultQualityAnalyzerService createDefaultQualityAnalyzer() {
    return new DefaultQualityAnalyzerService(violationsCalculatorService,
        scmAvailabilityCheckerServiceFactory,
        codeChangeProbabilityCalculatorFactory,
        secureChangeProbabilityCalculator,
        costsCalculator,
        qualityAnalysisRepository);
  }
}