nci-put-file.c

   1 /*
   2  * Copyright (c) 2016-2019 S. Gilles <sgilles@math.umd.edu>
   3  *
   4  * Permission to use, copy, modify, and/or distribute this software
   5  * for any purpose with or without fee is hereby granted, provided
   6  * that the above copyright notice and this permission notice appear
   7  * in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
  10  * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
  11  * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
  12  * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR
  13  * CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
  14  * OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
  15  * NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
  16  * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  17  */
  18 #include <errno.h>
  19 #include <locale.h>
  20 #include <stdint.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25
  26 #include <curl/curl.h>
  27 #include <yajl_parse.h>
  28
  29 #include "macros.h"
  30 #include "util.h"
  31
  32 int
  33 main(int argc, char **argv)
  34 {
  35         int ret = EINVAL;
  36         char *url_base = 0;
  37         char *auth_token = 0;
  38         char *course_id = 0;
  39         char *filename_arg = 0;
  40         char *filename = 0;
  41         char *foldername = 0;
  42         char *path_arg = 0;
  43         char *p = 0;
  44         size_t len = 0;
  45         char *built_uri = 0;
  46         int opt = 0;
  47
  48         setlocale(LC_ALL, "");
  49
  50         while ((opt = getopt(argc, argv, "c:n:p:")) != -1) {
  51                 switch (opt) {
  52                 case 'c':
  53                         course_id = optarg;
  54                         break;
  55                 case 'n':
  56                         filename_arg = optarg;
  57                         break;
  58                 case 'p':
  59                         path_arg = optarg;
  60                         break;
  61                 default:
  62                         break;
  63                 }
  64         }
  65
  66         if (!course_id) {
  67                 ret = EINVAL;
  68                 fprintf(stderr, "course-id is mandatory\n");
  69                 goto cleanup;
  70         }
  71
  72         if (!filename_arg) {
  73                 ret = EINVAL;
  74                 fprintf(stderr, "filename is mandatory\n");
  75                 goto cleanup;
  76         } else {
  77                 while (filename_arg &&
  78                        filename_arg[0] == '/') {
  79                         filename_arg++;
  80                 }
  81
  82                 len = strlen(filename_arg);
  83
  84                 if (!(filename = calloc(len + 1, sizeof *filename))) {
  85                         ret = errno;
  86                         perror(L("calloc"));
  87                         goto cleanup;
  88                 }
  89
  90                 sprintf(filename, "%s", filename_arg);
  91                 p = strrchr(filename, '/');
  92
  93                 if (p) {
  94                         foldername = filename;
  95                         *p = 0;
  96                         filename = p + 1;
  97                 }
  98         }
  99
 100         if (!filename ||
 101             !filename[0]) {
 102                 ret = EINVAL;
 103                 fprintf(stderr, "invalid filename\n");
 104                 goto cleanup;
 105         }
 106
 107         if (!path_arg) {
 108                 ret = EINVAL;
 109                 fprintf(stderr, "path is mandatory\n");
 110                 goto cleanup;
 111         }
 112
 113         curl_global_init(CURL_GLOBAL_DEFAULT);
 114
 115         if (!(url_base = get_url_base())) {
 116                 ret = ENOENT;
 117
 118                 /* Error should have already been printed */
 119                 goto cleanup;
 120         }
 121
 122         if (!(auth_token = get_auth_token())) {
 123                 ret = ENOENT;
 124
 125                 /* Error should have already been printed */
 126                 goto cleanup;
 127         }
 128
 129         /* Step 1 */
 130         len = snprintf(0, 0, "%s/api/v1/courses/%s/files", url_base, course_id);
 131
 132         if (len + 1 < len) {
 133                 ret = errno = EOVERFLOW;
 134                 perror(L(""));
 135                 goto cleanup;
 136         }
 137
 138         if (!(built_uri = malloc(len + 1))) {
 139                 ret = errno;
 140                 perror(L("malloc"));
 141                 goto cleanup;
 142         }
 143
 144         sprintf(built_uri, "%s/api/v1/courses/%s/files", url_base, course_id);
 145
 146         if ((ret = file_upload(built_uri, auth_token, filename, path_arg,
 147                                foldername))) {
 148                 goto cleanup;
 149         }
 150
 151         ret = 0;
 152 cleanup:
 153         free(url_base);
 154         free(auth_token);
 155         curl_global_cleanup();
 156
 157         return ret;
 158 }