it's kinda done
This commit is contained in:
parent
ff715db029
commit
0a8fe93297
2 changed files with 62 additions and 15 deletions
53
src/main.zig
53
src/main.zig
|
@ -15,6 +15,23 @@ fn parse_musicdef(allocator: mem.Allocator, tracks: *TrackMap, file: []const u8)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const forbidden_chars = "<>:\"/\\|?*!";
|
||||||
|
fn slugify(allocator: mem.Allocator, name: ?[]const u8, fallback: []const u8) ![]const u8 {
|
||||||
|
const notNullName = name orelse fallback;
|
||||||
|
|
||||||
|
// Make copy of name
|
||||||
|
var copiedName = try allocator.alloc(u8, notNullName.len);
|
||||||
|
|
||||||
|
for (0..copiedName.len) |i| {
|
||||||
|
if (std.mem.indexOfScalar(u8, forbidden_chars, notNullName[i])) |_| {
|
||||||
|
copiedName[i] = '-';
|
||||||
|
} else {
|
||||||
|
copiedName[i] = notNullName[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return copiedName;
|
||||||
|
}
|
||||||
|
|
||||||
pub fn main() !void {
|
pub fn main() !void {
|
||||||
// Get allocator
|
// Get allocator
|
||||||
var gpa = std.heap.GeneralPurposeAllocator(.{}){};
|
var gpa = std.heap.GeneralPurposeAllocator(.{}){};
|
||||||
|
@ -54,7 +71,7 @@ pub fn main() !void {
|
||||||
defer iter.deinit();
|
defer iter.deinit();
|
||||||
while (try iter.next()) |entry| {
|
while (try iter.next()) |entry| {
|
||||||
// Skip non-files
|
// Skip non-files
|
||||||
if (entry.kind != fs.File.Kind.file) {
|
if (entry.kind != .file) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -69,10 +86,15 @@ pub fn main() !void {
|
||||||
try parse_musicdef(keyAllocator, &trackInfos, filedata);
|
try parse_musicdef(keyAllocator, &trackInfos, filedata);
|
||||||
} else {
|
} else {
|
||||||
// Save file to hashmap of resolved files
|
// Save file to hashmap of resolved files
|
||||||
try trackLocations.put(try keyAllocator.dupe(u8, filename), entry.path);
|
try trackLocations.put(try keyAllocator.dupe(u8, filename), try keyAllocator.dupe(u8, entry.path));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Make target dirs
|
||||||
|
const outputDir = try fs.cwd().makeOpenPath("target", .{});
|
||||||
|
const originalsDir = try outputDir.makeOpenPath("original", .{});
|
||||||
|
const othersDir = try outputDir.makeOpenPath("others", .{});
|
||||||
|
|
||||||
// Print detected songs
|
// Print detected songs
|
||||||
var tracks = trackInfos.valueIterator();
|
var tracks = trackInfos.valueIterator();
|
||||||
while (tracks.next()) |trackInfo| {
|
while (tracks.next()) |trackInfo| {
|
||||||
|
@ -82,5 +104,32 @@ pub fn main() !void {
|
||||||
trackInfo.author,
|
trackInfo.author,
|
||||||
trackInfo.source,
|
trackInfo.source,
|
||||||
});
|
});
|
||||||
|
// Extract all the qualifying IDs
|
||||||
|
var ids = mem.splitScalar(u8, trackInfo.id, ',');
|
||||||
|
while (ids.next()) |id| {
|
||||||
|
const actualID = mem.trimLeft(u8, id, "\\");
|
||||||
|
|
||||||
|
// Map to track location
|
||||||
|
var soundIDBuffer: [10]u8 = undefined;
|
||||||
|
const soundID = try std.fmt.bufPrint(&soundIDBuffer, "O_{s}", .{actualID});
|
||||||
|
const location = trackLocations.get(soundID) orelse {
|
||||||
|
std.debug.panic("{s} has no track assigned", .{soundID});
|
||||||
|
};
|
||||||
|
|
||||||
|
// Copy file over, dir depending on originality (I-I mean, I'm not trying to be mean)
|
||||||
|
const sourceDirName = try slugify(allocator, trackInfo.source, "unknown");
|
||||||
|
defer allocator.free(sourceDirName);
|
||||||
|
const targetDir = if (trackInfo.is_original) originalsDir else try othersDir.makeOpenPath(sourceDirName, .{});
|
||||||
|
|
||||||
|
const author = try slugify(allocator, trackInfo.author, "unknown");
|
||||||
|
defer allocator.free(author);
|
||||||
|
const title = try slugify(allocator, trackInfo.title, trackInfo.id);
|
||||||
|
defer allocator.free(title);
|
||||||
|
|
||||||
|
const filename = try std.fmt.allocPrint(allocator, "{s} - {s}.ogg", .{ author, title });
|
||||||
|
defer allocator.free(filename);
|
||||||
|
|
||||||
|
try dir.copyFile(location, targetDir, filename, .{});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -21,7 +21,7 @@ is_original: bool,
|
||||||
|
|
||||||
pub fn parseLump(allocator: mem.Allocator, lump: []const u8) !Self {
|
pub fn parseLump(allocator: mem.Allocator, lump: []const u8) !Self {
|
||||||
// Get lines
|
// Get lines
|
||||||
var lines = mem.tokenize(u8, lump, "\n");
|
var lines = mem.tokenizeScalar(u8, lump, '\n');
|
||||||
|
|
||||||
// Next line is the ID, we need that ASAP!
|
// Next line is the ID, we need that ASAP!
|
||||||
const id = mem.trim(u8, lines.next().?, " \r");
|
const id = mem.trim(u8, lines.next().?, " \r");
|
||||||
|
@ -37,10 +37,9 @@ pub fn parseLump(allocator: mem.Allocator, lump: []const u8) !Self {
|
||||||
|
|
||||||
// Read rest of lump and assign to fields in struct
|
// Read rest of lump and assign to fields in struct
|
||||||
while (lines.next()) |line| {
|
while (lines.next()) |line| {
|
||||||
const sep = mem.indexOf(u8, line, "=");
|
const sep = mem.indexOfScalar(u8, line, '=') orelse continue;
|
||||||
if (sep) |sepIndex| {
|
const key = mem.trim(u8, line[0..sep], " \r");
|
||||||
const key = mem.trim(u8, line[0..sepIndex], " \r");
|
const val = mem.trim(u8, line[sep + 1 ..], " \r");
|
||||||
const val = mem.trim(u8, line[sepIndex + 1 ..], " \r");
|
|
||||||
const ownedVal = try allocator.dupe(u8, val);
|
const ownedVal = try allocator.dupe(u8, val);
|
||||||
const infoKey = std.meta.stringToEnum(InfoKey, key) orelse continue;
|
const infoKey = std.meta.stringToEnum(InfoKey, key) orelse continue;
|
||||||
switch (infoKey) {
|
switch (infoKey) {
|
||||||
|
@ -50,7 +49,6 @@ pub fn parseLump(allocator: mem.Allocator, lump: []const u8) !Self {
|
||||||
.OriginalComposers => info.originalComposers = ownedVal,
|
.OriginalComposers => info.originalComposers = ownedVal,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// Check if it's an original work
|
// Check if it's an original work
|
||||||
if (info.source) |source| {
|
if (info.source) |source| {
|
||||||
|
|
Loading…
Reference in a new issue