gosync/settings/arguments.go

86 lines
1.6 KiB
Go

package settings
import (
"os"
"runtime"
"strconv"
"strings"
"velvettear/gosync/help"
"velvettear/gosync/log"
)
// exported function(s)
func Initialize() {
os.Args = os.Args[1:]
var arguments []string
for index, arg := range os.Args {
switch strings.ToLower(arg) {
case "-h":
fallthrough
case "--help":
help.Print()
os.Exit(0)
case "-v":
fallthrough
case "--verbose":
setVerbose(true)
case "-c":
fallthrough
case "--concurrency":
var concurrency int
tmpIndex := index + 1
if tmpIndex < len(os.Args) {
tmp, error := strconv.Atoi(os.Args[tmpIndex])
if error == nil {
concurrency = tmp
}
}
if concurrency == 0 {
concurrency = runtime.NumCPU()
}
setConcurrency(concurrency)
case "-p":
fallthrough
case "--password":
tmpIndex := index + 1
if index > len(os.Args) {
break
}
setPassword(os.Args[tmpIndex])
case "-u":
fallthrough
case "--user":
tmpIndex := index + 1
if index > len(os.Args) {
break
}
setUser(os.Args[tmpIndex])
default:
arguments = append(arguments, arg)
}
}
if len(os.Args) < 2 {
log.Fatal("error: missing arguments")
}
setSource(arguments[0])
setTarget(arguments[1])
_, error := os.Stat(Source)
if os.IsNotExist(error) {
log.Fatal("given source does not exist", Source)
}
if !Verbose {
setVerbose(false)
}
}
// unexported function(s)
func removeArgument(index int) {
removeArguments(index, 0, 0)
}
func removeArguments(index int, before int, after int) {
// derp := index - 1 - before
copyArgs := os.Args[0 : index-before]
copyArgs = append(copyArgs, os.Args[index+1+after:]...)
os.Args = copyArgs
}