mirror of
https://github.com/StackExchange/dnscontrol.git
synced 2025-01-11 09:59:59 +08:00
113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
package printer
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/StackExchange/dnscontrol/models"
|
|
)
|
|
|
|
// CLI is an abstraction around the CLI.
|
|
type CLI interface {
|
|
Printer
|
|
StartDomain(domain string)
|
|
StartDNSProvider(name string, skip bool)
|
|
EndProvider(numCorrections int, err error)
|
|
StartRegistrar(name string, skip bool)
|
|
|
|
PrintCorrection(n int, c *models.Correction)
|
|
EndCorrection(err error)
|
|
PromptToRun() bool
|
|
}
|
|
|
|
// Printer is a simple abstraction for printing data. Can be passed to providers to give simple output capabilities.
|
|
type Printer interface {
|
|
Debugf(fmt string, args ...interface{})
|
|
Warnf(fmt string, args ...interface{})
|
|
}
|
|
|
|
var reader = bufio.NewReader(os.Stdin)
|
|
|
|
// ConsolePrinter is a handle for the console printer.
|
|
type ConsolePrinter struct{}
|
|
|
|
// StartDomain is called at the start of each domain.
|
|
func (c ConsolePrinter) StartDomain(domain string) {
|
|
fmt.Printf("******************** Domain: %s\n", domain)
|
|
}
|
|
|
|
// PrintCorrection is called to print/format each correction.
|
|
func (c ConsolePrinter) PrintCorrection(i int, correction *models.Correction) {
|
|
fmt.Printf("#%d: %s\n", i+1, correction.Msg)
|
|
}
|
|
|
|
// PromptToRun prompts the user to see if they want to execute a correction.
|
|
func (c ConsolePrinter) PromptToRun() bool {
|
|
fmt.Print("Run? (Y/n): ")
|
|
txt, err := reader.ReadString('\n')
|
|
run := true
|
|
if err != nil {
|
|
run = false
|
|
}
|
|
txt = strings.ToLower(strings.TrimSpace(txt))
|
|
if txt != "y" {
|
|
run = false
|
|
}
|
|
if !run {
|
|
fmt.Println("Skipping")
|
|
}
|
|
return run
|
|
}
|
|
|
|
// EndCorrection is called at the end of each correction.
|
|
func (c ConsolePrinter) EndCorrection(err error) {
|
|
if err != nil {
|
|
fmt.Println("FAILURE!", err)
|
|
} else {
|
|
fmt.Println("SUCCESS!")
|
|
}
|
|
}
|
|
|
|
// StartDNSProvider is called at the start of each new provider.
|
|
func (c ConsolePrinter) StartDNSProvider(provider string, skip bool) {
|
|
lbl := ""
|
|
if skip {
|
|
lbl = " (skipping)\n"
|
|
}
|
|
fmt.Printf("----- DNS Provider: %s...%s", provider, lbl)
|
|
}
|
|
|
|
// StartRegistrar is called at the start of each new registrar.
|
|
func (c ConsolePrinter) StartRegistrar(provider string, skip bool) {
|
|
lbl := ""
|
|
if skip {
|
|
lbl = " (skipping)\n"
|
|
}
|
|
fmt.Printf("----- Registrar: %s...%s", provider, lbl)
|
|
}
|
|
|
|
// EndProvider is called at the end of each provider.
|
|
func (c ConsolePrinter) EndProvider(numCorrections int, err error) {
|
|
if err != nil {
|
|
fmt.Println("ERROR")
|
|
fmt.Printf("Error getting corrections: %s\n", err)
|
|
} else {
|
|
plural := "s"
|
|
if numCorrections == 1 {
|
|
plural = ""
|
|
}
|
|
fmt.Printf("%d correction%s\n", numCorrections, plural)
|
|
}
|
|
}
|
|
|
|
// Debugf is called to print/format debug information.
|
|
func (c ConsolePrinter) Debugf(format string, args ...interface{}) {
|
|
fmt.Printf(format, args...)
|
|
}
|
|
|
|
// Warnf is called to print/format a warning.
|
|
func (c ConsolePrinter) Warnf(format string, args ...interface{}) {
|
|
fmt.Printf("WARNING: "+format, args...)
|
|
}
|