org.mybatis.generator.ext.api.MeldMergeShellCallback.java Source code

Introduction

Here is the source code for org.mybatis.generator.ext.api.MeldMergeShellCallback.java

Source

/**
 *    Copyright 2006-2015 the original author or authors.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */
package org.mybatis.generator.ext.api;

import org.apache.commons.exec.CommandLine;
import org.apache.commons.exec.DefaultExecutor;
import org.apache.commons.exec.ExecuteException;
import org.apache.commons.exec.ExecuteWatchdog;
import org.apache.tools.ant.util.FileUtils;
import org.mybatis.generator.exception.ShellException;
import org.mybatis.generator.internal.DefaultShellCallback;
import org.mybatis.generator.util.GenUtil;

import java.io.*;

/**
 * ???meld
 */
public class MeldMergeShellCallback extends DefaultShellCallback {

    @Override
    public boolean isMergeSupported() {
        return true;
    }

    /** ?
     *
     */
    public MeldMergeShellCallback() {
        super(true);
    }

    private void writeFile(File file, String content, String fileEncoding) throws IOException {
        FileOutputStream fos = new FileOutputStream(file, false);
        OutputStreamWriter osw;
        if (fileEncoding == null) {
            osw = new OutputStreamWriter(fos);
        } else {
            osw = new OutputStreamWriter(fos, fileEncoding);
        }

        BufferedWriter bw = new BufferedWriter(osw);
        bw.write(content);
        bw.close();
    }

    @Override
    public String mergeJavaFile(String newFileSource, String existingFileFullPath, String[] javadocTags,
            String fileEncoding) throws ShellException {
        try {
            File existingFile = new File(existingFileFullPath);
            File directory = existingFile.getParentFile();

            File tmpFile = GenUtil.getUniqueFileName(directory, existingFile.getName());
            writeFile(tmpFile, newFileSource, "UTF-8");
            Runtime rt = Runtime.getRuntime();
            Process proc = rt.exec("meld " + existingFileFullPath + " " + tmpFile.getAbsolutePath());
            int exitVal = proc.waitFor();
            System.out.println("Process exitValue: " + exitVal);

            org.apache.commons.io.FileUtils.deleteQuietly(tmpFile);
            return org.apache.commons.io.FileUtils.readFileToString(existingFile);
        } catch (Exception ex) {
            return null;
        }
    }

    public static boolean exec(String command) {
        CommandLine commandLine = CommandLine.parse(command);
        DefaultExecutor executor = new DefaultExecutor();
        int exitValue = DefaultExecutor.INVALID_EXITVALUE;
        ExecuteWatchdog watchdog = new ExecuteWatchdog(2000);
        executor.setWatchdog(watchdog);
        try {
            exitValue = executor.execute(commandLine);
        } catch (ExecuteException e) {
            exitValue = e.getExitValue();
        } catch (IOException e) {
            System.err.println(e.getMessage());
        }
        System.out.println("Exit :" + exitValue);
        return (exitValue == 0);
    }

    @Override
    public File getDirectory(String targetProject, String targetPackage) throws ShellException {
        return super.getDirectory(targetProject, targetPackage);
    }

    public static void main(String[] args) {
        exec("meld ");
    }
}