Added ffmpeg detection

mod/agi/moduleManager.go

@@ -2,8 +2,10 @@ package agi
 
 import (
 	"errors"
+	"log"
 
 	"imuslab.com/arozos/mod/agi/static"
+	apt "imuslab.com/arozos/mod/apt"
 )
 
 /*
@@ -50,5 +52,13 @@ func (g *Gateway) LoadAllFunctionalModules() {
 	g.ShareLibRegister()
 	g.IoTLibRegister()
 	g.AppdataLibRegister()
-	g.FFmpegLibRegister()
+
+	//Only register ffmpeg lib if host OS have ffmpeg installed
+	ffmpegExists, _ := apt.PackageExists("ffmpeg")
+	if ffmpegExists {
+		g.FFmpegLibRegister()
+	} else {
+		log.Println("[AGI] ffmpeg not installed on host OS. Bypassing module.")
+	}
+
 }

mod/apt/apt.go

@@ -1,160 +1,159 @@
-package apt
-
-import (
-	"encoding/json"
-	"errors"
-	"log"
-	"net/http"
-	"os"
-	"os/exec"
-	"runtime"
-	"strings"
-)
-
-/*
-	Pacakge management tool for Linux OS with APT
-
-	ONLY USABLE under Linux environment
-*/
-
-type AptPackageManager struct {
-	AllowAutoInstall bool
-}
-
-func NewPackageManager(autoInstall bool) *AptPackageManager {
-	return &AptPackageManager{
-		AllowAutoInstall: autoInstall,
-	}
-}
-
-//Install the given package if not exists. Set mustComply to true for "panic on failed to install"
-func (a *AptPackageManager) InstallIfNotExists(pkgname string, mustComply bool) error {
-	//Clear the pkgname
-	pkgname = strings.ReplaceAll(pkgname, "&", "")
-	pkgname = strings.ReplaceAll(pkgname, "|", "")
-
-	if a.AllowAutoInstall == false {
-		return errors.New("Package auto install is disabled")
-	}
-
-	installed, err := PackageExists(pkgname)
-	if err != nil {
-		log.Println(err.Error())
-	}
-
-	//log.Println(packageInfo)
-	if installed {
-		return nil
-	} else {
-		//Package not installed. Install if now if running in sudo mode
-		log.Println("Installing package " + pkgname + "...")
-		cmd := exec.Command("apt-get", "install", "-y", pkgname)
-		cmd.Stdout = os.Stdout
-		cmd.Stderr = os.Stderr
-		err := cmd.Run()
-		if err != nil {
-			if mustComply {
-				//Panic and terminate server process
-				log.Println("Installation failed on package: " + pkgname)
-				os.Exit(1)
-			} else {
-				log.Println("Installation failed on package: " + pkgname)
-			}
-			return err
-		}
-		return nil
-	}
-
-	return nil
-}
-
-func PackageExists(pkgname string) (bool, error) {
-	if runtime.GOOS == "windows" {
-		//Check if the command already exists in windows path paramters.
-		cmd := exec.Command("where", pkgname, "2>", "nul")
-		_, err := cmd.CombinedOutput()
-		if err != nil {
-			return false, errors.New("Package " + pkgname + " not found in Windows %PATH%.")
-		}
-		return true, nil
-	} else if runtime.GOOS == "darwin" {
-		//Mac OS. Check if package exists
-		cmd := exec.Command("whereis", pkgname)
-		out, err := cmd.CombinedOutput()
-		if err != nil {
-			return false, errors.New("Package " + pkgname + " not found in MacOS ENV variable.")
-		}
-
-		if strings.TrimSpace(string(out)) == "" {
-			//Package not found by whereis. Try brew
-			cmd := exec.Command("bash", "-c", "brew list | grep "+pkgname)
-			out, err = cmd.CombinedOutput()
-			if err != nil {
-				return false, errors.New("Package " + pkgname + " not found in MacOS ENV variable.")
-			}
-		}
-		if strings.TrimSpace(string(out)) != "" {
-			//Exists!
-			return true, nil
-		} else {
-			return false, errors.New("Package " + pkgname + " not installed on this Mac")
-		}
-
-
-	} else if runtime.GOOS == "linux" {
-		cmd := exec.Command("which", pkgname)
-		out, _ := cmd.CombinedOutput()
-
-		if len(string(out)) > 1 {
-			return true, nil
-		} else {
-			return false, errors.New("Package " + pkgname + " not installed on this Linux Host")
-		}
-
-	} else {
-		return false, errors.New("Unsupported Platform")
-	}
-
-	return false, errors.New("Unknown error occured when checking package installed")
-}
-
-func HandlePackageListRequest(w http.ResponseWriter, r *http.Request) {
-	if runtime.GOOS == "windows" {
-		w.Header().Set("Content-Type", "application/json")
-		w.Write([]byte("{\"error\":\"" + "Function disabled on Windows" + "\"}"))
-		return
-	}
-	cmd := exec.Command("apt", "list", "--installed")
-	out, err := cmd.CombinedOutput()
-	if err != nil {
-		w.Header().Set("Content-Type", "application/json")
-		w.Write([]byte("{\"error\":\"" + err.Error() + "\"}"))
-		return
-	}
-
-	results := [][]string{}
-	//Parse the output string
-	installedPackages := strings.Split(string(out), "\n")
-	for _, thisPackage := range installedPackages {
-		if len(thisPackage) > 0 {
-			packageInfo := strings.Split(thisPackage, "/")
-			packageName := packageInfo[0]
-			if len(packageInfo) >= 2 {
-				packageVersion := strings.Split(packageInfo[1], ",")[1]
-				if packageVersion[:3] == "now" {
-					packageVersion = packageVersion[4:]
-				}
-				if strings.Contains(packageVersion, "[installed") && packageVersion[len(packageVersion)-1:] != "]" {
-					packageVersion = packageVersion + ",automatic]"
-				}
-
-				results = append(results, []string{packageName, packageVersion})
-			}
-		}
-	}
-
-	jsonString, _ := json.Marshal(results)
-	w.Header().Set("Content-Type", "application/json")
-	w.Write(jsonString)
-	return
-}
+package apt
+
+import (
+	"encoding/json"
+	"errors"
+	"log"
+	"net/http"
+	"os"
+	"os/exec"
+	"runtime"
+	"strings"
+)
+
+/*
+	Pacakge management tool for Linux OS with APT
+
+	ONLY USABLE under Linux environment
+*/
+
+type AptPackageManager struct {
+	AllowAutoInstall bool
+}
+
+func NewPackageManager(autoInstall bool) *AptPackageManager {
+	return &AptPackageManager{
+		AllowAutoInstall: autoInstall,
+	}
+}
+
+// Install the given package if not exists. Set mustComply to true for "panic on failed to install"
+func (a *AptPackageManager) InstallIfNotExists(pkgname string, mustComply bool) error {
+	//Clear the pkgname
+	pkgname = strings.ReplaceAll(pkgname, "&", "")
+	pkgname = strings.ReplaceAll(pkgname, "|", "")
+
+	if a.AllowAutoInstall == false {
+		return errors.New("Package auto install is disabled")
+	}
+
+	installed, err := PackageExists(pkgname)
+	if err != nil {
+		log.Println(err.Error())
+	}
+
+	//log.Println(packageInfo)
+	if installed {
+		return nil
+	} else {
+		//Package not installed. Install if now if running in sudo mode
+		log.Println("Installing package " + pkgname + "...")
+		cmd := exec.Command("apt-get", "install", "-y", pkgname)
+		cmd.Stdout = os.Stdout
+		cmd.Stderr = os.Stderr
+		err := cmd.Run()
+		if err != nil {
+			if mustComply {
+				//Panic and terminate server process
+				log.Println("Installation failed on package: " + pkgname)
+				os.Exit(1)
+			} else {
+				log.Println("Installation failed on package: " + pkgname)
+			}
+			return err
+		}
+		return nil
+	}
+
+	return nil
+}
+
+func PackageExists(pkgname string) (bool, error) {
+	if runtime.GOOS == "windows" {
+		//Check if the command already exists in windows path paramters.
+		cmd := exec.Command("where.exe", pkgname, "2>", "nul")
+		_, err := cmd.CombinedOutput()
+		if err != nil {
+			return false, errors.New("Package " + pkgname + " not found in Windows %PATH%.")
+		}
+		return true, nil
+	} else if runtime.GOOS == "darwin" {
+		//Mac OS. Check if package exists
+		cmd := exec.Command("whereis", pkgname)
+		out, err := cmd.CombinedOutput()
+		if err != nil {
+			return false, errors.New("Package " + pkgname + " not found in MacOS ENV variable.")
+		}
+
+		if strings.TrimSpace(string(out)) == "" {
+			//Package not found by whereis. Try brew
+			cmd := exec.Command("bash", "-c", "brew list | grep "+pkgname)
+			out, err = cmd.CombinedOutput()
+			if err != nil {
+				return false, errors.New("Package " + pkgname + " not found in MacOS ENV variable.")
+			}
+		}
+		if strings.TrimSpace(string(out)) != "" {
+			//Exists!
+			return true, nil
+		} else {
+			return false, errors.New("Package " + pkgname + " not installed on this Mac")
+		}
+
+	} else if runtime.GOOS == "linux" {
+		cmd := exec.Command("which", pkgname)
+		out, _ := cmd.CombinedOutput()
+
+		if len(string(out)) > 1 {
+			return true, nil
+		} else {
+			return false, errors.New("Package " + pkgname + " not installed on this Linux Host")
+		}
+
+	} else {
+		return false, errors.New("Unsupported Platform")
+	}
+
+	return false, errors.New("Unknown error occured when checking package installed")
+}
+
+func HandlePackageListRequest(w http.ResponseWriter, r *http.Request) {
+	if runtime.GOOS == "windows" {
+		w.Header().Set("Content-Type", "application/json")
+		w.Write([]byte("{\"error\":\"" + "Function disabled on Windows" + "\"}"))
+		return
+	}
+	cmd := exec.Command("apt", "list", "--installed")
+	out, err := cmd.CombinedOutput()
+	if err != nil {
+		w.Header().Set("Content-Type", "application/json")
+		w.Write([]byte("{\"error\":\"" + err.Error() + "\"}"))
+		return
+	}
+
+	results := [][]string{}
+	//Parse the output string
+	installedPackages := strings.Split(string(out), "\n")
+	for _, thisPackage := range installedPackages {
+		if len(thisPackage) > 0 {
+			packageInfo := strings.Split(thisPackage, "/")
+			packageName := packageInfo[0]
+			if len(packageInfo) >= 2 {
+				packageVersion := strings.Split(packageInfo[1], ",")[1]
+				if packageVersion[:3] == "now" {
+					packageVersion = packageVersion[4:]
+				}
+				if strings.Contains(packageVersion, "[installed") && packageVersion[len(packageVersion)-1:] != "]" {
+					packageVersion = packageVersion + ",automatic]"
+				}
+
+				results = append(results, []string{packageName, packageVersion})
+			}
+		}
+	}
+
+	jsonString, _ := json.Marshal(results)
+	w.Header().Set("Content-Type", "application/json")
+	w.Write(jsonString)
+	return
+}