summaryrefslogtreecommitdiff
path: root/src/commands/echo.c
blob: 8b3953110be4fb6da54cac4af9441091fcdc5931 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#include "../command.h"

static struct {
    bool escape_codes;
    bool newline;
} flags;

static void print_with_escape_codes(const char* str) {
    
    size_t index = 0;
    while (true) {
        char c = str[index];
        index++;

        if (c == '\0') break;
        if (c != '\\') {
            putchar(c);
            continue;
        }

        char n = str[index];
        index++;

        switch (n) {
            case '\\':
                putchar('\\');
                break;
            case 'b':
                putchar('\b');
                break;
            case 'c':
                exit(EXIT_SUCCESS);
            case 'n':
                putchar('\n');
                break;
            case 'r':
                putchar('\r');
                break;
            case 't':
                putchar('\t');
                break;
            case 'v':
                putchar('\v');
                break;
            default:
                putchar(c);
                putchar(n);
        }
    }
}

static int short_arg(char c, char* next) {
    UNUSED(next);
    switch (c) {
        case 'e':
            flags.escape_codes = true;
            break;
        case 'E':
            flags.escape_codes = false;
            break;
        case 'n':
            flags.newline = false;
            break;
        default:
            flags.newline = true;
            flags.escape_codes = false;
            return ARG_IGNORE;
    };
    return ARG_UNUSED;
}

COMMAND(echo) {

    if (argc < 1) {
        return EXIT_SUCCESS;
    }

    flags.escape_codes = false;
    flags.newline = true;

    int start = parse_args(argc, argv, NULL, short_arg, NULL);

    for (int i = start; i < argc; i++) {
         if (flags.escape_codes) {
            print_with_escape_codes(argv[i]);
        } else {
            printf("%s", argv[i]);
        }

        if (i + 1 != argc) {
            putchar(' ');
        }
    } 

    if (flags.newline) {
        putchar('\n');
    }

    return EXIT_SUCCESS;
}