package cvss

//go:generate stringer -linecomment -type=v2Key

// CVSS 2.0 metric key.
type v2Key byte

const (
  v2AccessVector v2Key = iota // AV
  v2AccessComplexity // AC
  v2Authentication // Au
  v2ConfidentialityImpact // C
  v2IntegrityImpact // I
  v2AvailabilityImpact // A
  v2Exploitability // E
  v2RemediationLevel // RL
  v2ReportConfidence // RC
  v2CollateralDamagePotential // CDP
  v2TargetDistribution // TD
  v2ConfidentialityRequirement // CR
  v2IntegrityRequirement // IR
  v2AvailabilityRequirement // AR

  v2InvalidKey // invalid
)

// CVSS V2 metric key info lut
var v2Keys = map[v2Key]struct {
  Name string
  Category Category
} {
  v2AccessVector: { "Access Vector", Base },
  v2AccessComplexity: { "Access Complexity", Base },
  v2Authentication: { "Authentication", Base },
  v2ConfidentialityImpact: { "Confidentiality Impact", Base },
  v2IntegrityImpact: { "Integrity Impact", Base },
  v2AvailabilityImpact: { "Availability Impact", Base },
  v2Exploitability: { "Exploitability", Temporal },
  v2RemediationLevel: { "Remediation Level", Temporal },
  v2ReportConfidence: { "Report Confidence", Temporal },
  v2CollateralDamagePotential: { "Collateral Damage Potential", Environmental },
  v2TargetDistribution: { "Target Distribution", Environmental },
  v2ConfidentialityRequirement: { "Confidentiality Requirement", Environmental },
  v2IntegrityRequirement: { "Integrity Requirement", Environmental },
  v2AvailabilityRequirement: { "Availability Requirement", Environmental },
}

// // v2 metric key IDs lut
// var v2KeyIds = map[string]v2Key {
//   "AV": v2AccessVector,
//   "AC": v2AccessComplexity,
//   "Au": v2Authentication,
//   "C": v2ConfidentialityImpact,
//   "I": v2IntegrityImpact,
//   "A": v2AvailabilityImpact,
//   "E": v2Exploitability,
//   "RL": v2RemediationLevel,
//   "RC": v2ReportConfidence,
//   "CDP": v2CollateralDamagePotential,
//   "TD": v2TargetDistribution,
//   "CR": v2ConfidentialityRequirement,
//   "IR": v2IntegrityRequirement,
//   "AR": v2AvailabilityRequirement,
// }
// 
// // Get metric key from string.
// func getV2KeyFromString(s string) (v2Key, error) {
//   k, ok := v2KeyIds[s]
//   if ok {
//     return k, nil
//   } else {
//     return v2InvalidKey, newBadKey(V20, s)
//   }
// }

// Get metric key name.
func (k v2Key) Name() string {
  if data, ok := v2Keys[k]; ok {
    return data.Name
  } else {
    return "invalid"
  }
}

// Get metric key category.
func (k v2Key) Category() Category {
  if data, ok := v2Keys[k]; ok {
    return data.Category
  } else {
    return InvalidCategory
  }
}