package com.gton.util;

import java.io.*;
import org.jdom.input.*;
import org.jdom.output.*; //寫出DOM樹到文件
import org.jdom.*; //JDOM對象
import java.util.*;
import com.gton.bean.*;

public class MyXML {
    Document myDocument = null;
    Element testElement = null;
   public static String myPath = "E:\\MyWebApp\\Test_OA\\web\\admin\\test\\";
    public MyXML() {
        testElement = new Element("Test");
        myDocument = new Document(testElement);

    }

    public String setXML(ArrayList list) {
        for (int i = 0; i < list.size(); i++) {
            TestBean tb = (TestBean) list.get(i);
            Element subjectElemetn = new Element("subject");
            subjectElemetn.addContent(new Element("No").addContent(i + ""));
            subjectElemetn.addContent(new Element("Test_Theme").addContent(tb.
                    getTest_Theme()));
            subjectElemetn.addContent(new Element("Test_Result").addContent(tb.
                    getTest_Result()));
            subjectElemetn.addContent(new Element("Test_Score").addContent(tb.
                    getTest_Score() + ""));
            subjectElemetn.addContent(new Element("Test_Type").addContent(tb.
                    getTest_Type()));
            if (tb.getTest_Type().equals("1")) {
                subjectElemetn.addContent(new Element("Test_A").addContent(tb.
                        getTest_A()));
                subjectElemetn.addContent(new Element("Test_B").addContent(tb.
                        getTest_B()));
                subjectElemetn.addContent(new Element("Test_C").addContent(tb.
                        getTest_C()));
                subjectElemetn.addContent(new Element("Test_D").addContent(tb.
                        getTest_D()));
            }

            testElement.addContent(subjectElemetn);
        }
        XMLOutputter outputter = new XMLOutputter();
        return outputter.outputString(myDocument);
    }

    public String outXMLFile(ArrayList list,String date) {
        String path = myPath + date.substring(0, 4) + "\\" +
                      date.substring(4, 6) + "\\" + date.substring(6, 8) + "\\";
        for (int i = 0; i < list.size(); i++) {
            TestBean tb = (TestBean) list.get(i);
            Element subjectElemetn = new Element("subject");
            subjectElemetn.addContent(new Element("No").addContent(i + 1 + ""));
            subjectElemetn.addContent(new Element("Test_Theme").addContent(tb.
                    getTest_Theme()));
            subjectElemetn.addContent(new Element("Test_Result").addContent(tb.
                    getTest_Result()));
            subjectElemetn.addContent(new Element("Test_Score").addContent(tb.
                    getTest_Score() + ""));
            subjectElemetn.addContent(new Element("Test_Type").addContent(tb.
                    getTest_Type()));
            if (tb.getTest_Type().equals("1")) {
                subjectElemetn.addContent(new Element("Test_A").addContent(tb.
                        getTest_A()));
                subjectElemetn.addContent(new Element("Test_B").addContent(tb.
                        getTest_B()));
                subjectElemetn.addContent(new Element("Test_C").addContent(tb.
                        getTest_C()));
                subjectElemetn.addContent(new Element("Test_D").addContent(tb.
                        getTest_D()));
            }
            testElement.addContent(subjectElemetn);
        }
        try {
            File file = new File(path);
            if (!file.exists()) {
                file.mkdirs();
            }
            File xmlFile = new File(path + date + ".xml");
            FileWriter writer = null;
            XMLOutputter outputter = new XMLOutputter(Format.getCompactFormat().
                    setEncoding("gb2312"));
            writer = new FileWriter(xmlFile);
            outputter.output(myDocument, writer);
            writer.close();
        } catch (IOException ex) {
            ex.printStackTrace();
        }
        return date.substring(0, 4) + "\\" +
                      date.substring(4, 6) + "\\" + date.substring(6, 8) + "\\"+date+".xml";
    }
}