Add local caching of team logos to prevent invalid URLs

Author: achhabra2

.DS_Store

@@ -3,12 +3,15 @@ package main
 import (
 	"encoding/json"
 	"fmt"
+	"io"
 	"io/ioutil"
+	"log"
 	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
 )
 
-// const CircuitAPIURL = "https://indy-gaming-league-api.herokuapp.com/api/circuits?data={%22motherCircuitId%22:%225e4b290a420594ace7e97726%22,%22live%22:false,%22active%22:true,%22hidden%22:false,%20%22game%22:%22KILLER%20QUEEN%20BLACK%22}"
-// const CircuitAPIURL = "https://indy-gaming-league-api.herokuapp.com/api/circuits?data=%7B%22live%22:true,%22hidden%22:false,%22game%22:%22KILLER%20QUEEN%20BLACK%22%7D"
 const CircuitAPIURL = "https://indy-gaming-league-api.herokuapp.com/api/circuits?data=%7B%22motherCircuitId%22:%225ec8428a2cbb1c4439c29e6c%22,%22live%22:false,%22active%22:true,%22hidden%22:false,%22game%22:%22KILLER%20QUEEN%20BLACK%22%7D"
 
 type IGLCircuit struct {
@@ -41,3 +44,45 @@ func GetIGLCircuits(c chan []IGLCircuit) {
 
 	c <- result.Circuits
 }
+
+// DownloadFile will download a url to a local file. It's efficient because it will
+// write as it downloads and not load the whole file into memory.
+func DownloadFile(filepath string, url string) error {
+
+	// Get the data
+	resp, err := http.Get(url)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	// Create the file
+	out, err := os.Create(filepath)
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+
+	// Write the body to file
+	_, err = io.Copy(out, resp.Body)
+	return err
+}
+
+// UpdateTeamLogo will Take Team as an input and download the logo locally
+// as a png file
+func UpdateTeamLogo(t *Team) {
+
+	if t.Img == "/avatar.png" {
+		return
+	}
+
+	log.Printf("Downloading Logo %s\n", t.Img)
+	logoFile := fmt.Sprintf("%s.png", strings.Replace(t.Name, " ", "", -1))
+	err := DownloadFile(filepath.Join(logoPath, logoFile), t.Img)
+	if err != nil {
+		log.Println("Could not save logo image")
+		log.Println(err)
+		return
+	}
+	t.Img = "http://localhost:8080/logo/" + logoFile
+}

igl.go

@@ -7,6 +7,7 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"path/filepath"
 	"strconv"
 	"time"
 
@@ -39,6 +40,7 @@ type Stats struct {
 var h Team = Team{"Blue Team", PlaceholderImage, 1, 1, Stats{1, 1, 1, 1}}
 var a Team = Team{"Gold Team", PlaceholderImage, 1, 1, Stats{1, 1, 1, 1}}
 var s Scoreboard = Scoreboard{&h, &a, 0, 0, 0, 0}
+var logoPath string
 
 func setupLogs() {
 	f, err := os.OpenFile("./output.log", os.O_APPEND|os.O_CREATE|os.O_RDWR, 0666)
@@ -57,6 +59,9 @@ func main() {
 	// Configure log output to file
 	setupLogs()
 
+	// Create Directory For Logos
+	SetupLogoDirectory()
+
 	// Setup HTTP handler and websocket handler
 	StartHTTPServer()
 
@@ -151,24 +156,30 @@ type IGLApiResponse struct {
 	} `json:"data"`
 }
 
-// OLD JSON parser
-// results := result["data"].([]interface{})
-// teams := []Team{}
-// for _, v := range results {
-// 	t := v.(map[string]interface{})["team"]
-// 	s := v.(map[string]interface{})["stats"]
-// 	team := Team{}
-// 	team.Name = t.(map[string]interface{})["formattedName"].(string)
-// 	team.Tier = int(t.(map[string]interface{})["tier"].(float64))
-// 	team.Div = int(t.(map[string]interface{})["div"].(float64))
-// 	if t.(map[string]interface{})["logo"] != nil {
-// 		team.Img = t.(map[string]interface{})["logo"].(string)
-// 	} else {
-// 		team.Img = KQBAvatarImage
-// 	}
-// 	team.Stats.GamesWon, _ = strconv.Atoi(s.(map[string]interface{})["Games Won"].(string))
-// 	team.Stats.GamesLost, _ = strconv.Atoi(s.(map[string]interface{})["Games Lost"].(string))
-// 	team.Stats.MatchesWon, _ = strconv.Atoi(s.(map[string]interface{})["Matches Won"].(string))
-// 	team.Stats.MatchesLost, _ = strconv.Atoi(s.(map[string]interface{})["Matches Lost"].(string))
-// 	teams = append(teams, team)
-// }
+// SetupLogoDirectory creates a directory to cache the local logo files
+func SetupLogoDirectory() {
+
+	// Get Current Working Directory
+	path, err := os.Getwd()
+	if err != nil {
+		log.Println(err)
+	}
+	fmt.Println(path)
+
+	// Set Logo Path
+	logoPath = filepath.Join(path, "logo")
+	fmt.Println(logoPath)
+
+	// Create directory if it doesn't exist
+	if _, err := os.Stat(logoPath); os.IsNotExist(err) {
+		os.Mkdir(logoPath, 0755)
+	}
+}
+
+func tidyUp() {
+
+	// Remove logopath directory and files
+	os.RemoveAll(logoPath)
+	os.Remove("./output.log")
+	log.Println("Exiting...")
+}

main.go

@@ -226,7 +226,7 @@
         <article class="wrapper-left">
             <div class="one">
                 <figure class="image is-64x64 home-outline">
-                    <img id="home-img" src="https://bulma.io/images/placeholders/128x128.png" alt="Image">
+                    <img id="home-img" src="https://bulma.io/images/placeholders/128x128.png" alt="http://localhost:8080/avatar.png">
                 </figure>
             </div>
             <div class="two">
@@ -275,7 +275,7 @@
             </div>
             <div class="four">
                 <figure class="image is-64x64 away-outline">
-                    <img id="away-img" src="https://bulma.io/images/placeholders/128x128.png" alt="Image">
+                    <img id="away-img" src="https://bulma.io/images/placeholders/128x128.png" alt="http://localhost:8080/avatar.png">
                 </figure>
             </div>
         </article>

static/index.html

@@ -112,6 +112,8 @@ func IGLTeamSelection(w fyne.Window, apiUrl string) *fyne.Container {
 	saveButton := widget.NewButton("Start Scoreboard", func() {
 		log.Println("Saved.")
 		s = Scoreboard{&blueTeam, &goldTeam, 0, 0, 0, 0}
+		UpdateTeamLogo(&blueTeam)
+		UpdateTeamLogo(&goldTeam)
 		StartScoreboard(w)
 	})
 
@@ -304,7 +306,3 @@ func ProgressIndicator() *fyne.Container {
 	container := fyne.NewContainerWithLayout(layout.NewVBoxLayout(), label, infinite)
 	return container
 }
-
-func tidyUp() {
-	log.Println("Exiting...")
-}

ui.go

@@ -82,6 +82,7 @@ func StartHTTPServer() {
 	http.HandleFunc("/echo", echo)
 	box := packr.New("static", "./static")
 	http.Handle("/", http.FileServer(box))
+	http.Handle("/logo", http.FileServer(http.Dir(logoPath)))
 	go func() {
 		log.Fatal(http.ListenAndServe(":8080", nil))
 	}()