import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.Statement;

/** Apply lobster_engine_prompt_init.sql (INSERT IGNORE). */
public class ApplyLobsterPromptInit {

    private static final String HOST = "cq-cdb-8fjmemkb.sql.tencentcdb.com:27220";
    private static final String USER = "root";
    private static final String PASS = "Ylrz_1q2w3e4r5t6y";

    public static void main(String[] args) throws Exception {
        Class.forName("com.mysql.cj.jdbc.Driver");
        String db = args.length > 0 ? args[0] : "fs_tenant_cs1";
        String sql = Files.readString(Path.of("d:/ylrz_saas_new/java/sql/lobster_engine_prompt_init.sql"), StandardCharsets.UTF_8);
        if (sql.startsWith("")) sql = sql.substring(1);

        try (Connection c = DriverManager.getConnection(
                "jdbc:mysql://" + HOST + "/" + db + "?useSSL=false&serverTimezone=GMT%2B8&characterEncoding=utf8&allowMultiQueries=true",
                USER, PASS);
             Statement st = c.createStatement()) {
            st.execute(sql);
            try (ResultSet rs = st.executeQuery(
                    "SELECT COUNT(*) FROM lobster_system_prompt WHERE company_id IS NULL")) {
                rs.next();
                System.out.println("platform prompts in " + db + ": " + rs.getInt(1));
            }
            try (ResultSet rs = st.executeQuery(
                    "SELECT prompt_key FROM lobster_system_prompt WHERE company_id IS NULL AND prompt_key IN ('proactive_silent_greeting','tenant_product_claim_semantic','universal_deep_dialogue_render')")) {
                while (rs.next()) System.out.println("  ok: " + rs.getString(1));
            }
        }
        System.out.println("Done.");
    }
}