Старт

2025-12-04 12:20:47 +03:00
commit ff9301eddb
79 changed files with 4655 additions and 0 deletions
@@ -0,0 +1,126 @@
+package shine.db;
+
+
+import utils.config.AppConfig;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.nio.file.*;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+import java.sql.Statement;
+
+public class DatabaseInitializer {
+
+    public static void createNewDB(String[] args) {
+        AppConfig config = AppConfig.getInstance();
+        String dbPath = config.getParam("db.path");
+
+        if (dbPath == null || dbPath.isBlank()) {
+            System.err.println("Параметр db.path не задан в application.properties");
+            return;
+        }
+
+        Path dbFile = Paths.get(dbPath);
+        try {
+            // создаём директорию, если нужно
+            Path parent = dbFile.getParent();
+            if (parent != null && !Files.exists(parent)) {
+                Files.createDirectories(parent);
+            }
+
+            if (Files.exists(dbFile)) {
+                System.out.println("Файл базы данных уже существует: " + dbFile.toAbsolutePath());
+                System.out.print("Пересоздать БД (СТАРАЯ БУДЕТ УДАЛЕНА)? [y/N]: ");
+
+                BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
+                String answer = reader.readLine();
+                if (!"y".equalsIgnoreCase(answer) && !"yes".equalsIgnoreCase(answer)) {
+                    System.out.println("Операция отменена. БД не изменена.");
+                    return;
+                }
+
+                Files.delete(dbFile);
+                System.out.println("Старый файл БД удалён.");
+            }
+
+            createSchema("jdbc:sqlite:" + dbPath);
+            System.out.println("Новая БД успешно создана по пути: " + dbFile.toAbsolutePath());
+
+        } catch (IOException e) {
+            System.err.println("Ошибка работы с файлом БД: " + e.getMessage());
+        } catch (SQLException e) {
+            System.err.println("Ошибка создания схемы БД: " + e.getMessage());
+        }
+    }
+
+    private static void createSchema(String jdbcUrl) throws SQLException {
+        try {
+            Class.forName("org.sqlite.JDBC");
+        } catch (ClassNotFoundException e) {
+            throw new RuntimeException("SQLite JDBC driver not found", e);
+        }
+
+        try (Connection conn = DriverManager.getConnection(jdbcUrl);
+             Statement st = conn.createStatement()) {
+
+            // включаем внешние ключи на этом соединении (для инициализации тоже)
+            st.execute("PRAGMA foreign_keys = ON");
+
+            // 1. Таблица solana_users
+            st.executeUpdate("""
+                CREATE TABLE IF NOT EXISTS solana_users (
+                    login       TEXT    NOT NULL,
+                    loginId     INTEGER NOT NULL PRIMARY KEY,
+                    bchId       INTEGER NOT NULL,
+                    pubkey0     TEXT,
+                    pubkey1     TEXT,
+                    bchLimit    INTEGER         -- может быть NULL
+                );
+                """);
+
+            st.executeUpdate("""
+                CREATE INDEX IF NOT EXISTS idx_solana_users_login
+                ON solana_users (login);
+                """);
+
+            // 2. Таблица active_sessions
+            st.executeUpdate("""
+                CREATE TABLE IF NOT EXISTS active_sessions (
+                    sessionId        INTEGER NOT NULL PRIMARY KEY,
+                    session_pwd      TEXT    NOT NULL,
+                    loginId          INTEGER NOT NULL,
+                    time_ms          INTEGER NOT NULL,
+                    pubkey_num       INTEGER NOT NULL,
+                    push_endpoint    TEXT,
+                    push_p256dh_key  TEXT,
+                    push_auth_key    TEXT,
+                    FOREIGN KEY (loginId) REFERENCES solana_users(loginId)
+                );
+                """);
+
+            // 3. Таблица users_params
+            // Важно: пара (loginId, param) должна быть уникальна
+            st.executeUpdate("""
+                CREATE TABLE IF NOT EXISTS users_params (
+                    loginId        INTEGER NOT NULL,
+                    param          TEXT    NOT NULL,
+                    bch_channel_id INTEGER NOT NULL DEFAULT 0,
+                    value          TEXT,
+                    time_ms        INTEGER NOT NULL,
+                    pubkey_num     INTEGER NOT NULL,
+                    signature      TEXT,
+                    FOREIGN KEY (loginId) REFERENCES solana_users(loginId),
+                    UNIQUE (loginId, param)
+                );
+                """);
+
+            st.executeUpdate("""
+                CREATE INDEX IF NOT EXISTS idx_users_params_loginId
+                ON users_params (loginId);
+                """);
+        }
+    }
+}