riwiwa/muzi
1
0
Fork 0
mirror of https://github.com/riwiwa/muzi.git synced 2025-12-30 12:45:26 -08:00
Code Issues Packages Projects Releases Wiki Activity

Spotify zip extraction improved, extracts into different dirs depending

Browse Source 
on amount of zip files imported
This commit is contained in:
riwiwa
2025-10-07 15:04:50 -07:00
parent 7472fcbeb9
commit b37576d7f8
1 changed files with 54 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
muzi.c
View File

@@ -1,4 +1,6 @@
#include <cjson/cJSON.h>
#include <dirent.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
@@ -7,9 +9,9 @@
#include <sys/types.h>
#include <zip.h>
#define MAX_FILENAME_SIZE 255
// TODO:
// - unzip a given .zip archive of spotify data to a directory WITHOUT ROOT
// PERMS
// - for each json file in the directory {
// - for each json entry {
// - add entry to postgresql db
@@ -20,8 +22,9 @@
// - sql tables: "full history", "artists", "songs", "albums" (see ipad)
//
int extract(const char *path);
int extract(const char *path, const char *target);
int get_artist_plays(void);
int import_spotify(void);
int get_artist_plays(void) {
FILE *fp = fopen("/home/r/dl/spotify-data/rm35@gm - Spotify Extended "
@@ -71,7 +74,8 @@ int get_artist_plays(void) {
return 0;
}
int extract(const char *path) {
int extract(const char *path, const char *target) {
mkdir(target, 0777);
zip_t *za;
int err;
if ((za = zip_open(path, 0, &err)) == NULL) {
@@ -92,7 +96,9 @@ int extract(const char *path) {
continue;
}
// Create directories from zip file
char file_target[MAX_FILENAME_SIZE];
snprintf(file_target, MAX_FILENAME_SIZE, "%s/%s", target, st.name);
char *search = strchr(st.name, '/');
if (search != NULL) {
int index = search - st.name;
@@ -103,20 +109,28 @@ int extract(const char *path) {
end = j;
}
dir[end + 1] = '\0';
mkdir(dir, 0777);
char dir_target[MAX_FILENAME_SIZE];
snprintf(dir_target, MAX_FILENAME_SIZE, "%s/%s", target, dir);
mkdir(dir_target, 0777);
}
// Handle directories
if (file_target[strlen(file_target) - 1] == '/') {
mkdir(file_target, 0777); // Create directory if it doesn't exist
continue;
}
// Open file in archive
zip_file_t *zf = zip_fopen_index(za, i, 0);
if (!zf) {
fprintf(stderr, "Error opening file in zip: %s\n", st.name);
fprintf(stderr, "Error opening file in zip: %s\n", file_target);
continue;
}
// Create output file
FILE *outfile = fopen(st.name, "w+");
FILE *outfile = fopen(file_target, "w+");
if (!outfile) {
fprintf(stderr, "Error creating output file: %s\n", st.name);
fprintf(stderr, "Error creating output file: %s\n", file_target);
zip_fclose(zf);
continue;
}
@@ -135,4 +149,34 @@ int extract(const char *path) {
return 0;
}
int main(void) { extract("./rileyman35@gmail.com-my_spotify_data.zip"); }
int import_spotify(void) {
const char *path = "./spotify-data/zip";
const char *target_base = "./spotify-data/extracted";
DIR *dir = opendir(path);
if (dir == NULL) {
fprintf(stderr, "Error opening directory: %s (%d)\n", path, errno);
return errno;
}
struct dirent *entry = NULL;
while ((entry = readdir(dir)) != NULL) {
char full_name[MAX_FILENAME_SIZE];
snprintf(full_name, MAX_FILENAME_SIZE, "%s/%s", path, entry->d_name);
if (entry->d_type != DT_DIR) {
char *ext = strrchr(entry->d_name, '.');
if (strcmp(ext, ".zip") == 0) {
char *zip_dir_name = entry->d_name;
int len = strlen(zip_dir_name);
zip_dir_name[len - 4] = '\0';
char target[MAX_FILENAME_SIZE];
snprintf(target, MAX_FILENAME_SIZE, "%s/%s", target_base, zip_dir_name);
extract(full_name, target);
}
}
}
closedir(dir);
return 0;
}
int main(void) { import_spotify(); }