libnjb: gettr.c

# gettr.c

    #include "common.h"

    static int progress (u_int64_t sent, u_int64_t total, const char* buf, unsigned len, void *data)
    {
      int percent = (sent*100)/total;
    #ifdef __WIN32__
      printf("Progress: %I64u of %I64u (%d%%)\r", sent, total, percent);
    #else
      printf("Progress: %llu of %llu (%d%%)\r", sent, total, percent);
    #endif
      fflush(stdout);
      return 0;
    }

    static void usage (void)
    {
      fprintf(stderr, "gettr [ -s size ] <trackid> <filename>\n");
    }

    int main (int argc, char **argv)
    {
      njb_t njbs[NJB_MAX_DEVICES], *njb;
      int n, opt, debug;
      u_int32_t id, size;
      extern int optind;
      extern char *optarg;
      char *endptr;
      char *file;

      debug= 0;
      size= 0;
      while ( (opt= getopt(argc, argv, "D:s:")) != -1 ) {
        switch (opt) {
        case 'D':
          debug= atoi(optarg);
          break;
        case 's':
          size= strtoul(optarg, &endptr, 10);
          if ( *endptr != '\0' ) {
            fprintf(stderr, "illegal size value %s\n",
                    optarg);
            return 1;
          }
          break;
        default:
          usage();
          return 1;
        }
      }
      argc -= optind;
      argv += optind;

      if ( argc != 2 ) {
        usage();
        return 1;
      }

      id= strtoul(argv[0], &endptr, 10);
      if ( *endptr != 0 ) {
        fprintf(stderr, "illegal value %s\n", optarg);
        return 1;
      } else if ( ! id ) {
        fprintf(stderr, "bad song id %u\n", id);
        return 1;
      }

      file= argv[1];

      if ( debug ) NJB_Set_Debug(debug);

      if (NJB_Discover(njbs, 0, &n) == -1) {
        fprintf(stderr, "could not locate any jukeboxes\n");
        return 1;
      }

      if ( n == 0 ) {
        fprintf(stderr, "no NJB devices found\n");
        return 1;
      }

      njb = njbs;

      if ( NJB_Open(njb) == -1 ) {
        NJB_Error_Dump(njb,stderr);
        return 1;
      }

      NJB_Capture(njb);

      if ( ! size ) {
        njb_songid_t *tag;

        printf("Locating track %u\n", id);
        NJB_Reset_Get_Track_Tag(njb);
        while ( (tag = NJB_Get_Track_Tag(njb)) ) {
          if ( tag->trid == id ) {
            njb_songid_frame_t *frame;

            frame = NJB_Songid_Findframe(tag, FR_SIZE);
            size = frame->data.u_int32_val;
          }
          NJB_Songid_Destroy(tag);
        }

        /* Dump any pending errors */
        if (NJB_Error_Pending(njb)) {
          NJB_Error_Dump(njb, stderr);
        }

        if ( size ) {
          printf("%u bytes\n", size);
        } else {
          fprintf(stderr, "Track %u not found\n", id);
        }
      }

      if ( size ) {
        if ( NJB_Get_Track(njb, id, size, file, progress, NULL) == -1 ) {
          NJB_Error_Dump(njb, stderr);
        }
        printf("\n");
      }

      NJB_Release(njb);

      NJB_Close(njb);

      return 0;
    }