rustbasic_cli/
database.rs

1use std::fs;
2use rustbasic_core::Config;
3use rustbasic_core::database::connect;
4use base64::{Engine as _, engine::general_purpose};
5use rand::Rng;
6use regex::Regex;
7use colored::*;
8use sea_orm::ConnectionTrait;
9
10pub async fn clear_cache() {
11    println!("\n{}", "🧹 Cleaning Cache & Logs...".magenta().bold());
12
13    // 1. Clear Logs
14    let log_dir = "storage/logs";
15    if let Ok(entries) = fs::read_dir(log_dir) {
16        let mut count = 0;
17        for entry in entries.flatten() {
18            let path = entry.path();
19            if path.is_file() {
20                let _ = fs::OpenOptions::new()
21                    .write(true)
22                    .truncate(true)
23                    .open(&path);
24                count += 1;
25            }
26        }
27        println!("   {} Folder storage/logs telah dikosongkan. ({} file dibersihkan)", "✅ Logs:".green(), count);
28    } else {
29        println!("   {} Folder storage/logs tidak ditemukan.", "⚠️  Logs:".yellow());
30    }
31
32    // 2. Clear Sessions in DB
33    let cfg = Config::load();
34    let db = connect(&cfg).await;
35    
36    let truncate_sql = if cfg.db_connection == "mysql" {
37        "TRUNCATE TABLE sessions"
38    } else {
39        "DELETE FROM sessions"
40    };
41
42    match db.execute(sea_orm::Statement::from_string(cfg.db_backend(), truncate_sql.to_string())).await {
43        Ok(_) => println!("   {} Tabel sessions telah dikosongkan.", "✅ Sessions:".green()),
44        Err(e) => println!("   {} Gagal membersihkan tabel sessions. ({})", "❌ Error:".red(), e),
45    }
46
47    println!("\n{}", "✨ Cache berhasil dibersihkan!".green().bold());
48}
49
50pub fn generate_app_key() {
51    println!("\n{}", "🔑 Generating Application Key...".magenta().bold());
52
53    let mut key = [0u8; 32];
54    rand::rng().fill_bytes(&mut key);
55    
56    let encoded = general_purpose::STANDARD.encode(key);
57    let key_str = format!("base64:{}", encoded);
58    
59    let env_path = ".env";
60    match fs::read_to_string(env_path) {
61        Ok(content) => {
62            let re = Regex::new(r"(?m)^APP_KEY=.*").unwrap();
63            let new_content = if re.is_match(&content) {
64                re.replace(&content, &format!("APP_KEY={}", key_str)).to_string()
65            } else {
66                format!("{}\nAPP_KEY={}", content.trim_end(), key_str)
67            };
68
69            if let Err(e) = fs::write(env_path, new_content) {
70                println!("{} Gagal menulis ke file .env: {}", "❌ Error:".red(), e);
71            } else {
72                println!("{} {}", "✅ Application key set successfully:".green(), key_str.cyan());
73                println!("{}", "💡 Pastikan untuk tidak membagikan APP_KEY ini ke publik!".dimmed());
74            }
75        }
76        Err(_) => {
77            println!("{} File .env tidak ditemukan.", "❌ Error:".red());
78        }
79    }
80}
rustbasic_cli/database.rs

rustbasic_cli/
database.rs
