// Copyright (c) 2025-2026 Probo Inc <hello@probo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

package coredata

import (
	"encoding"
	"fmt"
)

type DataClassification string

const (
	DataClassificationPublic       DataClassification = "PUBLIC"
	DataClassificationInternal     DataClassification = "INTERNAL"
	DataClassificationConfidential DataClassification = "CONFIDENTIAL"
	DataClassificationSecret       DataClassification = "SECRET"
)

var (
	_ fmt.Stringer             = DataClassification("")
	_ encoding.TextMarshaler   = DataClassification("")
	_ encoding.TextUnmarshaler = (*DataClassification)(nil)
)

func DataClassifications() []DataClassification {
	return []DataClassification{
		DataClassificationPublic,
		DataClassificationInternal,
		DataClassificationConfidential,
		DataClassificationSecret,
	}
}

func (v DataClassification) IsValid() bool {
	switch v {
	case
		DataClassificationPublic,
		DataClassificationInternal,
		DataClassificationConfidential,
		DataClassificationSecret:
		return true
	}

	return false
}

func (v DataClassification) String() string {
	return string(v)
}

func (v DataClassification) MarshalText() ([]byte, error) {
	return []byte(v.String()), nil
}

func (v *DataClassification) UnmarshalText(text []byte) error {
	val := DataClassification(text)
	if !val.IsValid() {
		return fmt.Errorf("invalid DataClassification value: %q", string(text))
	}

	*v = val

	return nil
}