pub fn compress(cmd: &str, output: &str) -> Option<String> {
let trimmed = output.trim();
if trimmed.is_empty() {
return Some("ok".to_string());
}
if cmd.contains("generate") {
return Some(compress_generate(trimmed));
}
if cmd.contains("migrate") {
return Some(compress_migrate(trimmed));
}
if cmd.contains("db push") || cmd.contains("db pull") {
return Some(compress_db_sync(trimmed));
}
if cmd.contains("studio") {
return Some("Prisma Studio started".to_string());
}
if cmd.contains("format") {
return Some(compress_format(trimmed));
}
if cmd.contains("validate") {
return Some(compress_validate(trimmed));
}
Some(compact_lines(trimmed, 10))
}
fn compress_generate(output: &str) -> String {
let mut generated = Vec::new();
for line in output.lines() {
let trimmed = line.trim();
let plain = strip_ansi(trimmed);
if plain.contains("Generated") || plain.contains("generated") {
generated.push(plain);
}
}
if generated.is_empty() {
return strip_noise(output);
}
generated.join("\n")
}
fn compress_migrate(output: &str) -> String {
let mut results = Vec::new();
let mut migration_name = String::new();
for line in output.lines() {
let plain = strip_ansi(line.trim());
if plain.contains("migration") && plain.contains("created") {
migration_name = plain.clone();
}
if plain.contains("applied")
|| plain.contains("Already in sync")
|| plain.contains("Database is up to date")
{
results.push(plain);
}
}
if results.is_empty() && migration_name.is_empty() {
return strip_noise(output);
}
let mut parts = Vec::new();
if !migration_name.is_empty() {
parts.push(migration_name);
}
parts.extend(results);
parts.join("\n")
}
fn compress_db_sync(output: &str) -> String {
let lines: Vec<String> = output
.lines()
.map(|l| strip_ansi(l.trim()))
.filter(|l| !l.is_empty() && !l.contains("warn") && !l.starts_with("Prisma schema"))
.collect();
if lines.is_empty() {
return "ok (synced)".to_string();
}
lines.join("\n")
}
fn compress_format(output: &str) -> String {
if output.contains("already formatted") || output.contains("unchanged") {
return "ok (already formatted)".to_string();
}
strip_noise(output)
}
fn compress_validate(output: &str) -> String {
let plain = strip_ansi(output.trim());
if plain.contains("valid") && !plain.contains("invalid") {
return "ok (schema valid)".to_string();
}
compact_lines(&plain, 10)
}
fn strip_noise(output: &str) -> String {
output
.lines()
.map(|l| strip_ansi(l.trim()))
.filter(|l| !l.is_empty() && !l.contains("████") && !l.contains("▀") && !l.contains("━"))
.collect::<Vec<_>>()
.join("\n")
}
fn strip_ansi(s: &str) -> String {
let mut result = String::with_capacity(s.len());
let mut in_escape = false;
for c in s.chars() {
if c == '\x1b' {
in_escape = true;
} else if in_escape {
if c.is_ascii_alphabetic() {
in_escape = false;
}
} else {
result.push(c);
}
}
result
}
fn compact_lines(text: &str, max: usize) -> String {
let lines: Vec<&str> = text.lines().filter(|l| !l.trim().is_empty()).collect();
if lines.len() <= max {
return lines.join("\n");
}
format!(
"{}\n... ({} more lines)",
lines[..max].join("\n"),
lines.len() - max
)
}