155 lines
3.6 KiB
GDScript
155 lines
3.6 KiB
GDScript
class_name RuleSpacing
|
|
|
|
const SYMBOLS = [
|
|
r"\*\*=",
|
|
r"\*\*",
|
|
"<<=",
|
|
">>=",
|
|
"<<",
|
|
">>",
|
|
"==",
|
|
"!=",
|
|
">=",
|
|
"<=",
|
|
"&&",
|
|
r"\|\|",
|
|
r"\+=",
|
|
"-=",
|
|
r"\*=",
|
|
"/=",
|
|
"%=",
|
|
"&=",
|
|
r"\^=",
|
|
r"\|=",
|
|
"~=",
|
|
":=",
|
|
"->",
|
|
r"&",
|
|
r"\|",
|
|
r"\^",
|
|
"-",
|
|
r"\+",
|
|
"/",
|
|
r"\*",
|
|
">",
|
|
"<",
|
|
"-",
|
|
"%",
|
|
"=",
|
|
":",
|
|
",",
|
|
];
|
|
const KEYWORDS = [
|
|
"and",
|
|
"is",
|
|
"or",
|
|
"not",
|
|
]
|
|
|
|
|
|
static func apply(code: String) -> String:
|
|
var string_regex = RegEx.new()
|
|
string_regex.compile(r'(?<!#)("""|\'\'\'|"|\')((?:.|\n)*?)\1')
|
|
|
|
var string_matches = string_regex.search_all(code)
|
|
var string_map = {}
|
|
|
|
for i in range(string_matches.size()):
|
|
var match = string_matches[i]
|
|
var original = match.get_string()
|
|
var placeholder = "__STRING__%d__" % i
|
|
string_map[placeholder] = original
|
|
code = _replace(code, original, placeholder)
|
|
|
|
var comment_regex = RegEx.new()
|
|
comment_regex.compile("#.*")
|
|
var comment_matches = comment_regex.search_all(code)
|
|
var comment_map = {}
|
|
|
|
for i in range(comment_matches.size()):
|
|
var match = comment_matches[i]
|
|
var original = match.get_string()
|
|
var placeholder = "__COMMENT__%d__" % i
|
|
comment_map[placeholder] = original
|
|
code = _replace(code, original, placeholder)
|
|
|
|
var ref_regex = RegEx.new()
|
|
ref_regex.compile(r"\$[^.]*")
|
|
var ref_matches = ref_regex.search_all(code)
|
|
var ref_map = {}
|
|
|
|
for i in range(ref_matches.size()):
|
|
var match = ref_matches[i]
|
|
var original = match.get_string()
|
|
var placeholder = "__REF__%d__" % i
|
|
ref_map[placeholder] = original
|
|
code = _replace(code, original, placeholder)
|
|
|
|
code = _format_operators_and_commas(code)
|
|
|
|
for placeholder in ref_map:
|
|
code = code.replace(placeholder, ref_map[placeholder])
|
|
|
|
for placeholder in comment_map:
|
|
code = code.replace(placeholder, comment_map[placeholder])
|
|
|
|
for placeholder in string_map:
|
|
code = code.replace(placeholder, string_map[placeholder])
|
|
|
|
return code
|
|
|
|
|
|
static func _format_operators_and_commas(code: String) -> String:
|
|
var indent_regex = RegEx.create_from_string(r"^\s{4}")
|
|
var new_code = indent_regex.sub(code, "\t", true)
|
|
|
|
while (code != new_code):
|
|
code = new_code
|
|
new_code = indent_regex.sub(code, "\t", true)
|
|
|
|
var symbols_regex = "(" + ") | (".join(SYMBOLS) + ")"
|
|
symbols_regex = " * ?(" + symbols_regex + ") * "
|
|
var symbols_operator_regex = RegEx.create_from_string(symbols_regex)
|
|
code = symbols_operator_regex.sub(code, " $1 ", true)
|
|
|
|
# ": =" => ":="
|
|
code = RegEx.create_from_string(r": *=").sub(code, ":=", true)
|
|
|
|
# "a(" => "a ("
|
|
code = RegEx.create_from_string(r"(?<=[\w\)\]]) *([\(:,])(?!=)").sub(code, "$1", true)
|
|
|
|
# "( a" => "(a"
|
|
code = RegEx.create_from_string(r"([\(\{}]) *").sub(code, "$1", true)
|
|
|
|
# "a )" => "a)"
|
|
code = RegEx.create_from_string(r" *([\)\}])").sub(code, "$1", true)
|
|
|
|
# "if(" => "if ("
|
|
code = RegEx.create_from_string(r"\b(if|for|while|switch|match)\(").sub(code, "$1 (", true)
|
|
|
|
var keywoisrd_regex = r"|".join(KEYWORDS)
|
|
var keyword_operator_regex = RegEx.create_from_string(r"(?<=[ \)\]])(" + keywoisrd_regex + r")(?=[ \(\[])")
|
|
code = keyword_operator_regex.sub(code, " $1 ", true)
|
|
|
|
# tab "a\t=" => "a ="
|
|
code = RegEx.create_from_string(r"(\t*. * ?)\t * ").sub(code, "$1", true)
|
|
|
|
#trim
|
|
code = RegEx.create_from_string("[ \t]*\n").sub(code, "\n", true)
|
|
|
|
# " " => " "
|
|
code = RegEx.create_from_string(" +").sub(code, " ", true)
|
|
|
|
# "= -a" => "= -a"
|
|
code = RegEx.create_from_string(r"([=,(] ?)- ").sub(code, "$1-", true)
|
|
|
|
return code
|
|
|
|
|
|
static func _replace(text: String, what: String, forwhat: String) -> String:
|
|
var index := text.find(what)
|
|
|
|
if index != -1:
|
|
text = text.substr(0, index) + forwhat + text.substr(index + what.length())
|
|
|
|
return text
|