#!/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"),
]


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()