From a9d3b089553b5e57eb9aaa816114ac6c40d084f6 Mon Sep 17 00:00:00 2001
From: Ada <ada@gnous.eu>
Date: Thu, 28 Dec 2023 00:02:50 +0100
Subject: [PATCH] :sparkles: Plak expiration support

---
 db.go             |  2 +-
 main.go           | 14 +++++++++++-
 static/index.html |  2 +-
 utils.go          | 56 +++++++++++++++++++++++++++++++++++++++++++++++
 utils_test.go     | 43 ++++++++++++++++++++++++++++++++++++
 5 files changed, 114 insertions(+), 3 deletions(-)
 create mode 100644 utils_test.go

diff --git a/db.go b/db.go
index d6b60b7..a156541 100644
--- a/db.go
+++ b/db.go
@@ -35,7 +35,7 @@ func insertPaste(key string, content string, secret string, ttl time.Duration) {
 	}
 	err = db.HSet(ctx, key, "secret", hash.secret)
 	if ttl > -1 {
-		db.Do(ctx, key, ttl)
+		db.Expire(ctx, key, ttl)
 	}
 }
 
diff --git a/main.go b/main.go
index 2fdec8a..3064b46 100644
--- a/main.go
+++ b/main.go
@@ -8,6 +8,7 @@ import (
 	"log"
 	"net/http"
 	"strings"
+	"time"
 )
 
 var currentConfig Config
@@ -60,7 +61,18 @@ func handleRequest(w http.ResponseWriter, r *http.Request) {
 			secret := GenerateSecret()
 			url := "/" + GenerateUrl()
 			content := r.FormValue("content")
-			insertPaste(url, content, secret, -1)
+			rawExpiration := r.FormValue("exp")
+			expiration, err := ParseExpiration(rawExpiration)
+			if err != nil {
+				log.Println(err)
+			} else if expiration == 0 {
+				insertPaste(url, content, secret, -1)
+			} else if expiration == -1 {
+				w.WriteHeader(http.StatusBadRequest)
+			} else {
+				insertPaste(url, content, secret, time.Duration(expiration*int(time.Second)))
+			}
+
 			http.Redirect(w, r, url, http.StatusSeeOther)
 		} else {
 			w.WriteHeader(http.StatusMethodNotAllowed)
diff --git a/static/index.html b/static/index.html
index 40eb943..dfaa289 100644
--- a/static/index.html
+++ b/static/index.html
@@ -27,7 +27,7 @@
                     <input id="password" type="text">
                 </li>
                 <li><label for="exp">Expiration?</label>
-                    <input id="exp" type="date"></li>
+                    <input id="exp" name="exp" type="text" placeholder="vide, 0 ou 1d1h1m1s"></li>
                 <li>
                     <label for="type">Type</label>
                     <select id="type" name="type">
diff --git a/utils.go b/utils.go
index 2b2753e..5d23dce 100644
--- a/utils.go
+++ b/utils.go
@@ -3,8 +3,11 @@ package main
 import (
 	"crypto/rand"
 	"encoding/hex"
+	"fmt"
 	"log"
 	mathrand "math/rand"
+	"strconv"
+	"strings"
 )
 
 func GenerateUrl() string {
@@ -34,3 +37,56 @@ func UrlExist(url string) bool {
 func VerifySecret(url string, secret string) bool {
 	return secret == db.HGet(ctx, url, "secret").Val()
 }
+
+func parseIntBeforeSeparator(source *string, sep string) (int, error) { // return -1 if error, only accept positive number
+	var value int
+	var err error
+	if strings.Contains(*source, sep) {
+		value, err = strconv.Atoi(strings.Split(*source, sep)[0])
+		if err != nil {
+			log.Println(err)
+			return -1, fmt.Errorf("parseIntBeforeSeparator : \"%s\" : cannot parse value as int", *source)
+		}
+		if value < 0 { // Only positive value is correct
+			return -1, fmt.Errorf("parseIntBeforeSeparator : \"%s\" : format only take positive value", *source)
+		}
+		*source = strings.Join(strings.Split(*source, sep)[1:], "")
+	}
+	return value, nil
+}
+
+func ParseExpiration(source string) (int, error) { // return -1 if error
+	var expiration int
+	var tempOutput int
+	var err error
+	errMessage := "ParseExpiration : \"%s\" : invalid syntax"
+	if source == "0" {
+		return 0, nil
+	}
+	tempOutput, err = parseIntBeforeSeparator(&source, "d")
+	expiration = tempOutput * 86400
+	if err != nil {
+		log.Println(err)
+		return -1, fmt.Errorf(errMessage, source)
+	}
+	tempOutput, err = parseIntBeforeSeparator(&source, "h")
+	expiration += tempOutput * 3600
+	if err != nil {
+		log.Println(err)
+		return -1, fmt.Errorf(errMessage, source)
+	}
+	tempOutput, err = parseIntBeforeSeparator(&source, "m")
+	expiration += tempOutput * 60
+	if err != nil {
+		log.Println(err)
+		return -1, fmt.Errorf(errMessage, source)
+	}
+	tempOutput, err = parseIntBeforeSeparator(&source, "s")
+	expiration += tempOutput * 1
+	if err != nil {
+		log.Println(err)
+		return -1, fmt.Errorf(errMessage, source)
+	}
+
+	return expiration, nil
+}
diff --git a/utils_test.go b/utils_test.go
new file mode 100644
index 0000000..4beac4c
--- /dev/null
+++ b/utils_test.go
@@ -0,0 +1,43 @@
+package main
+
+import "testing"
+
+func TestParseExpirationFull(t *testing.T) { // test parseExpirationFull with all valid separator
+	result, _ := ParseExpiration("2d1h3m47s")
+	correctValue := 176627
+	if result != correctValue {
+		t.Fatal("Error in parseExpirationFull, want : ", correctValue, "got : ", result)
+	}
+}
+
+func TestParseExpirationMissing(t *testing.T) { // test parseExpirationFull with all valid separator
+	result, _ := ParseExpiration("1h47s")
+	correctValue := 3647
+	if result != correctValue {
+		t.Fatal("Error in ParseExpirationFull, want : ", correctValue, "got : ", result)
+	}
+}
+
+func TestParseExpirationNull(t *testing.T) { // test ParseExpirationFull with all valid separator
+	result, _ := ParseExpiration("0")
+	correctValue := 0
+	if result != correctValue {
+		t.Fatal("Error in ParseExpirationFull, want : ", correctValue, "got : ", result)
+	}
+}
+
+func TestParseExpirationNegative(t *testing.T) { // test ParseExpirationFull with all valid separator
+	result, _ := ParseExpiration("-42h1m4s")
+	correctValue := -1
+	if result != correctValue {
+		t.Fatal("Error in ParseExpirationFull, want : ", correctValue, "got : ", result)
+	}
+}
+
+func TestParseExpirationInvalid(t *testing.T) { // test ParseExpirationFull with all valid separator
+	result, _ := ParseExpiration("8h42h1m1d4s")
+	correctValue := -1
+	if result != correctValue {
+		t.Fatal("Error in ParseExpirationFull, want : ", correctValue, "got : ", result)
+	}
+}