com.ktanx.autocoder.task.DefaultFileHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ktanx-autocoder Show documentation

ktanx-autocoder is a java code generate kit.

The newest version!

package com.ktanx.autocoder.task;

import com.ktanx.autocoder.config.CodeTask;
import com.ktanx.autocoder.config.Configuration;
import com.ktanx.autocoder.config.Constants;
import com.ktanx.autocoder.config.TableMetaData;
import com.ktanx.autocoder.utils.LogUtils;
import com.ktanx.common.exception.KtanxException;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

public class DefaultFileHandler implements FileHandler {

    @Override
    public String writeToFile(String content, TableMetaData tableMetaData, CodeTask codeTask, Configuration configuration) {

        String targetDir = System.getProperty("user.dir");
        File rootPathFile = new File(targetDir);
        String moduleDir = codeTask.getProperties().get(Constants.MODULE_NAME);
        String rootPath = configuration.isChildMode() ? rootPathFile.getParentFile().getAbsolutePath()
                : rootPathFile.getAbsolutePath();
        targetDir = rootPath + File.separator + moduleDir;

        String generatedFileName = codeTask.getGeneratedFileName(tableMetaData.getName());
        File writeFile = new File(targetDir, generatedFileName);
        if (!configuration.isOverwrite() && writeFile.exists()) {
            LogUtils.log("File is exists : " + writeFile.getAbsolutePath());
            return null;
        }
        if (!writeFile.getParentFile().exists()) {
            writeFile.getParentFile().mkdirs();
        }
        try {
            FileWriter targetFileWriter = new FileWriter(writeFile);
            targetFileWriter.write(content);
            targetFileWriter.close();
        } catch (IOException e) {
            throw new KtanxException(e);
        }

        LogUtils.log("Generate file ：" + writeFile.getAbsoluteFile());
        return writeFile.getAbsolutePath();
    }
}