package youtubedl

import (
	"errors"
	"os/exec"
)

func install() error {
	if err := installPyPip3(); err != nil {
		return err
	}
	if err := installYtdl(); err != nil {
		return err
	}
	if err := installVTT(); err != nil {
		return err
	}
	return nil
}

func installPyPip3() error {
	hasPy := exec.Command("python3", "--version")
	hasPip := exec.Command("pip3", "--version")
	if err := hasPy.Run(); err != nil {
	} else if err := hasPip.Run(); err != nil {
	} else {
		return nil
	}
	for _, combo := range [][]string{
		[]string{"sudo", "apt", "install", "python3", "pip3"},
		[]string{"apt", "install", "python3", "pip3"},
		[]string{"apk", "add", "python3", "py3-pip"},
		[]string{"sudo", "apk", "add", "python3", "py3-pip"},
	} {
		cmd := exec.Command(combo[0], combo[1:]...)
		if err := cmd.Run(); err == nil {
			return nil
		}
	}
	return errors.New("cannot get python3 and pip3")
}

func installYtdl() error {
	cmd := exec.Command("youtube-dl", "--version")
	if err := cmd.Run(); err == nil {
		return err
	}
	cmd = exec.Command("sudo", "pip3", "install", "youtube-dl")
	if err := cmd.Run(); err != nil {
		cmd = exec.Command("pip3", "install", "youtube-dl")
		if err := cmd.Run(); err != nil {
			return err
		}
	}
	return nil
}

func installVTT() error {
	cmd := exec.Command("python3", "-m", "vtt_to_srt")
	if err := cmd.Run(); err == nil {
		return err
	}
	cmd = exec.Command("sudo", "pip3", "install", "vtt_to_srt3")
	if err := cmd.Run(); err != nil {
		cmd = exec.Command("pip3", "install", "vtt_to_srt3")
		if err := cmd.Run(); err != nil {
			return err
		}
	}
	return nil
}