bee/version.go

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"io"
	"os"
	path "path/filepath"
	"regexp"
)

var cmdVersion = &Command{
	UsageLine: "version",
	Short:     "show the Bee, Beego and Go version",
	Long: `
show the Bee, Beego and Go version

bee version
    bee   :1.2.3
    beego :1.4.2
    Go    :go version go1.3.3 linux/amd64

`,
}

const verboseVersionBanner = `______
| ___ \
| |_/ /  ___   ___
| ___ \ / _ \ / _ \
| |_/ /|  __/|  __/
\____/  \___| \___| v{{ .BeeVersion }}

Beego     : {{ .BeegoVersion }}
GoVersion : {{ .GoVersion }}
GOOS      : {{ .GOOS }}
GOARCH    : {{ .GOARCH }}
NumCPU    : {{ .NumCPU }}
GOPATH    : {{ .GOPATH }}
GOROOT    : {{ .GOROOT }}
Compiler  : {{ .Compiler }}
Date      : {{ Now "Monday, 2 Jan 2006" }}
`

const shortVersionBanner = `______
| ___ \
| |_/ /  ___   ___
| ___ \ / _ \ / _ \
| |_/ /|  __/|  __/
\____/  \___| \___| v{{ .BeeVersion }}
`

func init() {
	cmdVersion.Run = versionCmd
}

func versionCmd(cmd *Command, args []string) int {
	ShowVerboseVersionBanner()
	return 0
}

func ShowVerboseVersionBanner() {
	InitBanner(os.Stdout, bytes.NewBufferString(verboseVersionBanner))
}

func ShowShortVersionBanner() {
	InitBanner(os.Stdout, bytes.NewBufferString(shortVersionBanner))
}

func getBeegoVersion() string {
	gopath := os.Getenv("GOPATH")
	re, err := regexp.Compile(`VERSION = "([0-9.]+)"`)
	if err != nil {
		return ""
	}
	if gopath == "" {
		err = fmt.Errorf("You should set GOPATH env variable")
		return ""
	}
	wgopath := path.SplitList(gopath)
	for _, wg := range wgopath {
		wg, _ = path.EvalSymlinks(path.Join(wg, "src", "github.com", "astaxie", "beego"))
		filename := path.Join(wg, "beego.go")
		_, err := os.Stat(filename)
		if err != nil {
			if os.IsNotExist(err) {
				continue
			}
			ColorLog("[ERRO] Get `beego.go` has error\n")
		}
		fd, err := os.Open(filename)
		if err != nil {
			ColorLog("[ERRO] Open `beego.go` has error\n")
			continue
		}
		reader := bufio.NewReader(fd)
		for {
			byteLine, _, er := reader.ReadLine()
			if er != nil && er != io.EOF {
				return ""
			}
			if er == io.EOF {
				break
			}
			line := string(byteLine)
			s := re.FindStringSubmatch(line)
			if len(s) >= 2 {
				return s[1]
			}
		}

	}
	return "Beego not installed. Please install it first: https://github.com/astaxie/beego"
}