/*
 * The contents of this file are subject to the terms of the Common Development
 * and Distribution License (the License). You may not use this file except in
 * compliance with the License.
 *
 * You can obtain a copy of the License at http://www.netbeans.org/cddl.html
 * or http://www.netbeans.org/cddl.txt.
 *
 * When distributing Covered Code, include this CDDL Header Notice in each file
 * and include the License file at http://www.netbeans.org/cddl.txt.
 * If applicable, add the following below the CDDL Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyrighted [year] [name of copyright owner]"
 *
 * The Original Software is NetBeans. The Initial Developer of the Original
 * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
 * Microsystems, Inc. All Rights Reserved.
 */


package org.netbeans.modules.jbi.apisupport.project.wizard;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;
import javax.swing.JEditorPane;
import javax.swing.text.DefaultEditorKit;
import javax.swing.text.Document;
import javax.swing.text.EditorKit;
import javax.swing.text.StyledDocument;
import org.netbeans.modules.jbi.apisupport.JbiAdminSettings;
import org.netbeans.modules.jbi.apisupport.common.JbiComponentDescriptor;
import org.netbeans.modules.jbi.apisupport.common.JbiDescriptorFactory;
import org.netbeans.modules.jbi.apisupport.common.TemplateUtil;
import org.netbeans.modules.jbi.apisupport.common.Util;
import org.netbeans.modules.jbi.apisupport.common.wizard.ProjectWizardModel;
import org.netbeans.modules.jbi.apisupport.project.JbiCompProject;
import org.netbeans.modules.jbi.apisupport.project.JbiCompProjectGenerator;
import org.netbeans.modules.jbi.apisupport.project.JbiCompProjectProperties;
import org.netbeans.spi.project.support.ant.EditableProperties;
import org.openide.WizardDescriptor;
//import org.openide.cookies.SourceCookie;
import org.openide.filesystems.FileLock;
import org.openide.filesystems.FileObject;
import org.openide.filesystems.FileUtil;
import org.openide.loaders.DataFolder;
import org.openide.loaders.DataObject;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import org.netbeans.api.project.Project;
import org.netbeans.api.project.ProjectManager;
import org.netbeans.spi.project.support.ant.AntProjectHelper;
import org.netbeans.spi.project.ui.support.ProjectChooser;
//import org.openide.src.SourceElement;
import org.openide.text.FilterDocument;
import org.openide.text.IndentEngine;

/**
 * This Wizard model contains the data to generate a jbi component project for developing
 * a jbi component ( service engine or binding component )
 *
 * @author  chikkala
 */
public abstract class JbiComponentProjectWizardModel extends ProjectWizardModel {
    
    public static final String COMP_PRJ_WIZARD_TYPE = "COMP_PRJ_WIZARD_TYPE";
    
    public static final String SERVICE_ENGINE_COMP_TYPE = "service-engine";
    public static final String BINDING_COMPONENT_COMP_TYPE = "binding-component";
    
    public static final String SAMPLE_COMP_PACKAGE = "com.sample.component";
    public static final String SAMPLE_SE_PACKAGE = "com.sample.engine";
    public static final String SAMPLE_BC_PACKAGE = "com.sample.binding";
    
    public static final String SAMPLE_COMP_NAME = "JbiComponent";
    public static final String SAMPLE_SE_NAME = "ServiceEngine";
    public static final String SAMPLE_BC_NAME = "BindingComponent";
    
    private String mCompType;
    private String mName;
    private String mDescription;
    private String mPackageName;
    private String mTargetServerID;
    
    /** Creates a new instance of ComponentProjectWizardModel */
    public JbiComponentProjectWizardModel(WizardDescriptor wiz) {
        this(wiz, COMP_PRJ_WIZARD_TYPE );
    }
    /** Creates a new instance of ComponentProjectWizardModel */
    public JbiComponentProjectWizardModel(WizardDescriptor wiz, String wizType) {
        super(wiz,wizType);
    }
    
    public static JbiComponentProjectWizardModel createComponentProjectWizardModel(
            WizardDescriptor wiz, String compType) {
        JbiComponentProjectWizardModel wizModel =  null;
        if ( JbiComponentProjectWizardModel.BINDING_COMPONENT_COMP_TYPE.equals(compType)) {
            wizModel =  new BCProjectWizardModel(wiz);
            wizModel.setComponentType(compType);
        } else if (JbiComponentProjectWizardModel.SERVICE_ENGINE_COMP_TYPE.equals(compType)) {
            wizModel =  new SEProjectWizardModel(wiz);
            wizModel.setComponentType(compType);
        }
        
        return wizModel;
    }
    
    public void initializeOthers() {
        WizardDescriptor wiz = this.getWizardDescriptor();
        String templateName = "JbiComponent";
        DataObject templateDO = this.getTemplateDataObject();
        if ( templateDO != null ) {
            templateName = templateDO.getName();
        }
        
        this.setTargetName(templateName);
        this.setProjectName(templateName);
        //TODO fix it
        if ( this.getComponentType() == null  ) {
            this.setComponentType(SERVICE_ENGINE_COMP_TYPE);
        }
        
        if ( this.getComponentName() == null ) {
            initDescription(templateName);
        }
        
        if ( this.getPackageName() == null ) {
            initPackageName(templateName);
        }
        
        initTargetServerID();
    }
    
    public void instantiateOthers() throws IOException {
        createComponentProject();
    }
    
    public void uninitializeOthers() {
        WizardDescriptor wiz = this.getWizardDescriptor();
    }
    
    public static String getComponentDescription(String compName, String type) {
        
        String desc = "";
        
        if ( JbiComponentProjectWizardModel.SERVICE_ENGINE_COMP_TYPE.equals(type)) {
            desc = "Description of service engine : " + compName ;
        } else if (JbiComponentProjectWizardModel.BINDING_COMPONENT_COMP_TYPE.equals(type)) {
            desc = "Description of binding component : " + compName ;
        } else {
            desc = "Description of Jbi Component : " + compName;
        }
        
        return desc;
    }
    
    public static String getComponentName(String displayName, String type) {
        
        StringBuffer builder = new StringBuffer();
        boolean firstLetter = true;
        for (int i=0; i< displayName.length(); i++) {
            char c = displayName.charAt(i);
            if ((!firstLetter && Character.isJavaIdentifierPart(c)) || (firstLetter && Character.isJavaIdentifierStart(c))) {
                firstLetter = false;
                builder.append(c);
            }
        }
        
        String compName = JbiComponentProjectWizardModel.SAMPLE_COMP_NAME;
        
        if ( builder.length() == 0 ) {
            if ( JbiComponentProjectWizardModel.SERVICE_ENGINE_COMP_TYPE.equals(type)) {
                compName = JbiComponentProjectWizardModel.SAMPLE_SE_NAME;
            } else if (JbiComponentProjectWizardModel.BINDING_COMPONENT_COMP_TYPE.equals(type)) {
                compName = JbiComponentProjectWizardModel.SAMPLE_BC_NAME;
            }
        } else {
            compName = builder.toString();
        }
        return compName;
        
    }
    
    
    public static String getPackageName(String displayName, String type) {
        
        StringBuffer builder = new StringBuffer();
        boolean firstLetter = true;
        for (int i=0; i< displayName.length(); i++) {
            char c = displayName.charAt(i);
            if ((!firstLetter && Character.isJavaIdentifierPart(c)) || (firstLetter && Character.isJavaIdentifierStart(c))) {
                firstLetter = false;
                if (Character.isUpperCase(c)) {
                    c = Character.toLowerCase(c);
                }
                builder.append(c);
            }
        }
        
        String packageName = JbiComponentProjectWizardModel.SAMPLE_COMP_PACKAGE;
        
        if ( builder.length() == 0 ) {
            if ( JbiComponentProjectWizardModel.SERVICE_ENGINE_COMP_TYPE.equals(type)) {
                packageName = JbiComponentProjectWizardModel.SAMPLE_SE_PACKAGE;
            } else if (JbiComponentProjectWizardModel.BINDING_COMPONENT_COMP_TYPE.equals(type)) {
                packageName = JbiComponentProjectWizardModel.SAMPLE_BC_PACKAGE;
            }
        } else {
            packageName = builder.toString();
        }
        return packageName;
    }
    
    /**
     * Getter for property componentType.
     * @return Value of property componentType.
     */
    public String getComponentType() {
        return this.mCompType;
    }
    
    /**
     * Setter for property componentType.
     * @param componentType New value of property componentType.
     */
    public void setComponentType(String componentType) {
        this.mCompType = componentType;
    }
    
    /**
     * Getter for property name.
     * @return Value of property name.
     */
    public String getComponentName() {
        return this.mName;
    }
    
    /**
     * Setter for property name.
     * @param name New value of property name.
     */
    public void setComponentName(String name) {
        this.mName = name;
    }
    
    /**
     * Getter for property description.
     * @return Value of property description.
     */
    public String getComponentDescription() {
        return this.mDescription;
    }
    
    /**
     * Setter for property description.
     * @param description New value of property description.
     */
    public void setComponentDescription(String description) {
        this.mDescription = description;
    }
    
    /**
     * Getter for property packageName.
     * @return Value of property packageName.
     */
    public String getPackageName() {
        return mPackageName;
    }
    
    /**
     * Setter for property packageName.
     * @param packageName New value of property packageName.
     */
    public void setPackageName(String packageName) {
        this.mPackageName = packageName;
    }
    
    /**
     * Getter for property TargetServerID.
     * @return Value of property TargetServerID.
     */
    public String getTargetServerID() {
        return mTargetServerID;
    }
    
    /**
     * Setter for property TargetServerID.
     * @param packageName New value of property TargetServerID.
     */
    public void setTargetServerID(String targetServerID) {
        this.mTargetServerID = targetServerID;
    }
    
    protected void initTargetServerID() {
        String serverID = JbiAdminSettings.UNKNOWN_SERVER_ID;
        JbiAdminSettings[] list = JbiAdminSettings.JbiAdminSettingsImpl.getAllJbiAdminSettings();
        
        for ( int i=0; i < list.length; ++i) {
            serverID = list[i].getAppServerId();
            if (!JbiAdminSettings.UNKNOWN_SERVER_ID.equals(serverID)) {
                break;
            }
        }
        this.mTargetServerID = serverID;
    }
    
    protected void initDescription(String projectName ) {
        
        String type = this.getComponentType();
        String name = getComponentName(projectName, type);
        String desc = getComponentDescription(name, type);
        
        this.setComponentName(name);
        this.setComponentDescription(desc);
    }
    
    protected void initPackageName(String projectName ) {
        
        String type = this.getComponentType();
        String packageName = getPackageName(projectName, type);
        this.setPackageName(packageName);
    }
    
    protected abstract void createProjectSpecificArtifacts(FileObject prjDirFO) throws IOException;
    
    private void createComponentProject() throws IOException {
        FileObject prjDirFO = createProjectCommonArtifacts();
        createProjectSpecificArtifacts(prjDirFO);
        updateProjectProperties(prjDirFO);
    }
    
    private void updateProjectProperties(FileObject prjDirFO) throws IOException {
        
        Project p = ProjectManager.getDefault().findProject(prjDirFO);
        JbiCompProject jbiCompPrj = p.getLookup().lookup(JbiCompProject.class);
        AntProjectHelper prjHelper = jbiCompPrj.getAntProjectHelper();
        EditableProperties ep = prjHelper.getProperties(AntProjectHelper.PROJECT_PROPERTIES_PATH);
        
        String confDir = ep.getProperty(JbiCompProjectProperties.CONF_DIR);
        FileObject metaInfDirFO = FileUtil.createFolder(prjDirFO, confDir + "/META-INF");
        FileObject jbiXmlFO = metaInfDirFO.getFileObject("jbi.xml");
        String desc = "";
        String compClass = "";
        String btClass = "";
        if ( jbiXmlFO != null ) {
            JbiComponentDescriptor compDesc =
                    JbiDescriptorFactory.getInstance().createJbiComponentDescriptor(jbiXmlFO);
            desc = compDesc.getDescription();
            compClass = compDesc.getComponentClassName();
            btClass = compDesc.getBootstrapClassName();
        }
        
        ep.setProperty(JbiCompProjectProperties.JBI_COMPONENT_DESC, desc);
        ep.setProperty(JbiCompProjectProperties.JBI_COMPONENT_CLASS, compClass);
        ep.setProperty(JbiCompProjectProperties.JBI_COMPONENT_BT_CLASS, btClass);
        
        prjHelper.putProperties(AntProjectHelper.PROJECT_PROPERTIES_PATH, ep);
    }
    
    private FileObject createMavenProjectArtifacts(FileObject prjDirFO) throws IOException {
        
        String compName = this.getComponentName();
        String groupId = "open-jbi-components";
        String artifactId = compName;
        String version="1.0-SNAPSHOT";
        
        Map<String,String> tokenMap = new HashMap<String,String>();
        
        tokenMap.put("MVN_POM_GROUP_ID", groupId);
        tokenMap.put("MVN_POM_ARTIFACT_ID", artifactId);
        tokenMap.put("MVN_POM_VERSION", version);
        
        String pomXmlTemplatePath = "codegen/components/common/pom.xml";
        FileObject pomXmlFO = createFromTemplateFileObject(prjDirFO, pomXmlTemplatePath, null, tokenMap);
        
        String m2nbBuildXmlPath = "codegen/components/common/m2nbbuild.xml";
        FileObject m2nbBuildXmlFO = createFromTemplateFileObject(prjDirFO, m2nbBuildXmlPath, null, tokenMap);        

        String m2nbBuildImplXmlPath = "codegen/components/common/m2nbbuild-impl.xml";
        FileObject m2nbBuildImplXmlFO = createFromTemplateFileObject(prjDirFO, m2nbBuildImplXmlPath, null, tokenMap);        
        
        String m2nbBuildPropPath = "codegen/components/common/m2nbbuild.properties";
        FileObject m2nbBuildPropFO = createFromTemplateFileObject(prjDirFO, m2nbBuildPropPath, null, tokenMap);        
                
        return pomXmlFO;
    }
    /**
     * return project direcotry file object
     */
    private FileObject createProjectCommonArtifacts() throws IOException {
        
        File prjDirFile = this.getProjectDirectory();
        if (prjDirFile != null) {
            prjDirFile = FileUtil.normalizeFile(prjDirFile);
        }
        
        String prjName = this.getProjectName();
        List prjRefList = new ArrayList();
        String compType = this.getComponentType();
        String compName = this.getComponentName();
        String serverID = this.getTargetServerID();
        
        JbiCompProjectGenerator prjGen =
                new JbiCompProjectGenerator(prjDirFile, prjName, compType, compName);
        prjGen.setTargetServerID(serverID);
        prjGen.setProjectReferences(prjRefList);
        
        AntProjectHelper antPrjHelper = prjGen.createProject();
        FileObject prjDirFO = null;
        FileObject srcRoot = null;
        FileObject confRoot = null;
        
        prjDirFO = antPrjHelper.getProjectDirectory();
        
        // mount if not existing
        prjDirFO = FileUtil.toFileObject(prjDirFile);
        
        Project p = ProjectManager.getDefault().findProject(prjDirFO);
        
        File prjParentFile = null;
        
        prjParentFile = (prjDirFile != null) ? prjDirFile.getParentFile() : null;
        if (prjParentFile != null && prjParentFile.exists()) {
            ProjectChooser.setProjectsFolder(prjParentFile);
        }
        
        this.addCreatedFileObject(prjDirFO);
        
        srcRoot = prjDirFO.getFileObject(JbiCompProjectProperties.SRC_DIR_VALUE);        //NOI18N
        this.addCreatedFileObject(srcRoot);
        
        confRoot = prjDirFO.getFileObject(JbiCompProjectProperties.CONF_DIR_VALUE);        //NOI18N
        this.addCreatedFileObject(confRoot);
        
        
        Map<String, String> map = new HashMap<String, String>();
        map.put("JBI_COMP_NAME", this.getComponentName());
        
        FileObject jbiSupportFolder = Util.getJBISupportFolder();
        
//        String componentArtifactsZipTemplatePath = "codegen/components/common/component-common-artifacts.zip";        
//        FileObject compArtifactsZipFO = jbiSupportFolder.getFileObject(componentArtifactsZipTemplatePath);
//        Util.unZipArchive(compArtifactsZipFO.getInputStream(), prjDirFO);
        
        String commonZipTemplatePath = "codegen/components/common/common-artifacts.zip";
        FileObject commonZipFO = jbiSupportFolder.getFileObject(commonZipTemplatePath);
        Util.unZipArchive(commonZipFO.getInputStream(), prjDirFO);
        
        createMavenProjectArtifacts(prjDirFO);
        
        return prjDirFO;
        
    }
    
    protected Map createJbiXmlTokenMap(String rtClass, String btClass) {
        Map<String, String> map = new HashMap<String, String>();
        
        map.put("JBI_COMP_TYPE", this.getComponentType());
        
        map.put("JBI_COMP_NAME", this.getComponentName());
        map.put("JBI_COMP_DESC", this.getComponentDescription());
        
        map.put("JBI_COMP_RT_PACKAGE", this.getPackageName());
        map.put("JBI_COMP_RT_CLASS", rtClass);
        
        map.put("JBI_COMP_BT_PACKAGE", this.getPackageName());
        map.put("JBI_COMP_BT_CLASS", btClass);
        
        return map;
    }
    
    protected FileObject createJbiXml(FileObject prjDirFO, String jbiXmlTemplatePath, Map tokenMap ) throws IOException {
        
        FileObject confRoot = FileUtil.createFolder(prjDirFO, JbiCompProjectProperties.CONF_DIR_VALUE);
        // create META-INF dir
        FileObject metaInfFolder = FileUtil.createFolder(confRoot, "META-INF");
        
        FileObject jbiXmlFO = createFromTemplateFileObject(metaInfFolder, jbiXmlTemplatePath, null, tokenMap);
        return jbiXmlFO;
    }
    
    protected FileObject createPackage(FileObject srcRoot, String packageName) throws IOException {
        FileObject pkgFolder = srcRoot;
        
        if ( packageName != null ) {
            String pkgFileName = packageName.replace('.', '/');
            pkgFolder = FileUtil.createFolder( srcRoot, pkgFileName );
        }
        return pkgFolder;
    }
    
    protected StringBuffer readFromFO(FileObject fo) {
        InputStream inS = null;
        StringWriter writer = new StringWriter();
        PrintWriter out = new PrintWriter(writer);
        
        try {
            inS = fo.getInputStream();
            InputStreamReader reader = new InputStreamReader(inS);
            BufferedReader inReader = new BufferedReader(reader);
            
            for (String line = null; (line = inReader.readLine()) != null ;) {
                out.println(line);
            }
            out.close();
        } catch (Exception ex) {
            ex.printStackTrace();
        } finally {
            TemplateUtil.closeInputStream(inS);
            TemplateUtil.closeWriter(out);
            TemplateUtil.closeWriter(writer);
            
        }
        
        return writer.getBuffer();
    }
    
    protected StyledDocument createDocument() {
        EditorKit kit = JEditorPane.createEditorKitForContentType("text/x-java"); // NOI18N
        if (kit == null) {
            kit = new DefaultEditorKit();
        }
        Document doc = kit.createDefaultDocument();
        if (doc instanceof StyledDocument) {
            return (StyledDocument)doc;
        }
        return new FilterDocument(doc);
    }
    
    protected StringBuffer toFormattedJavaSource(StringBuffer javaSrc) {
        StringWriter sw = new StringWriter();
        StyledDocument doc = createDocument();
        IndentEngine indentator = IndentEngine.find(doc);
        PrintWriter pw = new PrintWriter(indentator.createWriter(doc, 0, sw));
        //    PrintWriter pw = new PrintWriter(sw);
        try {
            pw.print(javaSrc.toString());
        } catch (Exception ex ) {
            ex.printStackTrace();
        } finally {
            pw.close();
        }
        return sw.getBuffer();
    }
    
    protected FileObject createFromJavaTemplate(FileObject destFO,
            String jbiTemplatePath, String name) throws IOException {
        
        FileObject fo = createFromTemplateDataObject(destFO, jbiTemplatePath, name);
        
        DataObject dataObj = null;
        FileLock outLock = null;
        OutputStream outS = null;
        InputStream inS = null;
        
        try {
            dataObj = DataObject.find(fo);
//            SourceCookie srcCookie = (SourceCookie) dataObj.getCookie(SourceCookie.class);
//            SourceElement srcEl = srcCookie.getSource();
            // String javaSrc = srcEl.toString();
            StringBuffer javaSrc = readFromFO(fo);
            javaSrc = toFormattedJavaSource(javaSrc);
            // System.out.println(javaSrc);
            inS = new ByteArrayInputStream(javaSrc.toString().getBytes());
            outLock = fo.lock();
            outS = fo.getOutputStream(outLock);
            FileUtil.copy(inS, outS);
        } catch ( Exception ex) {
            ex.printStackTrace();
        } finally {
            TemplateUtil.closeInputStream(inS);
            TemplateUtil.closeOutputStream(outS);
            if ( outLock != null ) {
                outLock.releaseLock();
            }
        }
        return fo;
    }
    
    protected FileObject createFromTemplateDataObject(FileObject destFO,
            String jbiTemplatePath, String name ) throws IOException {
        FileObject jbiSupportFolder = Util.getJBISupportFolder();
        FileObject templateFO = jbiSupportFolder.getFileObject(jbiTemplatePath);
        DataObject templateDO = null;
        DataObject newDO = null;
        FileObject newFO = null;
        
        if ( templateFO == null ) {
            return null;
        }
        String newName = ( name != null ) ? name : templateFO.getName();
        try {
            templateDO = DataObject.find(templateFO);
            newDO = templateDO.createFromTemplate(DataFolder.findFolder(destFO),newName);
            newFO = newDO.getPrimaryFile();
        } catch (Exception ex) {
            ex.printStackTrace();
            newFO = FileUtil.copyFile(templateFO, destFO, newName);
        }
        return newFO;
    }
    
    protected FileObject createFromTemplateFileObject(FileObject destFO,
            String jbiTemplatePath, String name, Map tokenMap ) throws IOException {
        FileObject jbiSupportFolder = Util.getJBISupportFolder();
        FileObject templateFO = jbiSupportFolder.getFileObject(jbiTemplatePath);
        DataObject templateDO = null;
        DataObject newDO = null;
        FileObject newFO = null;
        
        if ( templateFO == null ) {
            System.out.println("createFromTemplateFileObject: template not found " + jbiTemplatePath);
            return null;
        }
        String newName = ( name != null ) ? name : templateFO.getName();
        
        HashMap map = new HashMap<String,String>();
        Map defMap = TemplateUtil.createDefaultTokenMap(newName, templateFO.getExt());
        map.putAll(defMap);
        map.putAll(tokenMap);
        
        newFO = TemplateUtil.createFromTemplate(templateFO, map, destFO, newName);
        
        return newFO;
    }
    
}