202 lines
7.8 KiB
Java
202 lines
7.8 KiB
Java
package org.gcube.dataanalysis.dataminer.poolmanager.util;
|
|
|
|
***REMOVED***
|
|
import org.gcube.dataanalysis.dataminer.poolmanager.datamodel.Dependency;
|
|
***REMOVED***
|
|
import org.tmatesoft.svn.core.SVNURL;
|
|
import org.tmatesoft.svn.core.auth.ISVNAuthenticationManager;
|
|
import org.tmatesoft.svn.core.internal.wc.SVNFileUtil;
|
|
import org.tmatesoft.svn.core.internal.wc.admin.SVNChecksumInputStream;
|
|
import org.tmatesoft.svn.core.io.ISVNEditor;
|
|
import org.tmatesoft.svn.core.io.SVNRepository;
|
|
import org.tmatesoft.svn.core.io.SVNRepositoryFactory;
|
|
import org.tmatesoft.svn.core.io.diff.SVNDeltaGenerator;
|
|
import org.tmatesoft.svn.core.wc.SVNWCUtil;
|
|
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.ByteArrayOutputStream;
|
|
***REMOVED***
|
|
import java.util.*;
|
|
|
|
/**
|
|
* Created by ggiammat on 5/9/17.
|
|
*/
|
|
public class SVNUpdater ***REMOVED***
|
|
|
|
private SVNRepository svnRepository;
|
|
private ServiceConfiguration configuration;
|
|
|
|
public SVNUpdater(ServiceConfiguration configuration) throws SVNException ***REMOVED***
|
|
this.svnRepository = SVNRepositoryFactory.create(SVNURL.parseURIEncoded(configuration.getSVNRepository()));
|
|
ISVNAuthenticationManager authManager = SVNWCUtil.createDefaultAuthenticationManager();
|
|
this.svnRepository.setAuthenticationManager(authManager);
|
|
this.configuration = configuration;
|
|
***REMOVED***
|
|
|
|
|
|
|
|
public void updateRPRotoDeps(Algorithm algorithm)***REMOVED***
|
|
this.updateSVN(this.configuration.getSVNRProtoOSDepsList(), algorithm.getOSDependencies());
|
|
this.updateSVN(this.configuration.getSVNRProtoCRANDepsList(), algorithm.getCranDependencies());
|
|
this.updateSVN(this.configuration.getSVNRProtoGitHubDepsList(), algorithm.getGitHubDependencies());
|
|
***REMOVED***
|
|
|
|
public void updateProdDeps(Algorithm algorithm)***REMOVED***
|
|
this.updateSVN(this.configuration.getSVNProdOSDepsList(), algorithm.getOSDependencies());
|
|
this.updateSVN(this.configuration.getSVNRProdCRANDepsList(), algorithm.getCranDependencies());
|
|
this.updateSVN(this.configuration.getSVNRProdGitHubDepsList(), algorithm.getGitHubDependencies());
|
|
***REMOVED***
|
|
|
|
|
|
public void updateSVNRProtoAlgorithmList(Algorithm algorithm, String targetVRE, String user, String env)***REMOVED***
|
|
this.updateSVNAlgorithmList(this.configuration.getSVNRProtoAlgorithmsList(), algorithm, targetVRE, user, env);
|
|
***REMOVED***
|
|
|
|
|
|
public void updateSVNProdAlgorithmList(Algorithm algorithm, String targetVRE, String user, String env)***REMOVED***
|
|
this.updateSVNAlgorithmList(this.configuration.getSVNProdAlgorithmsList(), algorithm, targetVRE, user, env);
|
|
***REMOVED***
|
|
|
|
public void updateSVNAlgorithmList(String file, Algorithm algorithm, String targetVRE, String user, String env)***REMOVED***
|
|
try ***REMOVED***
|
|
System.out.println("Updating algorithm list: "+ file);
|
|
final ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
|
|
svnRepository.getFile(file, SVNRepository.INVALID_REVISION, null, byteArrayOutputStream);
|
|
String lines[] = byteArrayOutputStream.toString().split("\\r?\\n");
|
|
|
|
***REMOVED***check if the algorithm is already in the list (match the class name)
|
|
for(String l: lines)***REMOVED***
|
|
if(l.contains(algorithm.getClazz()))***REMOVED***
|
|
System.out.println("Not updating algorithm list beacuse already present");
|
|
return;
|
|
***REMOVED***
|
|
***REMOVED***
|
|
|
|
***REMOVED***the algorithm is not in the list. Add it
|
|
List<String> newContent = new LinkedList<>(Arrays.asList(lines));
|
|
newContent.add(this.generateAlgorithmEntry(algorithm, targetVRE, user, env));
|
|
***REMOVED***Collections.sort(newContent);
|
|
|
|
|
|
final SVNDeltaGenerator deltaGenerator = new SVNDeltaGenerator();
|
|
|
|
byte[] originalContents = byteArrayOutputStream.toByteArray();
|
|
|
|
final ISVNEditor commitEditor = svnRepository.getCommitEditor("update algorithm list", null);
|
|
commitEditor.openRoot(-1);
|
|
commitEditor.openFile(file, -1);
|
|
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
|
|
for (String line : newContent) ***REMOVED***
|
|
baos.write(line.getBytes());
|
|
baos.write("\n".getBytes());
|
|
***REMOVED***
|
|
byte[] bytes = baos.toByteArray();
|
|
|
|
commitEditor.applyTextDelta(file,md5(originalContents));
|
|
|
|
final String checksum = deltaGenerator.sendDelta(file, new ByteArrayInputStream(originalContents), 0,
|
|
new ByteArrayInputStream(bytes), commitEditor, true);
|
|
commitEditor.closeFile(file, checksum);
|
|
commitEditor.closeEdit();
|
|
|
|
|
|
***REMOVED***catch(Exception ex)***REMOVED***
|
|
ex.printStackTrace();
|
|
***REMOVED***
|
|
|
|
finally ***REMOVED***
|
|
svnRepository.closeSession();
|
|
***REMOVED*** ***REMOVED***
|
|
|
|
|
|
public String generateAlgorithmEntry(Algorithm algorithm, String targetVRE, String user, String env)***REMOVED***
|
|
StringBuffer sb = new StringBuffer("| ");
|
|
sb.append(algorithm.getName() +" | ");
|
|
sb.append(user + " | ");
|
|
sb.append(algorithm.getCategory() + " | ");
|
|
sb.append(env + " | ");
|
|
sb.append("<notextile>./addAlgorithm.sh "+ algorithm.getName()+" " + algorithm.getCategory() + " " + algorithm.getClazz() + " " + targetVRE + " " + algorithm.getAlgorithmType() + " N " + algorithm.getPackageURL() + " \"" + algorithm.getDescription() + "\" </notextile> | ");
|
|
sb.append("none |");
|
|
return sb.toString();
|
|
***REMOVED***
|
|
|
|
|
|
public void updateSVN(String file, Collection<Dependency> deps) ***REMOVED***
|
|
try ***REMOVED***
|
|
System.out.println("Updating dependencies list: "+ file);
|
|
|
|
final ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
|
|
svnRepository.getFile(file, SVNRepository.INVALID_REVISION, null, byteArrayOutputStream);
|
|
String lines[] = byteArrayOutputStream.toString().split("\\r?\\n");
|
|
|
|
List<String> ldep = new LinkedList<>();
|
|
for(Dependency d: deps)***REMOVED***
|
|
ldep.add(d.getName());
|
|
***REMOVED***
|
|
List<String> aa = this.checkMatch(lines, ldep);
|
|
Collections.sort(aa);
|
|
|
|
final SVNDeltaGenerator deltaGenerator = new SVNDeltaGenerator();
|
|
|
|
byte[] originalContents = byteArrayOutputStream.toByteArray();
|
|
|
|
final ISVNEditor commitEditor = svnRepository.getCommitEditor("update dependencies", null);
|
|
commitEditor.openRoot(-1);
|
|
commitEditor.openFile(file, -1);
|
|
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
|
|
for (String line : aa) ***REMOVED***
|
|
baos.write(line.getBytes());
|
|
baos.write("\n".getBytes());
|
|
***REMOVED***
|
|
byte[] bytes = baos.toByteArray();
|
|
|
|
commitEditor.applyTextDelta(file,md5(originalContents));
|
|
|
|
final String checksum = deltaGenerator.sendDelta(file, new ByteArrayInputStream(originalContents), 0,
|
|
new ByteArrayInputStream(bytes), commitEditor, true);
|
|
commitEditor.closeFile(file, checksum);
|
|
commitEditor.closeEdit();
|
|
|
|
|
|
***REMOVED***catch(Exception ex)***REMOVED***
|
|
ex.printStackTrace();
|
|
***REMOVED***
|
|
|
|
finally ***REMOVED***
|
|
svnRepository.closeSession();
|
|
***REMOVED***
|
|
***REMOVED***
|
|
|
|
|
|
|
|
|
|
public static String md5(byte[] contents) ***REMOVED***
|
|
final byte[] tmp = new byte[1024];
|
|
final SVNChecksumInputStream checksumStream = new SVNChecksumInputStream(new ByteArrayInputStream(contents), "md5");
|
|
try ***REMOVED***
|
|
while (checksumStream.read(tmp) > 0) ***REMOVED***
|
|
***REMOVED***
|
|
***REMOVED***
|
|
return checksumStream.getDigest();
|
|
***REMOVED*** catch (IOException e) ***REMOVED***
|
|
***REMOVED***never happens
|
|
e.printStackTrace();
|
|
return null;
|
|
***REMOVED*** finally ***REMOVED***
|
|
SVNFileUtil.closeFile(checksumStream);
|
|
***REMOVED***
|
|
***REMOVED***
|
|
|
|
public List<String> checkMatch(String[] lines, List<String> ls) ***REMOVED***
|
|
Set<String> ss = new HashSet<String>(ls);
|
|
ss.addAll(Arrays.asList(lines));
|
|
return new ArrayList<>(ss);
|
|
***REMOVED***
|
|
|
|
|
|
***REMOVED***
|