intelli-shell 3.3.1

use std::{
    collections::{BTreeMap, HashSet},
    convert::Infallible,
    fmt::{Display, Formatter},
    mem,
    str::FromStr,
};

use heck::ToShoutySnakeCase;
use itertools::Itertools;
use percent_encoding::{AsciiSet, NON_ALPHANUMERIC, percent_decode_str, utf8_percent_encode};

use super::VariableValue;
use crate::utils::{
    COMMAND_VARIABLE_REGEX, COMMAND_VARIABLE_REGEX_ALT, SplitCaptures, SplitItem, flatten_str, flatten_variable_name,
};

/// A command containing variables
#[derive(Clone)]
#[cfg_attr(test, derive(Debug))]
pub struct CommandTemplate {
    /// The flattened root command (i.e., the first word)
    pub flat_root_cmd: String,
    /// The different parts of the command template, including variables or values
    pub parts: Vec<TemplatePart>,
}
impl CommandTemplate {
    /// Parses the given command as a [`CommandTemplate`]
    pub fn parse(cmd: impl AsRef<str>, alt: bool) -> Self {
        let cmd = cmd.as_ref();
        let regex = if alt {
            &COMMAND_VARIABLE_REGEX_ALT
        } else {
            &COMMAND_VARIABLE_REGEX
        };
        let splitter = SplitCaptures::new(regex, cmd);
        let parts = splitter
            .map(|e| match e {
                SplitItem::Unmatched(t) => TemplatePart::Text(t.to_owned()),
                SplitItem::Captured(c) => {
                    TemplatePart::Variable(Variable::parse(c.get(1).or(c.get(2)).unwrap().as_str()))
                }
            })
            .collect::<Vec<_>>();

        CommandTemplate {
            flat_root_cmd: flatten_str(cmd.split_whitespace().next().unwrap_or(cmd)),
            parts,
        }
    }

    /// Checks if the command has any variables without value
    pub fn has_pending_variable(&self) -> bool {
        self.parts.iter().any(|part| matches!(part, TemplatePart::Variable(_)))
    }

    /// Retrieves the previously selected values for the given flat variable name
    pub fn previous_values_for(&self, flat_variable_name: &str) -> Option<Vec<String>> {
        // Find all filled variables that match the flat name, collecting their unique values
        let values = self
            .parts
            .iter()
            .filter_map(|part| {
                if let TemplatePart::VariableValue(v, value) = part
                    && v.flat_name == flat_variable_name
                {
                    Some(value.clone())
                } else {
                    None
                }
            })
            .unique()
            .collect::<Vec<_>>();

        if values.is_empty() { None } else { Some(values) }
    }

    /// Retrieves the context for a variable in the command
    pub fn variable_context(&self) -> BTreeMap<String, String> {
        self.parts
            .iter()
            .filter_map(|part| {
                if let TemplatePart::VariableValue(v, value) = part
                    && !v.secret
                {
                    Some((v.flat_name.clone(), value.clone()))
                } else {
                    None
                }
            })
            .collect()
    }

    /// Counts the total number of variables in the template (both filled and unfilled)
    pub fn count_variables(&self) -> usize {
        self.parts
            .iter()
            .filter(|part| matches!(part, TemplatePart::Variable(_) | TemplatePart::VariableValue(_, _)))
            .count()
    }

    /// Returns the variable at a specific index (0-based)
    pub fn variable_at(&self, index: usize) -> Option<&Variable> {
        self.parts
            .iter()
            .filter_map(|part| match part {
                TemplatePart::Variable(v) | TemplatePart::VariableValue(v, _) => Some(v),
                _ => None,
            })
            .nth(index)
    }

    /// Updates the value of the variable at the given index, returning the previous value if any (0-based)
    pub fn set_variable_value(&mut self, index: usize, value: Option<String>) -> Option<String> {
        // Find the part that corresponds to the variable at the given index
        self.parts
            .iter_mut()
            .filter(|part| matches!(part, TemplatePart::Variable(_) | TemplatePart::VariableValue(_, _)))
            .nth(index)
            .and_then(|part| part.set_value(value))
    }

    /// Sets the values for the variables in the template, in order.
    ///
    /// This function will iterate through the variables and assign the values from the given slice in order starting
    /// from the first one until no more values are available.
    pub fn set_variable_values(&mut self, values: &[Option<String>]) {
        // Create an iterator for the values to consume them as we find variables
        let mut values_iter = values.iter();

        // Iterate through all parts of the command template
        for part in self.parts.iter_mut() {
            // We only care about parts that are variables
            if matches!(part, TemplatePart::Variable(_) | TemplatePart::VariableValue(_, _)) {
                // If we run out of values, stop. Otherwise, update the part.
                if let Some(value) = values_iter.next() {
                    part.set_value(value.clone());
                } else {
                    break;
                }
            }
        }
    }

    /// Creates a [VariableValue] for this command with the given flat variable name and value
    pub fn new_variable_value_for(
        &self,
        flat_variable_name: impl Into<String>,
        value: impl Into<String>,
    ) -> VariableValue {
        VariableValue::new(&self.flat_root_cmd, flat_variable_name, value)
    }
}
impl Display for CommandTemplate {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        for part in self.parts.iter() {
            write!(f, "{part}")?;
        }
        Ok(())
    }
}

/// Represents a part of a command, which can be either text, a variable, or a variable value
#[derive(Clone)]
#[cfg_attr(test, derive(Debug, PartialEq, Eq))]
pub enum TemplatePart {
    Text(String),
    Variable(Variable),
    VariableValue(Variable, String),
}
impl TemplatePart {
    /// Updates a `Variable` or `VariableValue` part based on the provided value, returning the previous value if any.
    /// - If `Some(value)` is provided, it becomes a `VariableValue`.
    /// - If `None` is provided, it becomes a `Variable`.
    /// - `Text` parts are ignored.
    pub fn set_value(&mut self, value: Option<String>) -> Option<String> {
        // We only care about parts that can hold a variable
        if !matches!(self, Self::Variable(_) | Self::VariableValue(_, _)) {
            return None;
        }

        // Temporarily replace self with a default to take ownership of the variable `v`
        let taken_part = mem::take(self);

        // Get previous value and determine new part
        let (new_part, previous_value) = match taken_part {
            Self::Variable(v) => (
                match value {
                    Some(val) => Self::VariableValue(v, val),
                    None => Self::Variable(v),
                },
                None,
            ),
            Self::VariableValue(v, old_val) => (
                match value {
                    Some(val) => Self::VariableValue(v, val),
                    None => Self::Variable(v),
                },
                Some(old_val),
            ),
            other => (other, None),
        };

        *self = new_part;
        previous_value
    }
}
impl Default for TemplatePart {
    fn default() -> Self {
        Self::Text(String::new())
    }
}
impl Display for TemplatePart {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            TemplatePart::Text(t) => write!(f, "{t}"),
            TemplatePart::Variable(v) => write!(f, "{{{{{}}}}}", v.display),
            TemplatePart::VariableValue(_, value) => write!(f, "{value}"),
        }
    }
}

/// Represents a variable from a command template
#[derive(Clone)]
#[cfg_attr(test, derive(Debug, PartialEq, Eq))]
pub struct Variable {
    /// The variable as it was displayed on the command (e.g., `"Opt1|Opt2:lower:kebab"`)
    pub display: String,
    /// Parsed variable values derived from `display` (e.g., `["Opt1","Opt2"]`)
    pub options: Vec<String>,
    /// Flattened variable names derived from `options` (e.g., `["opt1","opt2"]`)
    pub flat_names: Vec<String>,
    /// Flattened variable name derived from `flat_names` (e.g., `"opt1|opt2"`)
    pub flat_name: String,
    /// Parsed variable functions to apply (e.g., `["lower","kebab"]`)
    pub functions: Vec<VariableFunction>,
    /// Whether the variable is secret
    pub secret: bool,
}
impl Variable {
    /// Parses a variable text into its model
    pub fn parse(text: impl Into<String>) -> Self {
        let display: String = text.into();

        // Determine if the variable is secret or not
        let (content, secret) = match is_secret_variable(&display) {
            Some(inner) => (inner, true),
            None => (display.as_str(), false),
        };

        // Split the variable content in parts
        let parts: Vec<&str> = content.split(':').collect();
        let mut functions = Vec::new();
        let mut boundary_index = parts.len();

        // Iterate from right-to-left to find the boundary between options and functions
        if parts.len() > 1 {
            for (i, part) in parts.iter().enumerate().rev() {
                if let Ok(func) = VariableFunction::from_str(part) {
                    functions.push(func);
                    boundary_index = i;
                } else {
                    break;
                }
            }
        }

        // The collected functions are in reverse order
        functions.reverse();

        // Join the option parts back together, then split them by the pipe character
        let options_str = &parts[..boundary_index].join(":");
        let (options, flat_names) = if options_str.is_empty() {
            (vec![], vec![])
        } else {
            let mut options = Vec::new();
            let mut flat_names = Vec::new();
            let mut seen_options = HashSet::new();
            let mut seen_flat_names = HashSet::new();

            for option in options_str
                .split('|')
                .map(|o| o.trim())
                .filter(|o| !o.is_empty())
                .map(String::from)
            {
                if seen_options.insert(option.clone()) {
                    let flat_name = flatten_variable_name(&option);
                    if seen_flat_names.insert(flat_name.clone()) {
                        flat_names.push(flat_name);
                    }
                    options.push(option);
                }
            }

            (options, flat_names)
        };

        // Build back the flat name for this variable
        let flat_name = flat_names.iter().join("|");

        Self {
            display,
            options,
            flat_names,
            flat_name,
            functions,
            secret,
        }
    }

    /// Retrieves the env var names where the value for this variable might reside (in order of preference)
    pub fn env_var_names(&self, include_individual: bool) -> HashSet<String> {
        let mut names = HashSet::new();
        let env_var_name = self.display.to_shouty_snake_case();
        if !env_var_name.trim().is_empty() && env_var_name.trim() != "PATH" {
            names.insert(env_var_name.trim().to_owned());
        }
        let env_var_name_no_fn = self.flat_name.to_shouty_snake_case();
        if !env_var_name_no_fn.trim().is_empty() && env_var_name_no_fn.trim() != "PATH" {
            names.insert(env_var_name_no_fn.trim().to_owned());
        }
        if include_individual {
            names.extend(
                self.flat_names
                    .iter()
                    .map(|o| o.to_shouty_snake_case())
                    .filter(|o| !o.trim().is_empty())
                    .filter(|o| o.trim() != "PATH")
                    .map(|o| o.trim().to_owned()),
            );
        }
        names
    }

    /// Applies variable functions to the given text
    pub fn apply_functions_to(&self, text: impl Into<String>) -> String {
        let text = text.into();
        let mut result = text;
        for func in self.functions.iter() {
            result = func.apply_to(result);
        }
        result
    }

    /// Iterates every function to check if a char has to be replaced
    pub fn check_functions_char(&self, ch: char) -> Option<String> {
        let mut out: Option<String> = None;
        for func in self.functions.iter() {
            if let Some(ref mut out) = out {
                let mut new_out = String::from("");
                for ch in out.chars() {
                    if let Some(replacement) = func.check_char(ch) {
                        new_out.push_str(&replacement);
                    } else {
                        new_out.push(ch);
                    }
                }
                *out = new_out;
            } else if let Some(replacement) = func.check_char(ch) {
                out = Some(replacement);
            }
        }
        out
    }
}
impl FromStr for Variable {
    type Err = Infallible;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        Ok(Self::parse(s))
    }
}

/// Functions that can be applied to variable values
#[derive(Clone, Copy, Debug, PartialEq, Eq, strum::EnumString)]
pub enum VariableFunction {
    #[strum(serialize = "kebab")]
    KebabCase,
    #[strum(serialize = "snake")]
    SnakeCase,
    #[strum(serialize = "upper")]
    UpperCase,
    #[strum(serialize = "lower")]
    LowerCase,
    #[strum(serialize = "url")]
    UrlEncode,
}
impl VariableFunction {
    /// Applies this function to the given text
    pub fn apply_to(&self, input: impl AsRef<str>) -> String {
        let input = input.as_ref();
        match self {
            Self::KebabCase => replace_separators(input, '-'),
            Self::SnakeCase => replace_separators(input, '_'),
            Self::UpperCase => input.to_uppercase(),
            Self::LowerCase => input.to_lowercase(),
            Self::UrlEncode => idempotent_percent_encode(input, NON_ALPHANUMERIC),
        }
    }

    /// Checks if this char would be transformed by this function
    pub fn check_char(&self, ch: char) -> Option<String> {
        match self {
            Self::KebabCase | Self::SnakeCase => {
                let separator = if *self == Self::KebabCase { '-' } else { '_' };
                if ch != separator && is_separator(ch) {
                    Some(separator.to_string())
                } else {
                    None
                }
            }
            Self::UpperCase => {
                if ch.is_lowercase() {
                    Some(ch.to_uppercase().to_string())
                } else {
                    None
                }
            }
            Self::LowerCase => {
                if ch.is_uppercase() {
                    Some(ch.to_lowercase().to_string())
                } else {
                    None
                }
            }
            Self::UrlEncode => {
                if ch.is_ascii_alphanumeric() {
                    None
                } else {
                    Some(idempotent_percent_encode(&ch.to_string(), NON_ALPHANUMERIC))
                }
            }
        }
    }
}

/// Checks if a given variable is a secret (must not be stored), returning the inner variable name if it is
fn is_secret_variable(variable_name: &str) -> Option<&str> {
    if (variable_name.starts_with('*') && variable_name.ends_with('*') && variable_name.len() > 1)
        || (variable_name.starts_with('{') && variable_name.ends_with('}'))
    {
        Some(&variable_name[1..variable_name.len() - 1])
    } else {
        None
    }
}

/// Checks if a character is a separator
fn is_separator(c: char) -> bool {
    c == '-' || c == '_' || c.is_whitespace()
}

// This function replaces any sequence of separators with a single one
fn replace_separators(s: &str, separator: char) -> String {
    let mut result = String::with_capacity(s.len());

    // Split the string using the custom separator logic and filter out empty results
    let mut words = s.split(is_separator).filter(|word| !word.is_empty());

    // Join the first word without a separator
    if let Some(first_word) = words.next() {
        result.push_str(first_word);
    }
    // Append the separator and the rest of the words
    for word in words {
        result.push(separator);
        result.push_str(word);
    }

    result
}

/// Idempotent percent-encodes a string.
///
/// This function first checks if the input is already a correctly percent-encoded string
/// according to the provided `encode_set`.
/// - If it is, the input is returned
/// - If it is not (i.e., it's unencoded, partially encoded, or incorrectly encoded), the function encodes the entire
///   input string and returns a new `String`
pub fn idempotent_percent_encode(input: &str, encode_set: &'static AsciiSet) -> String {
    // Attempt to decode the input
    if let Ok(decoded) = percent_decode_str(input).decode_utf8() {
        // If successful, re-encode the decoded string using the same character set
        let re_encoded = utf8_percent_encode(&decoded, encode_set).to_string();

        // If the re-encoded string matches the original input, it means the input was already perfectly encoded
        if re_encoded == input {
            return re_encoded;
        }
    }

    // In all other cases (decoding failed, or the re-encoded string is different), encode it fully
    utf8_percent_encode(input, encode_set).to_string().to_string()
}

#[cfg(test)]
mod tests {
    use pretty_assertions::assert_eq;

    use super::*;
    #[test]
    fn test_parse_command_with_variables() {
        let cmd = CommandTemplate::parse("git commit -m {{{message}}} --author {{author:kebab}}", false);
        assert_eq!(cmd.flat_root_cmd, "git");
        assert_eq!(cmd.parts.len(), 4);
        assert_eq!(cmd.parts[0], TemplatePart::Text("git commit -m ".into()));
        assert!(matches!(cmd.parts[1], TemplatePart::Variable(_)));
        assert_eq!(cmd.parts[2], TemplatePart::Text(" --author ".into()));
        assert!(matches!(cmd.parts[3], TemplatePart::Variable(_)));
    }

    #[test]
    fn test_parse_command_no_variables() {
        let cmd = CommandTemplate::parse("echo 'hello world'", false);
        assert_eq!(cmd.flat_root_cmd, "echo");
        assert_eq!(cmd.parts.len(), 1);
        assert_eq!(cmd.parts[0], TemplatePart::Text("echo 'hello world'".into()));
    }

    #[test]
    fn test_has_pending_variable() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{var2}}", false);
        assert!(cmd.has_pending_variable());
        cmd.set_variable_value(0, Some("value1".to_string()));
        assert!(cmd.has_pending_variable());
        cmd.set_variable_value(1, Some("value2".to_string()));
        assert!(!cmd.has_pending_variable());
    }

    #[test]
    fn test_previous_values_for() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{var1}} {{var2}}", false);

        // 1. No values set yet
        assert_eq!(cmd.previous_values_for("var1"), None);
        assert_eq!(cmd.previous_values_for("var2"), None);

        // 2. Set one value for var1
        cmd.set_variable_value(0, Some("val1".into()));
        assert_eq!(cmd.previous_values_for("var1"), Some(vec!["val1".to_string()]));
        assert_eq!(cmd.previous_values_for("var2"), None);

        // 3. Set another value for var1
        cmd.set_variable_value(1, Some("val1".into()));
        assert_eq!(cmd.previous_values_for("var1"), Some(vec!["val1".to_string()]));

        // 4. Set a different value for var1
        cmd.set_variable_value(1, Some("val1_other".into()));
        let mut values = cmd.previous_values_for("var1").unwrap();
        values.sort();
        assert_eq!(values, vec!["val1".to_string(), "val1_other".to_string()]);

        // 5. Set value for var2
        cmd.set_variable_value(2, Some("val2".into()));
        assert_eq!(cmd.previous_values_for("var2"), Some(vec!["val2".to_string()]));
    }

    #[test]
    fn test_variable_context() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{{secret_var}}} {{var2}}", false);

        // Set value for the last variable
        cmd.set_variable_value(2, Some("value2".to_string()));
        let context_before_secret: Vec<_> = cmd.variable_context().into_iter().collect();
        assert_eq!(context_before_secret, vec![("var2".to_string(), "value2".to_string())]);

        // Set value for the secret variable
        cmd.set_variable_value(1, Some("secret_value".to_string()));
        let context_after_secret: Vec<_> = cmd.variable_context().into_iter().collect();
        // The secret variable value should not be in the context
        assert_eq!(context_after_secret, context_before_secret);
    }

    #[test]
    fn test_variable_context_is_empty() {
        let cmd = CommandTemplate::parse("cmd {{var1}}", false);
        let context: Vec<_> = cmd.variable_context().into_iter().collect();
        assert!(context.is_empty());
    }

    #[test]
    fn test_count_variables() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} middle {{var2}}", false);
        assert_eq!(cmd.count_variables(), 2);

        cmd.set_variable_value(0, Some("value1".to_string()));
        assert_eq!(cmd.count_variables(), 2);

        // Test with no variables
        let cmd_no_vars = CommandTemplate::parse("cmd no-vars", false);
        assert_eq!(cmd_no_vars.count_variables(), 0);
    }

    #[test]
    fn test_variable_at() {
        let cmd = CommandTemplate::parse("cmd {{var1}} middle {{var2}}", false);
        let var1 = Variable::parse("var1");
        let var2 = Variable::parse("var2");

        // Test valid indices
        assert_eq!(cmd.variable_at(0), Some(&var1));
        assert_eq!(cmd.variable_at(1), Some(&var2));

        // Test out-of-bounds index
        assert_eq!(cmd.variable_at(2), None);

        // Test with no variables
        let cmd_no_vars = CommandTemplate::parse("cmd no-vars", false);
        assert_eq!(cmd_no_vars.variable_at(0), None);
    }

    #[test]
    fn test_set_variable_value() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{var2}}", false);

        // 1. Set value for the first time
        let prev = cmd.set_variable_value(0, Some("val1".into()));
        assert_eq!(prev, None);
        let var1 = Variable::parse("var1");
        assert_eq!(cmd.parts[1], TemplatePart::VariableValue(var1.clone(), "val1".into()));

        // 2. Update value
        let prev = cmd.set_variable_value(0, Some("new_val1".into()));
        assert_eq!(prev, Some("val1".into()));
        assert_eq!(
            cmd.parts[1],
            TemplatePart::VariableValue(var1.clone(), "new_val1".into())
        );

        // 3. Unset value
        let prev = cmd.set_variable_value(0, None);
        assert_eq!(prev, Some("new_val1".into()));
        assert_eq!(cmd.parts[1], TemplatePart::Variable(var1.clone()));

        // 4. Set value on another index
        let prev = cmd.set_variable_value(1, Some("val2".into()));
        assert_eq!(prev, None);
        let var2 = Variable::parse("var2");
        assert_eq!(cmd.parts[3], TemplatePart::VariableValue(var2.clone(), "val2".into()));

        // 5. Test out of bounds
        let prev = cmd.set_variable_value(2, Some("val3".into()));
        assert_eq!(prev, None);
    }

    #[test]
    fn test_set_variable_values_full_update() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{var2}}", false);
        let values = vec![Some("value1".to_string()), Some("value2".to_string())];
        cmd.set_variable_values(&values);

        let var1 = Variable::parse("var1");
        let var2 = Variable::parse("var2");
        assert_eq!(cmd.parts[1], TemplatePart::VariableValue(var1, "value1".into()));
        assert_eq!(cmd.parts[3], TemplatePart::VariableValue(var2, "value2".into()));
        assert!(!cmd.has_pending_variable());
    }

    #[test]
    fn test_set_variable_values_partial_update() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{var2}} {{var3}}", false);
        let values = vec![Some("value1".to_string()), Some("value2".to_string())];
        cmd.set_variable_values(&values);

        let var1 = Variable::parse("var1");
        let var2 = Variable::parse("var2");
        let var3 = Variable::parse("var3");
        assert_eq!(cmd.parts[1], TemplatePart::VariableValue(var1, "value1".into()));
        assert_eq!(cmd.parts[3], TemplatePart::VariableValue(var2, "value2".into()));
        assert_eq!(cmd.parts[5], TemplatePart::Variable(var3));
        assert!(cmd.has_pending_variable());
    }

    #[test]
    fn test_set_variable_values_with_none_to_unset() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{var2}}", false);
        // Initially set both values
        cmd.set_variable_values(&[Some("val1".into()), Some("val2".into())]);
        assert!(!cmd.has_pending_variable());

        // Now, set_variable_values with a new set of values where the first is None (unset) and the second is updated
        let values = vec![None, Some("new_val2".to_string())];
        cmd.set_variable_values(&values);

        let var1 = Variable::parse("var1");
        let var2 = Variable::parse("var2");
        assert_eq!(cmd.parts[1], TemplatePart::Variable(var1));
        assert_eq!(cmd.parts[3], TemplatePart::VariableValue(var2, "new_val2".into()));
        assert!(cmd.has_pending_variable());
    }

    #[test]
    fn test_set_variable_values_empty_slice() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}} {{var2}}", false);
        let original_parts = cmd.parts.clone();
        cmd.set_variable_values(&[]);

        assert_eq!(cmd.parts, original_parts);
    }

    #[test]
    fn test_set_variable_values_more_values_than_variables() {
        let mut cmd = CommandTemplate::parse("cmd {{var1}}", false);
        let values = vec![Some("value1".to_string()), Some("ignored".to_string())];
        cmd.set_variable_values(&values);

        let var1 = Variable::parse("var1");
        assert_eq!(cmd.parts[1], TemplatePart::VariableValue(var1, "value1".into()));
        assert!(!cmd.has_pending_variable());
    }

    #[test]
    fn test_template_part_set_value() {
        let var = Variable::parse("v");

        // Test setting a value on a Variable
        let mut part1 = TemplatePart::Variable(var.clone());
        let prev1 = part1.set_value(Some("value".into()));
        assert_eq!(prev1, None);
        assert_eq!(part1, TemplatePart::VariableValue(var.clone(), "value".into()));

        // Test updating a value on a VariableValue
        let mut part2 = TemplatePart::VariableValue(var.clone(), "old".into());
        let prev2 = part2.set_value(Some("new".into()));
        assert_eq!(prev2, Some("old".into()));
        assert_eq!(part2, TemplatePart::VariableValue(var.clone(), "new".into()));

        // Test unsetting a value on a VariableValue
        let mut part3 = TemplatePart::VariableValue(var.clone(), "value".into());
        let prev3 = part3.set_value(None);
        assert_eq!(prev3, Some("value".into()));
        assert_eq!(part3, TemplatePart::Variable(var.clone()));

        // Test setting None on a Variable (should not change)
        let mut part4 = TemplatePart::Variable(var.clone());
        let prev4 = part4.set_value(None);
        assert_eq!(prev4, None);
        assert_eq!(part4, TemplatePart::Variable(var.clone()));

        // Test on a Text part (should not change)
        let mut part5 = TemplatePart::Text("text".into());
        let prev5 = part5.set_value(Some("value".into()));
        assert_eq!(prev5, None);
        assert_eq!(part5, TemplatePart::Text("text".into()));
    }

    #[test]
    fn test_parse_simple_variable() {
        let variable = Variable::parse("my_variable");
        assert_eq!(
            variable,
            Variable {
                display: "my_variable".into(),
                options: vec!["my_variable".into()],
                flat_names: vec!["my_variable".into()],
                flat_name: "my_variable".into(),
                functions: vec![],
                secret: false,
            }
        );
    }

    #[test]
    fn test_parse_secret_variable() {
        let variable = Variable::parse("{my_secret}");
        assert_eq!(
            variable,
            Variable {
                display: "{my_secret}".into(),
                options: vec!["my_secret".into()],
                flat_names: vec!["my_secret".into()],
                flat_name: "my_secret".into(),
                functions: vec![],
                secret: true,
            }
        );
    }

    #[test]
    fn test_parse_variable_with_multiple_options() {
        let variable = Variable::parse("Option 1 | option 1 | Option 2 | Option 2 | Option 3");
        assert_eq!(
            variable,
            Variable {
                display: "Option 1 | option 1 | Option 2 | Option 2 | Option 3".into(),
                options: vec![
                    "Option 1".into(),
                    "option 1".into(),
                    "Option 2".into(),
                    "Option 3".into()
                ],
                flat_names: vec!["option 1".into(), "option 2".into(), "option 3".into()],
                flat_name: "option 1|option 2|option 3".into(),
                functions: vec![],
                secret: false,
            }
        );
    }

    #[test]
    fn test_parse_variable_with_single_function() {
        let variable = Variable::parse("my_variable:kebab");
        assert_eq!(
            variable,
            Variable {
                display: "my_variable:kebab".into(),
                options: vec!["my_variable".into()],
                flat_names: vec!["my_variable".into()],
                flat_name: "my_variable".into(),
                functions: vec![VariableFunction::KebabCase],
                secret: false,
            }
        );
    }

    #[test]
    fn test_parse_variable_with_multiple_functions() {
        let variable = Variable::parse("my_variable:snake:upper");
        assert_eq!(
            variable,
            Variable {
                display: "my_variable:snake:upper".into(),
                options: vec!["my_variable".into()],
                flat_names: vec!["my_variable".into()],
                flat_name: "my_variable".into(),
                functions: vec![VariableFunction::SnakeCase, VariableFunction::UpperCase],
                secret: false,
            }
        );
    }

    #[test]
    fn test_parse_variable_with_options_and_functions() {
        let variable = Variable::parse("opt1|opt2:lower:kebab");
        assert_eq!(
            variable,
            Variable {
                display: "opt1|opt2:lower:kebab".into(),
                options: vec!["opt1".into(), "opt2".into()],
                flat_names: vec!["opt1".into(), "opt2".into()],
                flat_name: "opt1|opt2".into(),
                functions: vec![VariableFunction::LowerCase, VariableFunction::KebabCase],
                secret: false,
            }
        );
    }

    #[test]
    fn test_parse_variable_with_colon_in_options() {
        let variable = Variable::parse("key:value:kebab");
        assert_eq!(
            variable,
            Variable {
                display: "key:value:kebab".into(),
                options: vec!["key:value".into()],
                flat_names: vec!["key:value".into()],
                flat_name: "key:value".into(),
                functions: vec![VariableFunction::KebabCase],
                secret: false,
            }
        );
    }

    #[test]
    fn test_parse_variable_with_only_functions() {
        let variable = Variable::parse(":snake");
        assert_eq!(
            variable,
            Variable {
                display: ":snake".into(),
                options: vec![],
                flat_names: vec![],
                flat_name: "".into(),
                functions: vec![VariableFunction::SnakeCase],
                secret: false,
            }
        );
    }

    #[test]
    fn test_parse_variable_that_is_a_function_name() {
        let variable = Variable::parse("kebab");
        assert_eq!(
            variable,
            Variable {
                display: "kebab".into(),
                options: vec!["kebab".into()],
                flat_names: vec!["kebab".into()],
                flat_name: "kebab".into(),
                functions: vec![],
                secret: false,
            }
        );
    }

    #[test]
    fn test_variable_env_var_names() {
        // Simple variable
        let var1 = Variable::parse("my-variable");
        assert_eq!(var1.env_var_names(true), HashSet::from(["MY_VARIABLE".into()]));

        // Variable with options
        let var2 = Variable::parse("option1|option2");
        assert_eq!(
            var2.env_var_names(true),
            HashSet::from(["OPTION1_OPTION2".into(), "OPTION1".into(), "OPTION2".into()])
        );
        assert_eq!(var2.env_var_names(false), HashSet::from(["OPTION1_OPTION2".into()]));

        // Variable with functions
        let var3 = Variable::parse("my-variable:kebab:upper");
        assert_eq!(
            var3.env_var_names(true),
            HashSet::from(["MY_VARIABLE_KEBAB_UPPER".into(), "MY_VARIABLE".into()])
        );

        // Secret variable with asterisks
        let var4 = Variable::parse("*my-secret*");
        assert_eq!(var4.env_var_names(true), HashSet::from(["MY_SECRET".into()]));

        // Secret variable with braces
        let var5 = Variable::parse("{my-secret}");
        assert_eq!(var5.env_var_names(true), HashSet::from(["MY_SECRET".into()]));
    }

    #[test]
    fn test_variable_apply_functions_to() {
        // No functions, should not change the input
        let var_none = Variable::parse("text");
        assert_eq!(var_none.apply_functions_to("Hello World"), "Hello World");

        // Single function
        let var_upper = Variable::parse("text:upper");
        assert_eq!(var_upper.apply_functions_to("Hello World"), "HELLO WORLD");

        // Chained functions: kebab then upper
        let var_kebab_upper = Variable::parse("text:kebab:upper");
        assert_eq!(var_kebab_upper.apply_functions_to("Hello World"), "HELLO-WORLD");

        // Chained functions: snake then lower
        let var_snake_lower = Variable::parse("text:snake:lower");
        assert_eq!(var_snake_lower.apply_functions_to("Hello World"), "hello_world");
    }

    #[test]
    fn test_variable_check_functions_char() {
        // No functions, should always be None
        let var_none = Variable::parse("text");
        assert_eq!(var_none.check_functions_char('a'), None);
        assert_eq!(var_none.check_functions_char(' '), None);

        // Single function with a change
        let var_upper = Variable::parse("text:upper");
        assert_eq!(var_upper.check_functions_char('a'), Some("A".to_string()));

        // Single function with no change
        let var_lower = Variable::parse("text:lower");
        assert_eq!(var_lower.check_functions_char('a'), None);

        // Chained functions
        let var_upper_kebab = Variable::parse("text:upper:kebab");
        assert_eq!(var_upper_kebab.check_functions_char('a'), Some("A".to_string()));
        assert_eq!(var_upper_kebab.check_functions_char(' '), Some("-".to_string()));
        assert_eq!(var_upper_kebab.check_functions_char('-'), None);
    }

    #[test]
    fn test_variable_function_apply_to() {
        // KebabCase
        assert_eq!(VariableFunction::KebabCase.apply_to("some text"), "some-text");
        assert_eq!(VariableFunction::KebabCase.apply_to("Some Text"), "Some-Text");
        assert_eq!(VariableFunction::KebabCase.apply_to("some_text"), "some-text");
        assert_eq!(VariableFunction::KebabCase.apply_to("-"), "");
        assert_eq!(VariableFunction::KebabCase.apply_to("_"), "");

        // SnakeCase
        assert_eq!(VariableFunction::SnakeCase.apply_to("some text"), "some_text");
        assert_eq!(VariableFunction::SnakeCase.apply_to("Some Text"), "Some_Text");
        assert_eq!(VariableFunction::SnakeCase.apply_to("some-text"), "some_text");
        assert_eq!(VariableFunction::SnakeCase.apply_to("-"), "");
        assert_eq!(VariableFunction::SnakeCase.apply_to("_"), "");

        // UpperCase
        assert_eq!(VariableFunction::UpperCase.apply_to("some text"), "SOME TEXT");
        assert_eq!(VariableFunction::UpperCase.apply_to("SomeText"), "SOMETEXT");

        // LowerCase
        assert_eq!(VariableFunction::LowerCase.apply_to("SOME TEXT"), "some text");
        assert_eq!(VariableFunction::LowerCase.apply_to("SomeText"), "sometext");

        // Urlencode
        assert_eq!(VariableFunction::UrlEncode.apply_to("some text"), "some%20text");
        assert_eq!(VariableFunction::UrlEncode.apply_to("Some Text"), "Some%20Text");
        assert_eq!(VariableFunction::UrlEncode.apply_to("some-text"), "some%2Dtext");
        assert_eq!(VariableFunction::UrlEncode.apply_to("some_text"), "some%5Ftext");
        assert_eq!(
            VariableFunction::UrlEncode.apply_to("!@#$%^&*()"),
            "%21%40%23%24%25%5E%26%2A%28%29"
        );
        assert_eq!(VariableFunction::UrlEncode.apply_to("some%20text"), "some%20text");
    }

    #[test]
    fn test_variable_function_check_char() {
        // KebabCase
        assert_eq!(VariableFunction::KebabCase.check_char(' '), Some("-".to_string()));
        assert_eq!(VariableFunction::KebabCase.check_char('_'), Some("-".to_string()));
        assert_eq!(VariableFunction::KebabCase.check_char('-'), None);
        assert_eq!(VariableFunction::KebabCase.check_char('A'), None);

        // SnakeCase
        assert_eq!(VariableFunction::SnakeCase.check_char(' '), Some("_".to_string()));
        assert_eq!(VariableFunction::SnakeCase.check_char('-'), Some("_".to_string()));
        assert_eq!(VariableFunction::SnakeCase.check_char('_'), None);
        assert_eq!(VariableFunction::SnakeCase.check_char('A'), None);

        // UpperCase
        assert_eq!(VariableFunction::UpperCase.check_char('a'), Some("A".to_string()));
        assert_eq!(VariableFunction::UpperCase.check_char('A'), None);
        assert_eq!(VariableFunction::UpperCase.check_char(' '), None);

        // LowerCase
        assert_eq!(VariableFunction::LowerCase.check_char('A'), Some("a".to_string()));
        assert_eq!(VariableFunction::LowerCase.check_char('a'), None);
        assert_eq!(VariableFunction::LowerCase.check_char(' '), None);

        // UrlEncode
        assert_eq!(VariableFunction::UrlEncode.check_char(' '), Some("%20".to_string()));
        assert_eq!(VariableFunction::UrlEncode.check_char('!'), Some("%21".to_string()));
        assert_eq!(VariableFunction::UrlEncode.check_char('A'), None);
        assert_eq!(VariableFunction::UrlEncode.check_char('1'), None);
        assert_eq!(VariableFunction::UrlEncode.check_char('-'), Some("%2D".to_string()));
        assert_eq!(VariableFunction::UrlEncode.check_char('_'), Some("%5F".to_string()));
    }

    #[test]
    fn test_is_secret_variable() {
        // Test with asterisks
        assert_eq!(is_secret_variable("*secret*"), Some("secret"));
        assert_eq!(is_secret_variable("* another secret *"), Some(" another secret "));
        assert_eq!(is_secret_variable("**"), Some(""));

        // Test with braces
        assert_eq!(is_secret_variable("{secret}"), Some("secret"));
        assert_eq!(is_secret_variable("{ another secret }"), Some(" another secret "));
        assert_eq!(is_secret_variable("{}"), Some(""));

        // Test non-secret variables
        assert_eq!(is_secret_variable("not-secret"), None);
        assert_eq!(is_secret_variable("*not-secret"), None);
        assert_eq!(is_secret_variable("not-secret*"), None);
        assert_eq!(is_secret_variable("{not-secret"), None);
        assert_eq!(is_secret_variable("not-secret}"), None);
        assert_eq!(is_secret_variable(""), None);
        assert_eq!(is_secret_variable("*"), None);
        assert_eq!(is_secret_variable("{"), None);
        assert_eq!(is_secret_variable("}*"), None);
        assert_eq!(is_secret_variable("*{"), None);
    }
}