summaryrefslogtreecommitdiffstats
path: root/sandbox/kgoodson/SourceExplore/src/main/java/sourcehash/HashSPISource.java
blob: 6ac7067e46eb553d70d369cbcb2b9de9b81767c4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
package sourcehash;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.math.BigInteger;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Iterator;
import java.util.List;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;

import org.apache.tuscany.SourceExplore.ObjectFactory;
import org.apache.tuscany.SourceExplore.Package;
import org.apache.tuscany.SourceExplore.Projects;
import org.apache.tuscany.SourceExplore.Projects2;
import org.apache.tuscany.SourceExplore.Projects2.Project;



public class HashSPISource {
	
	public static void main(String[] args) throws Throwable {
	
		ObjectFactory fac = new ObjectFactory();
		

		String sourceRoot = "c:/Dev2/SCA";
		String inputXML = "src/main/resources/projects.xml";
		
		JAXBContext jaxbContext = JAXBContext
			.newInstance("org.apache.tuscany.SourceExplore");
			Unmarshaller um = jaxbContext.createUnmarshaller();

				
				File inputFile = new File(inputXML).getAbsoluteFile();
				Projects p = (Projects)um.unmarshal(inputFile);
				List<JAXBElement<String>> data = p.getProjectOrPackage();
				System.out.println(p);
				
				
				Projects2 p2 = fac.createProjects2();


				
				for (int i = 0; i < data.size(); i++) {
					Project proj = null;
					if(!"project".equals(data.get(i).getName().getLocalPart())) {
						throw new Exception("Expecting a project element " + i);
					} else {
						System.out.println("project: " + data.get(i).getValue());
						proj = fac.createProjects2Project();
						proj.setName(data.get(i).getValue());
						p2.getProject().add(proj);
					}
					for(i++;i<data.size() && "package".equals(data.get(i).getName().getLocalPart());i++) {
						if(!"package".equals(data.get(i).getName().getLocalPart())) {
							throw new Exception("Expecting a package element " + i);
						} else {
    						System.out.println("package: " + data.get(i).getValue());
    						Package pkg = fac.createPackage();
    						pkg.setName(data.get(i).getValue());
    						proj.getPackage().add(pkg);
    						processPackage(sourceRoot, proj, pkg);
						}
					}
					i--;
				}
				FileOutputStream fos = new FileOutputStream(new File("src/main/resources/out.xml").getAbsoluteFile());
				Marshaller m = jaxbContext.createMarshaller();
				m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
				m.marshal(p2, fos);
				fos.close();
			
				
	}

	private static void processPackage(String sourceRoot, Project proj,	Package pkg) {
		
		ObjectFactory fac = new ObjectFactory();
		
		String pkgPath = pkg.getName().replace('.', '/' );
	    String folderPath = sourceRoot +"/" + proj.getName() +"/src/main/java/" + pkgPath;
	    File folder = new File(folderPath);
	    File[] contents = folder.listFiles();
	    
	    List<org.apache.tuscany.SourceExplore.File> files = pkg.getFile();
	    if(contents!= null) {
	    for (File file : contents) {
	    	if(ofInterest(file)) {
	    		org.apache.tuscany.SourceExplore.File f = fac.createFile();
	    		f.setName(file.getPath());
	    		String digest = md5(file);
	    		f.setDigest(digest);
	    		files.add(f);
	    	}
			System.out.println(file.getName());
		}
	    }
	}

	private static String md5(File f) {
		MessageDigest digest = null;
		try {
			digest = MessageDigest.getInstance("MD5");
		} catch (NoSuchAlgorithmException e1) {
			// TODO Auto-generated catch block
			e1.printStackTrace();
		}

		InputStream is = null;
		String output = null;
		try {
			is = new FileInputStream(f);
		} catch (FileNotFoundException e1) {
			// TODO Auto-generated catch block
			e1.printStackTrace();
		}				
		byte[] buffer = new byte[8192];
		int read = 0;
		try {
			while( (read = is.read(buffer)) > 0) {
				digest.update(buffer, 0, read);
			}		
			byte[] md5sum = digest.digest();
			BigInteger bigInt = new BigInteger(1, md5sum);
			output = bigInt.toString(16);
			System.out.println("MD5: " + output);
		}
		catch(IOException e) {
			throw new RuntimeException("Unable to process file for MD5", e);
		}
		finally {
			try {
				is.close();
			}
			catch(IOException e) {
				throw new RuntimeException("Unable to close input stream for MD5 calculation", e);
			}
		}		
		return output;
	}

	private static boolean ofInterest(File file) {
		boolean result = false;
		
		if(!result) result = file.getPath().endsWith(".java"); 
		return result;
	}


}