abra/cli/record/new.go

package record

import (
	"context"
	"fmt"
	"strconv"

	"coopcloud.tech/abra/cli/internal"
	"coopcloud.tech/abra/pkg/dns"
	gandiPkg "coopcloud.tech/abra/pkg/dns/gandi"
	"coopcloud.tech/abra/pkg/formatter"
	"github.com/libdns/gandi"
	"github.com/libdns/libdns"
	"github.com/sirupsen/logrus"
	"github.com/urfave/cli"
)

// RecordNewCommand creates a new domain name record.
var RecordNewCommand = cli.Command{
	Name:      "new",
	Usage:     "Create a new domain record",
	Aliases:   []string{"n"},
	ArgsUsage: "<zone>",
	Flags: []cli.Flag{
		internal.DebugFlag,
		internal.NoInputFlag,
		internal.DNSProviderFlag,
		internal.DNSTypeFlag,
		internal.DNSNameFlag,
		internal.DNSValueFlag,
		internal.DNSTTLFlag,
		internal.DNSPriorityFlag,
		internal.OfflineFlag,
	},
	Before: internal.SubCommandBefore,
	Description: `
Create a new domain name record for a specific zone.

You must specify a zone (e.g. example.com) under which your domain name records
are listed. This zone must already be created on your provider account.

Example:

    abra record new foo.com -p gandi -t A -n myapp -v 192.168.178.44

You may also invoke this command in "wizard" mode and be prompted for input:

    abra record new
`,
	Action: func(c *cli.Context) error {
		zone, err := internal.EnsureZoneArgument(c)
		if err != nil {
			logrus.Fatal(err)
		}

		if err := internal.EnsureDNSProvider(); err != nil {
			logrus.Fatal(err)
		}

		var provider gandi.Provider
		switch internal.DNSProvider {
		case "gandi":
			provider, err = gandiPkg.New()
			if err != nil {
				logrus.Fatal(err)
			}
		default:
			logrus.Fatalf("%s is not a supported DNS provider", internal.DNSProvider)
		}

		if err := internal.EnsureDNSTypeFlag(c); err != nil {
			logrus.Fatal(err)
		}

		if err := internal.EnsureDNSNameFlag(c); err != nil {
			logrus.Fatal(err)
		}

		if err := internal.EnsureDNSValueFlag(c); err != nil {
			logrus.Fatal(err)
		}

		ttl, err := dns.GetTTL(internal.DNSTTL)
		if err != nil {
			return err
		}

		record := libdns.Record{
			Type:  internal.DNSType,
			Name:  internal.DNSName,
			Value: internal.DNSValue,
			TTL:   ttl,
		}

		if internal.DNSType == "MX" || internal.DNSType == "SRV" || internal.DNSType == "URI" {
			record.Priority = internal.DNSPriority
		}

		records, err := provider.GetRecords(context.Background(), zone)
		if err != nil {
			logrus.Fatal(err)
		}

		for _, existingRecord := range records {
			if existingRecord.Type == record.Type &&
				existingRecord.Name == record.Name &&
				existingRecord.Value == record.Value {
				logrus.Fatalf("%s record for %s already exists?", record.Type, zone)
			}
		}

		createdRecords, err := provider.SetRecords(
			context.Background(),
			zone,
			[]libdns.Record{record},
		)
		if err != nil {
			logrus.Fatal(err)
		}

		if len(createdRecords) == 0 {
			logrus.Fatal("provider library reports that no record was created?")
		}

		createdRecord := createdRecords[0]

		tableCol := []string{"type", "name", "value", "TTL", "priority"}
		table := formatter.CreateTable(tableCol)

		value := createdRecord.Value
		if len(createdRecord.Value) > 30 {
			value = fmt.Sprintf("%s...", createdRecord.Value[:30])
		}

		table.Append([]string{
			createdRecord.Type,
			createdRecord.Name,
			value,
			createdRecord.TTL.String(),
			strconv.Itoa(createdRecord.Priority),
		})

		table.Render()

		logrus.Info("record created")

		return nil
	},
}