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
|
#!/usr/bin/env python3
import sys
import json
import os
pkt_name = sys.argv[1]
operation = sys.argv[2]
product_path = "/usr/lib/code/product.json"
patch_path = "/usr/share/%s/patch.json" % pkt_name
cache_path = "/usr/share/%s/cache.json" % pkt_name
class term_colors:
WARNING = "\033[93m"
ENDC = "\033[0m"
if not os.path.exists(product_path):
print(
term_colors.WARNING
+ "WARN: "
+ term_colors.ENDC
+ product_path
+ " does not exist. You need to install extra/code in the official repository to use this package. Skipping..."
)
exit(0)
if not os.path.exists(cache_path):
with open(cache_path, "w") as file:
file.write("{}")
def fix_sign(undo=False):
"""Fix the error of failed signature verification."""
path = (
"/usr/lib/code/out/vs/code/electron-utility/sharedProcess/sharedProcessMain.js"
)
orig_text = 'import("node-ovsx-sign")'
fixed_text = 'import("@vscode/vsce-sign")'
if undo:
search_text = fixed_text
replace_text = orig_text
else:
search_text = orig_text
replace_text = fixed_text
try:
with open(path, "r", encoding="utf-8") as f:
content = f.read()
if search_text not in content:
return
new_content = content.replace(search_text, replace_text)
with open(path, "w", encoding="utf-8") as f:
_ = f.write(new_content)
except Exception as e:
print(f"[code-marketplace] ERROR: {e}")
def patch():
with open(product_path, "r") as product_file:
product_data = json.load(product_file)
with open(patch_path, "r") as patch_file:
patch_data = json.load(patch_file)
cache_data = {}
for key in patch_data.keys():
if key in product_data:
cache_data[key] = product_data[key]
product_data[key] = patch_data[key]
with open(product_path, "w") as product_file:
json.dump(product_data, product_file, indent="\t")
with open(cache_path, "w") as cache_file:
json.dump(cache_data, cache_file, indent="\t")
fix_sign(False)
def restore():
with open(product_path, "r") as product_file:
product_data = json.load(product_file)
with open(patch_path, "r") as patch_file:
patch_data = json.load(patch_file)
with open(cache_path, "r") as cache_file:
cache_data = json.load(cache_file)
for key in patch_data.keys():
if key in product_data:
del product_data[key]
for key in cache_data.keys():
product_data[key] = cache_data[key]
with open(product_path, "w") as product_file:
json.dump(product_data, product_file, indent="\t")
fix_sign(True)
if operation == "patch":
patch()
elif operation == "restore":
restore()
|