package Testlink;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintStream;
import java.util.ArrayList;
import java.util.List;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerConfigurationException;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

//定义操作XML文件的方法
public class XMLUtil {
//定义工作目录
private File workspaceFile;

//定义所有文件对象  
private File dirFile;

//定义buildXML文件夹  
private File buildXMLDirectory;

//html文件夹  
private File htmlFileDirectory;

//工作目录路径  
private String workspacePath;

//buildXML文件夹路径  
private String buildXMLDirectoryPath;

//定义build.xml文件路ing  
private String buildXMLFilePath;

//html文件夹路径  
private String htmlFileDirectoryPath;

//tap文件路径  
private String tapPath;

//测试结果节点  
private List<Node> testResult = new ArrayList();

//样本节点  
private List<Integer> sampleNumber = new ArrayList();

//此方法用户改变build.xml文件  
public void BuildXMLChange() {  
    //获取工作目录路径  
    workspaceFile = new File("");  
    workspacePath = workspaceFile.getAbsolutePath();

    //获取工作目录下所有文件  
    this.dirFile = new File(workspacePath);  
    File\[\] files = dirFile.listFiles();

    //定义buildXML文件夹路径  
    buildXMLDirectoryPath = workspacePath + "\\\\BuildXML";

    //获取buildXML文件夹所有文件  
    buildXMLDirectory = new File(buildXMLDirectoryPath);  
    File\[\] xmlFiles = this.buildXMLDirectory.listFiles();

    //获取build.xml文件路径  
    buildXMLFilePath = buildXMLDirectoryPath + "\\\\build.xml";

    if (files.length != 0) {

        File\[\] arrayOfFile1;  
        int j = (arrayOfFile1 = files).length;

        for (int i = 0; i < j; i++) {

            File file = arrayOfFile1\[i\];

            if (file.getName().equals("BuildXML")) {

                if (xmlFiles.length != 0) {

                    File\[\] arrayOfFile2;  
                    int m = (arrayOfFile2 = xmlFiles).length;

                    for (int k = 0; k < m; k++) {  
                        File xmlFile = arrayOfFile2\[k\];  
                        if (xmlFile.getName().equals("build.xml")) {

                            DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();  
                            try {  
                                //读取xml  
                                DocumentBuilder documentBuilder = documentBuilderFactory.newDocumentBuilder();  
                                Document document = documentBuilder.parse("file:///" + buildXMLFilePath);  
                                NodeList nodeList = document.getElementsByTagName("property");

                                Node property = nodeList.item(0);

                                NamedNodeMap nodeMap = property.getAttributes();  
                                Node node = nodeMap.getNamedItem("value");  
                                node.setTextContent(workspacePath);

                                //写入XML  
                                TransformerFactory factory = TransformerFactory.newInstance();  
                                Transformer former = factory.newTransformer();  
                                former.setOutputProperty("encoding", "GBK");  
                                try {  
                                    former.transform(new DOMSource(document),  
                                            new StreamResult(new File(buildXMLFilePath)));  
                                } catch (TransformerException e) {  
                                    e.printStackTrace();  
                                }  
                                System.out.println("build.xml文件不存在");  
                            } catch (ParserConfigurationException e) {  
                                e.printStackTrace();  
                            } catch (SAXException e) {  
                                e.printStackTrace();  
                            } catch (IOException e) {  
                                e.printStackTrace();  
                            } catch (TransformerConfigurationException e1) {  
                                e1.printStackTrace();  
                            }  
                        }  
                    }  
                } else {  
                    System.out.println("BuildXML文件夹为空");  
                }  
            }  
        }  
    } else {  
        System.out.println("文件夹内容为空");  
    }  
}

public void XMLToTAP() {  
    workspaceFile = new File("");  
    workspacePath = workspaceFile.getAbsolutePath();

    dirFile = new File(workspacePath);

    File\[\] files = dirFile.listFiles();

    htmlFileDirectoryPath = (workspacePath + "\\\\XMLFolder");

    htmlFileDirectory = new File(htmlFileDirectoryPath);

    File\[\] htmlFiles = htmlFileDirectory.listFiles();

    tapPath = (workspacePath + "\\\\TAPFolder");  
    if (files.length != 0) {  
        File\[\] arrayOfFile1;  
        int j = (arrayOfFile1 = files).length;  
        for (int i = 0; i < j; i++) {  
            File file = arrayOfFile1\[i\];  
            if (file.getName().equals("XMLFolder")) {  
                if (htmlFiles.length != 0) {  
                    File\[\] arrayOfFile2;  
                    int m = (arrayOfFile2 = htmlFiles).length;  
                    for (int k = 0; k < m; k++) {  
                        File xmlFile = arrayOfFile2\[k\];  
                        if (xmlFile.getName().endsWith(".xml")) {  
                            DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();  
                            try {  
                                DocumentBuilder documentBuilder = documentBuilderFactory.newDocumentBuilder();  
                                Document document = documentBuilder.parse(xmlFile);

                                Element rootElement = document.getDocumentElement();

                                NodeList nodeList = rootElement.getChildNodes();  
                                if (rootElement.hasChildNodes()) {  
                                    for (int x = 0; x < nodeList.getLength(); x++) {  
                                        if (nodeList.item(x).getNodeType() == 1) {  
                                            this.testResult.add(nodeList.item(x));  
                                        }  
                                    }  
                                } else {  
                                    System.out.println("接口测试结果为空");  
                                }  
                                for (int y = 0; y < this.testResult.size(); y++) {  
                                    if (((Node) this.testResult.get(y)).getNodeName().equals("sample")) {  
                                        this.sampleNumber.add(Integer.valueOf(y));  
                                    }  
                                }  
                                for (int z = 0; z < this.sampleNumber.size(); z++) {  
                                    NamedNodeMap namedNodeMap = ((Node) this.testResult  
                                            .get(((Integer) this.sampleNumber.get(z)).intValue())).getAttributes();  
                                    if (namedNodeMap.getNamedItem("s").getNodeValue().equals("false")) {  
                                        String lbValue = namedNodeMap.getNamedItem("lb").getNodeValue();

                                        String\[\] fileContent = lbValue.split("-");

                                        String fileNewPath = this.tapPath + "\\\\" + fileContent\[0\] + ".tap";

                                        File fileNew = new File(fileNewPath);

                                        FileWriter fWriter = new FileWriter(fileNew, true);  
                                        BufferedWriter bWriter = new BufferedWriter(fWriter);  
                                        bWriter.write("TAP version 13");  
                                        bWriter.newLine();  
                                        bWriter.write("1..1");  
                                        bWriter.newLine();  
                                        String resultStr = "not ok 1";  
                                        bWriter.write(resultStr);  
                                        bWriter.flush();  
                                        bWriter.close();  
                                        fWriter.close();  
                                    } else {  
                                        String lbValue = namedNodeMap.getNamedItem("lb").getNodeValue();

                                        String\[\] fileContent = lbValue.split("-");

                                        String fileNewPath = this.tapPath + "\\\\" + fileContent\[0\] + ".tap";

                                        File fileNew = new File(fileNewPath);

                                        FileWriter fWriter = new FileWriter(fileNew, true);  
                                        BufferedWriter bWriter = new BufferedWriter(fWriter);  
                                        bWriter.write("TAP version 13");  
                                        bWriter.newLine();  
                                        String resultStr = "1..1\\nok 1";  
                                        fWriter.write(resultStr);  
                                        bWriter.flush();  
                                        bWriter.close();  
                                        fWriter.close();  
                                    }  
                                }  
                            } catch (ParserConfigurationException e) {  
                                e.printStackTrace();  
                            } catch (SAXException e) {  
                                e.printStackTrace();  
                            } catch (IOException e) {  
                                e.printStackTrace();  
                            }  
                        } else {  
                            System.out.println("xml文件不存在");  
                        }  
                    }  
                } else {  
                    System.out.println("XML文件夹为空");  
                }  
            }  
        }  
    } else {  
        System.out.println("文件夹内容为空");  
    }  
}  

}