Homework 06: trit

homework06/trit.c

@@ -5,15 +5,16 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <ctype.h> // For isspace
 
 /* Constants */
 
 enum {
-    LOWER   = 1<<1,
-    UPPER   = 0,    // TODO: Modify
-    TITLE   = 0,    // TODO: Modify
-    STRIP   = 0,    // TODO: Modify
-    DELETE  = 0,    // TODO: Modify
+    LOWER    = 1<<1,
+    UPPER    = 1<<2,
+    TITLE    = 1<<3,
+    STRIP    = 1<<4,
+    DELETE   = 1<<5,
 };
 
 /* Functions */
@@ -21,26 +22,100 @@ enum {
 void usage(int status) {
     fprintf(stderr, "Usage: trit SET1 SET2\n\n");
     fprintf(stderr, "Post Translation filters:\n\n");
-    fprintf(stderr, "   -l      Convert to lowercase\n");
-    fprintf(stderr, "   -u      Convert to uppercase\n");
-    fprintf(stderr, "   -t      Convert to titlecase\n");
-    fprintf(stderr, "   -s      Strip trailing whitespace\n");
-    fprintf(stderr, "   -d      Delete letters in SET1\n");
+    fprintf(stderr, "  -l      Convert to lowercase\n");
+    fprintf(stderr, "  -u      Convert to uppercase\n");
+    fprintf(stderr, "  -t      Convert to titlecase\n");
+    fprintf(stderr, "  -s      Strip trailing whitespace\n");
+    fprintf(stderr, "  -d      Delete letters in SET1\n");
     exit(status);
 }
 
 void translate_stream(FILE *stream, const char *set1, const char *set2, int flags) {
-    // TODO
+    char line[256];
+
+    while (fgets(line, sizeof(line), stream) != NULL) {
+        // Remove newline char if present
+        size_t len = strlen(line);
+        if (len > 0 && line[len - 1] == '\n') {
+            line[len - 1] = '\0';
+        }
+
+        // Translate or delete part of string
+        char processed_line[1024];
+        strcpy(processed_line, line);
+
+        if (flags & DELETE) {
+            str_delete(processed_line, set1, line);
+            strcpy(processed_line, line);
+            set2 = "";
+        } else if (set1 != NULL && set2 != NULL) {
+            str_translate(processed_line, set1, set2, line);
+            strcpy(processed_line, line);
+        }
+
+        // Filters
+        if (flags & LOWER) {
+            str_lower(processed_line, line);
+            strcpy(processed_line, line);
+        }
+
+        if (flags & UPPER) {
+            str_upper(processed_line, line);
+            strcpy(processed_line, line);
+        }
+
+        if (flags & TITLE) {
+            str_title(processed_line, line);
+            strcpy(processed_line, line);
+        }
+
+        if (flags & STRIP) {
+            str_rstrip(processed_line, " \t\r\f\v", line); // all the whitespace characters I can think of (has no newline)
+            strcpy(processed_line, line);
+        }
+
+        printf("%s\n", processed_line);
+    }
 }
 
 /* Main Execution */
 
 int main(int argc, char *argv[]) {
-    // TODO: Parse command line arguments
+    int flags = 0;
+    char *set1 = NULL;
+    char *set2 = NULL;
+    int positional_args_count = 0;
 
-    // TODO: Translate standard input
-    return EXIT_SUCCESS;
+    for (int i = 1; i < argc; i++) {
+        if (strcmp(argv[i], "-l") == 0) {
+            flags |= LOWER;
+        } else if (strcmp(argv[i], "-u") == 0) {
+            flags |= UPPER;
+        } else if (strcmp(argv[i], "-t") == 0) {
+            flags |= TITLE;
+        } else if (strcmp(argv[i], "-d") == 0) {
+            flags |= DELETE;
+        } else if (strcmp(argv[i], "-s") == 0) {
+            flags |= STRIP;
+        } else if (strcmp(argv[i], "-h") == 0) {
+            usage(0);
+        } else {
+            positional_args_count++;
+            if (positional_args_count == 1) {
+                set1 = argv[i];
+            } else if (positional_args_count == 2) {
+                set2 = argv[i];
+            } else {
+                usage(1);
+            }
+        }
+    }
+
+    translate_stream(stdin, set1, set2, flags);
+
+    return 0;
 }
+/* Main Execution */
 
 /* vim: set sts=4 sw=4 ts=8 expandtab ft=c: */