package com.vci.ant.ext.tasks;
|
|
import java.io.File;
|
import java.io.FileWriter;
|
import java.io.IOException;
|
import java.io.StringWriter;
|
import java.util.ArrayList;
|
import java.util.Iterator;
|
import java.util.List;
|
import java.util.concurrent.TimeUnit;
|
|
import javax.swing.JFileChooser;
|
|
import org.apache.commons.io.FileUtils;
|
import org.apache.tools.ant.BuildException;
|
import org.apache.tools.ant.Location;
|
import org.apache.tools.ant.Project;
|
import org.apache.tools.ant.taskdefs.MatchingTask;
|
import org.apache.tools.ant.types.FileSet;
|
import org.dom4j.Attribute;
|
import org.dom4j.Document;
|
import org.dom4j.DocumentFactory;
|
import org.dom4j.DocumentHelper;
|
import org.dom4j.Element;
|
import org.dom4j.io.DocumentSource;
|
import org.dom4j.io.OutputFormat;
|
import org.dom4j.io.SAXReader;
|
import org.dom4j.io.SAXWriter;
|
import org.dom4j.io.XMLWriter;
|
|
/**
|
*
|
* <p>Title: 自动加载指定目录（包含子目录）下全部的.jar到项目的Classpath</p>
|
* <p>Description: 自动加载指定目录（包含子目录）下全部的.jar到项目的Classpath</p>
|
* <p></p>
|
* <p>Copyright: Copyright (c) 2017</p>
|
* <p>Company: VCI</p>
|
* @author xiongchao
|
* @time 2017-9-8
|
* @version 1.0
|
*/
|
public class AutoLoadJarToClasspath extends MatchingTask {
|
private String showChooseDialog = "true";
|
private String platformJarProjectPath = "";
|
|
public String getShowChooseDialog() {
|
return showChooseDialog;
|
}
|
|
public void setShowChooseDialog(String showChooseDialog) {
|
this.showChooseDialog = showChooseDialog;
|
}
|
|
public String getPlatformJarProjectPath() {
|
return platformJarProjectPath;
|
}
|
|
public void setPlatformJarProjectPath(String platformJarProjectPath) {
|
this.platformJarProjectPath = platformJarProjectPath;
|
}
|
|
@Override
|
public void execute() throws BuildException{
|
try{
|
Project project = super.getProject();
|
File workspaceFile = project.getBaseDir().getParentFile().getCanonicalFile();
|
String workspaceFullPath = workspaceFile.getCanonicalPath();
|
|
File projectFile = project.getBaseDir().getCanonicalFile();
|
String projectFullPath = projectFile.getCanonicalPath();
|
String projectDirName = projectFile.getName();
|
|
File f = null;
|
String jarPath = getPlatformJarProjectPath();
|
if(getShowChooseDialog() != null && "true".equalsIgnoreCase(getShowChooseDialog())){
|
try {
|
f = getChooseFile();
|
if(f == null){
|
System.out.println("选择无效");
|
}
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
return;
|
}
|
} else {
|
f = new File(jarPath);
|
if(f != null){
|
f = f.getCanonicalFile();
|
}
|
}
|
if(f == null){
|
return;
|
}
|
String classpathFileName = projectFullPath.replace("\\", "/") + "/.classpath";
|
List<Element> classpathEntrys = new ArrayList<Element>();
|
getJarFiles(f, workspaceFullPath, classpathEntrys);
|
SAXReader reader = new SAXReader(DocumentFactory.getInstance());
|
Document doc = reader.read(new File(classpathFileName));
|
Element root = doc.getRootElement();
|
@SuppressWarnings("unchecked")
|
Iterator<Element> elements = (Iterator<Element>)root.elementIterator();
|
while(elements.hasNext()){
|
Element e = elements.next();
|
if(isLibClasspathEntryElement(e)){
|
root.remove(e);
|
}
|
}
|
for (Element e : classpathEntrys) {
|
root.add(e);
|
}
|
|
OutputFormat format = OutputFormat.createPrettyPrint();
|
format.setEncoding(doc.getXMLEncoding());
|
XMLWriter write = new XMLWriter(new FileWriter(classpathFileName), format);
|
write.write(doc);
|
write.flush();
|
write.close();
|
System.out.println(classpathFileName + " 已更新，请刷新项目（" + projectDirName + "）！");
|
}catch(Throwable e){
|
e.printStackTrace();
|
throw new BuildException(e);
|
}
|
}
|
private boolean isLibClasspathEntryElement(Element e){
|
boolean res = false;
|
Attribute attrKind = e.attribute("kind");
|
Attribute attrPath = e.attribute("path");
|
if(attrKind != null && attrPath != null
|
&& "lib".equalsIgnoreCase(attrKind.getValue())
|
&& (attrPath.getValue().endsWith(".jar"))){
|
res = true;
|
}
|
return res;
|
}
|
|
private File getChooseFile() throws IOException{
|
String path = super.getProject().getBaseDir().getName();
|
if(path.contains("_")){
|
String[] paths = path.split("_");
|
paths[0] = "VCI-JAR";
|
StringBuffer sb = new StringBuffer();
|
for (int i = 0; i < paths.length; i++) {
|
sb.append(paths[i]);
|
if(i != paths.length - 1){
|
sb.append("_");
|
}
|
}
|
File f = new File(super.getProject().getBaseDir().getParent(), sb.toString());
|
System.out.println("a"+super.getProject().getBaseDir().getParent() + "/" + sb.toString());
|
path = f.getCanonicalPath();
|
System.out.println("c"+path);
|
} else {
|
path = super.getProject().getBaseDir().getParentFile().getCanonicalPath();
|
}
|
JFileChooser jfc = new JFileChooser(path);
|
jfc.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
|
jfc.showOpenDialog(null);
|
File f = jfc.getSelectedFile();
|
return f;
|
}
|
|
private void getJarFiles(File file, String workspaceFullPath,
|
List<Element> classpathEntrys) throws IOException{
|
File[] childFiles = file.listFiles();
|
if(childFiles == null){
|
return;
|
}
|
for (File childFile : childFiles) {
|
if(childFile.getName().equals(".svn")){
|
continue;
|
}
|
if(childFile.isDirectory()){
|
getJarFiles(childFile, workspaceFullPath, classpathEntrys);
|
} else if(childFile.getName().endsWith(".jar")){
|
Element e = DocumentHelper.createElement("classpathentry");
|
e.add(DocumentHelper.createAttribute(e, "kind", "lib"));
|
e.add(DocumentHelper.createAttribute(e, "path", childFile.getCanonicalPath().replace(workspaceFullPath, "").replace("\\", "/")));
|
classpathEntrys.add(e);
|
}
|
}
|
}
|
}
|
