Initial commit

1 files changed, 226 insertions, 0 deletions

diff --git a/main.c b/main.c
new file mode 100644
index 0000000..05c274d
--- /dev/null
+++ b/main.c
@@ -0,0 +1,226 @@
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <stdarg.h>
+#include <unistd.h>
+#include "fierrhea.h"
+#include "romero.h"
+#include "jenkins.h"
+
+#define INPUT_BUFSIZ 4096
+
+size_t START_Y = 0;
+const char *A_TAB = "    ";
+
+void repchar(char ch, size_t limit) {
+    while (limit > 0) {
+        putc(ch, stdout);
+        limit--;
+    }
+}
+
+size_t get_longest_line(char *s) {
+    size_t lengths[INPUT_BUFSIZ] = {0};
+    size_t len;
+    size_t line;
+
+    len = 0;
+    line = 0;
+    for (size_t i = 0; i < strlen(s); i++) {
+        if (s[i] == '\n') {
+            lengths[line] = len;
+            line++;
+            len = 0;
+            continue;
+        }
+        len++;
+    }
+
+    size_t longest;
+
+    longest = lengths[0];
+    for (size_t i = 0; i < sizeof(lengths) / sizeof(*lengths); i++) {
+        if (lengths[i] > longest) {
+            longest = lengths[i];
+        }
+    }
+    return longest;
+}
+
+void box_draw_top(size_t y, size_t longest) {
+    repchar(' ', y);
+    repchar('/', 1);
+    repchar('-', longest);
+    repchar('\\', 1);
+}
+
+void box_draw_bottom(size_t y, size_t longest) {
+    repchar(' ', y);
+    repchar('\\', 1);
+    repchar('-', longest);
+    repchar('/', 1);
+}
+
+void box_draw_next_line(size_t y) {
+    repchar(' ', y);
+    repchar('|', 1);
+}
+
+void box_draw_end_line(size_t longest, size_t len) {
+    repchar(' ', longest - len);
+    repchar('|', 1);
+}
+
+int boxprintf(const char *fmt, ...) {
+    int count;
+    va_list list;
+    va_start(list, fmt);
+    size_t len = 0;
+    char data[INPUT_BUFSIZ] = {0};
+    char output[INPUT_BUFSIZ] = {0};
+    vsnprintf(data, sizeof(data) - 1, fmt, list);
+
+    // convert tabs to spaces
+    for (size_t i = 0, n = 0; i < strlen(data) && n < sizeof(output); i++, n++) {
+        if (data[i] == '\t') {
+            strcat(&output[n], A_TAB);
+            n = n + strlen(A_TAB) - 1;
+            continue;
+        }
+        output[n] = data[i];
+    }
+
+    size_t longest = get_longest_line(output);
+
+    box_draw_top(START_Y, longest);
+    puts("");
+
+    for (size_t i = 0; i < strlen(output); i++) {
+        if (i == strlen(output) - 1) {
+            box_draw_end_line(longest, len);
+            continue;
+        }
+
+        if (len == 0) {
+            box_draw_next_line(START_Y);
+        }
+
+        if (output[i] == '\n') {
+            box_draw_end_line(longest, len);
+            putc('\n', stdout);
+            len = 0;
+            continue;
+        }
+
+        putc(output[i], stdout);
+
+        len++;
+    }
+    puts("");
+    box_draw_bottom(START_Y, longest);
+
+    va_end(list);
+    return count;
+}
+
+void driver_fierrhea(char *input) {
+    START_Y = 22;
+    printf("%s", FIERRHEA);
+    printf("\\  /\n");
+    printf("                      \\/\n");
+    boxprintf(input);
+    puts("");
+}
+
+void driver_jenkins(char *input) {
+    START_Y = 18;
+    boxprintf(input);
+    puts("");
+    printf("                   |  /\n");
+    printf("                   | /\n");
+    printf("                   |/\n");
+    puts(JENKINS);
+}
+
+void driver_romero(char *input) {
+    START_Y = 11;
+    boxprintf(input);
+    puts("");
+    printf("            \\  |\n");
+    printf("             \\ |\n");
+    printf("              \\|\n");
+    puts(ROMERO);
+}
+
+void usage(char *prog) {
+    printf("usage: %s [-d driver_name] [-] {input}\n"
+           "-h      Show this help message\n"
+           "-d      Driver name (fierrhea, jenkins, romero)\n"
+           "\n"
+           "-       Read from standard input\n"
+           "input   A quoted string\n", prog);
+}
+
+int main(int argc, char *argv[]) {
+    char driver_name[255] = {0};
+    char input[INPUT_BUFSIZ] = {0};
+    char *iptr = input;
+
+    // set default driver
+    strcpy(driver_name, "fierrhea");
+
+    int option;
+    while ((option = getopt (argc, argv, "hd:")) != -1) {
+        switch (option) {
+            case 'h':
+                usage(argv[0]);
+                exit(0);
+            case 'd':
+                strcpy(driver_name, optarg);
+                break;
+            case ':':
+                fprintf(stderr, "option requires value\n");
+                usage(argv[0]);
+                exit(1);
+                break;
+            case '?':
+                fprintf(stderr, "unknown option: %c\n", optopt);
+                usage(argv[0]);
+                exit(1);
+                break;
+            default:
+                usage(argv[0]);
+                exit(1);
+        }
+    }
+
+    if (optind < argc) {
+        strcpy(input, argv[optind]);
+    } else {
+        fprintf(stderr, "Missing input string or standard input (`-`)\n");
+        usage(argv[0]);
+        exit(1);
+    }
+
+    if (input[0] == '-') {
+       fread(input, 1, sizeof(input) - 1, stdin);
+    }
+
+    if (input[strlen(input) - 1] != '\n') {
+        input[strlen(input)] = '\n';
+    }
+
+    if (!strcmp(driver_name, "fierrhea")) {
+        driver_fierrhea(input);
+    } else if (!strcmp(driver_name, "jenkins")) {
+        driver_jenkins(input);
+    } else if (!strcmp(driver_name, "romero")) {
+        driver_romero(input);
+    } else {
+        fprintf(stderr, "invalid driver: '%s'\n", driver_name);
+        exit(1);
+    }
+
+    puts("");
+    return 0;
+}