gosync/settings/arguments.go

97 lines
1.7 KiB
Go
Raw Normal View History

2023-09-07 15:20:19 +02:00
package settings
import (
"os"
"runtime"
"strconv"
"strings"
"time"
2023-09-07 15:33:35 +02:00
"velvettear/gosync/help"
2023-09-07 15:20:19 +02:00
"velvettear/gosync/log"
)
// exported function(s)
func Initialize() {
os.Args = os.Args[1:]
var arguments []string
for _, arg := range os.Args {
arg = strings.ToLower(arg)
if arg != "-v" && arg != "--verbose" {
continue
}
setVerbose(true)
}
2023-09-08 13:29:45 +02:00
for index := 0; index < len(os.Args); index++ {
switch strings.ToLower(os.Args[index]) {
2023-09-07 15:33:35 +02:00
case "-h":
fallthrough
case "--help":
help.Print()
os.Exit(0)
2023-09-07 15:20:19 +02:00
case "-c":
fallthrough
case "--concurrency":
2023-09-08 13:29:45 +02:00
index++
if index < len(os.Args) {
concurrency, error := strconv.Atoi(os.Args[index])
if error != nil {
break
2023-09-07 15:20:19 +02:00
}
setConcurrency(concurrency)
2023-09-07 15:20:19 +02:00
}
case "-d":
fallthrough
case "--delay":
index++
if index < len(os.Args) {
delay, error := strconv.Atoi(os.Args[index])
if error != nil {
break
}
setDelay(time.Duration(delay) * time.Millisecond)
2023-09-07 15:20:19 +02:00
}
case "-p":
fallthrough
case "--password":
2023-09-08 13:29:45 +02:00
index++
2023-09-07 15:20:19 +02:00
if index > len(os.Args) {
break
}
2023-09-08 13:29:45 +02:00
setPassword(os.Args[index])
2023-09-07 15:20:19 +02:00
case "-u":
fallthrough
case "--user":
2023-09-08 13:29:45 +02:00
index++
2023-09-07 15:20:19 +02:00
if index > len(os.Args) {
break
}
2023-09-08 13:29:45 +02:00
setUser(os.Args[index])
2023-09-07 15:20:19 +02:00
default:
2023-09-08 13:29:45 +02:00
arguments = append(arguments, os.Args[index])
2023-09-07 15:20:19 +02:00
}
}
2023-09-07 15:33:35 +02:00
if len(os.Args) < 2 {
log.Fatal("error: missing arguments")
}
2023-09-07 15:20:19 +02:00
setSource(arguments[0])
setTarget(arguments[1])
if !SourceIsRemote() {
source, _ := strings.CutSuffix(Source, "/*")
_, error := os.Stat(source)
if os.IsNotExist(error) {
log.Fatal("given source does not exist", source)
}
2023-09-07 15:20:19 +02:00
}
setDefaults()
}
// unexported function(s)
func setDefaults() {
if Concurrency == 0 {
setConcurrency(runtime.NumCPU())
}
if Delay == 0 {
setDelay(100)
}
2023-09-07 15:20:19 +02:00
}