import os
from chia.plotters.bladebit import install_bladebit
from chia.plotters.madmax import install_madmax


def install_plotter(plotter, root_path):
    if plotter == "chiapos":
        print("Chiapos already installed. No action taken.")
        return
    elif plotter == "madmax":
        if not os.path.exists(root_path / "madmax-plotter/build/chia_plot"):
            print("Installing madmax plotter.")
            try:
                install_madmax(root_path)
            except Exception as e:
                print(f"Exception while installing madmax plotter: {e}")
            return
        else:
            print("Madmax plotter already installed.")
    elif plotter == "bladebit":
        if not os.path.exists(root_path / "bladebit/.bin/release/bladebit"):
            print("Installing bladebit plotter.")
            try:
                install_bladebit(root_path)
            except Exception as e:
                print(f"Exception while installing bladebit plotter: {e}")
                return
        else:
            print("Bladebit plotter already installed.")
    else:
        print("Unknown plotter. No action taken.")
        return