refactor: do cli.Exit in top level
Signed-off-by: HeshamTB <hishaminv@gmail.com>
This commit is contained in:
parent
e5e4641264
commit
4c6e251247
@ -2,6 +2,7 @@ package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"errors"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
@ -158,7 +159,7 @@ func createCliApp() *cli.App {
|
||||
Action: func(ctx *cli.Context, s string) error {
|
||||
_, err := netip.ParseAddr(s)
|
||||
if err != nil {
|
||||
return cli.Exit(fmt.Sprintf("Can not parse %s as a network IP", s), 1)
|
||||
return errors.New(fmt.Sprintf("Can not parse %s as a network IP", s))
|
||||
}
|
||||
return nil
|
||||
},
|
||||
@ -177,7 +178,7 @@ func createCliApp() *cli.App {
|
||||
app.Action = func(ctx *cli.Context) error {
|
||||
err := setup(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
return cli.Exit(err, 1)
|
||||
}
|
||||
run(ctx)
|
||||
return nil
|
||||
@ -206,7 +207,7 @@ func setup(ctx *cli.Context) error {
|
||||
slog.Info("Creating a private key")
|
||||
privateKey, err = wgtypes.GeneratePrivateKey()
|
||||
if err != nil {
|
||||
return cli.Exit(err, 1)
|
||||
return err
|
||||
}
|
||||
slog.Debug(fmt.Sprintf("new public key: %s", privateKey.PublicKey().String()))
|
||||
return nil
|
||||
@ -231,7 +232,7 @@ func setup(ctx *cli.Context) error {
|
||||
privateKeyStr := make([]byte, 45)
|
||||
n, err := privKeyFile.Read(privateKeyStr)
|
||||
if err != nil {
|
||||
return cli.Exit(err, 1)
|
||||
return err
|
||||
}
|
||||
if n != 45 {
|
||||
slog.Warn("Private key length did not math the expected 45!")
|
||||
@ -241,7 +242,7 @@ func setup(ctx *cli.Context) error {
|
||||
privateKey, err = wgtypes.ParseKey(string(privateKeyStr))
|
||||
slog.Debug("Keyfile opened for reading")
|
||||
if err != nil {
|
||||
return cli.Exit(err, 1)
|
||||
return err
|
||||
}
|
||||
slog.Debug("Private key parsed and is correct")
|
||||
}
|
||||
@ -256,7 +257,7 @@ func setup(ctx *cli.Context) error {
|
||||
if err != nil {
|
||||
slog.Warn("Error while initlizing Wireguard netlink and device!")
|
||||
slog.Warn("Ensure to run as root or with CAP_NET_ADMIN")
|
||||
return cli.Exit(err, 1)
|
||||
return err
|
||||
}
|
||||
|
||||
wgLink = wg
|
||||
@ -266,7 +267,7 @@ func setup(ctx *cli.Context) error {
|
||||
|
||||
dev, err := wgLink.Device(InterfaceName)
|
||||
if err != nil {
|
||||
return cli.Exit(err, 1)
|
||||
return err
|
||||
}
|
||||
slog.Info("Initialized Wireguard device using wgctrl")
|
||||
|
||||
@ -294,7 +295,7 @@ func setup(ctx *cli.Context) error {
|
||||
err = ipPool.Free(testVip)
|
||||
if err != nil {
|
||||
slog.Error("Could not free test Vip from IPPool!", err)
|
||||
return cli.Exit(err.Error(), 1)
|
||||
return err
|
||||
}
|
||||
slog.Debug("Test IP Freed")
|
||||
|
||||
@ -343,10 +344,12 @@ func testWgPeerAdd(wgLink *hvpnnode3.WGLink) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
peers, err := wgLink.GetAllPeers()
|
||||
if len(peers) != 0 {
|
||||
slog.Warn(fmt.Sprintf("Expected 0 peers, got %d", len(peers)))
|
||||
}
|
||||
|
||||
slog.Debug("Removed test peer")
|
||||
|
||||
return nil
|
||||
|
Loading…
Reference in New Issue
Block a user