domains.go

package main

import (
	"fmt"
	"log"
	"os"
	"text/tabwriter"

	"github.com/remind101/empire/pkg/heroku"
)

var cmdDomains = &Command{
	Run:      runDomains,
	Usage:    "domains",
	NeedsApp: true,
	Category: "domain",
	NumArgs:  0,
	Short:    "list domains",
	Long: `
Lists domains.

Examples:

    $ emp domains
    test.herokuapp.com
    www.test.com
`,
}

func runDomains(cmd *Command, args []string) {
	w := tabwriter.NewWriter(os.Stdout, 1, 2, 2, ' ', 0)
	defer w.Flush()

	appname := mustApp()
	cmd.AssertNumArgsCorrect(args)
	domains, err := client.DomainList(appname, &heroku.ListRange{
		Field: "hostname",
		Max:   1000,
	})
	must(err)

	for _, d := range domains {
		fmt.Fprintln(w, d.Hostname)
	}
}

var cmdDomainAdd = &Command{
	Run:      runDomainAdd,
	Usage:    "domain-add <domain>",
	NeedsApp: true,
	Category: "domain",
	NumArgs:  1,
	Short:    "add a domain",
}

func runDomainAdd(cmd *Command, args []string) {
	appname := mustApp()
	cmd.AssertNumArgsCorrect(args)
	domain := args[0]
	_, err := client.DomainCreate(appname, domain)
	must(err)
	log.Printf("Added %s to %s.", domain, appname)
}

var cmdDomainRemove = &Command{
	Run:      runDomainRemove,
	Usage:    "domain-remove <domain>",
	NeedsApp: true,
	Category: "domain",
	NumArgs:  1,
	Short:    "remove a domain",
}

func runDomainRemove(cmd *Command, args []string) {
	appname := mustApp()
	cmd.AssertNumArgsCorrect(args)
	domain := args[0]
	must(client.DomainDelete(appname, domain))
	log.Printf("Removed %s from %s.", domain, appname)
}