mirror of https://github.com/pikami/tiktok-dl.git
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"bufio"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
checkErr "./checkErr"
|
|
)
|
|
|
|
type delegateString func(string)
|
|
|
|
// CheckIfExists - Checks if file or directory exists
|
|
func CheckIfExists(path string) bool {
|
|
if _, err := os.Stat(path); os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// InitOutputDirectory - Creates output directory
|
|
func InitOutputDirectory(path string) {
|
|
if _, err := os.Stat(path); os.IsNotExist(err) {
|
|
os.MkdirAll(path, os.ModePerm)
|
|
}
|
|
}
|
|
|
|
// ReadFileToString - Reads file and returns content
|
|
func ReadFileToString(path string) string {
|
|
content, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return string(content)
|
|
}
|
|
|
|
// ReadFileLineByLine - Reads file line by line and calls delegate
|
|
func ReadFileLineByLine(path string, delegate delegateString) {
|
|
file, err := os.Open(path)
|
|
checkErr.CheckErr(err)
|
|
defer file.Close()
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
delegate(scanner.Text())
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// AppendToFile - Appends line to file
|
|
func AppendToFile(str string, filePath string) {
|
|
f, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
checkErr.CheckErr(err)
|
|
|
|
defer f.Close()
|
|
if _, err := f.WriteString(str + "\n"); err != nil {
|
|
checkErr.CheckErr(err)
|
|
}
|
|
}
|