package cmd

import (
  "fmt"

  "github.com/go-ping/ping"
  "github.com/spf13/cobra"
)

var (
  host string
  count int
)

var pingCmd = &cobra.Command{
  Use:   "ping",
  Short: "Ping test.",
  Long:  "Uses ping to test network latency and aliveness.",
  Run: func(cmd *cobra.Command, args []string) {
    pinger, err := ping.NewPinger(host)
    if err != nil {
    	panic(err)
    }
    pinger.Count = count
    pinger.OnFinish = func(stats *ping.Statistics) {
    	fmt.Printf("--- %s ping statistics ---\n", stats.Addr)
    	fmt.Printf("%d packets transmitted, %d packets received, %v%% packet loss\n",
    		stats.PacketsSent, stats.PacketsRecv, stats.PacketLoss)
    	fmt.Printf("round-trip min/avg/max/stddev = %v/%v/%v/%v\n",
    		stats.MinRtt, stats.AvgRtt, stats.MaxRtt, stats.StdDevRtt)
    }
    fmt.Printf("PING %s (%s):\n", pinger.Addr(), pinger.IPAddr())
    err = pinger.Run() // Blocks until finished.
    if err != nil {
    	panic(err)
    }
  },
}

func init() {
  // link flags to variables
  pingCmd.Flags().StringVarP(&host, "host", "a", "8.8.8.8", "Host to attempt pinging.")
  pingCmd.Flags().IntVarP(&count, "count", "c", 50, "Number of pings to send.")

  // add ping command to the main program
  rootCmd.AddCommand(pingCmd)
}