com.versioneye.utils.JsonUtils Maven / Gradle / Ivy

Go to download
package com.versioneye.utils;

import org.apache.maven.model.Dependency;
import org.apache.maven.project.MavenProject;
import org.codehaus.jackson.JsonParser;
import org.codehaus.jackson.map.ObjectMapper;
import org.eclipse.aether.artifact.Artifact;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Vector;

/**
 * Methods to deal with JSON.
 */
public class JsonUtils {

    public ByteArrayOutputStream dependenciesToJson(MavenProject project, List dependencies, String nameStrategy) throws Exception {
        List> dependencyHashes = getDependencyHashes(dependencies);
        ByteArrayOutputStream outstream = new ByteArrayOutputStream();
        toJson(outstream, getJsonPom(project, dependencyHashes, nameStrategy));
        return outstream;
    }

    public ByteArrayOutputStream artifactsToJson(List directDependencies) throws Exception {
        List> hashes = getHashes(directDependencies);
        ByteArrayOutputStream outstream = new ByteArrayOutputStream();
        toJson(outstream, hashes);
        return outstream;
    }

    public void dependenciesToJsonFile(String name, Map directDependencies, String file) throws Exception {
        File targetFile = getTargetFile(file);
        toJson(new FileOutputStream(targetFile), directDependencies);
    }

    public void dependenciesToJsonFile(MavenProject project, List directDependencies, String file, String nameStrategy) throws Exception {
        List> dependencyHashes = getHashes(directDependencies);
        File targetFile = getTargetFile(file);
        toJson(new FileOutputStream(targetFile), getJsonPom(project, dependencyHashes, nameStrategy));
    }

    public static void toJson(OutputStream output, Object input) throws Exception {
         ObjectMapper mapper = new ObjectMapper();
         mapper.configure(JsonParser.Feature.ALLOW_UNQUOTED_FIELD_NAMES, true);
         mapper.writeValue(output, input);
    }

    public List> getHashes(List directDependencies){
        List> hashes = (List>) new Vector>(directDependencies.size());
        hashes.addAll( generateHashForJsonOutput( directDependencies));
        return hashes;
    }

    public List> getDependencyHashes(List directDependencies){
        List> hashes = (List>) new Vector>(directDependencies.size());
        hashes.addAll( generateHashFromDependencyList( directDependencies));
        return hashes;
    }

    public static List> generateHashForJsonOutput(List input) {
         List> output = new Vector>(input.size());
         for (Artifact artifact : input) {
             HashMap hash = new HashMap(2);
             hash.put("version", artifact.getVersion());
             hash.put("name", artifact.getGroupId() + ":" + artifact.getArtifactId());
             output.add(hash);
         }
         return output;
    }

    public static List> generateHashFromDependencyList(List input) {
        List> output = new Vector>(input.size());
        for (Dependency dependency : input) {
            HashMap hash = new HashMap(2);
            hash.put("version", dependency.getVersion());
            hash.put("name", dependency.getGroupId() + ":" + dependency.getArtifactId());
            hash.put("scope", dependency.getScope() );
            output.add(hash);
        }
        return output;
    }

    public Map getJsonPom(MavenProject project, List> dependencyHashes, String nameStrategy){
        Map pom = new HashMap();
        pom.put("name", getNameFor(project, nameStrategy));
        pom.put("group_id", project.getGroupId());
        pom.put("artifact_id", project.getArtifactId());
        pom.put("language", "Java");
        pom.put("prod_type", "Maven2");
        pom.put("dependencies", dependencyHashes);
        return pom;
    }

    private String getNameFor(MavenProject project, String nameStrategy){
        String name = "project";
        if (nameStrategy == null || nameStrategy.isEmpty()){
            nameStrategy = "name";
        }
        if (nameStrategy.equals("name")){
            name = project.getName();
            if (name == null || name.isEmpty()){
                name = project.getArtifactId();
            }
        } else if (nameStrategy.equals("artifact_id")){
            name = project.getArtifactId();
        } else if (nameStrategy.equals("GA")){
            name = project.getGroupId() + "/" + project.getArtifactId();
        }
        return name;
    }

    private File getTargetFile(String file){
        File targetFile = new File(file);
        File parent = targetFile.getParentFile();
        if (!parent.exists()){
            parent.mkdirs();
        }
        return targetFile;
    }

}