zed/script/move-crate.py
2024-11-17 19:04:15 +01:00

108 lines
3.5 KiB
Python
Executable file

import sys
import os
import shutil
crate_name = sys.argv[1]
src_path = f"crates/{crate_name}/src"
dest_path = f"crates/zed_common/src/{crate_name}"
# Move the crate to zed_common
if os.path.exists(src_path):
shutil.move(src_path, dest_path)
print("Moved code to zed_common")
else:
print(f"Source path {src_path} does not exist")
src_mod_file = f"{dest_path}/{crate_name}.rs"
# Rename crate_name.rs to mod.rs
if os.path.exists(src_mod_file):
os.rename(src_mod_file, f"{dest_path}/mod.rs")
print("Created new mod.rs file")
else:
print(f"Source path {dest_path}.rs does not exist")
print("Fixing imports...")
# Look through all the text files, scan for use crate:: and replace with use crate::{crate_name}
for root, dirs, files in os.walk(dest_path):
for file in files:
if file.endswith(".rs"):
with open(os.path.join(root, file), "r") as f:
content = f.read()
new_content = content.replace("crate::", f"crate::{crate_name}::")
if new_content != content:
with open(os.path.join(root, file), "w") as f:
f.write(new_content)
print("Fixed imports")
# Remove crate from workspace file
with open("Cargo.toml", "r") as f:
lines = f.readlines()
new_lines = [line for line in lines if not (f"crates/{crate_name}\"" in line or f"{crate_name}.workspace = true" in line)]
if len(new_lines) != len(lines):
with open("Cargo.toml", "w") as f:
f.writelines(new_lines)
print(f"Removed {crate_name} from workspace file")
else:
print(f"Crate {crate_name} not found in workspace file")
# Check all the Cargo.toml files for {crate_name}.workspace = true and remove line
for root, dirs, files in os.walk("."):
if "Cargo.toml" in files:
cargo_toml_path = os.path.join(root, "Cargo.toml")
with open(cargo_toml_path, "r") as f:
lines = f.readlines()
new_lines = [line for line in lines if f"{crate_name}.workspace = true" not in line]
if len(new_lines) != len(lines):
with open(cargo_toml_path, "w") as f:
f.writelines(new_lines)
print(f"Removed {crate_name}.workspace = true from {cargo_toml_path}")
# Add module to lib.rs file in the correct ordering
with open("crates/zed_common/src/lib.rs", "r") as f:
lines = f.readlines()
mod_line = f"pub mod {crate_name};\n"
if mod_line not in lines:
# Find the correct position to insert the new module
insert_index = 0
for i, line in enumerate(lines):
if line.startswith("pub mod"):
if line > mod_line:
insert_index = i
break
else:
insert_index = i + 1
lines.insert(insert_index, mod_line)
with open("crates/zed_common/src/lib.rs", "w") as f:
f.writelines(lines)
print(f"Added {crate_name} module to lib.rs")
else:
print(f"Module {crate_name} already exists in lib.rs")
# Print the Cargo.toml content
cargo_toml_path = f"crates/{crate_name}/Cargo.toml"
if os.path.exists(cargo_toml_path):
with open(cargo_toml_path, "r") as f:
print(f"Contents of {cargo_toml_path}:")
print("-----------------------------")
print(f.read())
print("-----------------------------")
else:
print(f"Cargo.toml for {crate_name} not found")
# Remove the src path and print the cargo toml
if os.path.exists(f"crates/{crate_name}"):
shutil.rmtree(f"crates/{crate_name}")
print(f"Removed src directory from {crate_name}")
else:
print(f"Source directory for {crate_name} not found")