RepositoryRecord.java

package com.matrobot.gha.archive.repo;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import com.matrobot.gha.archive.event.EventRecord;

public class RepositoryRecord {

	public String repoName;
	public String language = "";
	public String homepage = "";
	public String createdAt = "";
	public int eventCount = 0;
	public int pushEventCount = 0;
	public int issueOpenEventCount = 0;
	public int forkEventCount = 0;
	public int openedPullCount = 0;
	public int closedPullCount = 0;
	public boolean isNew = false;
	public boolean isFork = false;
	
	public Set<String> committers = new HashSet<String>();
	public Set<String> community = new HashSet<String>();
	
	
	public RepositoryRecord(String name) {
		repoName = name;
	}

	
	/**
	 * Load activity.json
	 * 
	 * @param filename
	 * @throws IOException
	 */
	public static HashMap<String, RepositoryRecord> loadData(String filePath) throws IOException{
		
		List<RepositoryRecord> rows = new ArrayList<RepositoryRecord>();
		Gson gson = new Gson();
		Type datasetType = new TypeToken<Collection<RepositoryRecord>>(){}.getType();

		FileInputStream fis = new FileInputStream(filePath+"repositories.json");
		Reader reader = new InputStreamReader(fis, "UTF-8");
		rows = gson.fromJson(reader, datasetType);
		
		reader.close();

		HashMap<String, RepositoryRecord> dataset = new HashMap<String, RepositoryRecord>();
		for(RepositoryRecord row : rows){
			dataset.put(row.repoName, row);
		}
		
		return dataset;
	}
	
	/**
	 * @return Header for CSV file
	 */
	public static String getCSVHeaders(){
		return "name,push_count,event_count,committer_count,fork_count,community_size," + 
				"opened_pulls,closed_pulls,language,homepage,createdAt";
	}
	
	/**
	 * @return CSV
	 */
	public String toCSV(){
		
		String hp = homepage.replace("\"", "");
		return repoName + "," + pushEventCount + "," + eventCount + "," + committers.size() + "," +
				forkEventCount + "," + community.size() + "," + openedPullCount + "," + 
				closedPullCount + "," + language + ",\"" + hp + "\"," + createdAt; 
	}

	
	public void updateData(EventRecord event) {
		
		if(event.isCreateRepository()){
			isNew = true;
		}
		else if(event.type.equals("PushEvent")){
			addPushEvent(event);
		}
		else if(event.type.equals("PullRequestEvent")){
			addPullRequestEvent(event);
		}
		else if(event.type.equals("IssuesEvent")){
			issueOpenEventCount += 1;
		}
		else if(event.type.equals("ForkEvent")){
			forkEventCount += 1;
		}
		
		if(event.getActorLogin() != null){
			community.add(event.getActorLogin());
		}
		
		updateMetadata(event);
		eventCount += 1;
	}

	
	private void addPushEvent(EventRecord event) {
		
		if(event.payload.size > 0){
			pushEventCount += 1;
			for(String committer : event.getCommitters()){
				committers.add(committer);
			}
		}
	}

	
	private void addPullRequestEvent(EventRecord event) {
		
		if(event.payload.action.equals("opened")){
			openedPullCount += 1;
		}
		else if(event.payload.action.equals("closed")){
			closedPullCount += 1;
		}
	}


	/**
	 * Udate information about repository current state
	 */
	private void updateMetadata(EventRecord event) {

		if(event.repository != null){
			if(event.repository.language != null){
				language = event.repository.language;
			}
			if(event.repository.homepage != null){
				homepage = event.repository.homepage;
			}
			int index = event.repository.created_at.indexOf('T');
			if(index > 0){
				createdAt = event.repository.created_at.substring(0, index);
			}
		}
	}
	
}