Upgrade from Github

Author: calmh

build.sh

@@ -13,7 +13,12 @@ pkg() {
 
 	mkdir -p "$dst"
 	cp README.md LICENSE "$dst"
-	go build -o "$dst/psmcli" -ldflags "-w -X main.Version $version"
+	go build -o "$dst/psmcli" -ldflags "-w -X main.Version=$version"
+
+	if [[ -f ~/signingkeys/calmh.priv ]] ; then
+		[ -f $dst/psmcli ] && stsigtool sign ~/signingkeys/calmh.priv "$dst/psmcli" > "$dst/psmcli.sig"
+		[ -f $dst/psmcli.exe ] && stsigtool sign ~/signingkeys/calmh.priv "$dst/psmcli.exe" > "$dst/psmcli.exe.sig"
+	fi
 
 	if [[ "$GOOS" == "windows" ]] ; then
 		pushd build

main.go

@@ -9,22 +9,56 @@ import (
 	"fmt"
 	"net"
 	"os"
+	"regexp"
+	"runtime"
 	"strings"
 
 	"github.com/calmh/psmcli/completion"
+	"github.com/calmh/upgrade"
 	"golang.org/x/crypto/ssh/terminal"
 )
 
 var (
-	Version = "unknown-dev"
+	Version    = "unknown-dev"
+	signingKey = []byte(`-----BEGIN EC PUBLIC KEY-----
+MIGbMBAGByqGSM49AgEGBSuBBAAjA4GGAAQBYyrh+aRcAmqSAReHq00XgaJC5Zn3
+JC/rlXAM5M2ODmMElLypAtnmUYFJTnmQD1KSwV49GFwFy+iqzNa9AfQ4gQQB+RmV
+Q2n12crDe2nU9oI0aPZFkIlqrjA0Ky0jT8rpWhmuRc+Bq8XS4q8mY32RFSaceLXo
+N62RXYtkeHL+D41Ct+I=
+-----END EC PUBLIC KEY-----`)
 )
 
 func main() {
 	verbose := flag.Bool("v", false, "Verbose output")
+	doUpgrade := flag.Bool("upgrade", false, "Upgrade to latest version")
 	flag.Usage = usage
 	flag.Parse()
 	dst := flag.Arg(0)
 
+	if *doUpgrade {
+		rels, err := upgrade.GithubReleases("calmh/psmcli", Version, true, false)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		exp := regexp.MustCompile("-" + runtime.GOOS + "-" + runtime.GOARCH)
+		for _, rel := range rels {
+			assets := upgrade.MatchingAssets(exp, rel)
+			if len(assets) == 1 {
+				fmt.Println("Trying", assets[0].Name, "...")
+				err = upgrade.ToURL(assets[0].URL, signingKey)
+				if err == nil {
+					fmt.Println("Upgraded")
+					return
+				}
+				fmt.Println(err)
+			}
+		}
+
+		os.Exit(1)
+	}
+
 	if dst == "" {
 		usage()
 		os.Exit(2)
@@ -226,6 +260,7 @@ func usage() {
 	fmt.Println()
 	fmt.Println("Usage:")
 	fmt.Println("  psmcli [-v] <host:port>")
+	fmt.Println("  psmcli -upgrade")
 }
 
 func printResponse(res response) {