libmtp: sendtr.c

# sendtr.c

    #include <stdlib.h>
    #include <limits.h>
    #include <string.h>
    #include <libgen.h>
    #include <sys/stat.h>
    #include <sys/types.h>
    #include <fcntl.h>
    #ifdef HAVE_LANGINFO_H
    #include <langinfo.h>
    #endif

    #include "config.h"
    #include "common.h"
    #include "util.h"
    #include "connect.h"
    #include "libmtp.h"
    #include "pathutils.h"

    extern LIBMTP_folder_t *folders;
    extern LIBMTP_file_t *files;
    extern LIBMTP_mtpdevice_t *device;

    void sendtrack_usage (void)
    {
      fprintf(stderr, "usage: sendtr [ -D debuglvl ] [ -q ]\n");
      fprintf(stderr, "-t <title> -a <artist> -A <Album artist> -w <writer or composer>\n");
      fprintf(stderr, "    -l <album> -c <codec> -g <genre> -n <track number> -y <year>\n");
      fprintf(stderr, "       -d <duration in seconds> -s <storage_id> <local path> <remote path>\n");
      fprintf(stderr, "(-q means the program will not ask for missing information.)\n");
    }

    static char *prompt (const char *prompt, char *buffer, size_t bufsz, int required)
    {
      char *cp, *bp;

      while (1) {
        fprintf(stdout, "%s> ", prompt);
        if ( fgets(buffer, bufsz, stdin) == NULL ) {
          if (ferror(stdin)) {
            perror("fgets");
          } else {
            fprintf(stderr, "EOF on stdin\n");
          }
          return NULL;
        }

        cp = strrchr(buffer, '\n');
        if ( cp != NULL ) *cp = '\0';

        bp = buffer;
        while ( bp != cp ) {
          if ( *bp != ' ' && *bp != '\t' ) return bp;
          bp++;
        }

        if (! required) return bp;
      }
    }

    static int add_track_to_album(LIBMTP_album_t *albuminfo, LIBMTP_track_t *trackmeta)
    {
      LIBMTP_album_t *album;
      LIBMTP_album_t *found_album = NULL;
      int ret;

      /* Look for the album */
      album = LIBMTP_Get_Album_List(device);
      while(album != NULL) {
        if ((album->name != NULL &&
            album->artist != NULL &&
            !strcmp(album->name, albuminfo->name) &&
            !strcmp(album->artist, albuminfo->artist)) ||
              (album->name != NULL &&
            album->composer != NULL &&
            !strcmp(album->name, albuminfo->name) &&
            !strcmp(album->composer, albuminfo->composer))) {
          /* Disconnect this album for later use */
          found_album = album;
          album = album->next;
          found_album->next = NULL;
        } else {
          LIBMTP_album_t *tmp;

          tmp = album;
          album = album->next;
          LIBMTP_destroy_album_t(tmp);
        }
      }

      if (found_album != NULL) {
        uint32_t *tracks;

        tracks = (uint32_t *)malloc((found_album->no_tracks+1) * sizeof(uint32_t));
        printf("Album \"%s\" found: updating...\n", found_album->name);
        if (!tracks) {
          printf("failed malloc in add_track_to_album()\n");
          return 1;
        }
        found_album->no_tracks++;
        if (found_album->tracks != NULL) {
          memcpy(tracks, found_album->tracks, found_album->no_tracks * sizeof(uint32_t));
          free(found_album->tracks);
        }
        tracks[found_album->no_tracks-1] = trackmeta->item_id;
        found_album->tracks = tracks;
        ret = LIBMTP_Update_Album(device, found_album);
        LIBMTP_destroy_album_t(found_album);
      } else {
        uint32_t *trackid;

        trackid = (uint32_t *)malloc(sizeof(uint32_t));
        *trackid = trackmeta->item_id;
        albuminfo->tracks = trackid;
        albuminfo->no_tracks = 1;
        albuminfo->storage_id = trackmeta->storage_id;
        printf("Album doesn't exist: creating...\n");
        ret = LIBMTP_Create_New_Album(device, albuminfo);
        /* albuminfo will be destroyed later by caller */
      }

      if (ret != 0) {
        printf("Error creating or updating album.\n");
        printf("(This could be due to that your device does not support albums.)\n");
        LIBMTP_Dump_Errorstack(device);
        LIBMTP_Clear_Errorstack(device);
      } else {
        printf("success!\n");
      }
      return ret;
    }

    int sendtrack_function(char * from_path, char * to_path, char *partist, char *palbumartist, char *ptitle, char *pgenre, char *palbum, char *pcomposer, uint16_t tracknum, uint16_t length, uint16_t year, uint32_t storageid, uint16_t quiet)
    {
      char *filename, *parent;
      char artist[80], albumartist[80], title[80], genre[80], album[80], composer[80];
      char num[80];
      uint64_t filesize;
      uint32_t parent_id = 0;
      struct stat sb;
      LIBMTP_track_t *trackmeta;
      LIBMTP_album_t *albuminfo;
      int ret;

      printf("Sending track %s to %s\n", from_path, to_path);

      trackmeta = LIBMTP_new_track_t();
      albuminfo = LIBMTP_new_album_t();

      parent = dirname(strdup(to_path));
      filename = basename(strdup(to_path));
      parent_id = parse_path (parent,files,folders);
      if (parent_id == -1) {
        printf("Parent folder could not be found, skipping\n");
        return 1;
      }

      if (stat(from_path, &sb) == -1) {
        fprintf(stderr, "%s: ", from_path);
        perror("stat");
        return 1;
      }

      if (!S_ISREG(sb.st_mode))
        return 0;

      filesize = sb.st_size;
      trackmeta->filetype = find_filetype (from_path);
      if (!LIBMTP_FILETYPE_IS_TRACK(trackmeta->filetype)) {
        printf("Not a valid track codec: \"%s\"\n", LIBMTP_Get_Filetype_Description(trackmeta->filetype));
        return 1;
      }

      if ((ptitle == NULL) && (quiet == 0)) {
        if ( (ptitle = prompt("Title", title, 80, 0)) != NULL )
          if (!strlen(ptitle)) ptitle = NULL;
      }

      if ((palbum == NULL) && (quiet == 0)) {
        if ( (palbum = prompt("Album", album, 80, 0)) != NULL )
          if (!strlen(palbum)) palbum = NULL;
      }

      if ((palbumartist == NULL) && (quiet == 0)) {
        if ( (palbumartist = prompt("Album artist", albumartist, 80, 0)) != NULL )
          if (!strlen(palbumartist)) palbumartist = NULL;
      }

      if ((partist == NULL) && (quiet == 0)) {
        if ( (partist = prompt("Artist", artist, 80, 0)) != NULL )
          if (!strlen(partist)) partist = NULL;
      }

      if ((pcomposer == NULL) && (quiet == 0)) {
        if ( (pcomposer = prompt("Writer or Composer", composer, 80, 0)) != NULL )
          if (!strlen(pcomposer)) pcomposer = NULL;
      }

      if ((pgenre == NULL) && (quiet == 0)) {
        if ( (pgenre = prompt("Genre", genre, 80, 0)) != NULL )
          if (!strlen(pgenre)) pgenre = NULL;
      }

      if ((tracknum == 0) && (quiet == 0)) {
        char *pnum;
        if ( (pnum = prompt("Track number", num, 80, 0)) == NULL )
          tracknum = 0;
        else
          tracknum = strtoul(pnum, 0, 10);
      }

      if ((year == 0) && (quiet == 0)) {
        char *pnum;
        if ( (pnum = prompt("Year", num, 80, 0)) == NULL )
          year = 0;
        else
          year = strtoul(pnum, 0, 10);
      }

      if ((length == 0) && (quiet == 0)) {
        char *pnum;
        if ( (pnum = prompt("Length", num, 80, 0)) == NULL )
          length = 0;
        else
          length = strtoul(pnum, 0, 10);
      }

      printf("Sending track:\n");
      printf("Codec:     %s\n", LIBMTP_Get_Filetype_Description(trackmeta->filetype));
      if (ptitle) {
        printf("Title:     %s\n", ptitle);
        trackmeta->title = strdup(ptitle);
      }
      if (palbum) {
        printf("Album:     %s\n", palbum);
        trackmeta->album = strdup(palbum);
        albuminfo->name = strdup(palbum);
      }
      if (palbumartist) {
        printf("Album artist:    %s\n", palbumartist);
        albuminfo->artist = strdup(palbumartist);
      }
      if (partist) {
        printf("Artist:    %s\n", partist);
        trackmeta->artist = strdup(partist);
        if (palbumartist == NULL)
          albuminfo->artist = strdup(partist);
      }
      if (pcomposer) {
        printf("Writer or Composer:    %s\n", pcomposer);
        trackmeta->composer = strdup(pcomposer);
        albuminfo->composer = strdup(pcomposer);
      }
      if (pgenre) {
        printf("Genre:     %s\n", pgenre);
        trackmeta->genre = strdup(pgenre);
        albuminfo->genre = strdup(pgenre);
      }
      if (year > 0) {
        char tmp[80];
        printf("Year:      %d\n", year);
        snprintf(tmp, sizeof(tmp)-1, "%4d0101T0000.0", year);
        tmp[sizeof(tmp)-1] = '\0';
        trackmeta->date = strdup(tmp);
      }
      if (tracknum > 0) {
        printf("Track no:  %d\n", tracknum);
        trackmeta->tracknumber = tracknum;
      }
      if (length > 0) {
        printf("Length:    %d\n", length);
        // Multiply by 1000 since this is in milliseconds
        trackmeta->duration = length * 1000;
      }
      // We should always have this
      if (filename != NULL) {
        trackmeta->filename = strdup(filename);
      }
      trackmeta->filesize = filesize;
      trackmeta->parent_id = parent_id;
      {
        int rc;
        char *desc = NULL;
        LIBMTP_devicestorage_t *pds = NULL;

        if (0 != (rc=LIBMTP_Get_Storage(device, LIBMTP_STORAGE_SORTBY_NOTSORTED))) {
          perror("LIBMTP_Get_Storage()");
          exit(-1);
        }
        for (pds = device->storage; pds != NULL; pds = pds->next) {
          if (pds->id == storageid) {
            desc = strdup(pds->StorageDescription);
            break;
          }
        }
        if (NULL != desc) {
          printf("Storage ID: %s (%u)\n", desc, storageid);
          free(desc);
        } else
          printf("Storage ID: %u\n", storageid);
        trackmeta->storage_id = storageid;
      }

      printf("Sending track...\n");
      ret = LIBMTP_Send_Track_From_File(device, from_path, trackmeta, progress, NULL);
      printf("\n");
      if (ret != 0) {
        printf("Error sending track.\n");
        LIBMTP_Dump_Errorstack(device);
        LIBMTP_Clear_Errorstack(device);
        ret = 1;
      } else {
        printf("New track ID: %d\n", trackmeta->item_id);
      }

      /* Add here add to album call */
      if (palbum)
        ret = add_track_to_album(albuminfo, trackmeta);

      LIBMTP_destroy_album_t(albuminfo);
      LIBMTP_destroy_track_t(trackmeta);

      return ret;
    }

    int sendtrack_command (int argc, char **argv) {
      int opt;
      extern int optind;
      extern char *optarg;
      char *partist = NULL;
      char *palbumartist = NULL;
      char *pcomposer = NULL;
      char *ptitle = NULL;
      char *pgenre = NULL;
      char *pcodec = NULL;
      char *palbum = NULL;
      uint16_t tracknum = 0;
      uint16_t length = 0;
      uint16_t year = 0;
      uint16_t quiet = 0;
      uint32_t storageid = 0;
      while ( (opt = getopt(argc, argv, "qD:t:a:A:w:l:c:g:n:d:y:s:")) != -1 ) {
        switch (opt) {
        case 't':
          ptitle = strdup(optarg);
          break;
        case 'a':
          partist = strdup(optarg);
          break;
        case 'A':
          palbumartist = strdup(optarg);
          break;
        case 'w':
          pcomposer = strdup(optarg);
          break;
        case 'l':
          palbum = strdup(optarg);
          break;
        case 'c':
          pcodec = strdup(optarg); // FIXME: DSM check for MP3, WAV or WMA
          break;
        case 'g':
          pgenre = strdup(optarg);
          break;
        case 'n':
          tracknum = atoi(optarg);
          break;
        case 's':
          storageid = (uint32_t) strtoul(optarg, NULL, 0);
          break;
        case 'd':
          length = atoi(optarg);
          break;
        case 'y':
          year = atoi(optarg);
          break;
        case 'q':
          quiet = 1;
          break;
        default:
          sendtrack_usage();
        }
      }
      argc -= optind;
      argv += optind;

      if ( argc != 2 ) {
        printf("You need to pass a filename and destination.\n");
        sendtrack_usage();
        return 0;
      }

      checklang();

      printf("%s,%s,%s,%s,%s,%s,%s,%s,%d%d,%d,%u,%d\n",argv[0],argv[1],partist,palbumartist,ptitle,pgenre,palbum,pcomposer,tracknum, length, year, storageid, quiet);
      return sendtrack_function(argv[0],argv[1],partist,palbumartist,ptitle,pgenre,palbum,pcomposer, tracknum, length, year, storageid, quiet);
    }