package measurements

import (
	"fmt"
	"strings"
)

type SmartNvmeAttribute struct {
	AttributeId string `json:"attribute_id"` //json string from smartctl
	Name        string `json:"name"`
	Value       int64  `json:"value"`
	Threshold   int64  `json:"thresh"`

	TransformedValue int64   `json:"transformed_value"`
	Status           string  `json:"status,omitempty"`
	StatusReason     string  `json:"status_reason,omitempty"`
	FailureRate      float64 `json:"failure_rate,omitempty"`
}

func (sa *SmartNvmeAttribute) Flatten() map[string]interface{} {
	return map[string]interface{}{
		fmt.Sprintf("attr.%s.attribute_id", sa.AttributeId): sa.AttributeId,
		fmt.Sprintf("attr.%s.name", sa.AttributeId):         sa.Name,
		fmt.Sprintf("attr.%s.value", sa.AttributeId):        sa.Value,
		fmt.Sprintf("attr.%s.thresh", sa.AttributeId):       sa.Threshold,
	}
}
func (sa *SmartNvmeAttribute) Inflate(key string, val interface{}) {
	if val == nil {
		return
	}

	keyParts := strings.Split(key, ".")

	switch keyParts[2] {
	case "attribute_id":
		sa.AttributeId = val.(string)
	case "name":
		sa.Name = val.(string)
	case "value":
		sa.Value = val.(int64)
	case "thresh":
		sa.Threshold = val.(int64)
	}
}

//
////populate attribute status, using SMART Thresholds & Observed Metadata
//func (sa *SmartNvmeAttribute) PopulateAttributeStatus() {
//
//	//-1 is a special number meaning no threshold.
//	if sa.Threshold != -1 {
//		if smartMetadata, ok := metadata.NmveMetadata[sa.AttributeId]; ok {
//			//check what the ideal is. Ideal tells us if we our recorded value needs to be above, or below the threshold
//			if (smartMetadata.Ideal == "low" && sa.Value > sa.Threshold) ||
//				(smartMetadata.Ideal == "high" && sa.Value < sa.Threshold) {
//				sa.Status = SmartAttributeStatusFailed
//				sa.StatusReason = "Attribute is failing recommended SMART threshold"
//			}
//		}
//	}
//	//TODO: eventually figure out the critical_warning bits and determine correct error messages here.
//
//	//check if status is blank, set to "passed"
//	if len(sa.Status) == 0 {
//		sa.Status = SmartAttributeStatusPassed
//	}
//}