nikkuss/nix-output
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

work on bar display and stuff

Browse Source
This commit is contained in:
Nikkuss
2025-11-14 21:21:14 +00:00
parent b9c63c08b6
commit e958b461f6
3 changed files with 383 additions and 171 deletions
Download Patch File Download Diff File Expand all files Collapse all files

191
src/handlers/download.rs
View File

@@ -1,10 +1,11 @@
use std::{sync::LazyLock, time::Instant};
use std::{borrow::Cow, sync::LazyLock, time::Instant};
use indicatif::{HumanBytes, ProgressBar};
use owo_colors::OwoColorize;
use owo_colors::{OwoColorize, Style};
use crate::{
action::{Action, BuildStepId, ResultFields, StartFields},
download_display::{DOWNLOAD_STYLE, EXTRACT_STYLE, StyledString, style_bar},
estimator::Estimator,
handlers::{Handler, fetch::FetchHandler},
multibar::{BarSegment, MultiBar},
@@ -57,11 +58,9 @@ impl Handler for CopyHandler {
parent,
..
} if parent == &self.0 => {
let bar = state.add_pb(ProgressBar::new(1));
state.plug(DownloadHandler::new(
state.term_width,
*id,
bar,
path.to_string(),
));
// Handle global start action
@@ -77,7 +76,7 @@ pub struct DownloadHandler {
pub copy_id: BuildStepId,
pub download_estimator: Estimator,
pub extract_estimator: Estimator,
pub bar: ProgressBar,
pub bar: Option<ProgressBar>,
pub path: String,
pub download_expected: u64,
pub download_done: u64,
@@ -87,14 +86,13 @@ pub struct DownloadHandler {
}
impl DownloadHandler {
pub fn new(width: u16, copy_id: BuildStepId, bar: ProgressBar, path: String) -> Self {
bar.set_style(indicatif::ProgressStyle::with_template("{msg}").unwrap());
pub fn new(width: u16, copy_id: BuildStepId, path: String) -> Self {
let new = Self {
id: u64::MAX.into(),
copy_id,
download_estimator: Estimator::new(Instant::now()),
extract_estimator: Estimator::new(Instant::now()),
bar,
bar: None,
path,
download_expected: 0,
download_done: 0,
@@ -106,6 +104,9 @@ impl DownloadHandler {
new
}
pub fn draw_bar(&self, width: u16) {
let Some(bar) = &self.bar else {
return;
};
let name = nix_path::extract_package_name_string(&self.path)
.unwrap_or_else(|| "unknown".to_string());
let name = pad_string(&name, (width / 4) as usize);
@@ -113,55 +114,82 @@ impl DownloadHandler {
let status_len = (width / 4) as usize + 10;
if self.download_expected == 0 || self.extract_expected == 0 {
self.bar.set_message(format!("{status}"));
self.bar.tick();
bar.set_message(format!("{status}"));
bar.tick();
return;
}
let total_expected = self.download_expected + self.extract_expected;
let total_done = self.download_done + self.extract_done;
let work_per_sec = if self.download_done < self.download_expected {
StyledString::new(
Cow::Owned(format!(
"{}",
HumanBytes(self.download_estimator.steps_per_second(Instant::now()) as u64)
)),
DOWNLOAD_STYLE,
)
} else {
StyledString::new(
Cow::Owned(format!(
"{}",
HumanBytes(self.extract_estimator.steps_per_second(Instant::now()) as u64)
)),
EXTRACT_STYLE,
)
};
let dl_percent =
((self.download_done as f64 / self.download_expected as f64) * 100.0) as u64;
let ex_percent = ((self.extract_done as f64 / self.extract_expected as f64) * 100.0) as u64;
let download_per_sec =
HumanBytes(self.download_estimator.steps_per_second(Instant::now()) as u64).to_string();
let extract_per_sec =
HumanBytes(self.extract_estimator.steps_per_second(Instant::now()) as u64).to_string();
let download_done_human = HumanBytes(self.download_done).to_string();
let download_done_human = pad_string(&download_done_human, INFO_WIDTH)
.blue()
.bold()
.to_string();
let extract_done_human = HumanBytes(self.extract_done).to_string();
let extract_done_human = pad_string(&extract_done_human, INFO_WIDTH)
.green()
.bold()
.to_string();
let download_per_sec = pad_string(&format!("{download_per_sec}/s"), INFO_WIDTH)
.blue()
.bold()
.to_string();
let extract_per_sec = pad_string(&format!("{extract_per_sec}/s"), INFO_WIDTH)
.green()
.bold()
.to_string();
let work_per_sec = if self.download_done < self.download_expected {
download_per_sec
} else {
extract_per_sec
};
let display = format!("{work_per_sec} | {download_done_human} | {extract_done_human} ");
let display_length = (INFO_WIDTH * 3) + 9;
// + 6 to account for final format
let total_length = status_len + display_length + 4;
let min = dl_percent.min(ex_percent);
let dl = dl_percent.saturating_sub(min);
let len = total_length as u16;
let mbar = MultiBar([
BarSegment::Dynamic(&DONE_CHAR, min),
BarSegment::Dynamic(&DOWNLOAD_CHAR, dl),
BarSegment::Dynamic(" ", 100 - min - dl),
]);
let msg = style_bar(
StyledString::new(Cow::Owned("Downloading".to_string()), Style::new()),
StyledString::new(Cow::Owned(name), Style::new().purple()),
work_per_sec,
self.download_done as usize,
self.extract_done as usize,
mbar,
width,
);
// let total_expected = self.download_expected + self.extract_expected;
// let total_done = self.download_done + self.extract_done;
//
//
// let download_per_sec =
// HumanBytes(self.download_estimator.steps_per_second(Instant::now()) as u64).to_string();
// let extract_per_sec =
// HumanBytes(self.extract_estimator.steps_per_second(Instant::now()) as u64).to_string();
// let download_done_human = HumanBytes(self.download_done).to_string();
// let download_done_human = pad_string(&download_done_human, INFO_WIDTH)
// .blue()
// .bold()
// .to_string();
// let extract_done_human = HumanBytes(self.extract_done).to_string();
// let extract_done_human = pad_string(&extract_done_human, INFO_WIDTH)
// .green()
// .bold()
// .to_string();
// let download_per_sec = pad_string(&format!("{download_per_sec}/s"), INFO_WIDTH)
// .blue()
// .bold()
// .to_string();
// let extract_per_sec = pad_string(&format!("{extract_per_sec}/s"), INFO_WIDTH)
// .green()
// .bold()
// .to_string();
//
// let display = format!("{work_per_sec} | {download_done_human} | {extract_done_human} ");
// let display_length = (INFO_WIDTH * 3) + 9;
//
// // + 6 to account for final format
// let total_length = status_len + display_length + 4;
//
// if width <= len {
// self.bar.set_message(format!(
// "{}: {}/{}",
@@ -176,37 +204,32 @@ impl DownloadHandler {
// self.bar.tick();
// return;
// }
let msg = match width {
0..60 => {
format!(
"{}: {}/{}",
name,
total_done,
if total_expected == 0 {
"-".to_string()
} else {
total_expected.to_string()
}
)
}
_ => {
let bar = MultiBar([
BarSegment::Dynamic(&DONE_CHAR, min),
BarSegment::Dynamic(&DOWNLOAD_CHAR, dl),
BarSegment::Dynamic(" ", 100 - min - dl),
])
.scale((width.saturating_sub(total_length as u16)) as u64);
let bar = if width > display_length as u16 + status_len as u16 {
format!("[{}]", bar)
} else {
String::new()
};
format!("{status} {display} {bar}",)
}
};
self.bar.set_message(msg);
self.bar.tick();
// let msg = match width {
// 0..60 => {
// format!(
// "{}: {}/{}",
// name,
// total_done,
// if total_expected == 0 {
// "-".to_string()
// } else {
// total_expected.to_string()
// }
// )
// }
// _ => {
// .scale((width.saturating_sub(total_length as u16)) as u64);
// let bar = if width > display_length as u16 + status_len as u16 {
// format!("[{}]", bar)
// } else {
// String::new()
// };
//
// format!("{status} {display} {bar}",)
// }
// };
bar.set_message(msg);
bar.tick();
}
}
@@ -248,11 +271,19 @@ impl Handler for DownloadHandler {
self.extract_estimator.record(*done, Instant::now());
self.draw_bar(state.term_width);
}
if self.download_expected > 0 && self.extract_expected > 0 && self.bar.is_none() {
let bar = state.add_pb(ProgressBar::new(1));
bar.set_style(indicatif::ProgressStyle::with_template("{msg}").unwrap());
self.bar = Some(bar);
self.draw_bar(state.term_width);
}
Ok(true)
}
Action::Stop { id } if id == &self.copy_id => {
self.bar.finish_and_clear();
if let Some(bar) = &self.bar {
bar.finish_and_clear();
}
Ok(false)
}
_ => Ok(true),