Check if post id is correct, starts with 'C'

Wikidepia · Jul 7, 2024 · 277dc44 · 277dc44
1 parent 015447a
commit 277dc44
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/handlers/scraper/data.go b/handlers/scraper/data.go
@@ -54,6 +54,10 @@ type InstaData struct {
 }
 
 func GetData(postID string) (*InstaData, error) {
+	if postID[0] != 'C' {
+		return nil, errors.New("postID is not a valid Instagram post ID")
+	}
+
 	cacheInstaData, closer, err := DB.Get(utils.S2B(postID))
 	if err != nil && err != pebble.ErrNotFound {
 		log.Error().Str("postID", postID).Err(err).Msg("Failed to get data from cache")