dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 16:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: implement Knight Rider style spinner animation

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-05-08 14:08:06 -07:00
parent 7a3805d39f
commit 8f0fa6684d
1 changed files with 42 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

62
aider/utils.py
View file

@ -1,4 +1,3 @@
import itertools
import os import os
import platform import platform
import oslex import oslex
@ -251,29 +250,34 @@ def run_install(cmd):
class Spinner: class Spinner:
unicode_spinner = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
ascii_spinner = ["|", "/", "-", "\\"]
def __init__(self, text): def __init__(self, text):
self.text = text self.text = text
self.start_time = time.time() self.start_time = time.time()
self.last_update = 0 self.last_update = 0
self.visible = False self.visible = False
self.is_tty = sys.stdout.isatty() self.is_tty = sys.stdout.isatty()
self.tested = False
def test_charset(self): self.scanner_width = 7 # Width of the scanning area (e.g., 7 for "[---■--]")
if self.tested: self.scanner_pos = 0
return self.scanner_dir = 1 if self.scanner_width > 1 else 0 # Direction of scanner movement
self.tested = True
# Try unicode first, fall back to ascii if needed # Attempt to use Unicode characters for the scanner, fallback to ASCII
try: self.scanner_char = "#" # ASCII fallback
# Test if we can print unicode characters self.trail_char = "-" # ASCII fallback
print(self.unicode_spinner[0], end="", flush=True) if self.is_tty:
print("\r", end="", flush=True) try:
self.spinner_chars = itertools.cycle(self.unicode_spinner) # Test print Unicode chars and erase them to check encoding support
except UnicodeEncodeError: sys.stdout.write("■─")
self.spinner_chars = itertools.cycle(self.ascii_spinner) sys.stdout.flush()
sys.stdout.write("\b\b \b\b") # Backspace, write spaces, backspace
sys.stdout.flush()
self.scanner_char = "■" # Unicode
self.trail_char = "─" # Unicode
except UnicodeEncodeError:
pass # Stick to ASCII fallbacks
except Exception: # Catch other potential IO errors on strange terminals
pass
self.animation_len = self.scanner_width + 2 # For brackets like "[]"
def step(self): def step(self):
if not self.is_tty: if not self.is_tty:
@ -291,12 +295,30 @@ class Spinner:
if not self.visible: if not self.visible:
return return
self.test_charset() frame = [self.trail_char] * self.scanner_width
print(f"\r{self.text} {next(self.spinner_chars)}\r{self.text} ", end="", flush=True) # Ensure scanner_pos is within bounds for frame assignment
current_pos_in_frame = max(0, min(self.scanner_pos, self.scanner_width - 1))
if self.scanner_width > 0: # Only place char if width is positive
frame[current_pos_in_frame] = self.scanner_char
animation_segment = f"[{''.join(frame)}]"
print(f"\r{self.text} {animation_segment}", end="", flush=True)
# Update scanner position for the next frame
if self.scanner_width > 1:
self.scanner_pos += self.scanner_dir
if self.scanner_pos >= self.scanner_width - 1: # Reached or passed the end
self.scanner_pos = self.scanner_width - 1 # Pin to end
self.scanner_dir = -1 # Reverse direction
elif self.scanner_pos <= 0: # Reached or passed the beginning
self.scanner_pos = 0 # Pin to start
self.scanner_dir = 1 # Reverse direction
def end(self): def end(self):
if self.visible and self.is_tty: if self.visible and self.is_tty:
print("\r" + " " * (len(self.text) + 3)) clear_len = len(self.text) + 1 + self.animation_len # text + space + animation segment
print("\r" + " " * clear_len + "\r", end="", flush=True) # Clear line and reset cursor
self.visible = False
def find_common_root(abs_fnames): def find_common_root(abs_fnames):