indent.c

   1 /* depend: */
   2 /* cflags: */
   3 /* linker: debug.o */
   4
   5 #include <assert.h>
   6 #include <getopt.h>
   7 #include <malloc.h>
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11
  12 #include "debug.h"
  13
  14 /* macros */
  15
  16 #define CEIL(x, y) (((x) + (y) - 1) / (y))
  17 #define MIN(x, y) (((x) < (y)) ? (x) : (y))
  18 #define MAX(x, y) (((x) > (y)) ? (x) : (y))
  19
  20 //#define BUFFERSIZE 4096
  21 #define BUFFERSIZE 256
  22 #define TABSIZE 4
  23
  24 /* type definition */
  25
  26 typedef enum {
  27     e_unknown = 0,
  28     e_ansi,
  29     e_kr
  30 } cmode_t;
  31
  32 /* gobal variables */
  33
  34 char *progname = NULL;
  35
  36 /* help function */
  37
  38 void usage (int ret)
  39 {
  40     FILE *fd = ret ? stderr : stdout;
  41     fprintf (fd, "usage: %s [-i file] [-h] [-m k&r|ansi|c99] [-o file] [-v]\n", progname);
  42     fprintf (fd, " -i : input file\n");
  43     fprintf (fd, " -h : help message\n");
  44     fprintf (fd, " -m : indent mode\n");
  45     fprintf (fd, " -o : output file\n");
  46     fprintf (fd, " -v : verbose level (%d)\n", verbose);
  47
  48     exit (ret);
  49 }
  50
  51 /* indent function */
  52
  53 int indent (FILE *fin, FILE *fout, cmode_t cmode) {
  54     char bufin[BUFFERSIZE + 1] = {0};
  55     char bufout[BUFFERSIZE * TABSIZE + 1] = {0};
  56     size_t i, nb;
  57
  58     while (!feof (fin)) {
  59
  60         /* read file */
  61         nb = fread (bufin, 1, BUFFERSIZE, fin);
  62         VERBOSE (DEBUG, fprintf (stdout, "buffer in: %d\n", nb));
  63         if (errno != 0) {
  64             VERBOSE (ERROR, fprintf (stderr, "can't read file (%d)\n", errno));
  65             exit (1);
  66         }
  67
  68         /* process line */
  69         char *ptin = bufin;
  70         char *ptout = bufout;
  71         while (*ptin != '\0') {
  72             switch (*ptin) {
  73                 case '\t':
  74                     for (i = 0; i < TABSIZE; i++) {
  75                         *ptout++ = ' ';
  76                     }
  77                     break;
  78                 case '{':
  79                 case '}':
  80                 case ';':
  81                     *ptout++ = *ptin;
  82                     *ptout++ = '\n';
  83                     break;
  84                 default:
  85                     *ptout++ = *ptin;
  86             }
  87             ptin++;
  88         }
  89         ptout = '\0';
  90
  91         /* write file */
  92         VERBOSE (DEBUG, fprintf (stdout, "buffer out: %d\n", strlen (bufout)));
  93         ptout = bufout;
  94         while ((nb = fread (ptout, 1, strlen (ptout), fout)) != strlen (ptout)) {
  95             if (errno != 0) {
  96                 VERBOSE (ERROR, fprintf (stderr, "can't write file (%d)\n", errno));
  97                 exit (1);
  98             }
  99             ptout += nb;
 100         }
 101     }
 102
 103     /* close all */
 104     fclose (fin);
 105     fclose (fout);
 106
 107     return 0;
 108 }
 109
 110 /* main function */
 111
 112 int main (int argc, char *argv[])
 113 {
 114     cmode_t cmode = e_unknown;
 115     char *input = NULL;
 116     char *mode = "ansi";
 117     char *output = NULL;
 118
 119     /* get basename */
 120     char *pt = progname = argv[0];
 121     while (*pt) {
 122         if ((*pt == '/') || (*pt == '\\')) {
 123            progname = pt + 1;
 124         }
 125         pt++;
 126     }
 127
 128     int c;
 129     while ((c = getopt(argc, argv, "i:hm:o:v:")) != EOF) {
 130         switch (c) {
 131         case 'i':
 132             input = optarg;
 133             break;
 134         case 'm':
 135             mode = optarg;
 136             break;
 137         case 'o':
 138             output = optarg;
 139             break;
 140         case 'v':
 141             verbose = atoi (optarg);
 142             break;
 143         case 'h':
 144         default:
 145             usage (c != 'h');
 146         }
 147     }
 148     if (argc - optind != 0) {
 149         fprintf (stderr, "%s: invalid option -- %s\n", progname, argv[optind]);
 150         usage (1);
 151     }
 152
 153     /* check input */
 154     FILE *fin = NULL;
 155     if (input) {
 156         fin = fopen (input, "rb");
 157         if (!fin) {
 158             VERBOSE (ERROR, fprintf (stderr, "error: can't open file '%s'\n", input));
 159         }
 160     } else {
 161         fin = stdin;
 162     }
 163
 164     /* check output */
 165     FILE *fout = NULL;
 166     if (output) {
 167         fout = fopen (input, "wb");
 168         if (!fout) {
 169             VERBOSE (ERROR, fprintf (stderr, "error: can't open file '%s'\n", output));
 170             fclose (fin);
 171         }
 172     } else {
 173         fout = stdout;
 174     }
 175
 176     /* check mode */
 177     if (strcmp (mode, "ansi") == 0) {
 178         cmode = e_ansi;
 179     } else if (strcmp (mode, "k&r") == 0) {
 180         cmode = e_kr;
 181     } else {
 182         VERBOSE (ERROR, fprintf (stderr, "error: mode '%s' unknown\n", mode));
 183     }
 184
 185     return indent (fin, fout, cmode);
 186 }
 187
 188 // test: indent.exe -h
 189 // test: indent.exe -h | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
 190 // test: indent.exe -_ 2> /dev/null | awk 'END { if (NR == 0) { exit(0) } else exit (1) }'
 191 // test: indent.exe -_ 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
 192
 193 /* vim: set ts=4 sw=4 et: */