Compare commits

..

No commits in common. "280cba2f9a07e42c57c165e2ce206fadb0218503" and "6aac0bff2ba12a5e4c4c7b1ab6698e064f8157b9" have entirely different histories.

3 changed files with 39 additions and 65 deletions

View file

@ -15,17 +15,7 @@ pub fn index(request: *jetzig.Request, data: *jetzig.Data) !jetzig.View {
pub fn post(request: *jetzig.Request) !jetzig.View { pub fn post(request: *jetzig.Request) !jetzig.View {
var root = try request.data(.object); var root = try request.data(.object);
//const params = try request.params(); const params = try request.params();
const UploadParams = struct {
source: enum { LFMW, LFMS, Spotify },
earliest_date: ?[]const u8,
latest_date: ?[]const u8,
username: ?[]const u8,
};
const params = (try request.expectParams(UploadParams)).?;
const rule_file = try (std.fs.cwd().openFile("rules.json", .{ .mode = .read_only }) catch |err| switch (err) { const rule_file = try (std.fs.cwd().openFile("rules.json", .{ .mode = .read_only }) catch |err| switch (err) {
error.FileNotFound => std.fs.cwd().createFile("rules.json", .{ .read = true }), error.FileNotFound => std.fs.cwd().createFile("rules.json", .{ .read = true }),
else => err, else => err,
@ -36,39 +26,43 @@ pub fn post(request: *jetzig.Request) !jetzig.View {
const rule_list = std.json.parseFromSliceLeaky([]Data.Rule, request.allocator, rule_file_content, .{}) catch null; const rule_list = std.json.parseFromSliceLeaky([]Data.Rule, request.allocator, rule_file_content, .{}) catch null;
//var job = try request.job("process_scrobbles"); //var job = try request.job("process_scrobbles");
var job = try request.job("process_scrobbles2"); var job = try request.job("process_scrobbles2");
const source = params.getT(.integer, "t").?; // This param is required in HTML
// We can parse the dates better const latest_date = if (params.getT(.boolean, "adv-opt")) |_| blk: {
const latest_ts = if (params.latest_date) |ld| const date = params.getT(.string, "latest-date").?;
(try zeit.instant(.{ .source = .{ .iso8601 = ld } })).timestamp break :blk try zeit.Time.fromISO8601(date);
else } else (try zeit.instant(.{ .source = .now })).time();
(try zeit.instant(.{ .source = .now })).timestamp;
const earliest_ts = if (params.earliest_date) |ed| const earliest_date = if (params.getT(.boolean, "adv-opt")) |_| blk: {
(try zeit.instant(.{ .source = .{ .iso8601 = ed } })).timestamp const date = params.getT(.string, "earliest-date").?;
else break :blk try zeit.Time.fromISO8601(date);
(try zeit.instant(.{ .source = .{ .unix_timestamp = 0 } })).timestamp; } else (try zeit.instant(.{ .source = .{ .unix_timestamp = 0 } })).time();
const earliest_timestamp = earliest_date.instant().unixTimestamp();
const latest_timestamp = latest_date.instant().unixTimestamp();
var view_params = try root.put("scrobbles", .array); var view_params = try root.put("scrobbles", .array);
//var job_params = try job.params.put("scrobbles", .array);
var skipped_tracks: u64 = 0; var skipped_tracks: u64 = 0;
var limited_tracks: u64 = 0; var limited_tracks: u64 = 0;
const imported_scrobbles: []Data.UnifiedScrobble = switch (params.source) { const imported_scrobbles: []Data.UnifiedScrobble = switch (source) {
.LFMS, .Spotify => try Utils.scrobbleIngest(request.allocator, if (try request.file("upload")) |file| file.content else unreachable), 0, 1 => try Utils.scrobbleIngest(request.allocator, if (try request.file("upload")) |file| file.content else unreachable),
.LFMW => blk: { 2 => blk: {
const user_agent: []const u8 = "Zuletzt/0.0.1"; const user_agent: []const u8 = "Zuletzt/0.0.1";
var client = Client{ .allocator = request.allocator }; var client = Client{ .allocator = request.allocator };
var lastfm_response_buffer = std.ArrayList(u8).init(request.allocator); var lastfm_response_buffer = std.ArrayList(u8).init(request.allocator);
var scrobble_buffer = std.ArrayList(Data.UnifiedScrobble).init(request.allocator); var scrobble_buffer = std.ArrayList(Data.UnifiedScrobble).init(request.allocator);
const username = if (params.username) |un| un else "VAOTM"; const username = if (params.getT(.string, "username")) |un| un else "VAOTM";
var page: usize = 1; var page: usize = 1;
//var max_pages: ?usize = null; //var max_pages: ?usize = null;
while (true) : (page += 1) { while (true) : (page += 1) {
if (page > 91) break; if (page > 91) break;
const query: []const u8 = try std.fmt.allocPrint(request.allocator, "https://ws.audioscrobbler.com/2.0/?method=user.getrecenttracks&user={s}&api_key=b0c410a48a6078a651e0832699e3cd41&from={}&to={}&page={}&limit=1000&format=json", .{ username, @divFloor(earliest_ts, std.time.ns_per_s), @divFloor(latest_ts, std.time.ns_per_s), page }); const query: []const u8 = try std.fmt.allocPrint(request.allocator, "https://ws.audioscrobbler.com/2.0/?method=user.getrecenttracks&user={s}&api_key=b0c410a48a6078a651e0832699e3cd41&from={}&to={}&page={}&limit=1000&format=json", .{ username, earliest_timestamp, latest_timestamp, page });
const r = try client.fetch(.{ .response_storage = .{ .dynamic = &lastfm_response_buffer }, .location = .{ .url = query }, .method = .GET, .headers = .{ .user_agent = .{ .override = user_agent } } }); const r = try client.fetch(.{ .response_storage = .{ .dynamic = &lastfm_response_buffer }, .location = .{ .url = query }, .method = .GET, .headers = .{ .user_agent = .{ .override = user_agent } } });
std.log.debug("{}: {}", .{ page, r }); std.log.debug("{}: {}", .{ page, r });
if (@intFromEnum(r.status) == 500) { if (@intFromEnum(r.status) == 500) {
@ -85,6 +79,7 @@ pub fn post(request: *jetzig.Request) !jetzig.View {
break :blk try scrobble_buffer.toOwnedSlice(); break :blk try scrobble_buffer.toOwnedSlice();
}, },
else => unreachable,
}; };
var artists = try job.params.put("artists", .object); var artists = try job.params.put("artists", .object);
@ -94,10 +89,8 @@ pub fn post(request: *jetzig.Request) !jetzig.View {
var albumsongs = try job.params.put("albumsongs", .object); var albumsongs = try job.params.put("albumsongs", .object);
var albumsongsartists = try job.params.put("albumsongsartists", .object); var albumsongsartists = try job.params.put("albumsongsartists", .object);
var hash_buffer = [_]u8{undefined} ** 20; // A minimum i64 needs 19 digits + 1 negative sign
appends: for (imported_scrobbles) |scrobble| { appends: for (imported_scrobbles) |scrobble| {
if (scrobble.date > latest_ts or scrobble.date < earliest_ts) { if (scrobble.date > latest_timestamp * std.time.ns_per_s or scrobble.date < earliest_timestamp * std.time.ns_per_s) {
limited_tracks += 1; limited_tracks += 1;
continue :appends; continue :appends;
} }
@ -131,20 +124,18 @@ pub fn post(request: *jetzig.Request) !jetzig.View {
try album_hash_string.appendSlice(artist); try album_hash_string.appendSlice(artist);
const artist_hash = std.hash.Fnv1a_64.hash(artist); const artist_hash = std.hash.Fnv1a_64.hash(artist);
try stored_artist_hashes.append(artist_hash); try stored_artist_hashes.append(artist_hash);
//const signed_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(artist_hash))}); const signed_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(artist_hash))});
const signed_hash_string = try std.fmt.bufPrint(&hash_buffer, "{}", .{@as(i64, @bitCast(artist_hash))});
if (artists.get(signed_hash_string) == null) try artists.put(signed_hash_string, artist); if (artists.get(signed_hash_string) == null) try artists.put(signed_hash_string, artist);
} }
try album_hash_string.appendSlice(complete_scrobble.album); try album_hash_string.appendSlice(complete_scrobble.album);
const album_hash = std.hash.Fnv1a_64.hash(album_hash_string.items); const album_hash = std.hash.Fnv1a_64.hash(album_hash_string.items);
const signed_album_hash_string = try std.fmt.bufPrint(&hash_buffer, "{}", .{@as(i64, @bitCast(album_hash))}); const signed_album_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(album_hash))});
if (albums.get(signed_album_hash_string) == null) try albums.put(signed_album_hash_string, complete_scrobble.album); if (albums.get(signed_album_hash_string) == null) try albums.put(signed_album_hash_string, complete_scrobble.album);
for (stored_artist_hashes.items) |artist_hash| { for (stored_artist_hashes.items) |artist_hash| {
const artistalbum_hash = pair(artist_hash, album_hash); const artistalbum_hash = pair(artist_hash, album_hash);
//const signed_artistalbums_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(artistalbum_hash))}); const signed_artistalbums_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(artistalbum_hash))});
const signed_artistalbums_hash_string = try std.fmt.bufPrint(&hash_buffer, "{}", .{@as(i64, @bitCast(artistalbum_hash))});
if (tracks.get(signed_artistalbums_hash_string) == null) { if (tracks.get(signed_artistalbums_hash_string) == null) {
var artistalbum = try artistalbums.put(signed_artistalbums_hash_string, .object); var artistalbum = try artistalbums.put(signed_artistalbums_hash_string, .object);
try artistalbum.put("artist", @as(i64, @bitCast(artist_hash))); try artistalbum.put("artist", @as(i64, @bitCast(artist_hash)));
@ -156,28 +147,28 @@ pub fn post(request: *jetzig.Request) !jetzig.View {
try track_hash_string.appendSlice(complete_scrobble.album); try track_hash_string.appendSlice(complete_scrobble.album);
try track_hash_string.appendSlice(complete_scrobble.track); try track_hash_string.appendSlice(complete_scrobble.track);
const track_hash = std.hash.Fnv1a_64.hash(track_hash_string.items); const track_hash = std.hash.Fnv1a_64.hash(track_hash_string.items);
const signed_track_hash_string = try std.fmt.bufPrint(&hash_buffer, "{}", .{@as(i64, @bitCast(track_hash))}); const signed_track_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(track_hash))});
if (tracks.get(signed_track_hash_string) == null) try tracks.put(signed_track_hash_string, complete_scrobble.track); if (tracks.get(signed_track_hash_string) == null) try tracks.put(signed_track_hash_string, complete_scrobble.track);
const albumsong_hash = pair(album_hash, track_hash); const albumsong_hash = pair(album_hash, track_hash);
const signed_albumsong_hash_string = try std.fmt.bufPrint(&hash_buffer, "{}", .{@as(i64, @bitCast(albumsong_hash))}); const signed_albumsong_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(albumsong_hash))});
if (albumsongs.get(signed_albumsong_hash_string)) |albumsong| { if (albumsongs.get(signed_albumsong_hash_string)) |albumsong| {
var albumsong_scrobbles = albumsong.get("scrobbles"); var albumsong_scrobbles = albumsong.get("scrobbles");
try albumsong_scrobbles.?.append(@divFloor(complete_scrobble.date, std.time.ns_per_us)); // MICROSECONDS try albumsong_scrobbles.?.append(complete_scrobble.date);
} else { } else {
var albumsong = try albumsongs.put(signed_albumsong_hash_string, .object); var albumsong = try albumsongs.put(signed_albumsong_hash_string, .object);
try albumsong.put("album", @as(i64, @bitCast(album_hash))); try albumsong.put("album", @as(i64, @bitCast(album_hash)));
try albumsong.put("song", @as(i64, @bitCast(track_hash))); try albumsong.put("song", @as(i64, @bitCast(track_hash)));
var albumsong_scrobbles = try albumsong.put("scrobbles", .array); var albumsong_scrobbles = try albumsong.put("scrobbles", .array);
try albumsong_scrobbles.append(@divFloor(complete_scrobble.date, std.time.ns_per_us)); // MICROSECONDS try albumsong_scrobbles.append(complete_scrobble.date);
} }
for (complete_scrobble.artists_track) |artist| { for (complete_scrobble.artists_track) |artist| {
const artist_hash = std.hash.Fnv1a_64.hash(artist); const artist_hash = std.hash.Fnv1a_64.hash(artist);
const signed_artist_hash_string = try std.fmt.bufPrint(&hash_buffer, "{}", .{@as(i64, @bitCast(artist_hash))}); const signed_artist_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{@as(i64, @bitCast(artist_hash))});
if (artists.get(signed_artist_hash_string) == null) try artists.put(signed_artist_hash_string, artist); if (artists.get(signed_artist_hash_string) == null) try artists.put(signed_artist_hash_string, artist);
const albumsongsartist_hash = pair(albumsong_hash, artist_hash); const albumsongsartist_hash = pair(albumsong_hash, artist_hash);
const signed_albumsongsartist_hash_string = try std.fmt.bufPrint(&hash_buffer, "{}", .{@as(i64, @bitCast(albumsongsartist_hash))}); const signed_albumsongsartist_hash_string = try std.fmt.allocPrint(request.allocator, "{}", .{albumsongsartist_hash});
if (albumsongsartists.get(signed_albumsongsartist_hash_string) == null) { if (albumsongsartists.get(signed_albumsongsartist_hash_string) == null) {
var albumsongartist = try albumsongsartists.put(signed_albumsongsartist_hash_string, .object); var albumsongartist = try albumsongsartists.put(signed_albumsongsartist_hash_string, .object);
try albumsongartist.put("albumsong", @as(i64, @bitCast(albumsong_hash))); try albumsongartist.put("albumsong", @as(i64, @bitCast(albumsong_hash)));
@ -186,14 +177,12 @@ pub fn post(request: *jetzig.Request) !jetzig.View {
} }
} }
std.log.debug("Skipped {} tracks\nFiltered {} tracks by date", .{ skipped_tracks, limited_tracks }); std.log.debug("\nSkipped {} tracks\nFiltered {} tracks by date", .{ skipped_tracks, limited_tracks });
try job.schedule(); try job.schedule();
return request.render(.created); return request.render(.created);
} }
// Cantor Pairing Function
// https://en.wikipedia.org/wiki/Pairing_function
fn pair(a: u64, b: u64) u64 { fn pair(a: u64, b: u64) u64 {
return @mod(@divFloor((a +% b) *% (a +% b +% 1), 2) +% b, std.math.maxInt(u64)); return @divFloor((a +% b) *% (a +% b +% 1) +% b, 2);
} }

View file

@ -14,29 +14,14 @@
<label>Username</label> <label>Username</label>
<input type="text" name="username"/> <input type="text" name="username"/>
<fieldset> <fieldset>
<input type="radio" name="source" label="Last.fm" value="LFMS" required>Last.fm</input> <input type="radio" name="t" label="Last.fm" value="0" required>Last.fm</input>
<input type="radio" name="source" label="Spotify" value="Spotify" required>Spotify</input> <input type="radio" name="t" label="Spotify" value="1" required>Spotify</input>
<input type="radio" name="source" label="Last.fm (Web Auth)" value="LFMW" required>Last.fm (WebAuth)</input> <input type="radio" name="t" label="Last.fm (Web Auth)" value="2" required>Last.fm (WebAuth)</input>
<input type="checkbox" id='adv-opt' name="adv-opt" label="Advanced Options">Advanced Options</input> <input type="checkbox" name="adv-opt" label="Advanced Options">Advanced Options</input>
<div id="adv-opt-div" style="display: none;"> Limit to Scrobbles before: <input type="datetime-local" name="latest-date" label="date-before"></input>
Limit to Scrobbles before: <input type="datetime-local" name="latest_date" label="date-before"></input> Limit to Scrobbles after: <input type="datetime-local" name="earliest-date" label="date-after"></input>
Limit to Scrobbles after: <input type="datetime-local" name="earliest_date" label="date-after"></input>
</div>
</fieldset> </fieldset>
</form> </form>
</body> </body>
</html> </html>
<script>
const check = document.getElementById('adv-opt');
const box = document.getElementById('adv-opt-div')
check.addEventListener('click', function handleClick() {
if (check.checked) {
box.style.display = 'block'; // Element is shown
} else {
box.style.display = 'none'; // Element is hidden
}
});
</script>

View file

@ -64,7 +64,7 @@ const ScrobbleFields = enum {
ms_played, // Spotify playtime ms_played, // Spotify playtime
reason_end, // Spotify reason end,1_000 reason_end, // Spotify reason end,1_000
@"@attr", // LastFM now playing @"@attr", // LastFM now playing
irrelevant, // Not a field I care about irrelevant,
}; };
pub fn scrobbleIngest(allocator: std.mem.Allocator, input: []const u8) ![]Data.UnifiedScrobble { pub fn scrobbleIngest(allocator: std.mem.Allocator, input: []const u8) ![]Data.UnifiedScrobble {