step.plans.parser.yaml.YamlPlanConversionTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of step-plans-yaml-parser Show documentation
The newest version!
/*******************************************************************************
 * Copyright (C) 2020, exense GmbH
 *
 * This file is part of STEP
 *
 * STEP is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * STEP is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with STEP.  If not, see .
 ******************************************************************************/
package step.plans.parser.yaml;

import java.io.*;

public class YamlPlanConversionTool {

    public static void main(String[] args) {
        String folder = args.length > 0 ? args[0] : null;
        try (OutputStreamWriter writer = new OutputStreamWriter(System.out)) {
            writer.write("\n ---- STEP PLAN CONVERSION TOOL - STARTED ----\n");
            YamlPlanReader yamlPlanReader = new YamlPlanReader();

            if (folder == null || folder.isEmpty()) {
                writer.write("ERROR: Input folder is not specified in program arguments\n");
                return;
            }
            File inputFolder = new File(folder);
            if (!inputFolder.isDirectory()) {
                writer.write("ERROR: " + folder + " is not a directory\n");
                return;
            }

            File outputFolder = new File(inputFolder, "out");
            if (!outputFolder.exists()) {
                boolean outputFolderOk = outputFolder.mkdir();
                if (!outputFolderOk) {
                    writer.write("ERROR: " + outputFolder.getAbsolutePath() + " is not created\n");
                    return;
                }
            }

            writer.write("Converting plans in " + inputFolder.getAbsolutePath() + " ...\n");

            File[] plainTextPlans = inputFolder.listFiles((dir, name) -> name.matches(".*\\.plan"));
            if (plainTextPlans == null || plainTextPlans.length == 0) {
                writer.write("No plans have been found\n");
                return;
            }
            writer.write(plainTextPlans.length + " plan(s) will be converted to the yaml format\n");

            for (File plainTextPlan : plainTextPlans) {
                File outFile = new File(outputFolder, plainTextPlan.getName() + ".yml");
                try (FileInputStream fis = new FileInputStream(plainTextPlan); FileOutputStream fos = new FileOutputStream(outFile)) {
                    writer.write("Converting " + plainTextPlan.getName() + " ...\n");

                    // convert to simple yaml and save in output file
                    String fileName = plainTextPlan.getName();
                    int extensionIndex = fileName.lastIndexOf(".plan");
                    yamlPlanReader.convertFromPlainTextToYaml(extensionIndex > 0 ? fileName.substring(0, extensionIndex) : fileName, fis, fos);

                    writer.write("SUCCESS: " + outFile.getAbsolutePath() + "\n");
                } catch (Exception e) {
                    writer.write("ERROR: " + plainTextPlan.getName() + " hasn't been converted!\n");
                    e.printStackTrace();
                }
            }
            writer.write("---- STEP PLAN CONVERSION TOOL - FINISHED ----\n");
        } catch (IOException e) {
            System.out.println("ERROR: Internal error");
            e.printStackTrace();
        }
    }
}