Tomas/lunch-games
1
0
Fork 0
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

Added pretty download UI

Browse Source 
Closes #1
This commit is contained in:
Tomas
2025-11-27 03:11:44 +01:00
parent 229e0c8efd
commit 1e6f7f0ae6
5 changed files with 234 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

123
source/lunch/update.d
View File

@ -1,24 +1,27 @@
module lunch.update;
import std.datetime;
import std.stdio : writefln, File;
import std.stdio;
import std.traits;
import std.json;
import std.conv;
import std.file;
import std.file : dirEntries;
import std.path;
import std.file;
import std.math;
import std.string;
import std.digest.sha;
import std.array;
import std.format;
import std.algorithm;
import std.parallelism;
import std.datetime.stopwatch;
import lunch.conf;
import lunch.term;
import lunch.color;
import lunch.http;
import lunch.logger;
private struct RemoteFile
{
string file;
@ -87,6 +90,7 @@ private struct Action
private struct Actions
{
Action[] actions;
ulong total_download_size;
alias this = actions;
}
@ -100,7 +104,6 @@ private immutable(Local) _local;
private bool _actions_set = false;
private immutable(Actions) _actions;
private JSONValue[string] safe_object(JSONValue value) @trusted
{
if (value.type != JSONType.object)
@ -221,13 +224,11 @@ void wrapJSON(T)(ref T wrapper, JSONValue[string] json, bool ignore_missing = fa
}}
}
private string workdir(string path)
{
return chainPath(config.updater.workdir, path).array;
}
private void touch(string path, long mtime = Clock.currTime().toUnixTime)
{
infof("Touching %s", path);
@ -236,7 +237,6 @@ private void touch(string path, long mtime = Clock.currTime().toUnixTime)
setTimes(path, old_atime, SysTime.fromUnixTime(mtime));
}
private immutable(Remote) remote()
{
if (_remote_set)
@ -263,7 +263,6 @@ private immutable(Remote) remote()
return _remote;
}
private immutable(Local) local()
{
if (_local_set)
@ -302,7 +301,6 @@ private immutable(Local) local()
return _local;
}
private immutable(Actions) actions()
{
if (_actions_set)
@ -346,6 +344,7 @@ private immutable(Actions) actions()
if (!this_loc)
{
act ~= this_act;
act.total_download_size += this_rem.size;
continue;
}
@ -356,6 +355,7 @@ private immutable(Actions) actions()
if (this_loc.hash != this_rem.hash || this_loc.size != this_rem.size)
{
act ~= this_act;
act.total_download_size += this_rem.size;
}
else
{
@ -378,7 +378,6 @@ private immutable(Actions) actions()
return _actions;
}
public bool updateAvailable()
{
if (actions.length != 0)
@ -391,16 +390,77 @@ public void update()
{
info("Updating");
scope (exit) cast() _actions = [];
scope (exit) cast() _actions = Actions();
const dl_total = actions.actions.count!"a.what == b"(Action.download);
const dl_pad = cast(int)log10(cast(float)dl_total) + 1;
size_t dl_count = 0;
auto writeSW = StopWatch(AutoStart.yes);
auto speedSW = StopWatch(AutoStart.yes);
const dl_total = actions.actions.count!"a.what == b"(Action.download);
const dl_pad = cast(int)log10(cast(float)dl_total) + 1;
uint dl_count = 0;
ulong dl_size = 0;
const dl_spad = cast(int)log10(cast(float)actions.total_download_size) + 1;
ulong dl_speed = 0;
ulong dl_speed_old_size = 0;
const old_defaultpool = defaultPoolThreads;
scope (exit) defaultPoolThreads = old_defaultpool;
defaultPoolThreads = config.updater.parallel_downloads.to!uint - 1;
auto writeProgress = delegate()
{
string numbers = format(
"[%s%*d%s|%s%d%s]",
magenta, dl_spad, dl_size, reset,
magenta, actions.total_download_size, reset,
);
int barSpace = termWidth - dl_spad*2 - 3 - 2 - 10 - 11;
float progress = dl_size.to!float / actions.total_download_size;
long filledSpace = lround(progress * barSpace);
long emptySpace = lround((1.0 - progress) * barSpace);
string progress_bar = "["~yellow;
for (int n = 0; n < filledSpace; n++)
progress_bar ~= '=';
for (int n = 0; n < emptySpace; n++)
progress_bar ~= ' ';
progress_bar ~= reset~"]";
string percent = format(
"[%7.2f%%]", progress * 100
);
int speed_rank = 0;
float speed = dl_speed.to!float;
for (speed_rank = 0; speed >= 1000; speed_rank += 1)
speed /= 1000;
string unit = [
" B",
"KB",
"MB",
"GB",
"TB",
"PB",
"EB",
"ZB",
"YB",
"RB",
"QB",
][speed_rank];
string download_speed = (speed < 100)
? format(
"[%s%4.1f %s%s/s]", cyan, speed, reset, unit
)
: format(
"[%s%5.1f%s%s/s]", cyan, speed, reset, unit
);
writef("\r%s%s%s%s\b", numbers, download_speed, progress_bar, percent);
};
foreach (action; parallel(actions.actions, 1))
final switch(action.what)
{
@ -409,13 +469,36 @@ public void update()
remove(action.file);
break;
case Action.download:
info("Update %s", action.file);
mkdirRecurse(dirName(action.file));
synchronized
{
dl_count += 1;
writefln("[%*d|%*d] %s", dl_pad, dl_count, dl_pad, dl_total, action.file);
}
download(action.url, action.file);
download(action.url, action.file, (size_t amount){
synchronized
{
dl_size += amount;
if (speedSW.peek.total!"msecs" >= 1000)
{
long time = speedSW.peek.total!"msecs";
dl_speed = lround((dl_size - dl_speed_old_size) * time.to!float / 1000);
dl_speed_old_size = dl_size;
speedSW.reset();
}
if (writeSW.peek.total!"msecs" < 100)
return;
writeSW.reset();
}
writeProgress();
});
synchronized dl_count += 1;
writefln(
"\33[2K\r[%s%*d%s|%s%d%s] %s",
cyan, dl_pad, dl_count, reset,
cyan, dl_total, reset,
action.file.replace("/", dirSeparator).green,
);
writeProgress();
touch(action.file, action.mtime);
break;
}