Added option to download items from list

.gitignore

@@ -4,3 +4,4 @@ __debug_bin
 downloads
 *.exe
 tiktok-dl
+batch_file.txt

README.md

@@ -8,7 +8,8 @@ A simple tiktok video downloader written in go
 ## Basic usage
 Download the executable from `https://github.com/pikami/tiktok-dl/releases`\
 You can download all videos from user by running `./tiktok-dl [Options] TIKTOK_USERNAME`\
-You can download single video by running `./tiktok-dl [Options] VIDEO_URL`
+You can download single video by running `./tiktok-dl [Options] VIDEO_URL`\
+You can download items listed in a text file by running `./tiktok-dl [OPTIONS] -batch-file path/to/items.txt`
 
 ## Build instructions
 Clone this repository and run `go build` to build the executable.
@@ -17,6 +18,8 @@ Clone this repository and run `go build` to build the executable.
 * `-debug` - enables debug mode
 * `-output some_directory` - Output path (default "./downloads")
 * `-metadata` - Write video metadata to a .json file
+* `-batch-file` - File containing URLs/Usernames to download, one value per line. Lines starting with '#', are considered as comments and ignored.
 
 ## Acknowledgments
-This software uses the chromedp for web scraping, it can be found here: https://github.com/chromedp/chromedp
+This software uses the **chromedp** for web scraping, it can be found here: https://github.com/chromedp/chromedp\
+For releases the JS code is minified by using **terser** toolkit, it can be found here: https://github.com/terser/terser

main.go

@@ -8,6 +8,13 @@ import (
 func main() {
 	models.GetConfig()
 	url := models.Config.URL
+	batchFilePath := models.Config.BatchFilePath
+
+	// Batch file
+	if workflows.CanUseDownloadBatchFile(batchFilePath) {
+		workflows.DownloadBatchFile(batchFilePath)
+		return
+	}
 
 	// Single video
 	if workflows.CanUseDownloadSingleVideo(url) {

models/config.go

@@ -12,6 +12,7 @@ import (
 var Config struct {
 	URL           string
 	OutputPath    string
+	BatchFilePath string
 	Debug         bool
 	MetaData      bool
 }
@@ -19,30 +20,42 @@ var Config struct {
 // GetConfig - Returns Config object
 func GetConfig() {
 	outputPath := flag.String("output", "./downloads", "Output path")
+	batchFilePath := flag.String("batch-file", "", "File containing URLs/Usernames to download, one value per line. Lines starting with '#', are considered as comments and ignored.")
 	debug := flag.Bool("debug", false, "Enables debug mode")
 	metadata := flag.Bool("metadata", false, "Write video metadata to a .json file")
 	flag.Parse()
 
 	args := flag.Args()
-	if len(args) < 1 {
+	if len(args) < 1 && *batchFilePath == "" {
 		fmt.Println("Usage: tiktok-dl [OPTIONS] TIKTOK_USERNAME|TIKTOK_URL")
+		fmt.Println("  or:  tiktok-dl [OPTIONS] -batch-file path/to/users.txt")
 		os.Exit(2)
 	}
 
+	if len(args) > 0 {
 		Config.URL = flag.Args()[len(args)-1]
+	} else {
+		Config.URL = ""
+	}
 	Config.OutputPath = *outputPath
+	Config.BatchFilePath = *batchFilePath
 	Config.Debug = *debug
 	Config.MetaData = *metadata
 }
 
 // GetUsername - Get's username from passed URL param
 func GetUsername() string {
-	if match := strings.Contains(Config.URL, "/"); !match { // Not url
-		return strings.Replace(Config.URL, "@", "", -1)
+	return GetUsernameFromString(Config.URL)
+}
+
+// GetUsernameFromString - Get's username from passed param
+func GetUsernameFromString(str string) string {
+	if match := strings.Contains(str, "/"); !match { // Not url
+		return strings.Replace(str, "@", "", -1)
 	}
 
-	if match, _ := regexp.MatchString(".+tiktok\\.com/@.+", Config.URL); match { // URL
-		stripedSuffix := strings.Split(Config.URL, "@")[1]
+	if match, _ := regexp.MatchString(".+tiktok\\.com/@.+", str); match { // URL
+		stripedSuffix := strings.Split(str, "@")[1]
 		return strings.Split(stripedSuffix, "/")[0]
 	}
 

utils/fileio.go

@@ -1,10 +1,13 @@
 package utils
 
 import (
+	"bufio"
 	"io/ioutil"
 	"os"
 )
 
+type delegateString func(string)
+
 // CheckIfExists - Checks if file or directory exists
 func CheckIfExists(path string) bool {
 	if _, err := os.Stat(path); os.IsNotExist(err) {
@@ -30,3 +33,21 @@ func ReadFileToString(path string) string {
 
 	return string(content)
 }
+
+// ReadFileLineByLine - Reads file line by line and calls delegate
+func ReadFileLineByLine(path string, delegate delegateString) {
+	file, err := os.Open(path)
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		delegate(scanner.Text())
+	}
+
+	if err := scanner.Err(); err != nil {
+		panic(err)
+	}
+}

workflows/downloadBatchFile.go

@@ -0,0 +1,41 @@
+package workflows
+
+import (
+	models "../models"
+	utils "../utils"
+	"fmt"
+)
+
+// CanUseDownloadBatchFile - Check's if DownloadBatchFile can be used
+func CanUseDownloadBatchFile(batchFilePath string) bool {
+	return batchFilePath != ""
+}
+
+// DownloadBatchFile - Download items from batch file
+func DownloadBatchFile(batchFilePath string) {
+	if !utils.CheckIfExists(batchFilePath) {
+		panic(fmt.Sprintf("File path %s not found.", batchFilePath))
+	}
+
+	utils.ReadFileLineByLine(batchFilePath, downloadItem)
+}
+
+func downloadItem(batchItem string) {
+	if batchItem[0] == '#' {
+		return
+	}
+
+	// Single video
+	if CanUseDownloadSingleVideo(batchItem) {
+		DownloadSingleVideo(batchItem)
+		return
+	}
+
+	// Tiktok user
+	if CanUseDownloadUser(batchItem) {
+		DownloadUser(models.GetUsernameFromString(batchItem))
+		return
+	}
+
+	panic(fmt.Sprintf("Could not recognise URL format of string %s", batchItem))
+}

workflows/downloadVideo.go

@@ -16,7 +16,7 @@ func CanUseDownloadSingleVideo(url string) bool {
 
 // DownloadSingleVideo - Downloads single video
 func DownloadSingleVideo(url string) {
-	username := models.GetUsername()
+	username := models.GetUsernameFromString(url)
 	upload := client.GetVideoDetails(url)
 	downloadDir := fmt.Sprintf("%s/%s", models.Config.OutputPath, username)