package influxdb

import (
	"os"
	"runtime"
	"time"

	"github.com/influxdb/influxdb/influxql"
)

// GoDiagnostics captures basic information about the runtime.
type GoDiagnostics struct {
	GoMaxProcs   int
	NumGoroutine int
	Version      string
}

// NewGoDiagnostics returns a GoDiagnostics object.
func NewGoDiagnostics() *GoDiagnostics {
	return &GoDiagnostics{
		GoMaxProcs:   runtime.GOMAXPROCS(0),
		NumGoroutine: runtime.NumGoroutine(),
		Version:      runtime.Version(),
	}
}

// AsRow returns the GoDiagnostic object as an InfluxQL row.
func (g *GoDiagnostics) AsRow() *influxql.Row {
	return &influxql.Row{
		Name:    "go",
		Columns: []string{"time", "goMaxProcs", "numGoRoutine", "version"},
		Values: [][]interface{}{[]interface{}{time.Now().UTC(),
			g.GoMaxProcs, g.NumGoroutine, g.Version}},
	}
}

// SystemDiagnostics captures basic machine data.
type SystemDiagnostics struct {
	Hostname string
	PID      int
	OS       string
	Arch     string
	NumCPU   int
}

// NewSystemDiagnostics returns a SystemDiagnostics object.
func NewSystemDiagnostics() *SystemDiagnostics {
	hostname, err := os.Hostname()
	if err != nil {
		hostname = "unknown"
	}

	return &SystemDiagnostics{
		Hostname: hostname,
		PID:      os.Getpid(),
		OS:       runtime.GOOS,
		Arch:     runtime.GOARCH,
		NumCPU:   runtime.NumCPU(),
	}
}

// AsRow returns the GoDiagnostic object as an InfluxQL row.
func (s *SystemDiagnostics) AsRow() *influxql.Row {
	return &influxql.Row{
		Name:    "system",
		Columns: []string{"time", "hostname", "pid", "os", "arch", "numCPU"},
		Values: [][]interface{}{[]interface{}{time.Now().UTC(),
			s.Hostname, s.PID, s.OS, s.Arch, s.NumCPU}},
	}
}

// MemoryDiagnostics captures Go memory stats.
type MemoryDiagnostics struct {
	Alloc        int
	TotalAlloc   int
	Sys          int
	Lookups      int
	Mallocs      int
	Frees        int
	HeapAlloc    int
	HeapSys      int
	HeapIdle     int
	HeapInUse    int
	HeapReleased int
	HeapObjects  int
	PauseTotalNs int
	NumGC        int
}

// NewMemoryDiagnostics returns a MemoryDiagnostics object.
func NewMemoryDiagnostics() *MemoryDiagnostics {
	var m runtime.MemStats
	runtime.ReadMemStats(&m)

	return &MemoryDiagnostics{
		Alloc:        int(m.Alloc),
		TotalAlloc:   int(m.TotalAlloc),
		Sys:          int(m.Sys),
		Lookups:      int(m.Lookups),
		Mallocs:      int(m.Mallocs),
		Frees:        int(m.Frees),
		HeapAlloc:    int(m.HeapAlloc),
		HeapSys:      int(m.HeapSys),
		HeapIdle:     int(m.HeapIdle),
		HeapInUse:    int(m.HeapInuse),
		HeapReleased: int(m.HeapReleased),
		HeapObjects:  int(m.HeapObjects),
		PauseTotalNs: int(m.PauseTotalNs),
		NumGC:        int(m.NumGC),
	}
}

// AsRow returns the MemoryDiagnostics object as an InfluxQL row.
func (m *MemoryDiagnostics) AsRow() *influxql.Row {
	return &influxql.Row{
		Name: "memory",
		Columns: []string{"time", "alloc", "totalAlloc", "sys", "lookups", "mallocs", "frees", "heapAlloc",
			"heapSys", "heapIdle", "heapInUse", "heapReleased", "heapObjects", "pauseTotalNs", "numGG"},
		Values: [][]interface{}{[]interface{}{time.Now().UTC(),
			m.Alloc, m.TotalAlloc, m.Sys, m.Lookups, m.Mallocs, m.Frees, m.HeapAlloc,
			m.HeapSys, m.HeapIdle, m.HeapInUse, m.HeapReleased, m.HeapObjects, m.PauseTotalNs, m.NumGC}},
	}
}

// BuildDiagnostics capture basic build version information.
type BuildDiagnostics struct {
	Version    string
	CommitHash string
}

// AsRow returns the BuildDiagnostics object as an InfluxQL row.
func (b *BuildDiagnostics) AsRow() *influxql.Row {
	return &influxql.Row{
		Name:    "build",
		Columns: []string{"time", "version", "commitHash"},
		Values: [][]interface{}{[]interface{}{time.Now().UTC(),
			b.Version, b.CommitHash}},
	}
}