chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,219 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user