RepoActivityCmd.java

package com.matrobot.gha.archive.cmd;

import java.io.IOException;
import java.io.PrintStream;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;

import com.matrobot.gha.Configuration;
import com.matrobot.gha.ICommand;
import com.matrobot.gha.archive.event.EventReader;
import com.matrobot.gha.archive.event.FilteredEventReader;
import com.matrobot.gha.archive.event.IEventReader;
import com.matrobot.gha.archive.repo.FilteredRepoReader;
import com.matrobot.gha.archive.repo.IRepositoryReader;
import com.matrobot.gha.archive.repo.OrderedRepoReader;
import com.matrobot.gha.archive.repo.RepositoryReader;
import com.matrobot.gha.archive.repo.RepositoryRecord;

/**
 * Parse archive and create reports with repository info by months
 * 
 * @author Krzysztof Langner
 */
public class RepoActivityCmd implements ICommand{

	private IRepositoryReader reader;
	private HashMap<String, String> staticCSVFields = new HashMap<String, String>();
	
	
	@Override
	public void run(Configuration params) throws IOException {

		IEventReader eventReader = createEventReader(params);
		createRepoReader(params, eventReader);
		saveAsCSV(params.getOutputStream());
	}

	/**
	 * Filter by repository if repository param provided
	 */
	private IEventReader createEventReader(Configuration params) {
		IEventReader eventReader = new EventReader(params.getMonthFolders());
		if(params.getRepositories().size() > 0){
			FilteredEventReader filteredEventReader = new FilteredEventReader(eventReader);
			for(String repo : params.getRepositories()){
				filteredEventReader.addRepoFilter(repo);
			}
			
			eventReader = filteredEventReader;
		}
		return eventReader;
	}

	
	/**
	 * Create repository reader. 
	 * Add:
	 * - min activity filter
	 * - ordered reader
	 */
	private void createRepoReader(Configuration params, IEventReader eventReader) {
		RepositoryReader repoReader = new RepositoryReader(eventReader);
		reader = repoReader;
		if(params.getMinActivity() > 0){
			FilteredRepoReader filteredReader = new FilteredRepoReader(reader);
			filteredReader.setMinActivity(params.getMinActivity());
			reader = filteredReader;
		}
		
		if(params.getOrderBy() != null){
			OrderedRepoReader orderedReader = new OrderedRepoReader(reader);
			int orderBy = getOrderKey(params.getOrderBy());
			orderedReader.setField(orderBy);
			reader = orderedReader;
		}
	}

	
	private int getOrderKey(String orderBy) {

		int value = 0;
		
		if(orderBy.equals("forks")){
			value = OrderedRepoReader.SORT_BY_FORKS;
		}
		else if(orderBy.equals("community_size")){
			value = OrderedRepoReader.SORT_BY_COMMUNITY;
		}
		else if(orderBy.equals("pushes")){
			value = OrderedRepoReader.SORT_BY_PUSHES;
		}
		else if(orderBy.equals("events")){
			value = OrderedRepoReader.SORT_BY_EVENTS;
		}
		
		return value;
	}

	private void saveAsCSV(PrintStream printStream) throws UnsupportedEncodingException, IOException {

		for(String key : staticCSVFields.keySet()){
			printStream.print(key + ",");
		}
		printStream.println(RepositoryRecord.getCSVHeaders());
		RepositoryRecord record;
		while((record = reader.next()) != null){
			for(String value : staticCSVFields.values()){
				printStream.print(value + ",");
			}
			printStream.println(record.toCSV());
		}
	}

	
	/**
	 * Export data for matrobot.com website
	 */
	public static void main(String[] args) throws IOException {

		RepoActivityCmd app = new RepoActivityCmd();
		Configuration params = new Configuration("configs/export_repos.yaml");
		
		String[] date = params.getStartDate().split("-");
		if(date.length == 2){
			app.staticCSVFields.put("year", date[0]);
			app.staticCSVFields.put("month", date[1]);
		}
		app.run(params);
	}
	
}