up2GitX/share/tools.go

package share

import (
	"bufio"
	"encoding/json"
	"fmt"
	"io"
	"io/ioutil"
	NetHttp "net/http"
	"os"
	"path/filepath"
	"regexp"
	"strings"
	"sync"
	"time"

	"github.com/gookit/color"
	"github.com/gookit/gcli/v2/interact"
	"github.com/go-git/go-git/v5"
	"github.com/go-git/go-git/v5/config"
	"gopkg.in/src-d/go-git.v4/plumbing/transport/http"
	"github.com/bitly/go-simplejson"
)

type RepoLocal struct {
	path  string
	sizeM  float32
	alert  bool
	error string
}

const (
	WORKER = 5
)

func InvalidAlert(platform string) {
	fmt.Printf("Tell me which repos source your want to sync, Usage: ")
	color.Yellow.Printf("up2 %s /Users/Zoker/repos/ or up2 %s /Users/Zoker/repo.txt or up2 %s github:kesin\n", platform, platform, platform)
	fmt.Printf("See 'up2 %s -h' for more details\n", platform)
}

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

func GetGitDir(repoSource string) (repos []string, err error) {
	source, err := os.Stat(repoSource)
	if err != nil {
		return nil, err
	}

	// source is a dir
	if source.IsDir() {
		dir, err := ioutil.ReadDir(repoSource)
		if err != nil {
			return nil, err
		}

		pathSep := string(os.PathSeparator)
		for _, repo := range dir {
			if !repo.IsDir() {
				continue
			}
			repoPath := repoSource + pathSep + repo.Name() // todo check repo path valid
			if isGitRepo(repoPath) {                    // todo goroutine
				repos = append(repos, repoPath)
			}
		}
	// source is list
	} else {
		file, err := os.Open(repoSource)
		if err != nil {
			return nil, err
		}

		defer file.Close()
		bf := bufio.NewReader(file)

		for {
			path, _, err := bf.ReadLine()
			if err == io.EOF {
				break
			}
			if err != nil {
				return nil, err
			}
			if 0 == len(path) || string(path) == "\r\n" {
				continue
			}
			realPath := string(path)
			if isGitRepo(realPath) { // todo goroutine
				repos = append(repos, realPath)
			}
		}
	}
	return repos, nil
}

func isGitRepo(repoPath string) (isGit bool) {
	_, err := git.PlainOpen(repoPath)
	if err == nil {
		return true
	} else {
		return false
	}
}

func printRepos(repos []RepoLocal) {
	color.Yellow.Printf("\n%d repositories detected, please check below:\n\n", len(repos))

	alertFlag := false

	for i, repo := range repos {
		i = i + 1
		p := fmt.Sprintf("%d. %s", i, repo.path)
		fmt.Printf(p)
		alertFlag = alertFlag || repo.alert
		if repo.alert {
			   color.Red.Printf(" %.2f", repo.sizeM)
			   color.Red.Println("M")
		} else {
			   color.Green.Printf(" %.2f", repo.sizeM)
			   color.Green.Println("M")
		}

	}

	if alertFlag {
		color.Yellow.Println("Warning: some of your local repo is out of 1G, please make sure that you account have permission to sync repository that size more than 1G")
	}
}

func getRepoLocal(repos []string) (reposLocal []RepoLocal) {
	var wp sync.WaitGroup
	var mutex = &sync.Mutex{}
	paths := make(chan string)
	wp.Add(len(repos))
	for w := 1; w <= WORKER; w++ {
		go getRepoItemWorker(paths, &wp, &reposLocal, mutex)
	}
	for _, p := range repos {
		paths <- p
	}
	close(paths)
	wp.Wait()
	return reposLocal
}

func getRepoApi(repoDir, tokenFix string) (reposLocal []RepoLocal, repos []string) {
	var orgType string
	var repoArray []map[string]interface{}
	orgPath := repoDir[7:]
	platform := repoDir[:6]

	// check path type
	orgType = getGithubType(orgPath, tokenFix)
	if orgType == "" {
		return nil, nil
	}

	color.Green.Printf("Selected %s, type is %s, Start to fetch repository list...\n", orgPath, orgType)

	// get repos
	getGithubRepo(orgPath, orgType, &repoArray, "", tokenFix)

	count := len(repoArray)
	if count == 0 {
		color.Yellow.Printf("No repositories found in https://github.com/%s\nPlease check if path is valid\n", orgPath)
		return nil, nil
	} else { // TODO optimize code with local repo and struct repos
		for _, item := range repoArray {
			outOf1G := false
			size, _ := item["size"].(json.Number).Int64()
			if size > 1024*1024 {
				outOf1G = true
			}
			sizeMB := float32(size) / 1024.0
			repoPath := fmt.Sprintf("https://%s.com/%s/%s", platform, orgPath, item["name"].(string))
			reposLocal = append(reposLocal, RepoLocal{path: repoPath, sizeM: sizeMB, alert: outOf1G})
			repos = append(repos, repoPath)
		}
		return reposLocal, repos
	}
}

func getRepoItemWorker(paths <- chan string, wp *sync.WaitGroup, reposLocal *[]RepoLocal, mutex *sync.Mutex) {
	for path := range paths {
		size, outAlert, _ := repoSize(path)
		mutex.Lock()
		*reposLocal = append(*reposLocal, RepoLocal{path: path, sizeM: size, alert: outAlert})
		mutex.Unlock()
		wp.Done()
	}
}

func errGithub(body  *simplejson.Json) bool {
	mesg, err := body.Get("message").String()
	if err != nil {
		return false
	} else {
		color.Red.Println(mesg)
		return true
	}
}

func getGithubType(orgPath, tokenFix string) string {
	typeUrl := fmt.Sprintf("https://api.github.com/users/%s?from=up2Gitx%s", orgPath, tokenFix)
	result, _, err := GetByte(typeUrl)
	if err != nil {
		color.Red.Printf("Request Type error: %s \n", err.Error())
		return ""
	}

	typeRes, _ := simplejson.NewJson([]byte(result))
	errGithub(typeRes)

	typeStr, _ := typeRes.Get("type").String()
	if typeStr == "User" {
		return "users"
	} else if typeStr == "Organization" {
		 return "orgs"
	} else {
		color.Red.Printf("%s not exists, please enter a valid path!\n", orgPath)
		return ""
	}
}

func getGithubRepo(orgPath string, orgType string, repoArray *[]map[string]interface{}, url string, tokenFix string) {
	var apiUrl string
	if url != "" {
		apiUrl = url
	} else {
		apiUrl = fmt.Sprintf("https://api.github.com/%s/%s/repos?per_page=20%s", orgType, orgPath, tokenFix)
	}
	fmt.Printf("Fetching %s \n", apiUrl)

	result, header, err := GetByte(apiUrl)
	if err != nil {
		color.Red.Printf("Request Repositories error: %s \n", err.Error())
	}

	jsonRes, _ := simplejson.NewJson([]byte(result))
	errGithub(jsonRes)

	// TODO optimize type assertion remove for
	tmpRepos, _ := jsonRes.Array()
	for _, re := range tmpRepos {
		*repoArray = append(*repoArray, re.(map[string]interface {}))
	}

	pageLink := header.Get("link")
	if pageLink != "" {
		link := strings.Split(pageLink, ",")
		for _, l := range link {
			kv := strings.Split(l, ";")
			if strings.Contains(kv[1], "next") {
				getGithubRepo(orgPath, orgType, repoArray, kv[0][1 : len(kv[0])-1], "")
				break
			}
		}
	}

}

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

	outOf1G := false
	if size > 1024*1024*1024 {
		outOf1G = true
	}
	sizeMB := float32(size) / 1024.0 / 1024.0
	return sizeMB, outOf1G, err
}

func ReadyToAuth(args []string, platform string) ([]string, string, string) {
	repoDir := args[0]

	// if up to gitee from github|gitlab|bitbucket...
	if platform == "gitee" {
		reg, err := regexp.Compile(`^github:.+`)
		if err == nil && reg.MatchString(repoDir) {
			var tokenFix string
			if len(args) == 2 && args[1] != "" {
				tokenFix = fmt.Sprintf("&access_token=%s", args[1])
			}
			reposLocal, repos := getRepoApi(repoDir, tokenFix)
			if repos != nil {
				printRepos(reposLocal)
				if len(repos) > 1000 {
					color.Warn.Println("\nWarning: Gitee only support 1000 projects for each user, some of projects will not sync as expect")
				}
				inPut, _ := interact.ReadLine("\nCheck if this repositories are what you expected, ready to the next step? (y/n) ")
				if inPut == "y" {
					return repos, repoDir[:6], repoDir[7:]
				} else {
					ExitMessage()
				}
			}
			return nil, "", ""
		}
	}

	if FileExists(repoDir) {
		repos, _ := GetGitDir(repoDir)
		if len(repos) == 0 {
			color.Red.Printf("No git repositories detected in %s \n", repoDir)
		} else {
			reposLocal := getRepoLocal(repos)
			printRepos(reposLocal)
			inPut, _ := interact.ReadLine("\nCheck if this repositories are what you expected, ready to the next step? (y/n) ")
			if inPut == "y" {
				return repos, "local", ""
			} else {
				ExitMessage()
			}
		}
	} else {
		color.Red.Println("The path you provided is not a dir or not exists")
	}
	return nil, "", ""
}

func ExitMessage() {
	color.Yellow.Println("Bye, see you next time!")
}

func GetByte(url string) ([]byte, NetHttp.Header, error) {
	response, err := NetHttp.Get(url)
	if err != nil {
		color.Red.Printf("Request failed, Error: %s \n", err.Error())
		return nil, nil, err
	}
	defer response.Body.Close()

	body, _ := ioutil.ReadAll(response.Body)
	header := response.Header
	return body, header, nil
}

func Get(url string) (map[string]interface{}, error) {
	response, err := NetHttp.Get(url)
	if err != nil {
		color.Red.Printf("Request failed, Error: %s \n", err.Error())
		return nil, err
	}
	defer response.Body.Close()

	body, _ := ioutil.ReadAll(response.Body)

	var result map[string]interface{}
	json.Unmarshal(body, &result)
	return result, nil
}

func PostForm(uri string, params map[string]interface{}) (map[string]interface{}, error) {
	data := ""
	for k, v := range params {
		data += fmt.Sprintf("%s=%s&%s", k, v.(string), data)
	}

	response, err := NetHttp.Post(uri, "application/x-www-form-urlencoded", strings.NewReader(data))
	if err != nil {
		color.Red.Printf("Request failed, Error: %s \n", err.Error())
		return nil, err
	}
	defer response.Body.Close()

	body, _ := ioutil.ReadAll(response.Body)

	var result map[string]interface{}
	json.Unmarshal(body, &result)
	return result, nil
}

func PostFormByte(uri string, params map[string]interface{}) ([]byte, NetHttp.Header, error) {
	data := ""
	for k, v := range params {
		data += fmt.Sprintf("%s=%s&%s", k, v.(string), data)
	}

	response, err := NetHttp.Post(uri, "application/x-www-form-urlencoded", strings.NewReader(data))
	if err != nil {
		color.Red.Printf("Request failed, Error: %s \n", err.Error())
		return nil, nil, err
	}
	defer response.Body.Close()

	body, _ := ioutil.ReadAll(response.Body)
	header := response.Header

	return body, header, nil
}

func ShowProjectLists(host string, repos []string, path string) {
	for i, r := range repos {
		i = i + 1
		ra := strings.Split(r, "/")
		p := fmt.Sprintf("%d. https://%s/%s/%s", i, host, path, ra[len(ra) - 1])
		color.Yellow.Println(p)
	}
}

func AskPublic(npType string) string {
	namespace := []string{"Public (Anyone can see this repository)",
						  "Private (Only members can see this repository)"}
	if npType == "Enterprise" {
		namespace = append(namespace, "Inner public (Only enterprise members can see this repository)")
	}
	fmt.Printf("\n")
	ques := "Please choose those project's public type: (new projects will apply)"
	public := selectOne(namespace, ques)
	return public
}

func AskError() string {
	howTo := []string{"Exit and fix them",
		"Skip them"}
	ques := "There are errors on some projects, what would you like to do?"
	return selectOne(howTo, ques)
}

func AskExist() string {
	color.Notice.Println("\n", "WARNING: The exist project will remain private attribute as what it was!", "\n")
	howTo := []string{"Exit and fix them",
		"Skip them",
		"Overwrite the remote (same as git push --force, you need exactly know what you do before you select this item)"}
	ques := "The are some projects name already exists, what would you like to do?"
	return selectOne(howTo, ques)
}

func selectOne(items []string, ques string) string {
	return 	interact.SelectOne(ques, items, "",)
}

func SyncRepo(auth *http.BasicAuth ,source string, uri string, force string, tmpDir string) error {
	var forceStr, repoPath, rHead string
	// if project from api, clone it before pushing to target
	if tmpDir != "" {
		rHead = "remotes/origin"
		repoSplit := strings.Split(source, "/")
		repoName := repoSplit[len(repoSplit) - 1]
		repoPath = fmt.Sprintf("%s/%s", tmpDir, repoName)
		if r, err := git.PlainOpen(repoPath); err == nil { // fetch if repo exists
			r.Fetch(&git.FetchOptions{
				RefSpecs: []config.RefSpec{"refs/*:refs/*"},
			})
		} else {
			_, err := git.PlainClone(repoPath, true, &git.CloneOptions{
				URL:               source,
				RecurseSubmodules: git.DefaultSubmoduleRecursionDepth,
			})
			if err != nil {
				return err
			}
		}
	} else {
		rHead = "heads"
		repoPath = source
	}

	// generate a tmp remote
	remote := fmt.Sprintf("up2GitX-%d", time.Now().Unix())
	r, err := git.PlainOpen(repoPath)
	if err != nil {
		return err
	}

	// delete this remote after sync whether success or not
	defer deleteRemote(r, remote)
	_, err = r.CreateRemote(&config.RemoteConfig{
		Name: remote,
		URLs: []string{uri},
	})
	if err != nil {
		return err
	}

	switch force { // push force or not
	case "2":
		forceStr = "+"
	default:
		forceStr = ""
	}
	rHeadStrings := fmt.Sprintf("%srefs/%s/*:refs/%s/*", forceStr, rHead, "heads")
	rTagStrings := fmt.Sprintf("%srefs/%s/*:refs/%s/*", forceStr, "tags", "tags")
	rHeads := config.RefSpec(rHeadStrings)
	rTags := config.RefSpec(rTagStrings)

	err = r.Push(&git.PushOptions{RemoteName: remote,
		RefSpecs: []config.RefSpec{rHeads, rTags},
		Auth: auth})
	if err != nil {
		return err
	}
	return nil
}

func deleteRemote(r  *git.Repository, upRe string) {
	r.DeleteRemote(upRe)
}