Main update -> Updated arg handling

codepressed · codepressed · commit 5a64d9929f78 · 2023-02-03T18:23:00.000+01:00
diff --git a/README.md b/README.md
@@ -2,8 +2,9 @@
 * It is a console application with var args
 * Uses the traditional methodology to generate the xml (without a library), through
 of nodes and tree structure.
-* Can read ANY CSV File
-* Args are:
+* Can read ANY CSV File (comma-separated & semicolon-separated)
+
+Args are:
   - [0] Input file 
   - [1] Output file 
   - [2] Element node name and optionally, 
diff --git a/src/main/java/com/codepressed/CSVtoXML/Main.java b/src/main/java/com/codepressed/CSVtoXML/Main.java
@@ -2,10 +2,6 @@
 
 
 import org.w3c.dom.Document;
-import javax.xml.parsers.ParserConfigurationException;
-import javax.xml.transform.TransformerException;
-import java.io.IOException;
-import java.util.ArrayList;
 import java.util.List;
 import java.util.logging.Level;
 import java.util.logging.Logger;
@@ -23,35 +19,20 @@ public class Main {
     public static void main(String[] args) {
 
         //Arg validator
-        if (args.length == 0) {
+        if (args.length <2) {
             logger.log(Level.SEVERE, "No args were specified.");
-            System.exit(0);
+            System.exit(1);
         }
         //Vars Initialization
         String csvFile = args[0];
         String xmlFile = args[1];
-        String elementName;
-        String csvSplit = ",(?=([^\"]*\"[^\"]*\")*[^\"]*$)";
-
-        try {
-            elementName = args[2];
-        } catch (ArrayIndexOutOfBoundsException e) {
-            logger.log(Level.INFO, "Since you didn't specify a element name, 'element' will be the parental node.");
-            elementName = "element";
-        }
-
-        try {
-            if (args[3] == "-s")
-                csvSplit = ";(?=([^\"]*\"[^\"]*\")*[^\"]*$)";
-        } catch (ArrayIndexOutOfBoundsException e){
-        }
+        String elementName = args.length>=3 && !args[3].equals("-s") ? args[2] : "element";
+        String csvSplit = (args.length>3 && args[3].equals("-s") || args.length>4 && args[4].equals("-s")) ? ";(?=([^\"]*\"[^\"]*\")*[^\"]*$)":",(?=([^\"]*\"[^\"]*\")*[^\"]*$)";
 
-                List<String[]> elements;
-                elements = XMLutils.readCsvFile(csvFile, csvSplit);
-                Document xmlDoc;
-                xmlDoc = XMLutils.createXmlDocument(elements, elementName);
-                XMLutils.writeXmlDocumentToFile(xmlDoc, xmlFile);
-            }
+        List <String[]> elements = XMLutils.readCsvFile(csvFile, csvSplit);
+        Document xmlDoc = XMLutils.createXmlDocument(elements, elementName);
+        XMLutils.writeXmlDocumentToFile(xmlDoc, xmlFile);
+    }
     }
 
 
