hub/commands/utils.go

127 строки
2.5 KiB
Go

package commands
import (
"bufio"
"io/ioutil"
"os"
"path/filepath"
"strings"
"github.com/github/hub/git"
"github.com/github/hub/github"
"github.com/github/hub/utils"
)
type listFlag []string
func (l *listFlag) String() string {
return strings.Join([]string(*l), ",")
}
func (l *listFlag) Set(value string) error {
for _, flag := range strings.Split(value, ",") {
*l = append(*l, flag)
}
return nil
}
func isCloneable(file string) bool {
f, err := os.Open(file)
if err != nil {
return false
}
defer f.Close()
fi, err := f.Stat()
if err != nil {
return false
}
if fi.IsDir() {
gitf, err := os.Open(filepath.Join(file, ".git"))
if err == nil {
gitf.Close()
return true
} else {
return git.IsGitDir(file)
}
} else {
reader := bufio.NewReader(f)
line, err := reader.ReadString('\n')
if err == nil {
return strings.Contains(line, "git bundle")
} else {
return false
}
}
}
func gitRemoteForProject(project *github.Project) (foundRemote *github.Remote) {
remotes, err := github.Remotes()
utils.Check(err)
for _, remote := range remotes {
remoteProject, pErr := remote.Project()
if pErr == nil && remoteProject.SameAs(project) {
foundRemote = &remote
return
}
}
return nil
}
func isEmptyDir(path string) bool {
fullPath := filepath.Join(path, "*")
match, _ := filepath.Glob(fullPath)
return match == nil
}
func getTitleAndBodyFromFlags(messageFlag, fileFlag string) (title, body string, err error) {
if messageFlag != "" {
title, body = readMsg(messageFlag)
} else if fileFlag != "" {
title, body, err = readMsgFromFile(fileFlag)
}
return
}
func readMsgFromFile(filename string) (title, body string, err error) {
var content []byte
if filename == "-" {
content, err = ioutil.ReadAll(os.Stdin)
} else {
content, err = ioutil.ReadFile(filename)
}
if err != nil {
return
}
text := strings.Replace(string(content), "\r\n", "\n", -1)
title, body = readMsg(text)
return
}
func readMsg(message string) (title, body string) {
parts := strings.SplitN(message, "\n\n", 2)
title = strings.TrimSpace(strings.Replace(parts[0], "\n", " ", -1))
if len(parts) > 1 {
body = strings.TrimSpace(parts[1])
}
return
}
func runInLocalRepo(fn func(localRepo *github.GitHubRepo, project *github.Project, client *github.Client)) {
localRepo, err := github.LocalRepo()
utils.Check(err)
project, err := localRepo.CurrentProject()
utils.Check(err)
client := github.NewClient(project.Host)
fn(localRepo, project, client)
os.Exit(0)
}