mirror of https://github.com/pikami/tiktok-dl.git
29 lines
684 B
Go
29 lines
684 B
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
config "../models/config"
|
||
|
|
res "../resources"
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GetUsername - Get's username from passed URL param
|
||
|
|
func GetUsername() string {
|
||
|
|
return GetUsernameFromString(config.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/@.+", str); match { // URL
|
||
|
|
stripedSuffix := strings.Split(str, "@")[1]
|
||
|
|
return strings.Split(stripedSuffix, "/")[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
panic(fmt.Sprintf(res.ErrorCouldNotRecogniseURL, str))
|
||
|
|
}
|