From 9135b8cbd2e443e3e8b44ac36408410f762186c3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E1=B4=8B=E1=B4=87=CA=8F?=
 <24655118+gh0stkey@users.noreply.github.com>
Date: Wed, 18 Oct 2023 00:49:02 +0800
Subject: [PATCH] Add files via upload

---
 src/main/java/burp/rule/utils/RuleTool.java | 2 +-
 src/main/java/burp/rule/utils/YamlTool.java | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/burp/rule/utils/RuleTool.java b/src/main/java/burp/rule/utils/RuleTool.java
index d65ff87..1525d91 100644
--- a/src/main/java/burp/rule/utils/RuleTool.java
+++ b/src/main/java/burp/rule/utils/RuleTool.java
@@ -25,7 +25,7 @@ public class RuleTool {
             // 获取官方规则文件，在线更新写入
             FileOutputStream fileOutputStream = new FileOutputStream(this.rulesFilePath);
             fileOutputStream.write(httpResponse.body().bytes());
-            JOptionPane.showMessageDialog(null, "Config file updated successfully!", "Error",
+            JOptionPane.showMessageDialog(null, "Rules updated successfully!", "Info",
                     JOptionPane.INFORMATION_MESSAGE);
         } catch (Exception ignored) {
             JOptionPane.showMessageDialog(null, "Please check your network!", "Error",
diff --git a/src/main/java/burp/rule/utils/YamlTool.java b/src/main/java/burp/rule/utils/YamlTool.java
index ff8c7be..08de7be 100644
--- a/src/main/java/burp/rule/utils/YamlTool.java
+++ b/src/main/java/burp/rule/utils/YamlTool.java
@@ -1,6 +1,8 @@
 package burp.rule.utils;
 
 import java.util.Map;
+
+import burp.BurpExtender;
 import org.yaml.snakeyaml.DumperOptions;
 import org.yaml.snakeyaml.Yaml;
 import java.io.FileInputStream;