From 451ed1c69839c5f4b53a8087adbccff06dd524b3 Mon Sep 17 00:00:00 2001
From: Patrick G <geek1011@outlook.com>
Date: Thu, 27 Sep 2018 11:52:15 -0400
Subject: [PATCH] Improve update check

---
 bookbrowser.go | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/bookbrowser.go b/bookbrowser.go
index 9f15058b..6fc7dba6 100644
--- a/bookbrowser.go
+++ b/bookbrowser.go
@@ -1,9 +1,11 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
 	"io/ioutil"
 	"log"
+	"net/http"
 	"os"
 	"os/signal"
 	"path/filepath"
@@ -119,6 +121,7 @@ func main() {
 		if len(s.Indexer.BookList()) == 0 {
 			log.Fatalln("Fatal error: no books found")
 		}
+		checkUpdate()
 	}()
 
 	sigusr.Handle(func() {
@@ -131,3 +134,34 @@ func main() {
 		log.Fatalf("Error starting server: %s\n", err)
 	}
 }
+
+func checkUpdate() {
+	resp, err := http.Get("https://api.github.com/repos/geek1011/BookBrowser/releases/latest")
+	if err != nil {
+		return
+	}
+	defer resp.Body.Close()
+
+	buf, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return
+	}
+
+	if resp.StatusCode != 200 {
+		return
+	}
+
+	var obj struct {
+		URL string `json:"html_url"`
+		Tag string `json:"tag_name"`
+	}
+	if json.Unmarshal(buf, &obj) != nil {
+		return
+	}
+
+	if curversion != "dev" {
+		if !strings.HasPrefix(curversion, obj.Tag) {
+			log.Printf("Running version %s. Latest version is %s: %s\n", curversion, obj.Tag, obj.URL)
+		}
+	}
+}