// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

use std::str::FromStr;

use qsc_data_structures::target::TargetCapabilityFlags;

#[derive(Clone, Copy, Debug, PartialEq)]
pub enum Profile {
    Unrestricted,
    Base,
    Quantinuum,
}

impl Profile {
    #[must_use]
    pub fn to_str(&self) -> &'static str {
        match self {
            Self::Unrestricted => "Unrestricted",
            Self::Base => "Base",
            Self::Quantinuum => "Quantinuum",
        }
    }
}

impl From<Profile> for TargetCapabilityFlags {
    fn from(value: Profile) -> Self {
        match value {
            Profile::Unrestricted => Self::all(),
            Profile::Base => Self::empty(),
            Profile::Quantinuum => Self::Adaptive | Self::IntegerComputations | Self::QubitReset,
        }
    }
}

impl FromStr for Profile {
    type Err = ();

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "Quantinuum" | "quantinuum" => Ok(Self::Quantinuum),
            "Base" | "base" => Ok(Self::Base),
            "Unrestricted" | "unrestricted" => Ok(Self::Unrestricted),
            _ => Err(()),
        }
    }
}
