152 lines
3.8 KiB
Python
Executable file
152 lines
3.8 KiB
Python
Executable file
#!/usr/bin/python
|
|
|
|
# Author: @xstraok
|
|
# Modified by Observer KRypt0n_ <https://github.com/krypt0nn>
|
|
|
|
import os
|
|
import sys
|
|
import glob
|
|
import re
|
|
|
|
valid_commands = ["diff", "unused", "missing"]
|
|
|
|
if len(sys.argv) < 3:
|
|
print(f"Command format: ./fluentscan.py [command] [locale]\nAvailable commands: {valid_commands}")
|
|
|
|
sys.exit()
|
|
|
|
if sys.argv[1] not in valid_commands:
|
|
print(f"Invalid command \"{sys.argv[1]}\". Available commands: {valid_commands}")
|
|
|
|
sys.exit()
|
|
|
|
path = "assets/locales/" + sys.argv[2] + "/"
|
|
|
|
try:
|
|
open(path + "/main.ftl", "r").close()
|
|
|
|
except:
|
|
print(f"{path} does not exist")
|
|
|
|
sys.exit()
|
|
|
|
all_entries = {}
|
|
|
|
def dict_compare(d1, d2):
|
|
d1_keys = set(d1.keys())
|
|
d2_keys = set(d2.keys())
|
|
|
|
shared_keys = d1_keys.intersection(d2_keys)
|
|
|
|
added = d1_keys - d2_keys
|
|
removed = d2_keys - d1_keys
|
|
|
|
same = set(o for o in shared_keys if d1[o] == d2[o])
|
|
|
|
return added, removed, same
|
|
|
|
def to_dict(text):
|
|
result={}
|
|
|
|
for i in text:
|
|
if " =" in i:
|
|
try:
|
|
result[i.split()[0]] = ' '.join(i.split()[2:])
|
|
|
|
except:
|
|
pass
|
|
|
|
elif i:
|
|
result[list(result.keys())[-1]] += i
|
|
|
|
return result
|
|
|
|
def get_line_num(text,pattern):
|
|
line = 1
|
|
|
|
for i in text.split("\n"):
|
|
if pattern in i:
|
|
return line
|
|
|
|
line += 1
|
|
|
|
for filename in os.listdir("assets/locales/en"):
|
|
with open(os.path.join("assets/locales/en", filename), 'r') as locale_file:
|
|
created_locale = open(path + filename)
|
|
|
|
expected = to_dict(locale_file)
|
|
expected2 = to_dict(created_locale)
|
|
|
|
all_entries.update(expected)
|
|
|
|
added, removed, same = dict_compare(expected, expected2)
|
|
|
|
if sys.argv[1] == "unused" or sys.argv[1] == "missing":
|
|
files = glob.glob("src/**/*.rs", recursive=True)
|
|
|
|
used = []
|
|
vars = {}
|
|
|
|
for i in files:
|
|
with open(i, "r") as script:
|
|
text = script.read()
|
|
|
|
if sys.argv[1] == "unused":
|
|
for j in expected:
|
|
if f"\"{j}\"" in text:
|
|
used.append(j)
|
|
|
|
elif sys.argv[1] == "missing":
|
|
for j in text.split():
|
|
# TODO: ignore comments
|
|
if 'tr("' in j:
|
|
index = j.find('tr("')
|
|
|
|
var_name = re.sub('[^\\w-]+', '', j[index:].replace('tr("', '').replace("Some", ""))
|
|
|
|
# TODO: index multiple matches
|
|
vars[var_name] = [script.name, get_line_num(text,var_name)]
|
|
|
|
if sys.argv[1] == "unused":
|
|
for i in expected:
|
|
if i not in used:
|
|
print(f"[{locale_file.name}]\n"
|
|
" [Unused]\n"
|
|
f" {i}")
|
|
|
|
continue
|
|
|
|
if (added or removed or same) and sys.argv[1] == "diff":
|
|
print(f"[{created_locale.name[15:]}]")
|
|
|
|
if added:
|
|
print(" [Added]")
|
|
|
|
for i in added:
|
|
print(f" {i} = {expected[i]}")
|
|
|
|
if removed:
|
|
print(" [Removed]")
|
|
|
|
for i in removed:
|
|
print(f" {i} = {expected2[i]}")
|
|
|
|
#workaround
|
|
if same and same != "set()":
|
|
print(" [Untranslated]")
|
|
|
|
for i in same:
|
|
print(f" {i} = {expected[i]}")
|
|
|
|
print("")
|
|
|
|
if sys.argv[1] == "missing":
|
|
added, removed, same = dict_compare(vars, all_entries)
|
|
|
|
if not added:
|
|
print("Nothing is missing")
|
|
|
|
for i in added:
|
|
print(f"[{vars[i][0]}, line {vars[i][1]}]\n"
|
|
" [Missing]\n"
|
|
f" {i}")
|