wehub-resource-sync
220 lines
7.9 KiB
Python
220 lines
7.9 KiB
Python
import pathlib
|
|
import re
|
|
import click
|
|
from packaging.version import Version
|
|
import colorama
|
|
from typing import Dict
|
|
|
|
colorama.init()
|
|
|
|
|
|
def _iter_plugin_dirs(plugins_root: pathlib.Path) -> list[pathlib.Path]:
|
|
"""Return all plugin directories."""
|
|
return list(plugins_root.glob("livekit-plugins-*"))
|
|
|
|
|
|
def _read_pypi_name(pdir: pathlib.Path) -> str:
|
|
"""Read the actual PyPI package name from pyproject.toml or setup.py."""
|
|
pyproject = pdir / "pyproject.toml"
|
|
if pyproject.exists():
|
|
m = re.search(r'^name\s*=\s*"([^"]+)"', pyproject.read_text(), re.MULTILINE)
|
|
if m:
|
|
return m.group(1)
|
|
setup_py = pdir / "setup.py"
|
|
if setup_py.exists():
|
|
m = re.search(r'name\s*=\s*"([^"]+)"', setup_py.read_text())
|
|
if m:
|
|
return m.group(1)
|
|
return pdir.name
|
|
|
|
|
|
def _esc(*codes: int) -> str:
|
|
return "\033[" + ";".join(str(c) for c in codes) + "m"
|
|
|
|
def read_version(f: pathlib.Path) -> str:
|
|
"""Read __version__ = \"X.Y.Z\" from a Python file."""
|
|
text = f.read_text()
|
|
m = re.search(r'__version__\s*=\s*[\'"]([^\'"]+)[\'"]', text)
|
|
if not m:
|
|
raise ValueError(f"could not find __version__ in {f}")
|
|
return m.group(1)
|
|
|
|
def write_new_version(f: pathlib.Path, new_version: str) -> None:
|
|
"""Substitute a new __version__ = \"X.Y.Z\" in place of the existing one."""
|
|
text = f.read_text()
|
|
new_text = re.sub(
|
|
r'__version__\s*=\s*[\'"][^\'"]*[\'"]',
|
|
f'__version__ = "{new_version}"',
|
|
text,
|
|
count=1
|
|
)
|
|
f.write_text(new_text)
|
|
|
|
def bump_version(cur: str, bump_type: str) -> str:
|
|
v = Version(cur)
|
|
if bump_type == "release":
|
|
return v.base_version
|
|
if bump_type == "patch":
|
|
return f"{v.major}.{v.minor}.{v.micro + 1}"
|
|
if bump_type == "minor":
|
|
return f"{v.major}.{v.minor + 1}.0"
|
|
if bump_type == "major":
|
|
return f"{v.major + 1}.0.0"
|
|
raise ValueError(f"unknown bump type: {bump_type}")
|
|
|
|
def bump_prerelease(cur: str, bump_type: str) -> str:
|
|
v = Version(cur)
|
|
base = v.base_version
|
|
|
|
if bump_type == "dev":
|
|
next_dev = (v.dev + 1) if v.dev is not None else 0
|
|
return f"{base}.dev{next_dev}"
|
|
elif bump_type == "rc":
|
|
if v.pre and v.pre[0] == "rc":
|
|
next_rc = v.pre[1] + 1
|
|
else:
|
|
next_rc = 1
|
|
return f"{base}.rc{next_rc}"
|
|
else:
|
|
raise ValueError(f"unknown prerelease bump type: {bump_type}")
|
|
|
|
def update_plugins_pyproject_agents_version(new_agents_version: str) -> None:
|
|
plugins_root = pathlib.Path("livekit-plugins")
|
|
for pdir in _iter_plugin_dirs(plugins_root):
|
|
pyproject = pdir / "pyproject.toml"
|
|
if pyproject.exists():
|
|
old_text = pyproject.read_text()
|
|
pattern = r'("livekit-agents(?:\[.*?\])?\s*[=><!~]+\s*)([\w\.\-]+)(?=")'
|
|
|
|
def replacer(m: re.Match) -> str:
|
|
return f"{m.group(1)}{new_agents_version}"
|
|
|
|
new_text = re.sub(pattern, replacer, old_text)
|
|
if new_text != old_text:
|
|
pyproject.write_text(new_text)
|
|
print(f"Updated pyproject.toml in {pdir.name} to use livekit-agents {new_agents_version}")
|
|
|
|
def update_agents_pyproject_optional_dependencies(plugin_versions: Dict[str, str]) -> None:
|
|
agents_pyproject = pathlib.Path("livekit-agents/pyproject.toml")
|
|
if not agents_pyproject.exists():
|
|
print("Warning: livekit-agents/pyproject.toml not found")
|
|
return
|
|
|
|
old_text = agents_pyproject.read_text()
|
|
new_text = old_text
|
|
|
|
for pypi_name, new_version in plugin_versions.items():
|
|
if pypi_name.startswith("livekit-plugins-"):
|
|
# Match any dep key that references this PyPI package name
|
|
pattern = rf'(\w[\w-]*\s*=\s*\["{re.escape(pypi_name)}>=)([\w\.\-]+)(\"])'
|
|
|
|
def replacer(m: re.Match, _new_version=new_version) -> str:
|
|
return f"{m.group(1)}{_new_version}{m.group(3)}"
|
|
|
|
updated_text = re.sub(pattern, replacer, new_text)
|
|
if updated_text != new_text:
|
|
new_text = updated_text
|
|
print(f"Updated optional dependency {pypi_name} to version {new_version}")
|
|
|
|
if new_text != old_text:
|
|
agents_pyproject.write_text(new_text)
|
|
print("Updated livekit-agents/pyproject.toml optional-dependencies")
|
|
|
|
def update_versions(bump_type: str) -> None:
|
|
agents_ver = pathlib.Path("livekit-agents/livekit/agents/version.py")
|
|
plugins_root = pathlib.Path("livekit-plugins")
|
|
|
|
new_agents_version = None
|
|
plugin_versions = {}
|
|
|
|
if agents_ver.exists():
|
|
cur = read_version(agents_ver)
|
|
new = bump_version(cur, bump_type)
|
|
print(f"livekit-agents: {_esc(31)}{cur}{_esc(0)} -> {_esc(32)}{new}{_esc(0)}")
|
|
write_new_version(agents_ver, new)
|
|
new_agents_version = new
|
|
else:
|
|
print("Warning: No version.py found for livekit-agents.")
|
|
|
|
for pdir in _iter_plugin_dirs(plugins_root):
|
|
vf = pdir / "livekit" / "plugins" / pdir.name.split("livekit-plugins-")[1].replace("-", "_") / "version.py"
|
|
pypi_name = _read_pypi_name(pdir)
|
|
if vf.exists():
|
|
cur = read_version(vf)
|
|
new = bump_version(cur, bump_type)
|
|
print(f"{pypi_name}: {_esc(31)}{cur}{_esc(0)} -> {_esc(32)}{new}{_esc(0)}")
|
|
write_new_version(vf, new)
|
|
plugin_versions[pypi_name] = new
|
|
else:
|
|
print(f"Warning: version.py not found for {pypi_name} at {vf}")
|
|
|
|
if new_agents_version:
|
|
update_plugins_pyproject_agents_version(new_agents_version)
|
|
|
|
if plugin_versions:
|
|
update_agents_pyproject_optional_dependencies(plugin_versions)
|
|
|
|
def update_prerelease(prerelease_type: str) -> None:
|
|
agents_ver = pathlib.Path("livekit-agents/livekit/agents/version.py")
|
|
plugins_root = pathlib.Path("livekit-plugins")
|
|
|
|
new_agents_version = None
|
|
plugin_versions = {}
|
|
|
|
if agents_ver.exists():
|
|
cur = read_version(agents_ver)
|
|
new = bump_prerelease(cur, prerelease_type)
|
|
print(f"livekit-agents: {_esc(31)}{cur}{_esc(0)} -> {_esc(32)}{new}{_esc(0)}")
|
|
write_new_version(agents_ver, new)
|
|
new_agents_version = new
|
|
else:
|
|
print("Warning: No version.py for livekit-agents.")
|
|
|
|
for pdir in _iter_plugin_dirs(plugins_root):
|
|
vf = pdir / "livekit" / "plugins" / pdir.name.split("livekit-plugins-")[1].replace("-", "_") / "version.py"
|
|
pypi_name = _read_pypi_name(pdir)
|
|
if vf.exists():
|
|
cur = read_version(vf)
|
|
new_v = bump_prerelease(cur, prerelease_type)
|
|
print(f"{pypi_name}: {_esc(31)}{cur}{_esc(0)} -> {_esc(32)}{new_v}{_esc(0)}")
|
|
write_new_version(vf, new_v)
|
|
plugin_versions[pypi_name] = new_v
|
|
else:
|
|
print(f"Warning: version.py not found for {pypi_name} at {vf}")
|
|
|
|
if new_agents_version:
|
|
update_plugins_pyproject_agents_version(new_agents_version)
|
|
|
|
if plugin_versions:
|
|
update_agents_pyproject_optional_dependencies(plugin_versions)
|
|
|
|
@click.command("bump")
|
|
@click.option(
|
|
"--pre",
|
|
type=click.Choice(["rc", "dev", "none"]),
|
|
default="none",
|
|
help="Pre-release type. Use 'none' for normal bump, or 'rc'/'dev' for pre-release."
|
|
)
|
|
@click.option(
|
|
"--bump-type",
|
|
type=click.Choice(["patch", "minor", "major", "release"]),
|
|
default="patch",
|
|
help="Type of version bump to apply to every package. Use 'release' to strip pre-release suffixes. Defaults to patch."
|
|
)
|
|
def bump(pre: str, bump_type: str) -> None:
|
|
"""
|
|
Single command to do either normal or pre-release bumps.
|
|
|
|
For a normal release (with --pre=none), every package is bumped with the specified
|
|
--bump-type. For pre-release bumps (--pre=rc or --pre=dev), it updates the current
|
|
versions to a new RC or DEV version. In both cases, plugin pyproject.toml references
|
|
for 'livekit-agents' will be updated if that version changes.
|
|
"""
|
|
if pre == "none":
|
|
update_versions(bump_type)
|
|
else:
|
|
update_prerelease(pre)
|
|
|
|
if __name__ == "__main__":
|
|
bump()
|