From 7cd1b9d197c02e31e2f245644a4421de2103d81a Mon Sep 17 00:00:00 2001
From: Martin Angers <martin.n.angers@gmail.com>
Date: Mon, 24 Jul 2017 13:57:00 -0400
Subject: [PATCH] server: add checkupdates command

---
 server/checkupdates_cmd.go | 63 ++++++++++++++++++++++++++++++++++++++
 server/server.go           |  4 +++
 2 files changed, 67 insertions(+)
 create mode 100644 server/checkupdates_cmd.go

diff --git a/server/checkupdates_cmd.go b/server/checkupdates_cmd.go
new file mode 100644
index 0000000..0220606
--- /dev/null
+++ b/server/checkupdates_cmd.go
@@ -0,0 +1,63 @@
+package server
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
+
+	"github.com/harfangapps/regis-companion/resp"
+)
+
+type doer interface {
+	Do(req *http.Request) (*http.Response, error)
+}
+
+type checkUpdatesCmd struct {
+	client doer
+}
+
+const githubEndpoint = `https://api.github.com/repos/harfangapps/homebrew-harfangapps/releases/latest`
+
+// CHECKUPDATES
+func (c checkUpdatesCmd) Execute(cmdName string, req []string, s *Server) (interface{}, error) {
+	if len(req) != 1 {
+		return resp.Error(fmt.Sprintf("ERR wrong number of arguments for %v", cmdName)), nil
+	}
+
+	hreq, err := http.NewRequest("GET", githubEndpoint, nil)
+	if err != nil {
+		return resp.Error(fmt.Sprintf("ERR failed to create request: %v", err)), nil
+	}
+
+	res, err := c.client.Do(hreq)
+	if err != nil {
+		return resp.Error(fmt.Sprintf("ERR request failed: %v", err)), nil
+	}
+	defer res.Body.Close()
+
+	v, err := readRelease(res.Body)
+	if err != nil {
+		return resp.Error(fmt.Sprintf("ERR failed to read version: %v", err)), nil
+	}
+
+	// return true if the release is different than the current version
+	// (ideally, should be later than, but in practice different is enough)
+	return v != Version, nil
+}
+
+func readRelease(r io.Reader) (string, error) {
+	var obj struct {
+		TagName string `json:"tag_name"`
+	}
+
+	data, err := ioutil.ReadAll(r)
+	if err != nil {
+		return "", err
+	}
+	if err := json.Unmarshal(data, &obj); err != nil {
+		return "", err
+	}
+	return obj.TagName, nil
+}
diff --git a/server/server.go b/server/server.go
index ecd8a93..0e69714 100644
--- a/server/server.go
+++ b/server/server.go
@@ -5,6 +5,7 @@ import (
 	"expvar"
 	"fmt"
 	"net"
+	"net/http"
 	"sort"
 	"strings"
 	"sync"
@@ -45,6 +46,9 @@ var (
 
 func init() {
 	supportedCommands = map[string]command{
+		"checkupdates": checkUpdatesCmd{
+			client: &http.Client{Timeout: 10 * time.Second},
+		},
 		"command":       commandCmd{},
 		"gettunneladdr": getTunnelAddrCmd{},
 		"killtunnel":    killTunnelCmd{},