ipam/cmd/subnet-add.go

/*
Copyright © 2023 Laura Kalb <dev@lauka.net>
*/

package cmd

import (
    "fmt"
    "net/netip"
    "os"
    "os/user"
    "time"

    "github.com/spf13/cobra"
)

// addCmd represents the add command
var subnetaddCmd = &cobra.Command{
    Use:     "add subnet subnet-name [vlan]",
    Short:   "Add a new subnet",
    Long:    `Add a new subnet`,
    Args:    cobra.RangeArgs(2, 3),
    Aliases: []string{"a"},
    Run: func(cmd *cobra.Command, args []string) {
        var subnet string
        var vlanid string
        var netname string

        if len(args) == 2 {
            subnet = args[0]
            netname = args[1]
            vlanid = "-"
        }

        if len(args) == 3 {
            subnet = args[0]
            netname = args[1]
            vlanid = args[2]
        }

        // Parse subnet into ParseCIDR to test if it's a valid subnet
        // _, ipnet, err := net.ParseCIDR(subnet)
        ipnet, parseerr := netip.ParsePrefix(subnet)

        // Exit if subnet already exists, no need to add it then
        if SubnetExists(ipnet) {
            fmt.Printf("[ERROR] Subnet already exists: %v\n", subnet)
            os.Exit(1)
        }

        // Exit if parsed value is no valid IP
        if parseerr != nil {
            fmt.Println("[ERROR]", parseerr)
            os.Exit(1)
        }

        // Exit if parsed value is an IPv6 Address
        // TODO: Implement IPv6 support
        // if !ipnet.Addr().Is4() {
        //     fmt.Printf("[ERROR] IPv6 is not yet supported!\n")
        //     os.Exit(1)
        // }

        currentuser, _ := user.Current()

        subnetobject := Subnet{}
        subnetobject.Subnet = ipnet
        subnetobject.Name = netname
        subnetobject.Vlan = vlanid
        subnetobject.ChangedAt = time.Now()
        subnetobject.ChangedBy = currentuser.Username

        writeerr := subnetobject.WriteSubnet()

        if writeerr != nil {
            fmt.Println("[ERROR]", writeerr)
            os.Exit(1)
        }

        fmt.Printf("added subnet:\nnet: %v\nname: %v\nvlan: %v\n", subnet, netname, vlanid)
    },
}

func init() {
    subnetCmd.AddCommand(subnetaddCmd)

    // Here you will define your flags and configuration settings.

    // Cobra supports Persistent Flags which will work for this command
    // and all subcommands, e.g.:
    // addCmd.PersistentFlags().String("foo", "", "A help for foo")

    // Cobra supports local flags which will only run when this command
    // is called directly, e.g.:
    // addCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}