89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/csv"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"sort"
|
|
"strconv"
|
|
"time"
|
|
|
|
"code.stevenpolley.net/steven/cfcleaner/cf"
|
|
)
|
|
|
|
func init() {
|
|
|
|
}
|
|
|
|
func main() {
|
|
|
|
cfAPIKey := flag.String("cfapikey", "", "in cf, my profile -> API Tokens -> Create Token")
|
|
onlyZone := flag.String("onlyZone", "", "if specified, only the zone with this name will be processed. If omitted, all zones will be processed")
|
|
outFile := flag.String("outFile", "cfcleaner.csv", "the path of the output file")
|
|
flag.Parse()
|
|
|
|
if *cfAPIKey == "" {
|
|
fmt.Println("you must specify an API key with the -cfapikey=<key> parameter")
|
|
return
|
|
}
|
|
|
|
cfClient := cf.NewClient(*cfAPIKey)
|
|
|
|
zones, err := cfClient.GetZones()
|
|
if err != nil {
|
|
log.Fatalf("failed to get zones: %v", err)
|
|
}
|
|
|
|
outRows := make([]cf.Record, 0)
|
|
|
|
for _, zone := range zones {
|
|
if *onlyZone != "" && zone.Name != *onlyZone {
|
|
continue
|
|
}
|
|
log.Printf("processing zone '%s' with ID '%s'", zone.Name, zone.ID)
|
|
|
|
records, err := cfClient.GetRecords(zone)
|
|
if err != nil {
|
|
log.Printf("failed to get records for zone '%s' with ID '%s': %v", zone.Name, zone.ID, err)
|
|
}
|
|
|
|
for _, record := range records {
|
|
analytic, err := cfClient.GetRecordAnalytic(record)
|
|
if err != nil {
|
|
log.Printf("failed to get record report for record '%s' in zone '%s' with ID '%s': %v", record.Name, zone.Name, zone.ID, err)
|
|
continue
|
|
}
|
|
record.NumberQueries = analytic.Result.Totals.QueryCount
|
|
outRows = append(outRows, record)
|
|
}
|
|
}
|
|
|
|
sort.Slice(outRows, func(i, j int) bool {
|
|
return outRows[i].NumberQueries < outRows[j].NumberQueries
|
|
})
|
|
|
|
f, err := os.OpenFile(*outFile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
|
|
if err != nil {
|
|
log.Fatalf("failed to open output file '%s': %v", *outFile, err)
|
|
}
|
|
defer f.Close()
|
|
|
|
writer := csv.NewWriter(f)
|
|
defer writer.Flush()
|
|
err = writer.Write([]string{"Name", "Type", "NumberQueries", "CreatedOn", "ModifiedOn", "Comment", "Content"})
|
|
if err != nil {
|
|
log.Fatalf("failed to write to outFile: %v", err)
|
|
}
|
|
|
|
for _, row := range outRows {
|
|
err = writer.Write([]string{row.Name, row.Type, strconv.Itoa(row.NumberQueries), row.CreatedOn.Format(time.RFC3339), row.ModifiedOn.Format(time.RFC3339), row.Comment, row.Content})
|
|
if err != nil {
|
|
log.Fatalf("failed to write to outFile: %v", err)
|
|
}
|
|
}
|
|
|
|
fmt.Printf("wrote to file: %s\n", *outFile)
|
|
}
|