ipam/cmd/export.go

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

package cmd

import (
    "encoding/json"
    "fmt"
    "net/netip"
    "os"
    "path"
    "time"

    "github.com/spf13/cobra"
)

// exportCmd represents the export command
var exportCmd = &cobra.Command{
    Use:   "export",
    Short: "Export ipam configuration",
    Long: `Export current ipam contents to importable data format.
You can either export a single subnet or all subnets.`,
    Example: "ipam export\nipam export 192.168.0.0/24",
    Args:    cobra.RangeArgs(0, 1),
    Aliases: []string{"ex"},
    Run: func(cmd *cobra.Command, args []string) {
        var jsonoutput []Subnet
        if len(args) == 1 {
            prefix, parseerr := netip.ParsePrefix(args[0])
            if parseerr != nil {
                fmt.Println("[ERROR]", parseerr)
                os.Exit(1)
            }

            subnet, geterr := GetSubnet(prefix)
            if geterr != nil {
                fmt.Println("[ERROR]", geterr)
                os.Exit(1)
            }
            jsonoutput = append(jsonoutput, subnet)
        } else {
            subnetlist := ListSubnets()
            for _, net := range subnetlist {
                prefix, parseerr := netip.ParsePrefix(net)
                if parseerr != nil {
                    fmt.Println("[ERROR]", parseerr)
                    os.Exit(1)
                }
                subnet, geterr := GetSubnet(prefix)
                if geterr != nil {
                    fmt.Println("[ERROR]", geterr)
                    os.Exit(1)
                }
                jsonoutput = append(jsonoutput, subnet)
            }
        }

        //workingdir, _ := os.Getwd()
        //timestamp := time.Now().Format("2006-01-02_15-04")
        //exportfilename := workingdir + "/ipam_export_" + timestamp + ".json"

        var exportname string

        flagpath, _ := cmd.Flags().GetString("file")
        if path.IsAbs(flagpath) {
            exportname = flagpath
        } else {
            wd, _ := os.Getwd()
            exportname = path.Join(wd, flagpath)
        }

        data, _ := json.MarshalIndent(jsonoutput, "", "  ")

        file, fileerr := os.Create(exportname)
        if fileerr != nil {
            fmt.Println("[ERROR]", fileerr)
            os.Exit(1)
        }
        defer file.Close()

        _, writeerr := file.Write(data)
        if writeerr != nil {
            fmt.Println("[ERROR]", writeerr)
            os.Exit(1)
        }
        fmt.Printf("[INFO] Data was exported to file %v\n", exportname)
    },
}

func init() {
    rootCmd.AddCommand(exportCmd)

    timestamp := time.Now().Format("2006-01-02_15-04")
    exportCmd.Flags().StringP("file", "f", "./ipam_export_"+timestamp+".json", "File name for exported data.\nCan be both absolute or relative path.")
}