mirror of
https://github.com/sbrl/terrain50-cli.git
synced 2024-11-22 06:53:01 +00:00
Implement initial stream-slice subcommand
This commit is contained in:
parent
dfa4288abd
commit
dcc37dbe59
3 changed files with 88 additions and 0 deletions
15
src/Helpers/MathsHelpers.mjs
Normal file
15
src/Helpers/MathsHelpers.mjs
Normal file
|
@ -0,0 +1,15 @@
|
|||
"use strict";
|
||||
|
||||
/**
|
||||
* Convenience function for calculating the percentage betwween 2 given values.
|
||||
* @param {Number} count_so_far The fractional part (e.g. items done so far)
|
||||
* @param {Number} total The total part (e.g. the total number of items to do)
|
||||
* @param {Number} [range=100] The range to transform to (default: 100)
|
||||
* @return {Number} The calculated percentage.
|
||||
*/
|
||||
function percentage(count_so_far, total, range=100) {
|
||||
if(count_so_far == 0) return 0;
|
||||
return (count_so_far/total)*range;
|
||||
}
|
||||
|
||||
export { percentage };
|
48
src/Subcommands/stream-slice/index.mjs
Normal file
48
src/Subcommands/stream-slice/index.mjs
Normal file
|
@ -0,0 +1,48 @@
|
|||
"use strict";
|
||||
|
||||
import fs from 'fs';
|
||||
|
||||
import Terrain50 from 'terrain50';
|
||||
|
||||
import a from '../../Helpers/Ansi.mjs';
|
||||
import l from '../../Helpers/Log.mjs';
|
||||
|
||||
import { percentage } from '../../Helpers/MathsHelpers.mjs';
|
||||
|
||||
export default async function(settings) {
|
||||
// 1: Parse settings
|
||||
let stream_in = process.stdin;
|
||||
if(settings.cli.input !== "-") {
|
||||
l.log(`Reading from ${a.hicol}${settings.cli.input}${a.reset}`);
|
||||
stream_in = fs.createReadStream(settings.cli.input);
|
||||
}
|
||||
else
|
||||
l.log(`Reading from stdin`);
|
||||
|
||||
let stream_out = process.stdout;
|
||||
if(settings.cli.output !== "-") {
|
||||
l.log(`Writing to ${a.hicol}${settings.cli.output}${a.reset}`);
|
||||
stream_out = fs.createWriteStream(settings.cli.output);
|
||||
}
|
||||
else
|
||||
l.log(`Writing to stdout`);
|
||||
|
||||
let count = settings.cli.count,
|
||||
offset = settings.cli.offset;
|
||||
|
||||
// ------------------------------------------------------------------------
|
||||
|
||||
let i = -1;
|
||||
for await(let next of Terrain50.ParseStream(stream_in)) {
|
||||
i++;
|
||||
|
||||
if(i < offset) continue;
|
||||
|
||||
let is_last = i - offset >= count;
|
||||
await next.serialise(stream_out, is_last);
|
||||
|
||||
let percentage = percentage(i - offset, count);
|
||||
process.stderr.write(`Written ${i - offset} / count objects (~${count.toFixed(2)}%) \r`);
|
||||
}
|
||||
l.log(`Slicing complete`);
|
||||
}
|
25
src/Subcommands/stream-slice/meta.toml
Normal file
25
src/Subcommands/stream-slice/meta.toml
Normal file
|
@ -0,0 +1,25 @@
|
|||
description = "Extract a slice from a larger stream of objects"
|
||||
|
||||
[[arguments]]
|
||||
name = "input"
|
||||
description = "The input file to extract a slice from (default: stdin)"
|
||||
default_value = "-"
|
||||
type = "string"
|
||||
|
||||
[[arguments]]
|
||||
name = "output"
|
||||
description = "The output file to write to (default: stdout)"
|
||||
default_value = "-"
|
||||
type = "string"
|
||||
|
||||
[[arguments]]
|
||||
name = "count"
|
||||
description = "The number of objects to extract (default: 1)"
|
||||
default_value = 1
|
||||
type = "integer"
|
||||
|
||||
[[arguments]]
|
||||
name = "offset"
|
||||
description = "The number of objects to skip before beginning extraction (default: 0)"
|
||||
default_value = 0
|
||||
type = "integer"
|
Loading…
Reference in a new issue