命令行参数支持

Author: wangwenhua

README.md

@@ -1,13 +1,28 @@
-#### CSV文件转Excel
+### CSV文件转Excel
 
 将csv文件拖拽到可爱的初音MM上即会在相同的目录生成Excle文件
 
-可执行jar包
+#### 双击执行 
+可执行jar包，双击即可显示图形界面
 
-自定义窗口  
+**自定义窗口**  
 图片要是透明背景的  
 - images/frame.png -- 主窗口  
 - images/icon.png  -- 任务栏小图标  
 - images/success.png -- 弹出提示框   
 
 
+#### 命令行执行
+
+**命令行使用说明：**  
+> 参数1：输入的CSV文件  
+> 参数2：输出的Excel文件，默认输出为原文件相同的目录下  
+> 参数3：CSV文件编码，如：GBK，默认为UTF-8  
+
+如：
+```
+$ java -jar CSV转Excel工具.jar /opt/export/device.csv
+$ java -jar CSV转Excel工具.jar /opt/export/device.csv /opt/download/device.xlsx
+$ java -jar CSV转Excel工具.jar /opt/export/device.csv /opt/download/device.xlsx UTF-8
+```
+

src/main/java/com/wwh/excel/tools/Main.java

@@ -1,15 +1,62 @@
 package com.wwh.excel.tools;
 
 import java.awt.EventQueue;
+import java.awt.GraphicsEnvironment;
 
 import javax.swing.ImageIcon;
 
 import com.wwh.excel.tools.swing.ImageFrame;
 import com.wwh.excel.tools.swing.PopupDialog;
+import com.wwh.excel.tools.worker.Csv2Xlsx;
 
 public class Main {
     public static void main(String[] args) {
 
+        try {
+            boolean b = false;
+            if (args.length == 0) {
+                // 测试此环境是否支持显示器、键盘和鼠标
+                if (GraphicsEnvironment.isHeadless()) {
+                    System.out.println("无桌面环境，以无头模式运行！");
+                } else {
+                    launchFrame();
+                    b = true;
+                }
+            } else if (args.length == 1) { // 参数化运行
+                b = Csv2Xlsx.convert(args[0], null, null);
+            } else if (args.length == 2) {
+                b = Csv2Xlsx.convert(args[0], args[1], null);
+            } else if (args.length == 3) {
+                b = Csv2Xlsx.convert(args[0], args[1], args[2]);
+            }
+
+            if (!b) {
+                printUseage();
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+            printUseage();
+        }
+
+    }
+
+    private static void printUseage() {
+        System.out.println("");
+        System.out.println("########################################################################");
+        System.out.println("");
+        System.out.println("命令行使用说明：");
+        System.out.println("    参数1：输入的CSV文件");
+        System.out.println("    参数2：输出的Excel文件，默认输出为原文件相同的目录下");
+        System.out.println("    参数3：CSV文件编码，如：GBK，默认为UTF-8");
+        System.out.println("如：");
+        System.out.println("$ java -jar CSV转Excel工具.jar /opt/export/device.csv");
+        System.out.println("$ java -jar CSV转Excel工具.jar /opt/export/device.csv /opt/download/device.xlsx");
+        System.out.println("$ java -jar CSV转Excel工具.jar /opt/export/device.csv /opt/download/device.xlsx UTF-8");
+        System.out.println("");
+
+    }
+
+    private static void launchFrame() {
         EventQueue.invokeLater(new Runnable() {
             public void run() {
                 try {
@@ -25,6 +72,7 @@ public void run() {
                     frame.setVisible(true);
                 } catch (Exception e) {
                     e.printStackTrace();
+                    printUseage();
                 }
             }
         });

src/main/java/com/wwh/excel/tools/swing/ImageFrame.java

@@ -22,7 +22,7 @@
 import javax.swing.JLabel;
 import javax.swing.JOptionPane;
 
-import com.wwh.excel.tools.worker.Csv2Xls;
+import com.wwh.excel.tools.worker.Csv2Xlsx;
 
 public class ImageFrame extends JFrame {
 
@@ -119,7 +119,7 @@ public void drop(DropTargetDropEvent dtde)// 重写适配器的drop方法
                             return;
                         }
 
-                        Csv2Xls.convert(list.get(0));
+                        Csv2Xlsx.convert(list.get(0));
 
                         Point losPoint = ImageFrame.this.getLocationOnScreen();
                         popupDialog.showDelayHide(losPoint.x + dtde.getLocation().x, losPoint.y + dtde.getLocation().y);

src/main/java/com/wwh/excel/tools/worker/Csv2Xls.java renamed to src/main/java/com/wwh/excel/tools/worker/Csv2Xlsx.java

@@ -5,7 +5,9 @@
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.InputStreamReader;
+import java.nio.charset.Charset;
 
+import org.apache.commons.lang3.StringUtils;
 import org.apache.poi.ss.usermodel.Cell;
 import org.apache.poi.ss.usermodel.Row;
 import org.apache.poi.ss.usermodel.Sheet;
@@ -14,23 +16,75 @@
 
 import com.opencsv.CSVReader;
 
-public class Csv2Xls {
+/**
+ * 文件转换工具类
+ * 
+ * @author wwh
+ *
+ */
+public class Csv2Xlsx {
 
-    public static final String CHARSET = "UTF-8";
+    public static final String DEFAULT_CHARSET = "UTF-8";
+
+    /**
+     * @param inputFile  输入的CSV文件
+     * @param outputFile 输出文件
+     * @param charset    原文件编码格式
+     * @throws Exception
+     */
+    public static boolean convert(String inputFile, String outputFile, String charset) throws Exception {
+        if (StringUtils.isBlank(inputFile)) {
+            System.err.println("原文件不能为空！");
+            return false;
+        }
+
+        int lastIndexOf = inputFile.lastIndexOf(".");
+        if (lastIndexOf == -1) {
+            System.err.println("输入文件必须是.CSV格式的文件！");
+            return false;
+        }
+        String suffix = inputFile.substring(lastIndexOf);
+        if ("CSV".equalsIgnoreCase(suffix)) {
+            System.err.println("输入文件必须是.CSV格式的文件！");
+            return false;
+        }
+
+        // 判断文件是否存在
+        File input = new File(inputFile);
+        if (!input.exists()) {
+            System.err.println("原文件【" + inputFile + "】不存在！");
+            return false;
+        }
+        // 输出文件为空时默认在在当前目录生成同名的.xlsx文件
+        if (StringUtils.isBlank(outputFile)) {
+            outputFile = inputFile.subSequence(0, lastIndexOf) + ".xlsx";
+        }
+
+        if (StringUtils.isBlank(charset)) {
+            charset = DEFAULT_CHARSET;
+        }
+
+        return ConvertCsvToXlsx(inputFile, outputFile, charset);
+    }
 
     public static void convert(File inputFile) throws Exception {
         String sourceFile = inputFile.getAbsolutePath();
         String outFile = sourceFile.subSequence(0, sourceFile.lastIndexOf(".")) + ".xlsx";
-        ConvertCSVToXLS(sourceFile, outFile);
+        ConvertCsvToXlsx(sourceFile, outFile, DEFAULT_CHARSET);
     }
 
-    public static void ConvertCSVToXLS(String file, String destFile) throws Exception {
+    public static boolean ConvertCsvToXlsx(String file, String destFile, String charset) throws Exception {
+        if (!Charset.isSupported(charset)) {
+            System.out.println("不支持的字符集：" + charset);
+            return false;
+        }
+
         SXSSFWorkbook wb = null;
         CSVReader reader = null;
         try {
             wb = new SXSSFWorkbook(200);
             Sheet sh = wb.createSheet();
-            BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream(file), CHARSET));
+            BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream(file), charset));
 //            reader = new CSVReader(new FileReader(file));
             reader = new CSVReader(in);
 
@@ -56,6 +110,7 @@ public static void ConvertCSVToXLS(String file, String destFile) throws Exceptio
             wb.write(fileOut);
             fileOut.close();
             wb.dispose();
+
         } catch (Exception e) {
             e.printStackTrace();
             throw e;
@@ -67,5 +122,6 @@ public static void ConvertCSVToXLS(String file, String destFile) throws Exceptio
                 wb.dispose();
             }
         }
+        return true;
     }
 }