Added support for vm.tiktok.com urls

client/getRedirectUrl.go

@@ -0,0 +1,52 @@
+package client
+
+import (
+	"context"
+	"github.com/chromedp/chromedp"
+	"io/ioutil"
+	"os"
+	"time"
+
+	config "../models/config"
+	log "../utils/log"
+)
+
+func GetRedirectUrl(url string) (string, error) {
+	dir, err := ioutil.TempDir("", "chromedp-example")
+	if err != nil {
+		return "", err
+	}
+	defer os.RemoveAll(dir)
+
+	opts := append(chromedp.DefaultExecAllocatorOptions[:],
+		chromedp.DisableGPU,
+		chromedp.UserDataDir(dir),
+		chromedp.Flag("headless", !config.Config.Debug),
+	)
+
+	allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...)
+	defer cancel()
+
+	ctx, cancel := chromedp.NewContext(
+		allocCtx,
+		chromedp.WithLogf(log.Logf),
+	)
+	defer cancel()
+
+	ctx, cancel = context.WithTimeout(ctx, time.Duration(config.Config.Deadline)*time.Second)
+	defer cancel()
+
+	var jsOutput string
+	if err := chromedp.Run(ctx,
+		// Navigate to user's page
+		chromedp.Navigate(url),
+		// Wait until page loads
+		chromedp.WaitReady(`div`),
+		// Grab url links from our element
+		chromedp.EvaluateAsDevTools(`window.location.href`, &jsOutput),
+	); err != nil {
+		return "", err
+	}
+
+	return jsOutput, err
+}