mirror of https://github.com/pikami/tiktok-dl.git
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"github.com/chromedp/chromedp"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
models "../models"
|
|
utils "../utils"
|
|
)
|
|
|
|
// GetMusicUploads - Get all uploads by given music
|
|
func executeClientAction(url string, jsAction string) string {
|
|
dir, err := ioutil.TempDir("", "chromedp-example")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
opts := append(chromedp.DefaultExecAllocatorOptions[:],
|
|
chromedp.DisableGPU,
|
|
chromedp.UserDataDir(dir),
|
|
chromedp.Flag("headless", !models.Config.Debug),
|
|
)
|
|
|
|
allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...)
|
|
defer cancel()
|
|
|
|
ctx, cancel := chromedp.NewContext(
|
|
allocCtx,
|
|
chromedp.WithLogf(log.Printf),
|
|
)
|
|
defer cancel()
|
|
|
|
ctx, cancel = context.WithTimeout(ctx, time.Duration(models.Config.Deadline)*time.Second)
|
|
defer cancel()
|
|
|
|
var jsOutput string
|
|
err = chromedp.Run(ctx,
|
|
// Navigate to user's page
|
|
chromedp.Navigate(url),
|
|
// Execute url grabber script
|
|
chromedp.EvaluateAsDevTools(utils.ReadFileAsString("scraper.js"), &jsOutput),
|
|
chromedp.EvaluateAsDevTools(jsAction, &jsOutput),
|
|
// Wait until custom js finishes
|
|
chromedp.WaitVisible(`video_urls`),
|
|
// Grab url links from our element
|
|
chromedp.InnerHTML(`video_urls`, &jsOutput),
|
|
)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return jsOutput
|
|
}
|