pikapython/tools/pikaPackageManager/main.go

package main

import (
	"bufio"
	"fmt"
	"io"
	"net/http"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"time"

	"github.com/BurntSushi/toml"
	"github.com/go-git/go-git/v5"
	"github.com/go-git/go-git/v5/plumbing"
	cp "github.com/otiai10/copy"
)

var isShowSize = false
var isErr = 0
var isDebug = false

type Requerment_t struct {
	Name    string
	Version string
	Commit  string
}

type Package_t struct {
	Name           string
	Releases       []string
	ReleasesName   []string
	ReleasesCommit []string
}

type Config_t struct {
	Packages []Package_t
}

func main() {
	res := process()
	println()
	delay := 3
	if res != 0 {
		delay = 15
	}
	if res == 0 {
		fmt.Printf("\x1b[32;1m%s\x1b[0m\n", fmt.Sprintf("Succeed: %s", "Update OK !"))
	} else {
		fmt.Printf("\x1b[31;1m%s\x1b[0m\n", fmt.Sprintf("Error: %s", "Some error(s) occored. Please Check."))
	}
	fmt.Println()
	for i := delay; i >= 0; i-- {
		time.Sleep(1 * time.Second)
		fmt.Printf("    [This window will auto close after %2d second(s)]\r", i)
	}
}

func process() int {
	superPath := "/tmp"
	path := "/pikascript_packages"
	var repo *git.Repository

	go readPathSize(superPath + path)
	if !isDebug {
		repo = updatePikascript(superPath + path)
	}

	requestments, res := getRequestment("requestment.txt")
	if !res {
		fmt.Printf("Error! get requestment info faild.\n")
		return -1
	}
	fmt.Printf("\n")

	checkout_requestments(superPath+path, repo, requestments)
	save_requestments(requestments)

	// check main.py file is exist
	if _, err := os.Stat("main.py"); os.IsNotExist(err) {
		return isErr
	}

	fmt.Printf("\n")
	cmdRun("rust-msc-latest-win10.exe")
	return isErr
}

func save_requestments(requestments []Requerment_t) {
	os.Remove("requestment.txt")
	f, err := os.OpenFile("requestment.txt", os.O_CREATE, 0666)
	CheckIfError(err)
	for i, requestment := range requestments {
		f.WriteString(requestment.Name + "==" + requestment.Version)
		if i != len(requestments)-1 {
			f.WriteString("\r\n")
		}
	}
	f.Close()
}

func FilterDirsGlob(dir, suffix string) ([]string, error) {
	return filepath.Glob(filepath.Join(dir, suffix))
}

func MoveFile(sourcePath, destPath string) error {
	inputFile, err := os.Open(sourcePath)
	if err != nil {
		return fmt.Errorf("Couldn't open source file: %s", err)
	}
	outputFile, err := os.Create(destPath)
	if err != nil {
		inputFile.Close()
		return fmt.Errorf("Couldn't open dest file: %s", err)
	}
	defer outputFile.Close()
	_, err = io.Copy(outputFile, inputFile)
	inputFile.Close()
	if err != nil {
		return fmt.Errorf("Writing to output file failed: %s", err)
	}
	// The copy was successful, so now delete the original file
	err = os.Remove(sourcePath)
	if err != nil {
		return fmt.Errorf("Failed removing original file: %s", err)
	}
	return nil
}

func checkout_requestments(path string, repo *git.Repository, requerments []Requerment_t) {
	os.Mkdir("pikascript-lib", os.ModePerm)
	os.Mkdir("pikascript-core", os.ModePerm)
	os.Mkdir("pikascript-api", os.ModePerm)
	workTree, _ := repo.Worktree()
	for _, requerment := range requerments {
		/* update file */
		var packagePath string = path + "/" + requerment.Name + "/" + requerment.Name + "@" + requerment.Version
		var dirPath string = "pikascript-lib/" + requerment.Name
		if requerment.Name == "pikascript-core" {
			dirPath = "pikascript-core"
		}
		// fmt.Printf("    copy" + " " + packagePath + " " + dirPath + "\n")
		CheckIfError(cp.Copy(packagePath, dirPath))

		if requerment.Name == "pikascript-core" {
			CheckIfError(os.Rename(dirPath+"/rust-msc-latest-win10.exe", "./rust-msc-latest-win10.exe"))
		}

		pyFileList, _ := FilterDirsGlob(dirPath, "*.py")
		for i := range pyFileList {
			pyFileSource := strings.ReplaceAll(pyFileList[i], "\\", "/")
			pyFilePath := strings.Split(pyFileSource, "/")
			pyFileName := pyFilePath[len(pyFilePath)-1]
			fmt.Println("Installed: " + pyFileName + ": " + requerment.Version)
			CheckIfError(os.Rename(pyFileSource, pyFileName))
		}

		pyiFileList, _ := FilterDirsGlob(dirPath, "*.pyi")
		for i := range pyiFileList {
			pyFileSource := strings.ReplaceAll(pyiFileList[i], "\\", "/")
			pyFilePath := strings.Split(pyFileSource, "/")
			pyFileName := pyFilePath[len(pyFilePath)-1]
			fmt.Println("Installed: " + pyFileName + ": " + requerment.Version)
			CheckIfError(os.Rename(pyFileSource, pyFileName))
		}

		pyiFileList, _ = FilterDirsGlob(dirPath, "*.py.o")
		for i := range pyiFileList {
			pyFileSource := strings.ReplaceAll(pyiFileList[i], "\\", "/")
			pyFilePath := strings.Split(pyFileSource, "/")
			pyFileName := pyFilePath[len(pyFilePath)-1]
			fmt.Println("Installed: " + pyFileName + ": " + requerment.Version)
			CheckIfError(os.Rename(pyFileSource, pyFileName))
		}

	}
	err := workTree.Checkout(&git.CheckoutOptions{
		Hash:  plumbing.NewHash("master"),
		Force: true,
	})
	CheckIfError(err)
}
func CheckIfError(err error) {
	if err == nil {
		return
	}
	isErr = -1
	fmt.Printf("\x1b[31;1m%s\x1b[0m\n", fmt.Sprintf("Error: %s", err))
}

func getRequestment(path string) ([]Requerment_t, bool) {
	var requestments []Requerment_t
	requestment_file, _ := os.Open(path)
	defer requestment_file.Close()
	scanner := bufio.NewScanner(requestment_file)
	var count int
	for scanner.Scan() {
		var requestment Requerment_t
		var res int
		count++
		line := scanner.Text()
		line = strings.ReplaceAll(line, " ", "")
		line = strings.ReplaceAll(line, "\r", "")
		if line == "" {
			continue
		}
		if !strings.Contains(line, "==") {
			line, res = match_default_version(line)
		}
		if res != 0 {
			return nil, false
		}
		req_info := strings.Split(line, "==")
		if len(req_info) == 2 {
			requestment.Name = strings.Split(line, "==")[0]
			requestment.Version = strings.Split(line, "==")[1]
			fmt.Printf("Request: %s %s\n", requestment.Name, requestment.Version)
			requestments = append(requestments, requestment)
		}
	}
	return requestments, true
}

func http_get_str(url string) string {
	resp, err := http.Get(url)
	CheckIfError(err)
	// fsize, err := strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 32)
	fsize := 1024 * 1024
	data := make([]byte, fsize)
	n, _ := resp.Body.Read(data)
	data = data[0:n]
	return string(data)
}

func match_default_version(line string) (string, int) {
	var config Config_t
	var found int
	req_name := line
	toml_str := http_get_str("https://gitee.com/Lyon1998/pikascript/raw/master/packages.toml")
	_, err := toml.Decode(toml_str, &config)
	CheckIfError(err)
	for i_package, pkg := range config.Packages {
		i := 0
		for _, release := range pkg.Releases {
			pkg.ReleasesName = append(pkg.ReleasesName, strings.Split(release, " ")[0])
			pkg.ReleasesCommit = append(pkg.ReleasesCommit, strings.Split(release, " ")[1])
			i++
		}
		config.Packages[i_package] = pkg
	}
	var pkg Package_t
	packages := config.Packages
	for _, pkg = range packages {
		if req_name == pkg.Name {
			found = 1
			break
		}
	}
	if found == 1 {
		req_version := strings.Split(pkg.Releases[len(pkg.Releases)-1], " ")[0]
		return req_name + "==" + req_version, 0
	}
	return "", 1
}

func readPathSize(path string) {
	for {
		if !isShowSize {
			continue
		}
		size, _ := pathSize(path)
		fmt.Printf("Recived: %.3f MB \r", float64(size)/1024/1024)
		time.Sleep(time.Millisecond * 200)
	}
}

func updatePikascript(path string) *git.Repository {
	pathExist, err := PathExists(path + "/pikascript-core")
	if err != nil {
		fmt.Printf("PathExists(%s),err(%v)\n", path, err)
	}
	if !pathExist {
		/* clone the pikascript repo */
		fmt.Printf("Downloading pikascript to %s...\n", path)
		fmt.Printf("Need about 10 min(s)\n")
		fmt.Printf("Please wait...\n")
		/* remove pikascript folder */
		os.RemoveAll(path)
		isShowSize = true
		_, err = git.PlainClone(path, false, &git.CloneOptions{
			URL:      "https://gitee.com/lyon1998/pikascript_packages",
			Progress: os.Stdout,
		})
		if nil != err {
			fmt.Println(err)
		}
	}

	/* pull the pikascript repo */
	repo, _ := git.PlainOpen(path)
	w, _ := repo.Worktree()
	fmt.Println("Updating pikascript...")
	w.Pull(&git.PullOptions{RemoteName: "origin"})
	ref, _ := repo.Head()
	commit, _ := repo.CommitObject(ref.Hash())
	fmt.Println(commit)

	isShowSize = false

	err = w.Checkout(&git.CheckoutOptions{
		Hash:  plumbing.NewHash("master"),
		Force: true,
	})
	CheckIfError(err)

	return repo
}

func PathExists(path string) (bool, error) {
	_, err := os.Stat(path)
	if err == nil {
		return true, nil
	}
	if os.IsNotExist(err) {
		return false, nil
	}
	return false, err
}

func pathSize(path string) (int64, error) {
	var size int64
	err := filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if !info.IsDir() {
			size += info.Size()
		}
		return err
	})
	return size, err
}

func cmdRun(dir string) {
	cmd := exec.Command(dir)
	cmdReader, err := cmd.StdoutPipe()
	if err != nil {
		fmt.Fprintln(os.Stderr, "Error creating StdoutPipe for Cmd", err)
		return
	}

	scanner := bufio.NewScanner(cmdReader)
	scanner.Split(bufio.ScanBytes)

	go func() {
		for scanner.Scan() {
			fmt.Printf("%s", scanner.Text())
		}
	}()

	err = cmd.Start()
	if err != nil {
		fmt.Fprintln(os.Stderr, "Error starting Cmd", err)
		return
	}

	err = cmd.Wait()
	if err != nil {
		fmt.Fprintln(os.Stderr, "Error waiting for Cmd", err)
		return
	}
}