From 3e307658fa2e01adb515a3c1943e968f7206fcfd Mon Sep 17 00:00:00 2001
From: stefanodvx <69367859+stefanodvx@users.noreply.github.com>
Date: Mon, 21 Apr 2025 22:30:52 +0200
Subject: [PATCH] instagram: fixes share urls

---
 ext/instagram/main.go |  6 +++++-
 ext/instagram/util.go | 22 ++++++++++++++++++++--
 ext/pinterest/main.go |  2 +-
 ext/tiktok/main.go    |  2 +-
 util/misc.go          | 10 ++++++----
 5 files changed, 33 insertions(+), 9 deletions(-)

diff --git a/ext/instagram/main.go b/ext/instagram/main.go
index 21755bb..b3f5912 100644
--- a/ext/instagram/main.go
+++ b/ext/instagram/main.go
@@ -74,7 +74,11 @@ var ShareURLExtractor = &models.Extractor{
 
 	Run: func(ctx *models.DownloadContext) (*models.ExtractorResponse, error) {
 		client := util.GetHTTPClient(ctx.Extractor.CodeName)
-		redirectURL, err := util.GetLocationURL(client, ctx.MatchedContentURL, "")
+		redirectURL, err := util.GetLocationURL(
+			client,
+			ctx.MatchedContentURL,
+			igHeaders,
+		)
 		if err != nil {
 			return nil, fmt.Errorf("failed to get url location: %w", err)
 		}
diff --git a/ext/instagram/util.go b/ext/instagram/util.go
index 16ad34f..eb96502 100644
--- a/ext/instagram/util.go
+++ b/ext/instagram/util.go
@@ -18,8 +18,26 @@ import (
 	"github.com/bytedance/sonic"
 )
 
-var captionPattern = regexp.MustCompile(
-	`(?s)<meta property="og:title" content=".*?: &quot;(.*?)&quot;"`,
+var (
+	captionPattern = regexp.MustCompile(
+		`(?s)<meta property="og:title" content=".*?: &quot;(.*?)&quot;"`)
+
+	igHeaders = map[string]string{
+		"Accept":                    "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
+		"Accept-Language":           "en-GB,en;q=0.9",
+		"Cache-Control":             "max-age=0",
+		"Dnt":                       "1",
+		"Priority":                  "u=0, i",
+		"Sec-Ch-Ua":                 `Chromium";v="124", "Google Chrome";v="124", "Not-A.Brand";v="99`,
+		"Sec-Ch-Ua-Mobile":          "?0",
+		"Sec-Ch-Ua-Platform":        "macOS",
+		"Sec-Fetch-Dest":            "document",
+		"Sec-Fetch-Mode":            "navigate",
+		"Sec-Fetch-Site":            "none",
+		"Sec-Fetch-User":            "?1",
+		"Upgrade-Insecure-Requests": "1",
+		"User-Agent":                util.ChromeUA,
+	}
 )
 
 func BuildSignedPayload(contentURL string) (io.Reader, error) {
diff --git a/ext/pinterest/main.go b/ext/pinterest/main.go
index 6d0b55b..5b03358 100644
--- a/ext/pinterest/main.go
+++ b/ext/pinterest/main.go
@@ -48,7 +48,7 @@ var ShortExtractor = &models.Extractor{
 	Run: func(ctx *models.DownloadContext) (*models.ExtractorResponse, error) {
 		client := util.GetHTTPClient(ctx.Extractor.CodeName)
 		shortURL := fmt.Sprintf(shortenerAPIFormat, ctx.MatchedContentID)
-		location, err := util.GetLocationURL(client, shortURL, "")
+		location, err := util.GetLocationURL(client, shortURL, nil)
 		if err != nil {
 			return nil, fmt.Errorf("failed to get real url: %w", err)
 		}
diff --git a/ext/tiktok/main.go b/ext/tiktok/main.go
index 682f42b..b773288 100644
--- a/ext/tiktok/main.go
+++ b/ext/tiktok/main.go
@@ -47,7 +47,7 @@ var VMExtractor = &models.Extractor{
 
 	Run: func(ctx *models.DownloadContext) (*models.ExtractorResponse, error) {
 		client := util.GetHTTPClient(ctx.Extractor.CodeName)
-		redirectURL, err := util.GetLocationURL(client, ctx.MatchedContentURL, "")
+		redirectURL, err := util.GetLocationURL(client, ctx.MatchedContentURL, nil)
 		if err != nil {
 			return nil, fmt.Errorf("failed to get url location: %w", err)
 		}
diff --git a/util/misc.go b/util/misc.go
index caa67ab..5a146d5 100644
--- a/util/misc.go
+++ b/util/misc.go
@@ -21,7 +21,7 @@ var cookiesCache = make(map[string][]*http.Cookie)
 func GetLocationURL(
 	client models.HTTPClient,
 	url string,
-	userAgent string,
+	headers map[string]string,
 ) (string, error) {
 	if client == nil {
 		client = GetDefaultHTTPClient()
@@ -30,10 +30,12 @@ func GetLocationURL(
 	if err != nil {
 		return "", fmt.Errorf("failed to create request: %w", err)
 	}
-	if userAgent == "" {
-		userAgent = ChromeUA
+	for k, v := range headers {
+		req.Header.Set(k, v)
+	}
+	if req.Header.Get("User-Agent") == "" {
+		req.Header.Set("User-Agent", ChromeUA)
 	}
-	req.Header.Set("User-Agent", userAgent)
 	resp, err := client.Do(req)
 	if err != nil {
 		return "", fmt.Errorf("failed to send request: %w", err)