From 3f7b800423e91bc1508662a65a6dc4bec1eb1944 Mon Sep 17 00:00:00 2001
From: Noah Petherbridge <root@kirsle.net>
Date: Sat, 23 Dec 2017 13:32:26 -0800
Subject: [PATCH] Fix toggling on/off of Redis at runtime

---
 core/admin.go    | 26 ++++++++++++++++++++++++--
 core/markdown.go |  2 +-
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/core/admin.go b/core/admin.go
index f694cf5..fd8d4ab 100644
--- a/core/admin.go
+++ b/core/admin.go
@@ -1,10 +1,13 @@
 package core
 
 import (
+	"fmt"
 	"net/http"
 	"strconv"
 
 	"github.com/gorilla/mux"
+	"github.com/kirsle/blog/core/caches/null"
+	"github.com/kirsle/blog/core/caches/redis"
 	"github.com/kirsle/blog/core/forms"
 	"github.com/kirsle/blog/core/models/settings"
 	"github.com/urfave/negroni"
@@ -43,12 +46,12 @@ func (b *Blog) SettingsHandler(w http.ResponseWriter, r *http.Request) {
 			Title:        r.FormValue("title"),
 			AdminEmail:   r.FormValue("admin-email"),
 			URL:          r.FormValue("url"),
-			RedisEnabled: r.FormValue("redis-enabled") == "true",
+			RedisEnabled: len(r.FormValue("redis-enabled")) > 0,
 			RedisHost:    r.FormValue("redis-host"),
 			RedisPort:    redisPort,
 			RedisDB:      redisDB,
 			RedisPrefix:  r.FormValue("redis-prefix"),
-			MailEnabled:  r.FormValue("mail-enabled") == "true",
+			MailEnabled:  len(r.FormValue("mail-enabled")) > 0,
 			MailSender:   r.FormValue("mail-sender"),
 			MailHost:     r.FormValue("mail-host"),
 			MailPort:     mailPort,
@@ -78,6 +81,25 @@ func (b *Blog) SettingsHandler(w http.ResponseWriter, r *http.Request) {
 		} else {
 			// Save the settings.
 			settings.Save()
+
+			// Reset Redis configuration.
+			if settings.Redis.Enabled {
+				cache, err := redis.New(
+					fmt.Sprintf("%s:%d", settings.Redis.Host, settings.Redis.Port),
+					settings.Redis.DB,
+					settings.Redis.Prefix,
+				)
+				if err != nil {
+					b.Flash(w, r, "Error connecting to Redis: %s", err)
+					b.Cache = null.New()
+				} else {
+					b.Cache = cache
+				}
+			} else {
+				b.Cache = null.New()
+			}
+			b.DB.Cache = b.Cache
+
 			b.FlashAndReload(w, r, "Settings have been saved!")
 			return
 		}
diff --git a/core/markdown.go b/core/markdown.go
index be1448b..28953ba 100644
--- a/core/markdown.go
+++ b/core/markdown.go
@@ -115,7 +115,7 @@ func (b *Blog) Pygmentize(language, source string) (string, error) {
 	cacheKey := "pygmentize:" + hash
 
 	// Do we have it cached?
-	if cached, err := b.Cache.Get(cacheKey); err == nil {
+	if cached, err := b.Cache.Get(cacheKey); err == nil && len(cached) > 0 {
 		return string(cached), nil
 	}