增加导出查询数据功能

.idea/QuakeViewer.iml → QuakeViewer.iml

@@ -1,8 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<module type="JAVA_MODULE" version="4">
+<module version="4">
   <component name="AdditionalModuleElements">
     <content url="file://$MODULE_DIR$" dumb="true">
-      <sourceFolder url="file://$MODULE_DIR$" isTestSource="false" />
       <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
       <sourceFolder url="file://$MODULE_DIR$/src/com.rubby.quakeviewer/view" type="java-resource" />
     </content>

README.md

@@ -12,9 +12,9 @@ java -jar QuakeViewer.jar
 
 ![image-20231214162439986](README.assets/image-20231214162439986.png)
 
-可以右键复制URL、ip等内容：
+可以右键复制URL、ip等内容，可以导出选中的数据、全部查询数据：
 
-![image-20231214162544959](README.assets/image-20231214162544959.png)
+![image-20231216005847289](README.assets/image-20231216005847289.png)
 
 双击可以实现跳转：
 

pom.xml

@@ -29,6 +29,16 @@
             <artifactId>controlsfx</artifactId>
             <version>8.40.18</version>
         </dependency>
+        <dependency>
+            <groupId>com.opencsv</groupId>
+            <artifactId>opencsv</artifactId>
+            <version>4.1</version>
+        </dependency>
+        <dependency>
+            <groupId>com.alibaba</groupId>
+            <artifactId>fastjson</artifactId>
+            <version>1.2.78</version>
+        </dependency>
     </dependencies>
 
 </project>

src/com.rubby.quakeviewer/controller/QuakeViewerController.java

@@ -6,6 +6,10 @@
 import com.rubby.quakeviewer.util.CreateConfigFile;
 import com.rubby.quakeviewer.util.JsonRequest;
 import com.rubby.quakeviewer.util.ReadFile;
+
+import java.io.*;
+import java.nio.charset.StandardCharsets;
+import java.text.SimpleDateFormat;
 import javafx.application.Platform;
 import javafx.collections.FXCollections;
 import javafx.concurrent.Task;
@@ -28,9 +32,9 @@
 import org.controlsfx.control.textfield.TextFields;
 
 import java.awt.*;
-import java.io.File;
+
+import com.opencsv.CSVWriter;
 import java.io.FileWriter;
-import java.io.IOException;
 import java.math.BigInteger;
 import java.net.URI;
 import java.nio.file.Files;
@@ -231,6 +235,108 @@ public void onClikCopyCompany(ActionEvent actionEvent) {
         clipboardContent.putString(content);
         Clipboard.getSystemClipboard().setContent(clipboardContent);
     }
+
+    @FXML
+    public void onExportDatas(ActionEvent actionEvent) throws IOException {
+        String timestamp = new SimpleDateFormat("yyyyMMddHHmmss").format(new Date());
+        String fileName = "导出数据_" + timestamp + ".csv";
+
+        List<assets> slectedAssets = table.getSelectionModel().getSelectedItems();
+
+        FileOutputStream os = new FileOutputStream(fileName);
+        os.write(0xef); //加上这句话
+        os.write(0xbb); //加上这句话
+        os.write(0xbf); //加上这句话
+        try (CSVWriter writer = new CSVWriter(new OutputStreamWriter(os, StandardCharsets.UTF_8))) {
+            // 写入CSV文件头
+            String[] header = {"序号", "Url", "Ip","Port","Domain","Service","Title","Cms","Icp","Company"};
+            writer.writeNext(header);
+
+            // 遍历资产列表，写入每一行数据
+            for (assets asset : slectedAssets) {
+                String[] row = {
+                        String.valueOf(asset.getCount()),
+                        asset.getUrl(),
+                        asset.getIp(),
+                        asset.getPort(),
+                        asset.getDomain(),
+                        asset.getService(),
+                        asset.getTitle(),
+                        asset.getCms(),
+                        asset.getIcp(),
+                        asset.getCompany()
+                };
+                writer.writeNext(row);
+            }
+            Alert alert = new Alert(Alert.AlertType.INFORMATION);
+            alert.setTitle("导出成功");
+            alert.setHeaderText(null);
+            alert.setContentText("CSV文件导出成功，文件名: " + fileName);
+
+            // 显示弹窗
+            alert.showAndWait();
+        } catch (IOException e) {
+            e.printStackTrace();
+            Alert alert = new Alert(Alert.AlertType.INFORMATION);
+            alert.setTitle("导出失败");
+            alert.setHeaderText(null);
+            alert.setContentText("CSV文件导出失败：" + e.getMessage());
+
+            // 显示弹窗
+            alert.showAndWait();
+        }
+    }
+
+
+    @FXML
+    public void onExportAllDatas(ActionEvent actionEvent) throws IOException {
+        String timestamp = new SimpleDateFormat("yyyyMMddHHmmss").format(new Date());
+        String fileName = "导出数据_" + timestamp + ".csv";
+
+        List<assets> slectedAssets = table.getItems();
+        FileOutputStream os = new FileOutputStream(fileName);
+        os.write(0xef); //加上这句话
+        os.write(0xbb); //加上这句话
+        os.write(0xbf); //加上这句话
+        try (CSVWriter writer = new CSVWriter(new OutputStreamWriter(os, StandardCharsets.UTF_8))) {
+            // 写入CSV文件头
+            String[] header = {"序号", "Url", "Ip","Port","Domain","Service","Title","Cms","Icp","Company"};
+            writer.writeNext(header);
+
+            // 遍历资产列表，写入每一行数据
+            for (assets asset : slectedAssets) {
+                String[] row = {
+                        String.valueOf(asset.getCount()),
+                        asset.getUrl(),
+                        asset.getIp(),
+                        asset.getPort(),
+                        asset.getDomain(),
+                        asset.getService(),
+                        asset.getTitle(),
+                        asset.getCms(),
+                        asset.getIcp(),
+                        asset.getCompany()
+                };
+                writer.writeNext(row);
+            }
+            Alert alert = new Alert(Alert.AlertType.INFORMATION);
+            alert.setTitle("导出成功");
+            alert.setHeaderText(null);
+            alert.setContentText("CSV文件导出成功，文件名: " + fileName);
+
+            // 显示弹窗
+            alert.showAndWait();
+        } catch (IOException e) {
+            e.printStackTrace();
+            Alert alert = new Alert(Alert.AlertType.INFORMATION);
+            alert.setTitle("导出失败");
+            alert.setHeaderText(null);
+            alert.setContentText("CSV文件导出失败：" + e.getMessage());
+
+            // 显示弹窗
+            alert.showAndWait();
+        }
+    }
 //    @FXML
 //    public void jumpToURL(MouseEvent mouseEvent) {
 //        assets seletedAssets = table.getSelectionModel().getSelectedItem();