update: fix donottouch branch

handlers/shawty.go

@@ -30,16 +30,27 @@ func Shawty(w http.ResponseWriter, r *http.Request) {
 		longUrl = "https://" + longUrl
 	}
 
+	errorTempl := template.Must(template.ParseFiles("./partial-html/short-link-error.html"))
+
 	// Parse the URL to validate it and check its scheme
 	parsedUrl, err := url.Parse(longUrl)
 	if err != nil || parsedUrl.Scheme != "https" {
-		w.Write([]byte("Invalid URL. Only HTTPS schema is allowed"))
+		asserts.NoErr(errorTempl.Execute(w, "Invalid URL. Only HTTPS schema is allowed"), "Failed to execute template short-link-error.html")
 		return
 	}
 
-	// Check if URL is valid by checking if it contains `.`
+	// Check if URL contains a TLD (Top-Level Domain)
 	if !strings.Contains(longUrl, ".") {
-		w.Write([]byte("Enter a valid URL"))
+		asserts.NoErr(errorTempl.Execute(w, "The URL doesn't contain TLD (Top-Level Domain)"), "Failed to execute template short-link-error.html")
+		return
+	} else if split := strings.SplitN(longUrl, ".", 2); split[1] == "" {
+		asserts.NoErr(errorTempl.Execute(w, "The URL doesn't contain TLD (Top-Level Domain)"), "Failed to execute template short-link-error.html")
+		return
+	}
+
+	// Check the lenght of the URL
+	if len(longUrl) > 1000 {
+		asserts.NoErr(errorTempl.Execute(w, "The URL is too long, Max URL lenght is 1000 characters"), "Failed to execute template short-link-error.html")
 		return
 	}
 

partial-html/short-link-error.html

@@ -0,0 +1,3 @@
+<div class="text-lg font-mono mb-2 mt-10 text-center">
+  <p class="text-red-400">{{ . }}</p>
+</div>