From 1ddf005a2388d80531d3bd2ecb8c732d34335ea3 Mon Sep 17 00:00:00 2001
From: Antoni Sawicki <tenox@google.com>
Date: Mon, 26 Oct 2020 01:32:09 -0700
Subject: [PATCH] if to switch to improve readability

---
 wrp.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/wrp.go b/wrp.go
index 369fbe5..10c577f 100644
--- a/wrp.go
+++ b/wrp.go
@@ -356,7 +356,8 @@ func (w wrpReq) capture() {
 	ismap[mappath] = w
 	var ssize string
 	var sw, sh int
-	if w.imgType == "gif" {
+	switch w.imgType {
+	case "gif":
 		i, err := png.Decode(bytes.NewReader(pngcap))
 		if err != nil {
 			log.Printf("%s Failed to decode screenshot: %s\n", w.req.RemoteAddr, err)
@@ -379,7 +380,7 @@ func (w wrpReq) capture() {
 		sw = i.Bounds().Max.X
 		sh = i.Bounds().Max.Y
 		log.Printf("%s Encoded GIF image: %s, Size: %s, Colors: %d, %dx%d\n", w.req.RemoteAddr, imgpath, ssize, w.colors, sw, sh)
-	} else if w.imgType == "png" {
+	case "png":
 		pngbuf := bytes.NewBuffer(pngcap)
 		img[imgpath] = *pngbuf
 		cfg, _, _ := image.DecodeConfig(pngbuf)