package pl.edu.icm.saos.batch.jobs;

import org.springframework.batch.core.Job;
import org.springframework.batch.core.Step;
import org.springframework.batch.core.configuration.annotation.JobBuilderFactory;
import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
import org.springframework.batch.core.launch.support.RunIdIncrementer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import pl.edu.icm.saos.importer.common.JudgmentImportDownloadWriter;
import pl.edu.icm.saos.importer.common.JudgmentImportProcessWriter;
import pl.edu.icm.saos.importer.common.JudgmentWithCorrectionList;
import pl.edu.icm.saos.importer.notapi.common.JsonImportDownloadProcessor;
import pl.edu.icm.saos.importer.notapi.common.JsonImportDownloadReader;
import pl.edu.icm.saos.importer.notapi.common.JsonJudgmentImportProcessProcessor;
import pl.edu.icm.saos.importer.notapi.common.JsonJudgmentItem;
import pl.edu.icm.saos.importer.notapi.common.JudgmentImportProcessReader;
import pl.edu.icm.saos.importer.notapi.common.NotApiImportDownloadStepExecutionListener;
import pl.edu.icm.saos.importer.notapi.common.content.ContentDownloadStepExecutionListener;
import pl.edu.icm.saos.importer.notapi.common.content.ContentProcessChunkListener;
import pl.edu.icm.saos.importer.notapi.supremecourt.judgment.json.SourceScJudgment;
import pl.edu.icm.saos.importer.notapi.supremecourt.judgment.process.ScjImportProcessStepExecutionListener;
import pl.edu.icm.saos.persistence.model.SupremeCourtJudgment;
import pl.edu.icm.saos.persistence.model.importer.notapi.RawSourceScJudgment;

/**
 * @author Łukasz Dumiszewski
 */
@Configuration
public class ScjImportJobConfiguration {
   
    
    @Autowired
    private JobBuilderFactory jobs;

    @Autowired
    private StepBuilderFactory steps;
    
    
    
    
    @Autowired
    private JsonImportDownloadReader scjImportDownloadReader;

    @Autowired
    private JsonImportDownloadProcessor<RawSourceScJudgment> scjImportDownloadProcessor;

    @Autowired
    private JudgmentImportDownloadWriter scjImportDownloadWriter;
    
    @Autowired
    private NotApiImportDownloadStepExecutionListener scjImportDownloadStepExecutionListener;
    
    @Autowired
    private ContentDownloadStepExecutionListener scjContentDownloadStepExecutionListener;
    
    
    
    @Autowired
    private JudgmentImportProcessReader<RawSourceScJudgment> scjImportProcessReader;

    @Autowired
    private JsonJudgmentImportProcessProcessor<SourceScJudgment, SupremeCourtJudgment> scjImportProcessProcessor;
    
       
    @Autowired
    private ScjImportProcessStepExecutionListener scjImportProcessStepExecutionListener;
    
    @Autowired
    private ContentProcessChunkListener contentProcessChunkListener;
    

    @Bean
    public JudgmentImportProcessWriter<SupremeCourtJudgment> scjImportProcessWriter() {
        return new JudgmentImportProcessWriter<>();
    }
    
    /**
     * Only download 
     */
    @Bean
    public Job scJudgmentImportDownloadJob() {
        return jobs.get("IMPORT_SC_JUDGMENTS_download").start(scJudgmentImportDownloadStep()).incrementer(new RunIdIncrementer()).build();
    }
    
    
    @Bean
    protected Step scJudgmentImportDownloadStep() {
        return steps.get("scJudgmentImportDownloadStep").<JsonJudgmentItem, RawSourceScJudgment> chunk(20)
            .reader(scjImportDownloadReader)
            .processor(scjImportDownloadProcessor)
            .writer(scjImportDownloadWriter)
            .listener(scjImportDownloadStepExecutionListener)
            .listener(scjContentDownloadStepExecutionListener)
            .build();
    } 
    

    /**
     * Only process
     */
    @Bean
    public Job scJudgmentImportProcessJob() {
        return jobs.get("IMPORT_SC_JUDGMENTS_process").start(scJudgmentImportProcessStep()).incrementer(new RunIdIncrementer()).build();
    }
   
    @Bean
    protected Step scJudgmentImportProcessStep() {
        return steps.get("scJudgmentImportProcessStep").<RawSourceScJudgment, JudgmentWithCorrectionList<SupremeCourtJudgment>> chunk(20)
            .reader(scjImportProcessReader)
            .processor(scjImportProcessProcessor)
            .writer(scjImportProcessWriter())
            .listener(scjImportProcessStepExecutionListener)
            .listener(contentProcessChunkListener)
            .build();
    } 
    
    /**
     * Download and process
     */
    @Bean
    public Job scJudgmentImportJob() {
        return jobs.get("IMPORT_SC_JUDGMENTS").start(scJudgmentImportDownloadStep()).next(scJudgmentImportProcessStep()).incrementer(new RunIdIncrementer()).build();
    }
   
    
}