allowed = Arrays.asList("csv", "json", "sql");
private static void PrintBanner() {
- System.out.println("===================================");
- System.out.println(" " + TITLE + " v" + VERSION);
- System.out.println(" Author: " + AUTHOR);
- System.out.println(" " + DESCRIPTION);
- System.out.println("===================================\n");
+ System.out.println();
+ System.out.println("╭─────────────────────────────────────────────────────────────────────────╮");
+ System.out.println("│ (๑˃̵ᴗ˂̵)و 欢迎来到灵儿的安全世界呀～♡ (づ｡◕‿‿◕｡)づ ");
+ System.out.println("│ ");
+ System.out.println("│ 工具名称：" + TITLE + " v" + VERSION);
+ System.out.println("│ 作者昵称：" + AUTHOR + " (๑•̀ㅂ•́)و✧");
+ System.out.println("│ 简介说明：" + DESCRIPTION);
+ System.out.println("│ ");
+ System.out.println("│ 📧 邮箱联系: " + EMAIL);
+ System.out.println("│ 💻 GitHub : " + GITHUB);
+ System.out.println("│ 📂 GitCode : " + GITCODE);
+ System.out.println("│ ");
+ System.out.println("│ 🛠️ 使用方法: " + USAGE);
+ System.out.println("│ ");
+ System.out.println("│ 💌 小声明哟～我是乖乖的好孩子，这工具是帮大家做好事的～ ");
+ System.out.println("│ 不是让坏蛋乱用的哦 (ง •̀_•́)ง ");
+ System.out.println("│ 如果有人坏坏地用了，后果自负哒～我不背锅哦～请保持正义感～♡ ");
+ System.out.println("╰─────────────────────────────────────────────────────────────────────────╯");
+ System.out.println();
}
public static void main(String[] args) {
try {
Main.PrintBanner();
- int pages = userUtil.askForPositiveInt("📄 想爬多少页咧～？（建议填个50以上更爽！）：");
+ /**
+ * 嘻嘻～这里是数据库模式设置哦 ♡
+ * 你可以从 {@link xin.ctkqiang.dto.Database Database} 枚举里选一个最爱滴数据库：
+ *
+ * - {@link xin.ctkqiang.dto.Database#MYSQL MYSQL} ：咱们平时用的 MySQL
+ * 数据库，默认就选它啦～
+ * - {@link xin.ctkqiang.dto.Database#SQLITE SQLITE} ：轻巧灵活的 SQLite
+ * 嵌入式数据库，萌萌哒！
+ *
+ *
+ * 小提示✨：默认模式是 MYSQL 哦，要用 SQLite 的话，记得把下面的代码改成
+ * Configuration.setDatabaseMode(Database.SQLITE); 才能生效呢～
+ *
+ * 举个栗子给你看👇：
+ *
+ * 乖乖用【默认】 MySQL 嘛 ~
+ *
+ *
+ * Configuration.setDatabaseMode(Database.MYSQL);
+ *
+ *
+ * 想用 SQLite ~
+ *
+ *
+ * Configuration.setDatabaseMode(Database.SQLITE);
+ *
+ *
+ * 嘻嘻，这样就能轻轻松松切换数据库, 你棒棒啦！🎀🌸💖
+ */
+ Configuration.setDatabaseMode(Database.MYSQL);
+
+ int pages = userUtil.askForPositiveInt("📄 想爬多少页咧～？【建议填个50以上更爽！】");
boolean isExport = userUtil.askForYesNo("💾 要不要顺便导出数据嘞？（支持 .csv/.json/.sql哦）✨：");
if (isExport) {
diff --git a/src/main/java/xin/ctkqiang/config/Configuration.java b/src/main/java/xin/ctkqiang/config/Configuration.java
index b87fd99..2c08799 100644
--- a/src/main/java/xin/ctkqiang/config/Configuration.java
+++ b/src/main/java/xin/ctkqiang/config/Configuration.java
@@ -1,13 +1,28 @@
package xin.ctkqiang.config;
+import xin.ctkqiang.dto.Database;
+
public class Configuration {
+ private static String DatabaseMode = Database.MYSQL.getValue();
+
public static final String DB_NAME = "ExploitDB";
public static final String DB_URL = "jdbc:mysql://localhost:3306/" + DB_NAME
+ "?serverTimezone=UTC&allowPublicKeyRetrieval=true&useSSL=false";
- public static final String DB_USER = "root"; // 改成你的数据库用户名
- public static final String DB_PASSWORD = ""; // 改成你的数据库密码
+ public static final String DB_SQLITE_URL = "jdbc:sqlite:ling.sqlite";
+
+ public static final String DB_USER = "root";
+ public static final String DB_PASSWORD = "";
public static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver";
public static final boolean DEBUG = false;
+
+ public static String getDatabaseMode() {
+ return DatabaseMode;
+ }
+
+ public static void setDatabaseMode(Database databaseMode) {
+ DatabaseMode = databaseMode.getValue();
+ System.out.println(String.format("🎀 你当前选择的数据库模式是:「%s」～酱酱 ♪(๑˃ᴗ˂)ﻭ \n", getDatabaseMode()));
+ }
}
diff --git a/src/main/java/xin/ctkqiang/controller/DatabaseController.java b/src/main/java/xin/ctkqiang/controller/DatabaseController.java
index d3a9bb5..dbbc4a3 100644
--- a/src/main/java/xin/ctkqiang/controller/DatabaseController.java
+++ b/src/main/java/xin/ctkqiang/controller/DatabaseController.java
@@ -18,6 +18,7 @@
import com.fasterxml.jackson.databind.ObjectMapper;
import xin.ctkqiang.config.Configuration;
+import xin.ctkqiang.dto.Database;
import xin.ctkqiang.dto.Exploit;
/**
@@ -32,7 +33,7 @@ public class DatabaseController {
/** 数据库名称 */
protected final static String DB_NAME = Configuration.DB_NAME;
- /** 数据库连接URL */
+ /** MYSQL 数据库连接URL */
protected final static String DB_URL = Configuration.DB_URL;
/** 数据库用户名 */
@@ -44,6 +45,9 @@ public class DatabaseController {
/** 数据库驱动类名 */
protected final static String DB_DRIVER = Configuration.DB_DRIVER;
+ /** SQLITE 数据库连接URL */
+ protected final static String DB_SQLITE_URL = Configuration.DB_SQLITE_URL;
+
/**
* 静态初始化块
* 在类加载时尝试加载数据库驱动
@@ -63,7 +67,35 @@ public class DatabaseController {
* @throws SQLException 如果连接数据库时发生错误
*/
protected Connection getConnection() throws SQLException {
- return DriverManager.getConnection(DB_URL, DB_USER, DB_PASSWORD);
+ String mode = Configuration.getDatabaseMode();
+ String upperMode = mode.toUpperCase();
+ System.out.println("💡 切换数据库模式 -> " + upperMode);
+
+ switch (upperMode) {
+ case "SQLITE":
+ try {
+ Class.forName("org.sqlite.JDBC");
+ System.out.println("📦 嘻嘻～SQLite 驱动已加载好惹！我已经抱好数据库小可爱啦～ฅ(＾・ω・＾ฅ)");
+ } catch (ClassNotFoundException e) {
+ System.err.println("❌ 哎呀呀！SQLite JDBC 驱动不见惹～人家找不到驱动怎么贴贴数据库嘛喵呜呜｡ﾟ(ﾟ´ω`ﾟ)ﾟ｡");
+ throw new SQLException("找不到 SQLite 驱动喵～要不要检查一下依赖有没有加对呀？");
+ }
+ return DriverManager.getConnection(DatabaseController.DB_SQLITE_URL);
+
+ case "MYSQL":
+ default:
+ try {
+ Class.forName(DatabaseController.DB_DRIVER);
+ System.out.println("📦 MySQL 驱动加载完成！我打扮好了，要去连接数据库小哥哥啦～(๑•̀ㅂ•́)و✧");
+ } catch (ClassNotFoundException e) {
+ System.err.println("❌ 呜呜呜 MySQL JDBC 驱动不见了～是不是打包的时候忘记带灵儿一起走啦 >///< ");
+ throw new SQLException("MySQL 驱动加载失败了喵～快检查一下 `.jar` 有没有遗漏吧！");
+ }
+ return DriverManager.getConnection(
+ DatabaseController.DB_URL,
+ DatabaseController.DB_USER,
+ DatabaseController.DB_PASSWORD);
+ }
}
/**
@@ -109,20 +141,52 @@ public void CreateTableIfNotExists() {
+ ");";
try {
- // 确保数据库驱动已加载
- Class.forName(DatabaseController.DB_DRIVER);
-
- // 创建数据库连接并执行建表语句
- try (Connection conn = DriverManager.getConnection(
- DatabaseController.DB_URL,
- DatabaseController.DB_USER,
- DatabaseController.DB_PASSWORD);
- Statement stmt = conn.createStatement()) {
- stmt.execute(query);
- System.out.println("💾 数据表【内容信息】初始化成功！");
+
+ String mode = Configuration.getDatabaseMode();
+ Database dbMode;
+
+ try {
+ dbMode = Database.fromValue(mode);
+ } catch (IllegalArgumentException e) {
+ dbMode = null;
+ }
+
+ if (dbMode == null) {
+ System.out.println("🥴 吃嗯的！这个数据库模式不认识喔：\"" + mode + "\" 妹妹我拒绝连接！");
+ return;
+ }
+
+ switch (dbMode) {
+ case SQLITE:
+ // 加载 SQLite JDBC 驱动
+ Class.forName("org.sqlite.JDBC");
+
+ // 构建连接（数据库文件叫 ling.sqlite）
+ try (Connection conn = DriverManager.getConnection(DatabaseController.DB_SQLITE_URL);
+ Statement stmt = conn.createStatement()) {
+ stmt.execute(query);
+ System.out.println("🎀 SQLite 模式下，表结构初始化好啦！已经准备好可爱的记录了喵～");
+ }
+
+ break;
+ case MYSQL:
+ default:
+ // 确保数据库驱动已加载
+ Class.forName(DatabaseController.DB_DRIVER);
+
+ // 创建数据库连接并执行建表语句
+ try (Connection conn = DriverManager.getConnection(
+ DatabaseController.DB_URL,
+ DatabaseController.DB_USER,
+ DatabaseController.DB_PASSWORD);
+ Statement stmt = conn.createStatement()) {
+ stmt.execute(query);
+ System.out.println("💾 MySQL 模式下，数据表【内容信息】初始化成功～♡");
+ }
}
+
} catch (Exception e) {
- System.err.println("❌ 数据表创建失败：" + e.getMessage());
+ System.err.println("❌ 表创建失败惹呜呜呜呜：妹抖崩坏 > <\n" + e.getMessage());
e.printStackTrace();
}
}
diff --git a/src/main/java/xin/ctkqiang/controller/UserUtil.java b/src/main/java/xin/ctkqiang/controller/UserUtil.java
index a8412b7..817212f 100644
--- a/src/main/java/xin/ctkqiang/controller/UserUtil.java
+++ b/src/main/java/xin/ctkqiang/controller/UserUtil.java
@@ -15,7 +15,7 @@ public int askForPositiveInt(String prompt) {
int result = DEFAULT;
while (true) {
- System.out.print(prompt + " (默认 " + DEFAULT + "): ");
+ System.out.print(prompt + " (默认 " + DEFAULT + "): 🌼 ");
String input = scanner.nextLine().trim();
if (input.isEmpty()) {
diff --git a/src/main/java/xin/ctkqiang/dto/Database.java b/src/main/java/xin/ctkqiang/dto/Database.java
new file mode 100644
index 0000000..dd3429f
--- /dev/null
+++ b/src/main/java/xin/ctkqiang/dto/Database.java
@@ -0,0 +1,25 @@
+package xin.ctkqiang.dto;
+
+public enum Database {
+ MYSQL("MySQL"),
+ SQLITE("SQLite");
+
+ private final String value;
+
+ Database(String value) {
+ this.value = value;
+ }
+
+ public String getValue() {
+ return value;
+ }
+
+ public static Database fromValue(String value) {
+ for (Database db : values()) {
+ if (db.value.equalsIgnoreCase(value)) {
+ return db;
+ }
+ }
+ throw new IllegalArgumentException("Unknown database type: " + value);
+ }
+}