From e57d61d2759702c574fbc8839da8527efb2ffbff Mon Sep 17 00:00:00 2001
From: bel <me@me.me>
Date: Wed, 29 Jan 2020 04:11:19 +0000
Subject: [PATCH] Initial commit

---
 youtubedl/install.go | 50 ++++++++++++++++++++++++++++++++++++--------
 1 file changed, 41 insertions(+), 9 deletions(-)

diff --git a/youtubedl/install.go b/youtubedl/install.go
index fe905a4..2d87ea5 100644
--- a/youtubedl/install.go
+++ b/youtubedl/install.go
@@ -1,27 +1,56 @@
 package youtubedl
 
-import "os/exec"
+import (
+	"errors"
+	"os/exec"
+)
 
 func install() error {
-	cmd := exec.Command("youtube-dl", "--version")
-	if err := cmd.Run(); err == nil {
+	if err := installPyPip3(); err != nil {
 		return err
 	}
-	cmd = exec.Command("sudo", "pip3", "install", "youtube-dl")
-	if err := cmd.Run(); err == nil {
+	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 {
-		return err
+	if err := cmd.Run(); err != nil {
+		cmd = exec.Command("pip3", "install", "youtube-dl")
+		if err := cmd.Run(); err != nil {
+			return err
+		}
 	}
 	return nil
 }
@@ -32,8 +61,11 @@ func installVTT() error {
 		return err
 	}
 	cmd = exec.Command("sudo", "pip3", "install", "vtt_to_srt3")
-	if err := cmd.Run(); err == nil {
-		return err
+	if err := cmd.Run(); err != nil {
+		cmd = exec.Command("pip3", "install", "vtt_to_srt3")
+		if err := cmd.Run(); err != nil {
+			return err
+		}
 	}
 	return nil
 }