51 lines
995 B
C++
51 lines
995 B
C++
#include "compiler/util/string_helpers.h"
|
|
#include <vector>
|
|
|
|
namespace tree_sitter {
|
|
namespace util {
|
|
|
|
using std::string;
|
|
using std::vector;
|
|
using std::set;
|
|
|
|
void str_replace(string *input, const string &search, const string &replace) {
|
|
size_t pos = 0;
|
|
while (1) {
|
|
pos = input->find(search, pos);
|
|
if (pos == string::npos)
|
|
break;
|
|
input->erase(pos, search.length());
|
|
input->insert(pos, replace);
|
|
pos += replace.length();
|
|
}
|
|
}
|
|
|
|
string escape_string(string input) {
|
|
str_replace(&input, "\"", "\\\"");
|
|
str_replace(&input, "\n", "\\n");
|
|
return input;
|
|
}
|
|
|
|
string escape_char(char character) {
|
|
switch (character) {
|
|
case '\0':
|
|
return "\\0";
|
|
case '"':
|
|
return "\\\"";
|
|
case '\'':
|
|
return "\\'";
|
|
case '\n':
|
|
return "\\n";
|
|
case '\r':
|
|
return "\\r";
|
|
case '\t':
|
|
return "\\t";
|
|
case '\\':
|
|
return "\\\\";
|
|
default:
|
|
return string() + character;
|
|
}
|
|
}
|
|
|
|
} // namespace util
|
|
} // namespace tree_sitter
|