From 5febcceecb34a69bac14e58c3db4a8695b91f21e Mon Sep 17 00:00:00 2001
From: rishavvajpayee <reshavvajpayee@gmail.com>
Date: Tue, 29 Oct 2024 13:50:59 +0530
Subject: [PATCH] added commands left to headers

---
 internal/server/http.go | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/internal/server/http.go b/internal/server/http.go
index b450bb3..cd4f631 100644
--- a/internal/server/http.go
+++ b/internal/server/http.go
@@ -4,12 +4,14 @@ import (
 	"context"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"log/slog"
 	"net/http"
 	"strconv"
 	"strings"
 	"time"
 
+	"server/config"
 	"server/internal/db"
 	"server/internal/middleware"
 	"server/internal/server/utils"
@@ -188,6 +190,26 @@ func (s *HTTPServer) getNextCleanupTime() (int64, error) {
 }
 
 func (s *HTTPServer) getCommandsLeft() (int64, error) {
-	// clarification required
-	return 1, nil
+	configValue := config.LoadConfig()
+	currentWindow := time.Now().Unix() / int64(configValue.Server.RequestWindowSec)
+	key := fmt.Sprintf("request_count:%d", currentWindow)
+
+	val, err := s.DiceClient.Client.Get(context.Background(), key).Result()
+	if err != nil {
+		if errors.Is(err, dicedb.Nil) {
+			return 1000, nil
+		}
+		return 0, err
+	}
+
+	requestCount, err := strconv.ParseInt(val, 10, 64)
+	if err != nil {
+		return 0, err
+	}
+
+	remaining := int64(1000) - requestCount
+	if remaining < 0 {
+		remaining = 0
+	}
+	return remaining, nil
 }