nali/cmd/cdn.go

package cmd

import (
	"github.com/zu1k/nali/internal/app"

	"github.com/spf13/cobra"
)

// cdnCmd represents the cdn command
var cdnCmd = &cobra.Command{
	Use:   "cdn",
	Short: "Query cdn service provider",
	Long:  `Query cdn service provider`,
	Run: func(cmd *cobra.Command, args []string) {
		if update {
			app.UpdateDB()
		}

		app.InitCDNDB()
		app.CDN(args)
	},
}

var (
	update = false
)

func init() {
	rootCmd.AddCommand(cdnCmd)
	cdnCmd.Flags().BoolVarP(&update, "update", "u", false, "Update CDN database")
}