universe@23: /*
universe@23:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
universe@23:  *
universe@35:  * Copyright 2016 Mike Becker. All rights reserved.
universe@23:  *
universe@23:  * Redistribution and use in source and binary forms, with or without
universe@23:  * modification, are permitted provided that the following conditions are met:
universe@23:  *
universe@23:  *   1. Redistributions of source code must retain the above copyright
universe@23:  *      notice, this list of conditions and the following disclaimer.
universe@23:  *
universe@23:  *   2. Redistributions in binary form must reproduce the above copyright
universe@23:  *      notice, this list of conditions and the following disclaimer in the
universe@23:  *      documentation and/or other materials provided with the distribution.
universe@23:  *
universe@23:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@23:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@23:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@23:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@23:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@23:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@23:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@23:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@23:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@23:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@23:  * POSSIBILITY OF SUCH DAMAGE.
universe@23:  *
universe@23:  */
universe@52: 
universe@52: #include <unistd.h>
universe@52: 
universe@22: #include "c2html.h"
universe@52: #include "highlighter.h"
universe@1: 
universe@39: #include "ucx/list.h"
universe@1: 
universe@1: void printhelp() {
universe@19:     printf("Formats source code using HTML.\n\nUsage:\n"
universe@19:         "  c2html [Options] FILE\n\n"
universe@19:         " Options:\n"
universe@19:         "  -h                    Prints this help message\n"
universe@19:         "  -j                    Highlight Java instead of C source code\n"
universe@19:         "  -o <output>           Output file (stdout, if not specified)\n"
universe@22:         "  -H <header>           Prepend header file\n"
universe@22:         "  -F <footer>           Append footer file\n"
universe@19:         "  -p                    Disable highlighting (plain text)\n"
olaf@24:         "  -l                    Disable line numbers\n"
universe@37:         "  -V, -v                Prints version and exits\n"
universe@19:         "\n");
universe@1: }
universe@1: 
universe@50: void formatlines(highlighter_func highlighter,
universe@46:         UcxList *in, write_func out, void *stream, int showlineno) {
olaf@24:     
universe@44:     /* compute width of line numbering */
universe@44:     int lnw;
universe@44:     if (showlineno) {
universe@44:         size_t lines = ucx_list_size(in);
universe@44:         lnw = 1;
universe@44:         int p = 1;
universe@44:         while ((p*=10) < lines) lnw++;
universe@44:     }
universe@44:     
universe@44:     /* start monospace formatting */
universe@44:     out("<pre>\n", 1, 6, stream);
universe@39: 
universe@44:     /* process lines */
universe@39:     size_t lineno = 0;
universe@51:     HighlighterData *hd = new_highlighter_data();
universe@51:     UcxBuffer *line = ucx_buffer_new(NULL, 1024, UCX_BUFFER_AUTOEXTEND);
universe@51:     if(!line || !hd) {
universe@51:         perror("Error allocating buffer for output");
universe@51:         return;
universe@51:     }
universe@46:     
universe@39:     UCX_FOREACH(sourceline, in) {
universe@44:         /* increase line number and clean line buffer */
universe@39:         lineno++;
universe@44:         ucx_buffer_clear(line);
universe@44:         
universe@44:         /* write line number */
universe@44:         if (showlineno) {
universe@44:             ucx_bprintf(line, "<span class=\"c2html-lineno\">"
universe@44:                     "<a name=\"l%d\" href=\"#l%d\">%*d </a></span> ",
universe@44:                     lineno, lineno, lnw, lineno);
universe@44:         }
universe@44:         
universe@45:         /* process code line */
universe@51:         highlighter(sourceline->data, line, hd);
olaf@24:         
universe@50:         /* write code line */
universe@44:         out(line->space, 1, line->size, stream);
olaf@24:     }
olaf@24:     
universe@44:     /* end monospace formatting */
universe@44:     out("</pre>\n", 1, 7, stream);
universe@44:     
universe@44:     /* cleanup and return */
universe@51:     free_highlighter_data(hd);
universe@39:     ucx_buffer_free(line);
olaf@24: }
olaf@24: 
universe@42: #define FILEBUF_SIZE 4096
universe@42: 
universe@39: enum source_type {
universe@39:     SOURCE_C,
universe@39:     SOURCE_JAVA,
universe@39:     SOURCE_PLAIN
universe@39: };
universe@39: 
universe@1: int main(int argc, char** argv) {
universe@43: 
universe@43:     /* Default settings */
universe@39:     Settings settings;
universe@22:     memset(&settings, 0, sizeof(settings));
olaf@24:     settings.showlinenumbers = 1;
universe@39:     enum source_type sourcetype = SOURCE_C;
universe@19: 
universe@43:     /* Parse command line */
universe@19:     char optc;
universe@37:     while ((optc = getopt(argc, argv, "hljo:pH:F:vV")) != -1) {
universe@19:         switch (optc) {
universe@19:             case 'o':
universe@19:                 if (!(optarg[0] == '-' && optarg[1] == 0)) {
universe@19:                     settings.outfilename = optarg;
universe@19:                 }
universe@19:                 break;
universe@22:             case 'F':
universe@22:                 settings.footerfile = optarg;
universe@22:                 break;
universe@22:             case 'H':
universe@22:                 settings.headerfile = optarg;
universe@22:                 break;
universe@19:             case 'j':
universe@39:                 sourcetype = SOURCE_JAVA;
universe@19:                 break;
universe@19:             case 'p':
universe@39:                 sourcetype = SOURCE_PLAIN;
universe@19:                 break;
olaf@24:             case 'l':
olaf@24:                 settings.showlinenumbers = 0;
olaf@24:                 break;
universe@19:             case 'h':
universe@19:                 printhelp();
universe@38:                 return EXIT_SUCCESS;
universe@37:             case 'v':
universe@37:             case 'V':
universe@37: #ifdef VERSION_DEVELOP
universe@37:                 printf("%d.%d (unstable)\n", VERSION_MAJOR, VERSION_MINOR);
universe@37: #else
universe@37:                 printf("%d.%d\n", VERSION_MAJOR, VERSION_MINOR);
universe@37: #endif
universe@38:                 return EXIT_SUCCESS;
universe@19:             default:
universe@38:                 return EXIT_FAILURE;
universe@11:         }
universe@19:     }
universe@19: 
universe@19:     if (optind != argc-1) {
universe@11:         printhelp();
universe@43:         return EXIT_FAILURE;
universe@19:     } else {
universe@47:         /* Choose highlighter */
universe@46:         highlighter_func hltr = NULL;
universe@39:         switch (sourcetype) {
universe@39:             case SOURCE_C:
universe@48:                 hltr = c_highlighter;
olaf@24:                 break;
universe@39:             case SOURCE_JAVA:
universe@48:                 hltr = java_highlighter;
olaf@24:                 break;
universe@39:             case SOURCE_PLAIN:
universe@48:                 hltr = plain_highlighter;
olaf@24:                 break;
universe@39:             default: /* should be unreachable */
universe@39:                 fprintf(stderr, "error in enum source_type\n");
universe@43:                 return EXIT_FAILURE;
olaf@24:         }
universe@42:         
universe@43:         /* Open output file */
universe@43:         settings.infilename = argv[optind];
universe@43:         FILE *fout;
universe@43:         if (settings.outfilename) {
universe@43:             fout = fopen(settings.outfilename, "w");
universe@43:             if (!fout) {
universe@43:                 perror("Error opening output file");
universe@43:                 return EXIT_FAILURE;
universe@43:             }
universe@43:         } else {
universe@43:             fout = stdout;
universe@43:         }
universe@43:         
universe@43:         /* Allocate file buffer  */
universe@43:         char *filebuf = malloc(FILEBUF_SIZE);
universe@43:         if (!filebuf) {
universe@43:             perror("Error allocating file buffer");
universe@43:             return EXIT_FAILURE;
universe@43:         }
universe@43:         
universe@43:         /* Prepend header file */
universe@42:         {
universe@42:             FILE *headerfile = fopen(settings.headerfile, "r");
universe@42:             if (!headerfile) {
universe@42:                 perror("Error opening header file");
universe@43:                 if (fout != stdout) {
universe@43:                     fclose(fout);
universe@43:                 }
universe@43:                 return EXIT_FAILURE;
universe@42:             }
universe@42:             ucx_stream_copy(headerfile, fout,
universe@42:                     (read_func) fread, (write_func) fwrite,
universe@42:                     filebuf, FILEBUF_SIZE, (size_t)-1);
universe@42:             fclose(headerfile);
universe@42:         }
universe@19: 
universe@43:         /* Process input file */
universe@39:         FILE *inputfile = fopen(settings.infilename, "r");
universe@19:         if (inputfile) {
universe@42:             UcxBuffer *content = ucx_buffer_new(NULL,
universe@42:                     FILEBUF_SIZE*2, UCX_BUFFER_AUTOEXTEND);
universe@39:             {
universe@42:                 ucx_stream_copy(inputfile, content, (read_func) fread,
universe@39:                         (write_func) ucx_buffer_write,
universe@42:                         filebuf, FILEBUF_SIZE, (size_t)-1);
universe@39:             }
universe@39:             fclose(inputfile);
universe@39:             
universe@42:             UcxList *inputlines = ucx_list_append(NULL, content->space);
universe@42:             for (size_t i = 1 ; i < content->size ; i++) {
universe@42:                 if (content->space[i] == '\r') {
universe@42:                     content->space[i] = '\n'; i++;
universe@39:                 }
universe@42:                 if (content->space[i] == '\n' && i+1 < content->size) {
universe@42:                     ucx_list_append(inputlines, content->space+i+1);
universe@39:                 }
universe@39:             }
universe@39:             
universe@46:             formatlines(hltr, inputlines,
universe@46:                     (write_func) fwrite, fout, settings.showlinenumbers);
universe@45:             
universe@42:             ucx_buffer_free(content);
universe@22:         } else {
universe@22:             perror("Error opening input file");
universe@43:             if (fout != stdout) {
universe@43:                 fclose(fout);
universe@43:             }
universe@43:             return EXIT_FAILURE;
olaf@24:         }
olaf@24:         
universe@43:         /* Append footer file */
universe@42:         {
universe@42:             FILE *footerfile = fopen(settings.footerfile, "r");
universe@42:             if (!footerfile) {
universe@42:                 perror("Error opening footer file");
universe@43:                 if (fout != stdout) {
universe@43:                     fclose(fout);
universe@43:                 }
universe@43:                 return EXIT_FAILURE;
universe@42:             }
universe@42:             ucx_stream_copy(footerfile, fout,
universe@42:                     (read_func) fread, (write_func) fwrite,
universe@42:                     filebuf, FILEBUF_SIZE, (size_t)-1);
universe@42:             fclose(footerfile);
universe@22:         }
universe@43: 
universe@22:         
universe@42:         free(filebuf);
universe@19: 
universe@43:         return EXIT_SUCCESS;
universe@11:     }
universe@1: }
universe@1: