From 1040830ead2d6a88044369cfc2e6f095a50bf536 Mon Sep 17 00:00:00 2001
From: Tarun Koyalwar <tarun@projectdiscovery.io>
Date: Fri, 21 Jul 2023 20:34:42 +0530
Subject: [PATCH] fix build test

---
 ratelimit_var.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/ratelimit_var.go b/ratelimit_var.go
index 246cd68..57edf60 100644
--- a/ratelimit_var.go
+++ b/ratelimit_var.go
@@ -6,6 +6,7 @@ import (
 	"strconv"
 	"strings"
 	"time"
+	"unicode"
 
 	stringsutil "github.com/projectdiscovery/utils/strings"
 	timeutil "github.com/projectdiscovery/utils/time"
@@ -147,8 +148,17 @@ func parseRateLimit(s string) (RateLimit, error) {
 	if err != nil {
 		return RateLimit{}, errors.New("parse error: " + err.Error())
 	}
+	timeValue := sArr[1]
+	if len(timeValue) > 0 {
+		// check if time is given ex: 1s
+		// if given value is just s (add prefix 1)
+		firstChar := timeValue[0]
+		if !unicode.IsDigit(rune(firstChar)) {
+			timeValue = "1" + timeValue
+		}
+	}
 
-	duration, err := timeutil.ParseDuration(sArr[1])
+	duration, err := timeutil.ParseDuration(timeValue)
 	if err != nil {
 		return RateLimit{}, errors.New("parse error: " + err.Error())
 	}