util/batch.hh

   1 /*
   2  * Copyright © 2021  Behdad Esfahbod
   3  *
   4  *  This is part of HarfBuzz, a text shaping library.
   5  *
   6  * Permission is hereby granted, without written agreement and without
   7  * license or royalty fees, to use, copy, modify, and distribute this
   8  * software and its documentation for any purpose, provided that the
   9  * above copyright notice and the following two paragraphs appear in
  10  * all copies of this software.
  11  *
  12  * IN NO EVENT SHALL THE COPYRIGHT HOLDER BE LIABLE TO ANY PARTY FOR
  13  * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES
  14  * ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN
  15  * IF THE COPYRIGHT HOLDER HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH
  16  * DAMAGE.
  17  *
  18  * THE COPYRIGHT HOLDER SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING,
  19  * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
  20  * FITNESS FOR A PARTICULAR PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS
  21  * ON AN "AS IS" BASIS, AND THE COPYRIGHT HOLDER HAS NO OBLIGATION TO
  22  * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
  23  */
  24
  25 #ifndef BATCH_HH
  26 #define BATCH_HH
  27
  28 #include "options.hh"
  29
  30 typedef int (*main_func_t) (int argc, char **argv);
  31
  32 template <typename main_t, bool report_status=false>
  33 int
  34 batch_main (int argc, char **argv)
  35 {
  36   if (argc == 2 && !strcmp (argv[1], "--batch"))
  37   {
  38     int ret = 0;
  39     char buf[4092];
  40     while (fgets (buf, sizeof (buf), stdin))
  41     {
  42       size_t l = strlen (buf);
  43       if (l && buf[l - 1] == '\n') buf[l - 1] = '\0';
  44
  45       char *args[64];
  46       argc = 0;
  47       args[argc++] = argv[0];
  48       char *p = buf, *e;
  49       args[argc++] = p;
  50       while ((e = strchr (p, ';')) && argc < (int) ARRAY_LENGTH (args))
  51       {
  52         *e++ = '\0';
  53         while (*e == ';')
  54           e++;
  55         args[argc++] = p = e;
  56       }
  57
  58       int result = main_t () (argc, args);
  59
  60       if (report_status)
  61         fprintf (stdout, result == 0 ? "success\n" : "failure\n");
  62       fflush (stdout);
  63
  64       ret = MAX (ret, result);
  65     }
  66     return ret;
  67   }
  68
  69   int ret = main_t () (argc, argv);
  70   if (report_status && ret != 0)
  71     fprintf (stdout, "error: Operation failed. Probably a bug. File github issue.\n");
  72   return ret;
  73 }
  74
  75 #endif