first process

[indent.git] / indent.c

/* depend: */
/* cflags: */
/* linker: debug.o */

#include <assert.h>
#include <getopt.h>
#include <malloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "debug.h"

/* macros */

#define CEIL(x, y) (((x) + (y) - 1) / (y))
#define MIN(x, y) (((x) < (y)) ? (x) : (y))
#define MAX(x, y) (((x) > (y)) ? (x) : (y))

//#define BUFFERSIZE 4096
#define BUFFERSIZE 256
#define TABSIZE 4

/* type definition */

typedef enum {
    e_unknown = 0,
    e_ansi,
    e_kr
} cmode_t;

/* gobal variables */

char *progname = NULL;

/* help function */

void usage (int ret)
{
    FILE *fd = ret ? stderr : stdout;
    fprintf (fd, "usage: %s [-i file] [-h] [-m k&r|ansi|c99] [-o file] [-v]\n", progname);
    fprintf (fd, " -i : input file\n");
    fprintf (fd, " -h : help message\n");
    fprintf (fd, " -m : indent mode\n");
    fprintf (fd, " -o : output file\n");
    fprintf (fd, " -v : verbose level (%d)\n", verbose);

    exit (ret);
}

/* indent function */

int indent (FILE *fin, FILE *fout, cmode_t cmode) {
    char bufin[BUFFERSIZE + 1] = {0};
    char bufout[BUFFERSIZE * TABSIZE + 1] = {0};
    size_t i, nb;

    while (!feof (fin)) {

        /* read file */
        nb = fread (bufin, 1, BUFFERSIZE, fin);
        VERBOSE (DEBUG, fprintf (stdout, "buffer: %d\n", nb));
        if (errno != 0) {
            VERBOSE (ERROR, fprintf (stderr, "can't read file\n"));
            exit (1);
        }

        /* process line */
        char *ptin = bufin;
        char *ptout = bufout;
        while (*ptin != '\0') {
            switch (*ptin) {
                case '\t':
                    for (i = 0; i < TABSIZE; i++) {
                        *ptout++ = ' ';
                    }
                    break;
                case '{':
                case '}':
                case ';':
                    *ptout++ = *ptin;
                    *ptout++ = '\n';
                    break;
                default:
                    *ptout++ = *ptin;
            }
            ptin++;
        }
        ptout = '\0';

        /* write file */
        ptout = bufout;
        while ((nb = fread (ptout, 1, strlen (ptout), fout)) != strlen (ptout)) {
            if (errno != 0) {
                VERBOSE (ERROR, fprintf (stderr, "can't write file\n"));
                exit (1);
            }
            ptout += nb;
        }
    }

    /* close all */
    fclose (fin);
    fclose (fout);

    return 0;
}

/* main function */

int main (int argc, char *argv[]) 
{
    cmode_t cmode = e_unknown;
    char *input = NULL;
    char *mode = "ansi";
    char *output = NULL;
   
    /* get basename */
    char *pt = progname = argv[0];
    while (*pt) {
        if ((*pt == '/') || (*pt == '\\')) {
           progname = pt + 1;
        }
        pt++;
    }

    int c;
    while ((c = getopt(argc, argv, "i:hm:o:v:")) != EOF) {
        switch (c) {
        case 'i':
            input = optarg;
            break;
        case 'm':
            mode = optarg;
            break;
        case 'o':
            output = optarg;
            break;
        case 'v':
            verbose = atoi (optarg);
            break;
        case 'h':
        default:
            usage (c != 'h');
        }
    }
    if (argc - optind != 0) {
        fprintf (stderr, "%s: invalid option -- %s\n", progname, argv[optind]);
        usage (1);
    }

    /* check input */
    FILE *fin = NULL;
    if (input) {
        fin = fopen (input, "rb");
        if (!fin) {
            VERBOSE (ERROR, fprintf (stderr, "error: can't open file '%s'\n", input));
        }
    } else {
        fin = stdin;
    }

    /* check output */
    FILE *fout = NULL;
    if (output) {
        fout = fopen (input, "wb");
        if (!fout) {
            VERBOSE (ERROR, fprintf (stderr, "error: can't open file '%s'\n", output));
            fclose (fin);
        }
    } else {
        fout = stdout;
    }

    /* check mode */
    if (strcmp (mode, "ansi") == 0) {
        cmode = e_ansi;
    } else if (strcmp (mode, "k&r") == 0) {
        cmode = e_kr;
    } else {
        VERBOSE (ERROR, fprintf (stderr, "error: mode '%s' unknown\n", mode));
    }

    return indent (fin, fout, cmode);
}

// test: indent.exe -h
// test: indent.exe -h | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
// test: indent.exe -_ 2> /dev/null | awk 'END { if (NR == 0) { exit(0) } else exit (1) }'
// test: indent.exe -_ 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'

/* vim: set ts=4 sw=4 et: */