56 lines
1.7 KiB
Python
Executable file
56 lines
1.7 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
from pathlib import Path
|
|
import json
|
|
import itertools
|
|
import subprocess
|
|
from io import StringIO
|
|
|
|
VSCODE_CONFIG_FILE = Path.home().joinpath(".config/Code/storage.json")
|
|
|
|
SEARCH_DIRS = [
|
|
Path.home().joinpath("Projects"),
|
|
Path.home().joinpath("Repositories"),
|
|
Path.home().joinpath("SR"),
|
|
]
|
|
|
|
|
|
def get_search_project_dirs():
|
|
for path in SEARCH_DIRS:
|
|
if not path.exists():
|
|
continue
|
|
for subdir in path.iterdir():
|
|
if subdir.is_dir():
|
|
yield subdir
|
|
|
|
|
|
def get_vscode_project_dirs():
|
|
vscode_config = json.loads(VSCODE_CONFIG_FILE.read_text())
|
|
|
|
for recent_path in itertools.chain(*vscode_config['openedPathsList'].values()):
|
|
if not recent_path:
|
|
continue
|
|
recent_path_path = Path(recent_path.replace('file://', ''))
|
|
if recent_path_path.exists():
|
|
yield recent_path_path
|
|
|
|
|
|
def main():
|
|
project_paths = sorted(set(itertools.chain(get_search_project_dirs(), get_vscode_project_dirs())), key=lambda p: p.name.lower())
|
|
try:
|
|
project_paths.remove(Path.home()) # Don't try and edit home dir
|
|
except ValueError:
|
|
pass
|
|
project_paths_display = [str(project.relative_to(Path.home())) for project in project_paths]
|
|
|
|
selected_project = subprocess.run(["rofi", "-dmenu", "-no-case-sensitive", "-format", "i"], input="\n".join(project_paths_display).encode(), stdout=subprocess.PIPE)
|
|
selected_project.check_returncode()
|
|
|
|
selected_index = int(selected_project.stdout.strip())
|
|
if selected_index == -1:
|
|
return # This is hit when enter is hit on no results
|
|
subprocess.run(['code', project_paths[selected_index]])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|