From d87b462b1d696116c2ae6540cbc61ea10b79e21b Mon Sep 17 00:00:00 2001
From: Eliot Whalan <ewhal@pantsu.cat>
Date: Mon, 4 Jul 2016 09:59:31 +1000
Subject: [PATCH] Create paste duratation function

---
 main.go | 57 +++++++++++++++++++++------------------------------------
 1 file changed, 21 insertions(+), 36 deletions(-)

diff --git a/main.go b/main.go
index c4c744c..50da3aa 100644
--- a/main.go
+++ b/main.go
@@ -78,6 +78,25 @@ func hash(paste string) string {
 	sha := base64.URLEncoding.EncodeToString(hasher.Sum(nil))
 	return sha
 }
+func durationFromExpiry(expiry string) time.Duration {
+	switch expiry {
+	case "5 minutes":
+		return time.Minute * 5
+	case "1 hour":
+		return time.Hour + 1 // XXX: did you mean '*'?
+	case "1 day":
+		return time.Hour * 24
+	case "1 week":
+		return time.Hour * 24 * 7
+	case "1 month":
+		return time.Hour * 24 * 30
+	case "1 year":
+		return time.Hour * 24 * 365
+	case "forever":
+		return time.Hour * 24 * (365 * 20)
+	}
+	return time.Hour * 24 * (365 * 20)
+}
 
 func save(raw string, lang string, title string, expiry string) []string {
 	db, err := sql.Open("mysql", DATABASE)
@@ -101,43 +120,9 @@ func save(raw string, lang string, title string, expiry string) []string {
 	} else {
 		url = ADDRESS + "/p/" + id + "/" + lang
 	}
-	now := time.Now()
-	var expiryTime string
+	const timeFormat = "2006-01-02 15:04:05"
+	expiryTime := time.Now().Add(durationFromExpiry(expiry)).Format(timeFormat)
 
-	switch expiry {
-	case "5 minutes":
-		expiryTime = now.Add(time.Minute * 5).Format("2006-01-02 15:04:05")
-		break
-
-	case "1 hour":
-		expiryTime = now.Add(time.Hour + 1).Format("2006-01-02 15:04:05")
-		break
-
-	case "1 day":
-		expiryTime = now.Add(time.Hour * 24 * 1).Format("2006-01-02 15:04:05")
-		break
-
-	case "1 week":
-		expiryTime = now.Add(time.Hour * 24 * 7).Format("2006-01-02 15:04:05")
-		break
-
-	case "1 month":
-		expiryTime = now.Add(time.Hour * 24 * 30).Format("2006-01-02 15:04:05")
-		break
-
-	case "1 year":
-		expiryTime = now.Add(time.Hour * 24 * 365).Format("2006-01-02 15:04:05")
-		break
-
-	case "forever":
-		expiryTime = now.Add(time.Hour * 24 * (365 * 20)).Format("2006-01-02 15:04:05")
-		break
-
-	default:
-		expiryTime = now.Add(time.Hour * 24 * (365 * 20)).Format("2006-01-02 15:04:05")
-		break
-
-	}
 	delKey := uniuri.NewLen(40)
 	dataEscaped := html.EscapeString(raw)