dan/nekos
1
0
Fork 0
mirror of https://github.com/danbulant/nekos synced 2026-07-06 11:30:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: gif support

Browse source
This commit is contained in:
Daniel Bulant 2021-10-03 19:51:56 +02:00
parent 0b677ffb61
commit 813487d3a2
1 changed files with 45 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
nekos.js
View file

@ -1,4 +1,6 @@
#!/usr/bin/env zx #!/usr/bin/env zx
/// <reference path="node_modules/zx/globals.d.ts" />
const { performance } = require("perf_hooks");
const tty = require("tty"); const tty = require("tty");
$.verbose = false; $.verbose = false;
@ -26,7 +28,9 @@ Usage:
Options: Options:
--help Show this screen --help Show this screen
--loop [interval] Loops the image to be repeated in select interval`); --loop [interval] Loops the image to be repeated in select interval (randomizes the image each time), in ms. Defaults to 2000ms.
--gif renders all GIF frames, according to their delay(s). Looped by default. When enabled, --loop is ignored. Option ignored if url isn't GIF.
--url Shows current image's URL before the image`);
process.exit(0); process.exit(0);
} }
@ -35,6 +39,9 @@ const istty = tty.isatty(process.stdout.fd);
const heightOut = await $`stty size | awk '{print $1}'`; const heightOut = await $`stty size | awk '{print $1}'`;
const height = parseInt(heightOut.stdout); const height = parseInt(heightOut.stdout);
var sigPreventable = false;
var sigint = false;
async function showImage(type) { async function showImage(type) {
const imgurl = `https://nekos.life/api/v2/img/` + type; const imgurl = `https://nekos.life/api/v2/img/` + type;
const urlOut = await $`curl -fsSL ${imgurl} | jq -r ".url"`; const urlOut = await $`curl -fsSL ${imgurl} | jq -r ".url"`;
@ -43,19 +50,55 @@ async function showImage(type) {
console.error("Type not found"); console.error("Type not found");
process.exit(1); process.exit(1);
} }
const args = [ const args = [
istty && "--fill", istty && "--fill",
argv.colors !== false && "--colors" argv.colors !== false && "--colors"
].filter(t => t); ].filter(t => t);
await $`curl -fsSL ${url} | jp2a --height=${height} ${args} -`.pipe(process.stdout);
if(argv.url) console.log(url);
if(!argv.gif || !url.endsWith(".gif")) {
await $`curl -fsSL ${url} | convert - jpeg:- | jp2a --height=${height-1} ${args} -`.pipe(process.stdout);
} else {
const tempdir = await fs.mkdtemp(path.join(os.tmpdir(), "nekos-"));
await $`curl -fsSL ${url} -o ${tempdir + "/src.gif"}`;
await $`convert -coalesce ${tempdir + "/src.gif"} ${tempdir + "/img.jpg"}`;
const frameLens = (await $`magick identify -format "%T\n" ${tempdir + "/src.gif"}`).stdout.split("\n").map(t => 1 / parseInt(t));
sigPreventable = true;
var frame = 0;
while(true) {
if(frame >= frameLens.length - 2) frame = 0;
if(sigint) break;
const start = performance.now();
try {
await $`jp2a ${args} --height=${height-1} ${tempdir + "/img-" + frame + ".jpg"}`.pipe(process.stdout);
} catch(e) {
break;
}
const end = performance.now();
console.log(`${frame}/${frameLens.length} ${frameLens[frame]}ms ${end-start}ms`);
await sleep(frameLens[frame] - (end - start));
frame++;
}
await fs.unlink(tempdir);
}
} }
process.on("SIGINT", (s) => {
if(!sigPreventable) process.exit(1);
sigint = true;
});
await showImage(type); await showImage(type);
if(argv.loop) { if(argv.loop) {
const offset = parseInt(argv.loop); const offset = parseInt(argv.loop);
const delay = isNaN(offset) ? 2000 : offset; const delay = isNaN(offset) ? 2000 : offset;
sigPreventable = true;
while(true) { while(true) {
if(sigint) break;
await sleep(delay); await sleep(delay);
if(sigint) break;
await showImage(type); await showImage(type);
} }
} }