Add check updates functionality to fix #35

internal/config/config.go

@@ -71,7 +71,7 @@ type Config struct {
 	path   string
 	Socket string `json:",omitempty"`
 
-	CheckVersion bool `json:",omitempty"`
+	CheckUpdates bool `json:",omitempty"`
 	Proxied      bool
 	ExposePorts  []string `json:",omitempty"`
 	NAT          *bool

ui/check_updates.go

@@ -0,0 +1,63 @@
+package ui
+
+import (
+	"encoding/json"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/NHAS/wag/internal/config"
+)
+
+type githubResponse struct {
+	Body       string
+	Prerelease bool   `json:"prerelease"`
+	TagName    string `json:"tag_name"`
+	Published  string `json:"published_at"`
+	Url        string `json:"html_url"`
+}
+
+type Update struct {
+	New           bool
+	UpdateVersion string
+	Message       []string
+	Url           string
+	Released      string
+}
+
+var (
+	mostRecentUpdate *Update
+	lastChecked      time.Time
+)
+
+func getUpdate() Update {
+
+	if !config.Values().CheckUpdates {
+		return Update{}
+	}
+
+	if time.Now().After(lastChecked.Add(15*time.Minute)) || mostRecentUpdate == nil {
+		resp, err := http.Get("https://api.github.com/repos/NHAS/wag/releases/latest")
+		if err != nil {
+			return Update{}
+		}
+		defer resp.Body.Close()
+
+		var gr githubResponse
+		err = json.NewDecoder(resp.Body).Decode(&gr)
+		if err != nil {
+			return Update{}
+		}
+
+		mostRecentUpdate = &Update{
+			UpdateVersion: gr.TagName,
+			Message:       strings.Split(gr.Body, "\r\n"),
+			Url:           gr.Url,
+			Released:      gr.Published,
+		}
+	}
+
+	mostRecentUpdate.New = strings.Split(config.Version, "-")[0] != mostRecentUpdate.UpdateVersion
+
+	return *mostRecentUpdate
+}

ui/structs.go

@@ -1,6 +1,7 @@
 package ui
 
 type Page struct {
+	Update
 	Description string
 	Title       string
 	User        string

ui/templates/menus.html

@@ -170,24 +170,34 @@ <h6 class="collapse-header">Tools:</h6>
                     <!-- Topbar Navbar -->
                     <ul class="navbar-nav ml-auto">
 
+
+                        {{if .Url}}
                         <li class="nav-item dropdown no-arrow mx-1">
                             <a class="nav-link dropdown-toggle" href="#" id="alertsDropdown" role="button"
                                 data-toggle="dropdown" aria-haspopup="true" aria-expanded="false">
                                 <i class="icon-bell"></i>
                                 <!-- Counter - Alerts -->
+                                {{if .New}}
                                 <span class="badge badge-danger badge-counter">1</span>
+                                {{end}}
                             </a>
                             <!-- Dropdown - Alerts -->
                             <div class="dropdown-list dropdown-menu dropdown-menu-right shadow animated--grow-in"
                                 aria-labelledby="alertsDropdown">
                                 <a class="dropdown-item d-flex align-items-center" href="#">
                                     <div>
-                                        <div class="small text-gray-500">{{.UpdateDate}}</div>
-                                        <span class="font-weight-bold">Wag {{.NewVersion}} is now available!</span>
+                                        <div class="small text-gray-500">{{.Released}}</div>
+                                        <span class="font-weight-bold">{{.UpdateVersion}}</span>
+                                        {{range $val := .Message}}
+                                        <p>{{$val}}</p>
+                                        {{end}}
                                     </div>
                                 </a>
                             </div>
                         </li>
+                        {{end}}
+
+
 
                         <!-- Nav Item - User Information -->
                         <li class="nav-item dropdown no-arrow">

ui/ui_webserver.go

@@ -213,6 +213,7 @@ func populateDashboard(w http.ResponseWriter, r *http.Request) {
 
 	d := Dashboard{
 		Page: Page{
+			Update:      getUpdate(),
 			Description: "Dashboard",
 			Title:       "Dashboard",
 			User:        u.Username,
@@ -403,6 +404,7 @@ func StartWebServer(errs chan<- error) {
 				XDPState string
 			}{
 				Page: Page{
+					Update:      getUpdate(),
 					Description: "Firewall state page",
 					Title:       "Firewall",
 					User:        u.Username,
@@ -599,6 +601,7 @@ func StartWebServer(errs chan<- error) {
 
 			d := GeneralSettings{
 				Page: Page{
+					Update:      getUpdate(),
 					Description: "Wag settings",
 					Title:       "Settings - General",
 					User:        u.Username,
@@ -755,6 +758,7 @@ func changePassword(w http.ResponseWriter, r *http.Request) {
 
 	d := ChangePassword{
 		Page: Page{
+			Update:      getUpdate(),
 			Description: "Change password page",
 			Title:       "Change password",
 			User:        u.Username,