viktor/infra
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add flag to print only allocated ip for vpn use case

Browse source
This commit is contained in:
viktorbarzin 2021-03-25 19:54:58 +00:00
parent 1cd6e97bf7
commit 1d85e9ddb2
1 changed files with 16 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

23
cli/main.go
View file

@ -11,8 +11,9 @@ import (
)
const (
useCaseFlagName = "use-case"
repoRootFlagName = "repo-root"
useCaseFlagName = "use-case"
repoRootFlagName = "repo-root"
printResultOnlyFlagName = "result-only"
)
var (
@ -28,17 +29,22 @@ func main() {
func run() error {
useCase := flag.String(useCaseFlagName, "", fmt.Sprintf("Use case to run. Available use cases are: %+v", validUseCases))
printResultOnly := flag.Bool(printResultOnlyFlagName, false, "Whether or not to print only the result (allocated ip) or print full command logging")
// repoRootParam := flag.String(repoRootFlagName, "", fmt.Sprintf("Path to the root of the infra repository."))
// VPN flags
vpnClientName := flag.String(vpnClientNameFlagName, "", fmt.Sprintf("Friendly VPN user name."))
vpnClientPubKey := flag.String(vpnClientPubKeyFlagName, "", fmt.Sprintf("VPN client public key."))
flag.Set("logtostderr", "true")
flag.Set("stderrthreshold", "WARNING")
flag.Set("v", "2")
// Flag definitions above!
flag.Parse()
if !*printResultOnly {
flag.Set("logtostderr", "true")
flag.Set("stderrthreshold", "WARNING")
flag.Set("v", "2")
}
// if *repoRootParam == "" {
// return fmt.Errorf("'-%s' flag must not be empty", repoRootFlagName)
// }
@ -65,12 +71,12 @@ func run() error {
switch *useCase {
case vpnUseCaseFlagName:
// get last used ip and increment
lastIP, err := getAndUpdateIP(gitFs, vpnLastIPConfFileRelative)
ip, err := getAndUpdateIP(gitFs, vpnLastIPConfFileRelative)
if err != nil {
return errors.Wrapf(err, "failed to get valid last ip from file %s", vpnLastIPConfFileRelative)
}
// insert new vpn client config
err = addVPNClient(gitFs, *vpnClientName, *vpnClientPubKey, vpnClientsConfFileRelative, lastIP)
err = addVPNClient(gitFs, *vpnClientName, *vpnClientPubKey, vpnClientsConfFileRelative, ip)
if err != nil {
return errors.Wrapf(err, "failed to add vpn client")
}
@ -78,6 +84,9 @@ func run() error {
if _, err = worktree.Commit("Added new VPN client config", &git.CommitOptions{All: true, Author: &object.Signature{Name: "Webhook Handler Bot"}}); err != nil {
return errors.Wrapf(err, "failed to commit")
}
if *printResultOnly {
println(ip)
}
default:
err = errors.New(fmt.Sprintf("unsupported use case: %s", *useCase))
}