aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/tokenizer.h
blob: c223e438a91f71dac94304657a750d54d4fc5a37 (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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
// A specialized tokenizer for tokenizing the fish language. In the future, the tokenizer should be
// extended to support marks, tokenizing multiple strings and disposing of unused string segments.
#ifndef FISH_TOKENIZER_H
#define FISH_TOKENIZER_H

#include <stdbool.h>
#include <stddef.h>

#include "common.h"

/// Token types.
enum token_type {
    TOK_NONE,             /// Tokenizer not yet constructed
    TOK_ERROR,            /// Error reading token
    TOK_STRING,           /// String token
    TOK_PIPE,             /// Pipe token
    TOK_END,              /// End token (semicolon or newline, not literal end)
    TOK_REDIRECT_OUT,     /// redirection token
    TOK_REDIRECT_APPEND,  /// redirection append token
    TOK_REDIRECT_IN,      /// input redirection token
    TOK_REDIRECT_FD,      /// redirection to new fd token
    TOK_REDIRECT_NOCLOB,  /// redirection token
    TOK_BACKGROUND,       /// send job to bg token
    TOK_COMMENT           /// comment token
};

/// Tokenizer error types.
enum tokenizer_error {
    TOK_ERROR_NONE,
    TOK_UNTERMINATED_QUOTE,
    TOK_UNTERMINATED_SUBSHELL,
    TOK_UNTERMINATED_SLICE,
    TOK_UNTERMINATED_ESCAPE,
    TOK_OTHER
};

/// Flag telling the tokenizer to accept incomplete parameters, i.e. parameters with mismatching
/// paranthesis, etc. This is useful for tab-completion.
#define TOK_ACCEPT_UNFINISHED 1

/// Flag telling the tokenizer not to remove comments. Useful for syntax highlighting.
#define TOK_SHOW_COMMENTS 2

/// Flag telling the tokenizer to not generate error messages, which we need to do when tokenizing
/// off of the main thread (since wgettext is not thread safe).
#define TOK_SQUASH_ERRORS 4

/// Ordinarily, the tokenizer ignores newlines following a newline, or a semicolon. This flag tells
/// the tokenizer to return each of them as a separate END.
#define TOK_SHOW_BLANK_LINES 8

typedef unsigned int tok_flags_t;

struct tok_t {
    // The text of the token, or an error message for type error.
    wcstring text;
    // The type of the token.
    token_type type;
    // If an error, this is the error code.
    enum tokenizer_error error;
    // If an error, this is the offset of the error within the token. A value of 0 means it occurred
    // at 'offset'.
    size_t error_offset;
    // Offset of the token.
    size_t offset;
    // Length of the token.
    size_t length;

    tok_t() : type(TOK_NONE), error(TOK_ERROR_NONE), error_offset(-1), offset(-1), length(-1) {}
};

/// The tokenizer struct.
class tokenizer_t {
    // No copying, etc.
    tokenizer_t(const tokenizer_t &);
    void operator=(const tokenizer_t &);

    /// A pointer into the original string, showing where the next token begins.
    const wchar_t *buff;
    /// A copy of the original string.
    const wchar_t *orig_buff;
    /// The last token.
    wcstring last_token;
    /// Type of last token.
    enum token_type last_type;
    /// Offset of last token.
    size_t last_pos;
    /// Whether there are more tokens.
    bool has_next;
    /// Whether incomplete tokens are accepted.
    bool accept_unfinished;
    /// Whether comments should be returned.
    bool show_comments;
    /// Whether all blank lines are returned.
    bool show_blank_lines;
    /// Last error.
    tokenizer_error error;
    /// Last error offset, in "global" coordinates (relative to orig_buff).
    size_t global_error_offset;
    /// Whether we are squashing errors.
    bool squash_errors;
    /// Whether to continue the previous line after the comment.
    bool continue_line_after_comment;

    void call_error(enum tokenizer_error error_type, const wchar_t *where,
                    const wchar_t *error_message);
    void read_string();
    void read_comment();
    void tok_next();

   public:
    /// Constructor for a tokenizer. b is the string that is to be tokenized. It is not copied, and
    /// should not be freed by the caller until after the tokenizer is destroyed.
    ///
    /// \param b The string to tokenize
    /// \param flags Flags to the tokenizer. Setting TOK_ACCEPT_UNFINISHED will cause the tokenizer
    /// to accept incomplete tokens, such as a subshell without a closing parenthesis, as a valid
    /// token. Setting TOK_SHOW_COMMENTS will return comments as tokens
    tokenizer_t(const wchar_t *b, tok_flags_t flags);

    /// Returns the next token by reference. Returns true if we got one, false if we're at the end.
    bool next(struct tok_t *result);
};

/// Returns only the first token from the specified string. This is a convenience function, used to
/// retrieve the first token of a string. This can be useful for error messages, etc. On failure,
/// returns the empty string.
wcstring tok_first(const wcstring &str);

/// Helper function to determine redirection type from a string, or TOK_NONE if the redirection is
/// invalid. Also returns the fd by reference.
enum token_type redirection_type_for_string(const wcstring &str, int *out_fd = NULL);

/// Helper function to determine which fd is redirected by a pipe.
int fd_redirected_by_pipe(const wcstring &str);

/// Helper function to return oflags (as in open(2)) for a redirection type.
int oflags_for_redirection_type(enum token_type type);

enum move_word_style_t {
    move_word_style_punctuation,      // stop at punctuation
    move_word_style_path_components,  // stops at path components
    move_word_style_whitespace        // stops at whitespace
};

/// Our state machine that implements "one word" movement or erasure.
class move_word_state_machine_t {
   private:
    bool consume_char_punctuation(wchar_t c);
    bool consume_char_path_components(wchar_t c);
    bool is_path_component_character(wchar_t c);
    bool consume_char_whitespace(wchar_t c);

    int state;
    move_word_style_t style;

   public:
    explicit move_word_state_machine_t(move_word_style_t st);
    bool consume_char(wchar_t c);
    void reset();
};

#endif