downloadQJ/utils/getScenes.go

package utils

import (
	"encoding/json"
	"io"
	"net/http"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"time"
)

var myClient = &http.Client{Timeout: 10 * time.Second}

func GetJson(url string, target interface{}) error {
	r, err := myClient.Get(url)
	if err != nil {
		return err
	}
	defer r.Body.Close()

	return json.NewDecoder(r.Body).Decode(target)
}

func DownloadFile(path string, url string) (err error) {

	dir := filepath.Dir(path)

	err = os.MkdirAll(dir, 0644)
	if err != nil {
		return err
	}
	out, err := os.Create(path)
	if err != nil {
		return err
	}
	defer out.Close()

	// Get the data
	resp, err := http.Get(url)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	// Writer the body to file
	_, err = io.Copy(out, resp.Body)
	if err != nil {
		return err
	}

	return nil
}

func IsCommandAvailable(name string) bool {
	cmd := exec.Command("command", "-v", name)
	if err := cmd.Run(); err != nil {
		return false
	}
	return true
}

func DetectSource(path string, dest string) (url string) {
	saveP := strings.Split(path, "?")[0]

	if strings.Contains(saveP, "https://4dkk.4dage.com/720yun_fd_manage") {
		p := strings.ReplaceAll(saveP, "https://4dkk.4dage.com/720yun_fd_manage", "")
		url = dest + p
	}
	if strings.Contains(saveP, "https://4dkk.4dage.com/scene_edit_data") {
		p := strings.ReplaceAll(saveP, "https://4dkk.4dage.com", "")
		url = dest + p
	}

	return url
}

func EnsureDir(dirName string) error {

	err := os.MkdirAll(dirName, os.ModeDir)

	if err == nil || os.IsExist(err) {
		return nil
	} else {
		return err
	}
}