tools

mpd_update_library.c (4241B)

---

 #include <ctype.h>
 #include <mpd/client.h>
 #include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 
 #define DEFAULT_HOST "localhost"
 #define DEFAULT_PORT 6600
 #define DEFAULT_BACKOFF 5
 
 struct mpd_connection *conn() {
   const char *host = getenv("MPD_HOST");
   if (host == NULL || strlen(host) == 0)
     host = DEFAULT_HOST;
 
   unsigned port = DEFAULT_PORT;
   const char *port_str = getenv("MPD_PORT");
   if (port_str != NULL && strlen(port_str) > 0)
     port = (unsigned)atoi(port_str);
 
   struct mpd_connection *c = mpd_connection_new(host, port, 0);
   if (c == NULL) {
     fprintf(stderr, "Failed to create MPD connection object\n");
     return NULL;
   }
 
   if (mpd_connection_get_error(c) != MPD_ERROR_SUCCESS) {
     fprintf(stderr, "Error connecting to MPD (%s:%u): %s\n", host, port,
             mpd_connection_get_error_message(c));
     mpd_connection_free(c);
     return NULL;
   }
   return c;
 }
 
 char *trim_whitespace(char *str) {
   char *end;
   while (isspace((unsigned char)*str))
     str++;
   if (*str == 0)
     return str;
   end = str + strlen(str) - 1;
   while (end > str && isspace((unsigned char)*end))
     end--;
   end[1] = '\0';
   return str;
 }
 
 int main(int argc, char *argv[]) {
   if (argc > 1 &&
       (strcmp(argv[1], "-h") == 0 || strcmp(argv[1], "--help") == 0)) {
     printf("Usage: %s [input_file]\n", argv[0]);
     return 0;
   }
 
   FILE *input = stdin;
   if (argc > 1) {
     input = fopen(argv[1], "r");
     if (input == NULL) {
       perror("Error opening file");
       return 1;
     }
   } else if (isatty(STDIN_FILENO)) {
     fprintf(stderr, "Reading from stdin... (Press Ctrl+D to finish)\n");
   }
 
   char **top_dirs = NULL;
   int dir_count = 0;
   char *line = NULL;
   size_t len = 0;
   ssize_t read_bytes;
 
   while ((read_bytes = getline(&line, &len, input)) != -1) {
     char *trimmed = trim_whitespace(line);
     if (strlen(trimmed) == 0 || trimmed[0] == '#')
       continue;
 
     char *slash = strchr(trimmed, '/');
     if (slash != NULL)
       *slash = '\0';
 
     bool found = false;
     for (int i = 0; i < dir_count; i++) {
       if (strcmp(top_dirs[i], trimmed) == 0) {
         found = true;
         break;
       }
     }
 
     if (!found) {
       char **temp_dirs = realloc(top_dirs, sizeof(char *) * (dir_count + 1));
       if (temp_dirs == NULL) {
         perror("realloc");
         break;
       }
       top_dirs = temp_dirs;
       top_dirs[dir_count] = strdup(trimmed);
       if (top_dirs[dir_count] == NULL) {
         perror("strdup");
         break;
       }
       dir_count++;
     }
   }
   free(line);
   if (input != stdin)
     fclose(input);
 
   if (dir_count == 0) {
     printf("No directories found in input.\n");
     for (int i = 0; i < dir_count; i++)
       free(top_dirs[i]);
     free(top_dirs);
     return 0;
   }
 
   printf("Found %d unique top-level directories to update.\n", dir_count);
 
   struct mpd_connection *c = conn();
   if (c == NULL) {
     for (int i = 0; i < dir_count; i++)
       free(top_dirs[i]);
     free(top_dirs);
     return 1;
   }
 
   int backoff = DEFAULT_BACKOFF;
   const char *backoff_str = getenv("MPD_UPDATE_BACKOFF");
   if (backoff_str != NULL && strlen(backoff_str) > 0)
     backoff = atoi(backoff_str);
 
   int updated_count = 0;
   int error_count = 0;
 
   for (int i = 0; i < dir_count; i++) {
     printf("Updating: '%s'\n", top_dirs[i]);
     unsigned job_id = mpd_run_update(c, top_dirs[i]);
     if (job_id == 0) {
       fprintf(stderr, "  ✗ Failed: %s\n", mpd_connection_get_error_message(c));
       error_count++;
       mpd_connection_clear_error(c);
     } else {
       printf("  ✓ Success (Job ID: %u)\n", job_id);
       updated_count++;
     }
     if (i < dir_count - 1 && backoff > 0)
       sleep(backoff);
   }
 
   printf("Update completed!\n");
   printf("Total directories processed: %d\n", dir_count);
   printf("Successfully updated: %d\n", updated_count);
   printf("Errors: %d\n", error_count);
 
   struct mpd_status *status = mpd_run_status(c);
   if (status != NULL) {
     if (mpd_status_get_update_id(status) > 0)
       printf("MPD is currently updating...\n");
     mpd_status_free(status);
   }
 
   for (int i = 0; i < dir_count; i++)
     free(top_dirs[i]);
   free(top_dirs);
   mpd_connection_free(c);
   return 0;
 }