Check for new versions and upgrade from webgui. closes #1

commander

@@ -199,15 +199,13 @@ case $txt in
 "server-shutdown")
     halt
     ;;
+"version-update")
+    /home/labca/labca/install &>>$LOGFILE
+    ;;
 *)
     echo "Unknown command '$txt'. ERROR!"
     exit 1
     ;;
 esac
 
 echo "ok"
-
-
-# TODO:
-# upgrade the labca stuff
-

gui/main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"bufio"
 	"bytes"
+	"context"
 	"crypto/aes"
 	"crypto/cipher"
 	"crypto/rand"
@@ -11,7 +12,9 @@ import (
 	"errors"
 	"fmt"
 	"github.com/biz/templates"
+	"github.com/dustin/go-humanize"
 	_ "github.com/go-sql-driver/mysql"
+	"github.com/google/go-github/github"
 	"github.com/gorilla/mux"
 	"github.com/gorilla/securecookie"
 	"github.com/gorilla/sessions"
@@ -39,20 +42,23 @@ import (
 )
 
 const (
-	writeWait  = 10 * time.Second
-	pongWait   = 60 * time.Second
-	pingPeriod = (pongWait * 9) / 10
+	writeWait      = 10 * time.Second
+	pongWait       = 60 * time.Second
+	pingPeriod     = (pongWait * 9) / 10
+	updateInterval = 24 * time.Hour
 )
 
 var (
-	appSession    *sessions.Session
-	restartSecret string
-	sessionStore  *sessions.CookieStore
-	tmpls         *templates.Templates
-	version       string
-	dbConn        string
-	dbType        string
-	isDev         bool
+	appSession      *sessions.Session
+	restartSecret   string
+	sessionStore    *sessions.CookieStore
+	tmpls           *templates.Templates
+	version         string
+	dbConn          string
+	dbType          string
+	isDev           bool
+	updateAvailable bool
+	updateChecked   time.Time
 
 	upgrader = websocket.Upgrader{
 		ReadBufferSize:  1024,
@@ -232,6 +238,43 @@ func errorHandler(w http.ResponseWriter, r *http.Request, err error, status int)
 	}
 }
 
+func checkUpdates(forced bool) ([]string, []string) {
+	var versions []string
+	var descriptions []string
+
+	if forced || updateChecked.Add(updateInterval).Before(time.Now()) {
+		latest := ""
+		newer := true
+
+		client := github.NewClient(nil)
+
+		if releases, _, err := client.Repositories.ListReleases(context.Background(), "hakwerk", "labca", nil); err == nil {
+			for i := 0; i < len(releases); i++ {
+				release := releases[i]
+				if !*release.Draft {
+					if !*release.Prerelease || isDev {
+						if latest == "" {
+							latest = *release.Name
+						}
+						if *release.Name == version {
+							newer = false
+						}
+						if newer {
+							versions = append(versions, *release.Name)
+							descriptions = append(descriptions, *release.Body)
+						}
+					}
+				}
+			}
+
+			updateChecked = time.Now()
+			updateAvailable = (len(releases) > 0) && (latest != version)
+		}
+	}
+
+	return versions, descriptions
+}
+
 func rootHandler(w http.ResponseWriter, r *http.Request) {
 	if !viper.GetBool("config.complete") {
 		http.Redirect(w, r, r.Header.Get("X-Request-Base")+"/setup", http.StatusFound)
@@ -240,6 +283,10 @@ func rootHandler(w http.ResponseWriter, r *http.Request) {
 
 	dashboardData, err := CollectDashboardData(w, r)
 	if err == nil {
+		checkUpdates(false)
+		dashboardData["UpdateAvailable"] = updateAvailable
+		dashboardData["UpdateChecked"] = humanize.RelTime(updateChecked, time.Now(), "", "")
+
 		render(w, r, "dashboard", dashboardData)
 	}
 }
@@ -798,6 +845,24 @@ func (res *Result) ManageComponents(w http.ResponseWriter, r *http.Request, acti
 	}
 }
 
+func _checkUpdatesHandler(w http.ResponseWriter, r *http.Request) {
+	res := struct {
+		Success         bool
+		UpdateAvailable bool
+		UpdateChecked   string
+		Versions        []string
+		Descriptions    []string
+		Errors          map[string]string
+	}{Success: true, Errors: make(map[string]string)}
+
+	res.Versions, res.Descriptions = checkUpdates(true)
+	res.UpdateAvailable = updateAvailable
+	res.UpdateChecked = humanize.RelTime(updateChecked, time.Now(), "", "")
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(res)
+}
+
 func _managePostDispatch(w http.ResponseWriter, r *http.Request, action string) bool {
 	if action == "backup-restore" || action == "backup-delete" || action == "backup-now" {
 		_backupHandler(w, r)
@@ -829,6 +894,11 @@ func _managePostDispatch(w http.ResponseWriter, r *http.Request, action string)
 		return true
 	}
 
+	if action == "version-check" {
+		_checkUpdatesHandler(w, r)
+		return true
+	}
+
 	return false
 }
 
@@ -858,6 +928,8 @@ func _managePost(w http.ResponseWriter, r *http.Request) {
 		"update-config",
 		"update-email",
 		"send-email",
+		"version-check",
+		"version-update",
 	} {
 		if a == action {
 			actionKnown = true
@@ -878,7 +950,7 @@ func _managePost(w http.ResponseWriter, r *http.Request) {
 		res.Message = "Command failed - see LabCA log for any details"
 	}
 
-	if action != "server-restart" && action != "server-shutdown" {
+	if action != "server-restart" && action != "server-shutdown" && action != "version-update" {
 		res.ManageComponents(w, r, action)
 	}
 
@@ -890,6 +962,10 @@ func _manageGet(w http.ResponseWriter, r *http.Request) {
 	manageData := make(map[string]interface{})
 	manageData["RequestBase"] = r.Header.Get("X-Request-Base")
 
+	checkUpdates(false)
+	manageData["UpdateAvailable"] = updateAvailable
+	manageData["UpdateChecked"] = humanize.RelTime(updateChecked, time.Now(), "", "")
+
 	components := _parseComponents(getLog(w, r, "components"))
 	for i := 0; i < len(components); i++ {
 		if components[i].Name == "NGINX Webserver" {
@@ -2292,6 +2368,8 @@ func init() {
 	dbType = viper.GetString("db.type")
 
 	version = viper.GetString("version")
+
+	updateAvailable = false
 }
 
 func main() {

gui/setup.sh

@@ -9,6 +9,7 @@ if [ ! -e bin/labca ]; then
     go get github.com/biz/templates
     go get github.com/go-sql-driver/mysql
     go get github.com/dustin/go-humanize
+    go get github.com/google/go-github/github
     go get github.com/gorilla/mux
     go get github.com/gorilla/securecookie
     go get github.com/gorilla/sessions

gui/templates/views/dashboard.tmpl

@@ -127,7 +127,13 @@
         <div class="col-lg-4">
             <div class="panel panel-default">
                 <div class="panel-heading">
-                    <i class="fa fa-fw fa-desktop"></i> System Overview
+                    <span class="pull-left"><i class="fa fa-fw fa-desktop"></i> System Overview</span>
+                    {{ if .UpdateAvailable }}
+                    <span class="pull-right">
+                        <a class="update" href="{{ .RequestBase }}/manage"><i class="fa fa-fw fa-rocket"></i> <i>update available!</i></a>
+                    </span>
+                    {{ end }}
+                    <div class="clearfix"></div>
                 </div>
                 <div class="panel-body">
                     <div class="table-responsive">