import os
import subprocess
from typing import Generator

os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"

root_path = os.path.dirname(os.path.realpath(__file__))
inputs_path = os.path.join(root_path, 'inputs')

if os.name == 'nt':
    plugin_path = os.path.join(root_path, '..', 'plugin.bat')
else:
    plugin_path = os.path.join(root_path, '..', 'plugin.py')


def get_files(path, end: str) -> Generator[str, None, None]:
    for r, dirs, files in os.walk(path):
        for filename in [f for f in files if f.endswith(end)]:
            yield os.path.join(r, filename)


def get_directories(path):
    for root, directories, files in os.walk(path):
        for directory in directories:
            yield directory


def relative(file: str, path: str):
    return os.path.join(os.path.dirname(file), path)


def read_relative(file: str, path: str):
    with open(relative(file, path)) as fh:
        return fh.read()


def protoc_plugin(path: str, output_dir: str):
    subprocess.run(
        f"protoc --plugin=protoc-gen-custom={plugin_path} --custom_out={output_dir} --proto_path={path} {path}/*.proto",
        shell=True,
    )


def protoc_reference(path: str, output_dir: str):
    subprocess.run(f"protoc --python_out={output_dir} --proto_path={path} {path}/*.proto", shell=True)