package storage

import (
	"github.com/sqshq/sampler/config"
	"github.com/sqshq/sampler/console"
	"gopkg.in/yaml.v2"
	"log"
	"runtime"
)

type Statistics struct {
	Version         string
	OS              string
	WindowWidth     int            `yaml:"ww"`
	WindowHeight    int            `yaml:"wh"`
	LaunchCount     int            `yaml:"lc"`
	ComponentsCount map[string]int `yaml:"cc"`
}

const statisticsFileName = "statistics.yml"

func UpdateStatistics(config config.Config, width int, height int) *Statistics {

	statistics := new(Statistics)

	if fileExists(statisticsFileName) {
		file := readStorageFile(getPlatformStoragePath(statisticsFileName))
		err := yaml.Unmarshal(file, statistics)
		if err != nil {
			log.Fatalf("Failed to read statistics file: %v", err)
		}
		statistics.ComponentsCount = countComponentsPerType(config)
		statistics.WindowWidth = width
		statistics.WindowWidth = height
		statistics.LaunchCount += 1
	} else {
		statistics = &Statistics{
			Version:         console.AppVersion,
			OS:              runtime.GOOS,
			LaunchCount:     1,
			WindowWidth:     width,
			WindowHeight:    height,
			ComponentsCount: countComponentsPerType(config),
		}
	}

	file, err := yaml.Marshal(statistics)
	if err != nil {
		log.Fatalf("Failed to marshal statistics file: %v", err)
	}

	saveStorageFile(file, statisticsFileName)

	return statistics
}

func countComponentsPerType(config config.Config) map[string]int {
	m := make(map[string]int)
	m["runcharts"] = len(config.RunCharts)
	m["sparkLines"] = len(config.SparkLines)
	m["barcharts"] = len(config.BarCharts)
	m["gauges"] = len(config.Gauges)
	m["asciiboxes"] = len(config.AsciiBoxes)
	m["textboxes"] = len(config.TextBoxes)
	return m
}