import os
import re
import sys
from typing import Callable, Set
v = sys.argv[1]
tag = sys.argv[2]
our_crates = [
"chia-bls",
"clvm-traits",
"chia-traits",
"chia_py_streamable_macro",
"chia_streamable_macro",
"chia-protocol",
"chia-tools",
"clvm-utils",
"clvm-derive",
"chia-wallet",
"chia-client",
"chia-ssl",
"fuzz",
"chia-wallet/fuzz",
"clvm-utils/fuzz",
]
def crates_with_changes() -> Set[str]:
ret = set()
for c in our_crates:
diff = os.popen(f"git diff {tag} -- {c}").read().strip()
if len(diff) > 0:
ret.add(c)
ret.add("wheel")
return ret
def update_cargo(name: str, crates: Set[str]) -> None:
subst = ""
with open(f"{name}/Cargo.toml") as f:
for line in f:
split = line.split()
if split == []:
subst += line
continue
if split[0] == "version" and name in crates:
line = f'version = "{v}"\n'
elif split[0] in crates:
line = re.sub('version = "(>?=?)\d+\.\d+\.\d+"', f'version = "\\g<1>{v}"', line)
subst += line
with open(f"{name}/Cargo.toml", "w") as f:
f.write(subst)
crates = crates_with_changes()
crates.add(".")
crates.add("chia")
print("bumping version of crates:")
for c in crates:
print(f" - {c}")
for c in our_crates:
update_cargo(c, crates)
update_cargo(".", crates)
update_cargo("wheel", crates)