mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-11-23 22:30:07 +00:00
print all collisions found before failing
This commit is contained in:
parent
ccfdc8dd97
commit
b456dc5342
2 changed files with 56 additions and 28 deletions
|
@ -2,16 +2,14 @@
|
|||
|
||||
from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
|
||||
from pathlib import Path
|
||||
from typing import List
|
||||
from dataclasses import dataclass
|
||||
import itertools
|
||||
import re
|
||||
|
||||
|
||||
ERROR_MESSAGE_TEMPLATE = """Found alias collision
|
||||
Alias `%s` defined in `%s` already exists as alias `%s` in `%s`.
|
||||
Consider renaming your alias.
|
||||
"""
|
||||
ERROR_MESSAGE_TEMPLATE = (
|
||||
"Alias `%s` defined in `%s` already exists as alias `%s` in `%s`."
|
||||
)
|
||||
|
||||
|
||||
def dir_path(path_string: str) -> Path:
|
||||
|
@ -42,41 +40,62 @@ class Alias:
|
|||
module: Path
|
||||
|
||||
|
||||
def find_aliases_in_file(file: Path) -> List[Alias]:
|
||||
@dataclass(frozen=True)
|
||||
class Collision:
|
||||
|
||||
existing_alias: Alias
|
||||
new_alias: Alias
|
||||
|
||||
|
||||
def find_aliases_in_file(file: Path) -> list[Alias]:
|
||||
matches = re.findall(r"^alias (.*)='(.*)'", file.read_text(), re.M)
|
||||
return [Alias(match[0], match[1], file) for match in matches]
|
||||
|
||||
|
||||
def find_all_aliases(path: Path) -> List:
|
||||
def find_all_aliases(path: Path) -> list:
|
||||
files = list(path.rglob("*.zsh"))
|
||||
aliases = [find_aliases_in_file(file) for file in files]
|
||||
return list(itertools.chain(*aliases))
|
||||
|
||||
|
||||
def check_for_duplicates(aliases: List[Alias]) -> None:
|
||||
def check_for_duplicates(aliases: list[Alias]) -> list[Collision]:
|
||||
elements = {}
|
||||
collisions = []
|
||||
for alias in aliases:
|
||||
if alias.alias in elements:
|
||||
existing = elements[alias.alias]
|
||||
raise ValueError(
|
||||
ERROR_MESSAGE_TEMPLATE
|
||||
% (
|
||||
f"{alias.alias}={alias.value}",
|
||||
alias.module.name,
|
||||
f"{existing.alias}={existing.value}",
|
||||
existing.module.name,
|
||||
)
|
||||
)
|
||||
collisions.append(Collision(existing, alias))
|
||||
else:
|
||||
elements[alias.alias] = alias
|
||||
return collisions
|
||||
|
||||
|
||||
def print_collisions(collisions: dict[Alias, Alias]) -> None:
|
||||
if collisions:
|
||||
print(f"Found {len(collisions)} alias collisions:\n")
|
||||
for collision in collisions:
|
||||
print(
|
||||
ERROR_MESSAGE_TEMPLATE
|
||||
% (
|
||||
f"{collision.new_alias.alias}={collision.new_alias.value}",
|
||||
collision.new_alias.module.name,
|
||||
f"{collision.existing_alias.alias}={collision.existing_alias.value}",
|
||||
collision.existing_alias.module.name,
|
||||
)
|
||||
)
|
||||
print("\nConsider renaming your aliases.")
|
||||
else:
|
||||
print("Found no collisions")
|
||||
|
||||
|
||||
def main():
|
||||
"""main"""
|
||||
args = parse_arguments()
|
||||
aliases = find_all_aliases(args.folder)
|
||||
check_for_duplicates(aliases)
|
||||
print("Found no collisions")
|
||||
collisions = check_for_duplicates(aliases)
|
||||
print_collisions(collisions)
|
||||
if collisions:
|
||||
exit(-1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
|
@ -7,6 +7,7 @@ from check_alias_collision import (
|
|||
find_aliases_in_file,
|
||||
check_for_duplicates,
|
||||
Alias,
|
||||
Collision,
|
||||
)
|
||||
|
||||
|
||||
|
@ -73,21 +74,29 @@ class CheckAliasCollisionTest(TestCase):
|
|||
result = find_aliases_in_file(Path("conditional.zsh"))
|
||||
self.assertListEqual([], result)
|
||||
|
||||
def test_check_for_duplicates__no_duplicates_should_not_raise(self) -> None:
|
||||
check_for_duplicates(
|
||||
def test_check_for_duplicates__no_duplicates_should_return_empty_dict(self) -> None:
|
||||
result = check_for_duplicates(
|
||||
[
|
||||
Alias("g", "git", Path("git.zsh")),
|
||||
Alias("ga", "git add", Path("git.zsh")),
|
||||
Alias("gaa", "git add --all", Path("git.zsh")),
|
||||
]
|
||||
)
|
||||
self.assertTrue(True)
|
||||
self.assertListEqual(result, [])
|
||||
|
||||
def test_check_for_duplicates__duplicates_should_raise(self) -> None:
|
||||
with self.assertRaises(ValueError):
|
||||
check_for_duplicates(
|
||||
def test_check_for_duplicates__duplicates_should_have_one_collision(self) -> None:
|
||||
result = check_for_duplicates(
|
||||
[
|
||||
Alias("gc", "git commit", Path("git.zsh")),
|
||||
Alias("gc", "git clone", Path("git.zsh")),
|
||||
]
|
||||
)
|
||||
self.assertListEqual(
|
||||
result,
|
||||
[
|
||||
Collision(
|
||||
Alias("gc", "git commit", Path("git.zsh")),
|
||||
Alias("gc", "git clone", Path("git.zsh")),
|
||||
)
|
||||
],
|
||||
)
|
||||
|
|
Loading…
Reference in a new issue