package com.jasson.business.document;

import java.sql.*;
import java.util.Properties;
import java.io.*;
import java.sql.*;

public class Coder {

    public String pathName="D:\\JDYM\\src\\com\\jasson\\business\\document\\";
    public String className="";
    public String tableName="";
    private ResultSet rs = null;
    private Statement stmt = null;
    private ResultSetMetaData rsD;
    public Coder() {
    }
    public void writeInfoToFile(){
        try {
            //DbConn dbConn = new DbConn();
            //Connection conn = dbConn.getConnection();
            String url="jdbc:microsoft:sqlserver://devserver:1433;DatabaseName=AgentFlow";
            //String url="jdbc:microsoft:sqlserver://localhost:1433;DatabaseName=test";
            String dbDriver = "com.microsoft.jdbc.sqlserver.SQLServerDriver";
            Properties props = new Properties();
            Class.forName(dbDriver);
            props.put("user","sa");
            props.put("password","");
            Connection conn = DriverManager.getConnection(url, props);
            String sql="select * from "+tableName+" where 1<>1";
            String fileStreamStr="package com.jasson.business.document;\r\n\r\n";
            fileStreamStr+="/**\r\n";
            fileStreamStr+=" * <p>Title: "+className+"</p>\r\n";
            fileStreamStr+=" * <p>Description: </p>\r\n";
            fileStreamStr+=" * <p>Copyright: Copyright (c) 2003,by wingroup</p>\r\n";
            fileStreamStr+=" * <p>Company: wingroup</p>\r\n";
            fileStreamStr+=" * @author jarrywen\r\n";
            fileStreamStr+=" * @version 1.0\r\n";
            fileStreamStr+=" */\r\n\r\n";
            fileStreamStr+="import java.sql.Date;\r\n\r\n";
            fileStreamStr+="public class "+className+" {\r\n";
            stmt= conn.createStatement();
            rs=stmt.executeQuery(sql);
            rsD=rs.getMetaData();
            String fieldType="String";
            for(int i=1;i<rsD.getColumnCount()+1;i++){
                fieldType=getFieldType(rsD,i);
                fileStreamStr+="\tprivate "+fieldType+" "+rsD.getColumnLabel(i)+";\r\n";
            }
            //struct
            fileStreamStr+="\tpublic "+className+"( ";
            for(int i=1;i<rsD.getColumnCount()+1;i++){
                fieldType=getFieldType(rsD,i);
                if(i==1)
                    fileStreamStr+=fieldType+" "+rsD.getColumnLabel(i);
                else
                    fileStreamStr+=", "+fieldType+" "+rsD.getColumnLabel(i);
            }
            fileStreamStr+=" ){\r\n";
            for(int i=1;i<rsD.getColumnCount()+1;i++){
                fieldType=getFieldType(rsD,i);
                fileStreamStr+="\t\tthis."+rsD.getColumnLabel(i)+" = "+rsD.getColumnLabel(i)+";\r\n";
            }
            fileStreamStr+="\t}\r\n";

            fileStreamStr+="\tpublic "+className+"() {\r\n";
            fileStreamStr+="\t}\r\n";
            stmt= conn.createStatement();
            rs=stmt.executeQuery(sql);
            rsD=rs.getMetaData();
            for(int i=1;i<rsD.getColumnCount()+1;i++){
                fieldType=getFieldType(rsD,i);
                fileStreamStr+="\tpublic "+fieldType+" get"+rsD.getColumnLabel(i)+"(){\r\n";
                fileStreamStr+="\t\treturn "+rsD.getColumnLabel(i)+";\r\n";
                fileStreamStr+="\t}\r\n";
                fileStreamStr+="\tpublic void set"+rsD.getColumnLabel(i)+"("+fieldType+" "+rsD.getColumnLabel(i)+"){\r\n";
                fileStreamStr+="\t\tthis."+rsD.getColumnLabel(i)+" = "+rsD.getColumnLabel(i)+";\r\n";
                fileStreamStr+="\t}\r\n";
            }
            fileStreamStr+="}\r\n";
            //System.out.println(fileStreamStr);
            File file = new File (pathName+className+".java");
            FileWriter out = new FileWriter(file);
            out.write(fileStreamStr);
            out.close();
        }
        catch (Exception ex) {
            ex.printStackTrace();
        }
    }
    private String getFieldType(ResultSetMetaData rsD,int i){
        try {
            String fieldType="String";
            if (rsD.getColumnTypeName(i).equalsIgnoreCase("int")||rsD.getColumnTypeName(i).equalsIgnoreCase("smallint")||rsD.getColumnTypeName(i).equalsIgnoreCase("tinyint")){
                fieldType="int";
            }
            else{
                if (rsD.getColumnTypeName(i).equalsIgnoreCase("datetime")){
                    fieldType="Date";
                }
            }
            return fieldType;
        }
        catch (Exception ex) {
            ex.printStackTrace();
            return "";
        }
    }
    public static void main(String[] args) {
        Coder coder = new Coder();
        coder.pathName="D:\\JDYM\\src\\com\\jasson\\business\\document\\";
        coder.className="PackageInfo";
        coder.tableName="TPackage";
        coder.writeInfoToFile();
        coder.className="RecipientAliasInfo";
        coder.tableName="TRecipientAlias";
        coder.writeInfoToFile();
        /*coder.className="SendTargetInfo";
        coder.tableName="TSendTarget";
        coder.writeInfoToFile();
        coder.className="DocPhraseInfo";
        coder.tableName="TPhrase";
        coder.writeInfoToFile();
        coder.className="ArchiveInfo";
        coder.tableName="TArchiveDoc";
        coder.writeInfoToFile();
        coder.className="DocAttachmentInfo";
        coder.tableName="TAttachment";
        coder.writeInfoToFile();
        coder.className="DocCoworkOpinionInfo";
        coder.tableName="TCoworkOpinion";
        coder.writeInfoToFile();
        coder.className="DocSendInfo";
        coder.tableName="TSendDoc";
        coder.writeInfoToFile();
        coder.className="DocImageInfo";
        coder.tableName="TImage";
        coder.writeInfoToFile();
        coder.className="DocBorrowInfo";
        coder.tableName="TBorrowDoc";
        coder.writeInfoToFile();
        coder.className="DocExtensionInfo";
        coder.tableName="TExtension";
        coder.writeInfoToFile();
        coder.className="DocJudgmentInfo";
        coder.tableName="TJudgment";
        coder.writeInfoToFile();*/
    }
}