138 lines
4.1 KiB
D
138 lines
4.1 KiB
D
import std.stdio, std.regex, std.json, std.file, std.datetime, std.conv, std.process, std.net.curl, std.string;
|
|
import qobuz.api;
|
|
|
|
int main(string[] args)
|
|
{
|
|
string VERSION = "1.0";
|
|
|
|
if (args.length != 2) {
|
|
writefln("Usage: %s <album id or url>", args[0]);
|
|
return -1;
|
|
}
|
|
|
|
auto path = thisExePath();
|
|
path = path.replaceFirst(regex("qobuz-get$"), "magic.json"); // HACK
|
|
string json;
|
|
try {
|
|
json = readText(path);
|
|
} catch (Exception e) {
|
|
writeln("Could not open magic.json!");
|
|
}
|
|
auto magic = parseJSON(json);
|
|
|
|
// strip url part if we have it
|
|
string id;
|
|
auto urlPart = regex("^https?://play.qobuz.com/album/");
|
|
if (args[1].matchFirst(urlPart)) {
|
|
id = args[1].replaceFirst(urlPart, "");
|
|
} else {
|
|
id = args[1];
|
|
}
|
|
|
|
writeln("Looking up album...");
|
|
auto album = getAlbum(magic, id);
|
|
|
|
string title, artist, genre, year;
|
|
JSONValue[] tracks;
|
|
|
|
try {
|
|
title = album["title"].str;
|
|
artist = album["artist"]["name"].str;
|
|
genre = album["genres_list"][0].str;
|
|
auto releaseTime = SysTime.fromUnixTime(album["released_at"].integer, UTC());
|
|
year = releaseTime.year.text;
|
|
|
|
writefln("[ %s - %s (%s, %s) ]", artist, title, genre, year);
|
|
|
|
tracks = album["tracks"]["items"].array();
|
|
} catch (Exception e) {
|
|
writeln("Could not parse album data!");
|
|
return -4;
|
|
}
|
|
|
|
string dirName = artist~" - "~title~" ("~year~") [WEB FLAC]";
|
|
try {
|
|
mkdir(dirName);
|
|
} catch (Exception e) {
|
|
writeln("Could not create directory: `"~dirName~"`. Does it exist already?");
|
|
return -9;
|
|
}
|
|
|
|
foreach (i, track; tracks) {
|
|
auto num = (i+1).text;
|
|
string url, trackName;
|
|
try {
|
|
trackName = track["title"].str;
|
|
if (num.length < 2)
|
|
num = "0"~num;
|
|
writef(" [%s] %s... ", num, trackName);
|
|
stdout.flush;
|
|
url = getDownloadUrl(magic, track["id"].integer.text);
|
|
} catch (Exception e) {
|
|
writeln("Failed to parse track data!");
|
|
return -7;
|
|
}
|
|
|
|
try {
|
|
auto pipes = pipeProcess([magic["ffmpeg"].str, "-i", "-", "-metadata", "title="~trackName, "-metadata", "artist="~artist,
|
|
"-metadata", "album="~title, "-metadata", "year="~year, "-metadata", "track="~num, "-metadata", "genre="~genre,
|
|
"-metadata", "comment=qobuz-get "~VERSION, dirName~"/"~num~" "~trackName~".flac"], Redirect.stdin | Redirect.stderr | Redirect.stdout);
|
|
foreach (chunk; byChunkAsync(url, 1024)) {
|
|
pipes.stdin.rawWrite(chunk);
|
|
pipes.stdin.flush;
|
|
}
|
|
pipes.stdin.close;
|
|
wait(pipes.pid);
|
|
} catch (Exception e) {
|
|
writeln("Failed to download track! Check that ffmpeg is properly configured.");
|
|
return -8;
|
|
}
|
|
writeln("Done!");
|
|
}
|
|
|
|
// Get album art
|
|
write("Getting album art... ");
|
|
stdout.flush;
|
|
download(id.getArtUrl, dirName~"/cover.jpg");
|
|
writeln("Done!");
|
|
|
|
string choice;
|
|
while (choice != "n" && choice != "y") {
|
|
write("Generate spectrals? [y/n] ");
|
|
stdout.flush;
|
|
choice = readln().chomp;
|
|
}
|
|
if (choice == "y") {
|
|
auto full = execute([magic["sox"].str, dirName~"/01 "~tracks[0]["title"].str~".flac", "-n", "remix", "1", "spectrogram",
|
|
"-x", "3000", "-y", "513", "-z", "120", "-w", "Kaiser", "-o", "SpecFull.png"]);
|
|
auto zoom = execute([magic["sox"].str, dirName~"/01 "~tracks[0]["title"].str~".flac", "-n", "remix", "1", "spectrogram",
|
|
"-X", "500", "-y", "1025", "-z", "120", "-w", "Kaiser", "-S", "0:30", "-d", "0:04", "-o", "SpecZoom.png"]);
|
|
if (full.status != 0 || zoom.status != 0)
|
|
writeln("Generating spectrals failed! Is sox configured properly?");
|
|
else
|
|
writeln("SpecFull.png and SpecZoom.png written.");
|
|
}
|
|
|
|
choice = null;
|
|
while (choice != "n" && choice != "y") {
|
|
write("Create .torrent file? [y/n] ");
|
|
stdout.flush;
|
|
choice = readln().chomp;
|
|
}
|
|
if (choice == "y") {
|
|
write("Announce URL: ");
|
|
stdout.flush;
|
|
string announce = readln().chomp;
|
|
|
|
auto t = execute([magic["mktorrent"].str, "-l", "20", "-a", announce, dirName]);
|
|
if (t.status != 0)
|
|
writeln("Creating .torrent file failed! Is mktorrent configured properly?");
|
|
else
|
|
writeln("'"~dirName~".torrent' created.");
|
|
}
|
|
|
|
writeln("All done, exiting.");
|
|
|
|
return 0;
|
|
}
|