git.ozlabs.org Git - ccan/blob - tools/_infotojson/infotojson.c

   1 /* This extract info from _info.c and create json file and also optionally store to db */
   2 #include "infotojson.h"
   3
   4 /* Is A == B ? */
   5 #define streq(a,b) (strcmp((a),(b)) == 0)
   6
   7 /* Does A start with B ? */
   8 #define strstarts(a,b) (strncmp((a),(b),strlen(b)) == 0)
   9
  10 /* This version adds one byte (for nul term) */
  11 static void *grab_file(void *ctx, const char *filename)
  12 {
  13         unsigned int max = 16384, size = 0;
  14         int ret, fd;
  15         char *buffer;
  16
  17         if (streq(filename, "-"))
  18                 fd = dup(STDIN_FILENO);
  19         else
  20                 fd = open(filename, O_RDONLY, 0);
  21
  22         if (fd < 0)
  23                 return NULL;
  24
  25         buffer = talloc_array(ctx, char, max+1);
  26         while ((ret = read(fd, buffer + size, max - size)) > 0) {
  27                 size += ret;
  28                 if (size == max)
  29                         buffer = talloc_realloc(ctx, buffer, char, max*=2 + 1);
  30         }
  31         if (ret < 0) {
  32                 talloc_free(buffer);
  33                 buffer = NULL;
  34         } else
  35                 buffer[size] = '\0';
  36         close(fd);
  37         return buffer;
  38 }
  39
  40 /* This is a dumb one which copies.  We could mangle instead. */
  41 static char **split(const char *text)
  42 {
  43         char **lines = NULL;
  44         unsigned int max = 64, num = 0;
  45
  46         lines = talloc_array(text, char *, max+1);
  47
  48         while (*text != '\0') {
  49                 unsigned int len = strcspn(text, "\n");
  50                 lines[num] = talloc_array(lines, char, len + 1);
  51                 memcpy(lines[num], text, len);
  52                 lines[num][len] = '\0';
  53                 text += len + 1;
  54                 if (++num == max)
  55                         lines = talloc_realloc(text, lines, char *, max*=2 + 1);
  56         }
  57         lines[num] = NULL;
  58         return lines;
  59 }
  60
  61 /*combin desc into an array to write to db*/
  62 static char *combinedesc(char **desc)
  63 {
  64         unsigned int i = 0, size = 0;;
  65         char *combine;
  66
  67         for(i = 0; desc[i]; i++)
  68                 size += strlen(desc[i]);
  69
  70         combine = (char *)palloc((size + i)* sizeof(char));
  71         strcpy(combine, desc[0]);
  72
  73         for(i = 1; desc[i]; i++) {
  74                 strcat(combine, "\n");
  75                 strcat(combine, desc[i]);
  76         }
  77         strreplace(combine,'\'',' ');
  78         return combine;
  79 }
  80
  81 /*creating json structure for storing to file/db*/
  82 struct json * createjson(char **infofile, char *author)
  83 {
  84         struct json *jsonobj;
  85         unsigned int modulename;
  86
  87         if(infofile == NULL || author == NULL) {
  88                 printf("Error Author or Info file is NULL\n");
  89                 exit(1);
  90         }
  91
  92         jsonobj = (struct json *)palloc(sizeof(struct json));
  93
  94         jsonobj->author = author;
  95
  96         modulename =  strchr(infofile[0], '-') - infofile[0];
  97         jsonobj->module = (char *)palloc(sizeof(char) * (modulename - 1));
  98         strncpy(jsonobj->module, infofile[0], modulename - 1);
  99         jsonobj->module[modulename - 1] = '\0';
 100
 101         jsonobj->title = infofile[0];
 102         jsonobj->desc = &infofile[1];
 103
 104         return jsonobj;
 105 }
 106
 107 /*extracting title and description from _info.c files*/
 108 char **extractinfo(char **file)
 109 {
 110         char **infofile = NULL;
 111         unsigned int count = 0, j = 0, size = 0;
 112         bool printing = false;
 113
 114         while(file[size++]);
 115         infofile = (char **) palloc(size * sizeof(char *));
 116
 117         for (j = 0; j < size - 1; j++) {
 118                 if (streq(file[j], "/**")) {
 119                         printing = true;
 120                 }
 121                 else if (streq(file[j], " */"))
 122                         printing = false;
 123                 else if (printing) {
 124                         if (strstarts(file[j], " * "))
 125                                 infofile[count++] = file[j] + 3;
 126                         else if (strstarts(file[j], " *"))
 127                                 infofile[count++] = file[j] + 2;
 128                         else {
 129                                 printf("Error in comments structure\n%d",j);
 130                                 exit(1);
 131                         }
 132                 }
 133         }
 134         infofile[count] = NULL;
 135         return infofile;
 136 }
 137
 138 /*storing json structure to json file*/
 139 int storejsontofile(struct json *jsonobj, char *file)
 140 {
 141         FILE *fp;
 142         unsigned int j = 0;
 143         fp = fopen(file, "wt");
 144
 145         fprintf(fp,"\"Module\":\"%s\",\n",jsonobj->module);
 146         fprintf(fp,"\"Title\":\"%s\",\n",jsonobj->title);
 147         fprintf(fp,"\"Author\":\"%s\",\n",jsonobj->author);
 148         fprintf(fp,"\"Description\":[\n");
 149         while(jsonobj->desc[j++])
 150                 fprintf(fp,"{\n\"str\":\"%s\"\n},\n",jsonobj->desc[j - 1]);
 151         fprintf(fp,"]\n");
 152         fclose(fp);
 153         return 1;
 154
 155 }
 156
 157 /*storing json structure to db*/
 158 int storejsontodb(struct json *jsonobj, char *db)
 159 {
 160         char *cmd, *query;
 161         sqlite3 *handle;
 162         char *errstr;
 163         struct db_query *q;
 164
 165         handle = db_open(db);
 166
 167         query = aprintf("SELECT module from search where module=\"%s\";", jsonobj->module);
 168         q = db_query(handle, query);
 169         if (!q->num_rows)
 170                 cmd = aprintf("INSERT INTO search VALUES(\"%s\",\"%s\",\"%s\",'%s\');",
 171                         jsonobj->module, jsonobj->author, jsonobj->title, combinedesc(jsonobj->desc));
 172         else
 173                 cmd = aprintf("UPDATE search set author=\"%s\", title=\"%s\", desc='%s\' where module=\"%s\";",
 174                         jsonobj->author, jsonobj->title, combinedesc(jsonobj->desc), jsonobj->module);
 175
 176         db_command(handle, cmd);
 177         db_close(handle);
 178         return 1;
 179 }
 180
 181 int main(int argc, char *argv[])
 182 {
 183         char *file;
 184         char **lines;
 185         char **infofile;
 186
 187         struct json *jsonobj = NULL;
 188
 189         if(argc < 4) {
 190                 printf("usage: infotojson infofile jsonfile author sqlitedb\n");
 191                 return 1;
 192         }
 193
 194         file = grab_file(NULL, argv[1]);
 195         if (!file)
 196                 err(1, "Reading file %s", argv[1]);
 197
 198         lines = split(file);
 199
 200         //extract info from lines
 201         infofile = extractinfo(lines);
 202
 203         //create json obj
 204         jsonobj = createjson(infofile, argv[3]);
 205
 206         //store to file
 207         storejsontofile(jsonobj, argv[2]);
 208
 209         if(argv[4] != NULL)
 210                 storejsontodb(jsonobj, argv[4]);
 211
 212         talloc_free(file);
 213         return 0;
 214 }