vivacity-resolver 0.20.0

//! Port of `Composer\Json\JsonManipulator` (docs/reference/JsonManipulator.php)
//! for the scope reached by `require` and `remove`: textual editing of
//! composer.json through regular expressions, so as to touch only the
//! modified part and preserve the file's formatting.
//!
//! The patterns are Composer's, compiled by pcre2 with the same flags
//! (`s`, `x`, `i`, never `u`), over bytes. Manipulated values are
//! `serde_json::Value`s following the `vivacity_core::phpjson` convention: an
//! object reduced to `STDCLASS_MARKER` is an `ArrayObject` (formatted
//! `{...}` even when empty), an empty object is an empty PHP array
//! (formatted `[]`).
//!
//! Not ported: `repositories`, lists (`addListItem`...), `addProperty`, the
//! `policy.*` branch of `removeConfigSetting`.
//!
//! Known deviations from PHP, all on manifests `json_decode` rejects or
//! only half accepts: the `pcre.backtrack_limit` threshold (pcre2 does not
//! expose `match_limit`; the limit error does follow Composer's `catch`
//! path, the threshold differs); an out-of-range float (`1e999`, `INF` on
//! the PHP side, rejected here); a lone UTF-16 surrogate or more than 512
//! nesting levels (PHP carries on with `null` and adds a duplicate root
//! key, here an error).

use pcre2::bytes::{Captures, Regex, RegexBuilder};
use serde::Deserialize as _;
use serde_json::{Map, Value};
use vivacity_core::phpjson::{php_json_encode_with, STDCLASS_MARKER};

use crate::platform::is_platform_package;
use crate::version::preg_quote;

/// `PCRE2_ERROR_MATCHLIMIT` (pcre2.h), the equivalent of
/// `PREG_BACKTRACK_LIMIT_ERROR` on the PHP side.
const PCRE2_ERROR_MATCHLIMIT: i32 = -47;

/// `JsonFile::INDENT_DEFAULT`.
const INDENT_DEFAULT: &str = "    ";

/// `JsonManipulator::DEFINES`: the JSON grammar as named subpatterns.
const DEFINES: &str = r#"(?(DEFINE)
       (?<number>    -? (?= [1-9]|0(?!\d) ) \d++ (?:\.\d++)? (?:[eE] [+-]?+ \d++)? )
       (?<boolean>   true | false | null )
       (?<string>    " (?:[^"\\]*+ | \\ ["\\bfnrt\/] | \\ u [0-9A-Fa-f]{4} )* " )
       (?<array>     \[  (?:  (?&json) \s*+ (?: , (?&json) \s*+ )*+  )?+  \s*+ \] )
       (?<pair>      \s*+ (?&string) \s*+ : (?&json) \s*+ )
       (?<object>    \{  (?:  (?&pair)  (?: , (?&pair)  )*+  )?+  \s*+ \} )
       (?<json>      \s*+ (?: (?&number) | (?&boolean) | (?&string) | (?&array) | (?&object) ) )
    )"#;

#[derive(Debug, thiserror::Error)]
pub enum ManipulatorError {
    /// `The json file must be an object ({})`.
    #[error("The json file must be an object ({{}})")]
    NotAnObject,
    /// `JsonFile::parseJson` failed (ParsingException on the Composer side).
    #[error("The input does not contain valid JSON\n{0}")]
    Parse(String),
    /// A regular expression failed where Composer would let the exception
    /// propagate.
    #[error("regex: {0}")]
    Regex(String),
    /// Path Composer ends with an exception (`LogicException`,
    /// `InvalidArgumentException`, `TypeError`).
    #[error("{0}")]
    Logic(String),
    /// Method or branch outside the ported scope.
    #[error("unsupported: {0}")]
    Unsupported(String),
}

type Result<T> = std::result::Result<T, ManipulatorError>;

/// PCRE flags of a pattern (the letters after the delimiter in PHP).
#[derive(Clone, Copy, Default)]
struct Flags {
    s: bool,
    x: bool,
    i: bool,
    m: bool,
}

const SX: Flags = Flags {
    s: true,
    x: true,
    i: false,
    m: false,
};
const X: Flags = Flags {
    s: false,
    x: true,
    i: false,
    m: false,
};
const IX: Flags = Flags {
    s: false,
    x: true,
    i: true,
    m: false,
};
const I: Flags = Flags {
    s: false,
    x: false,
    i: true,
    m: false,
};
const S: Flags = Flags {
    s: true,
    x: false,
    i: false,
    m: false,
};
const NONE: Flags = Flags {
    s: false,
    x: false,
    i: false,
    m: false,
};

fn compile(pattern: &str, flags: Flags) -> Result<Regex> {
    RegexBuilder::new()
        .dotall(flags.s)
        .extended(flags.x)
        .caseless(flags.i)
        .multi_line(flags.m)
        .build(pattern)
        .map_err(|e| ManipulatorError::Regex(format!("{e} in `{pattern}`")))
}

/// `Preg::isMatch`: the captures, or `None` if nothing matches.
fn captures<'s>(re: &Regex, subject: &'s str) -> Result<Option<Captures<'s>>> {
    re.captures(subject.as_bytes())
        .map_err(|e| ManipulatorError::Regex(e.to_string()))
}

/// `Preg::isMatch` inside a `catch` of `PREG_BACKTRACK_LIMIT_ERROR`: the
/// pcre2 match limit (`PCRE2_ERROR_MATCHLIMIT`) follows this path
/// (`false`), any other error propagates.
fn captures_or_limit<'s>(re: &Regex, subject: &'s str) -> Result<Option<Captures<'s>>> {
    match re.captures(subject.as_bytes()) {
        Ok(c) => Ok(c),
        Err(e) if e.code() == PCRE2_ERROR_MATCHLIMIT => Ok(None),
        Err(e) => Err(ManipulatorError::Regex(e.to_string())),
    }
}

/// Named group: `None` if it did not participate (`PREG_UNMATCHED_AS_NULL`).
fn named<'s>(caps: &Captures<'s>, name: &str) -> Option<&'s str> {
    caps.name(name)
        .and_then(|m| std::str::from_utf8(m.as_bytes()).ok())
}

fn group<'s>(caps: &Captures<'s>, i: usize) -> Option<&'s str> {
    caps.get(i)
        .and_then(|m| std::str::from_utf8(m.as_bytes()).ok())
}

/// `Preg::replaceCallback` on all occurrences.
fn replace_all(
    re: &Regex,
    subject: &str,
    mut f: impl FnMut(&Captures<'_>) -> Result<String>,
) -> Result<(String, usize)> {
    let mut out = String::with_capacity(subject.len());
    let mut last = 0;
    let mut count = 0;
    for caps in re.captures_iter(subject.as_bytes()) {
        let caps = caps.map_err(|e| ManipulatorError::Regex(e.to_string()))?;
        let m = caps
            .get(0)
            .ok_or_else(|| ManipulatorError::Regex("no group 0".into()))?;
        out.push_str(&subject[last..m.start()]);
        out.push_str(&f(&caps)?);
        last = m.end();
        count += 1;
    }
    out.push_str(&subject[last..]);
    Ok((out, count))
}

/// `Preg::replace` with a literal replacement string (Composer runs its
/// replacements through `addcslashes(..., '\\$')`, which amounts to
/// inserting them verbatim).
fn replace_literal(re: &Regex, subject: &str, replacement: &str) -> Result<(String, usize)> {
    replace_all(re, subject, |_| Ok(replacement.to_owned()))
}

/// `JsonFile::encode` of a scalar or a key (`JSON_UNESCAPED_SLASHES |
/// JSON_UNESCAPED_UNICODE`; `JSON_PRETTY_PRINT` has no effect here).
fn encode(value: &Value) -> Result<String> {
    php_json_encode_with(value, vivacity_core::phpjson::FLAGS_JSONFILE)
        .map_err(|e| ManipulatorError::Logic(format!("JSON encoding failed: {e}")))
}

fn encode_str(s: &str) -> Result<String> {
    encode(&Value::String(s.to_owned()))
}

/// Maximum depth of `json_decode` (its default `$depth` parameter).
const PHP_JSON_DEPTH: usize = 512;

/// `json_decode`: serde with two deviations corrected: the recursion limit
/// (128 in serde, 512 in PHP, hence a preliminary pass measuring nesting
/// outside strings) and the `-0` literal, integer `0` for PHP but float
/// `-0.0` for serde (rewritten before parsing).
fn json_decode(s: &str) -> std::result::Result<Value, String> {
    let (normalized, depth) = scan_json(s);
    if depth > PHP_JSON_DEPTH {
        return Err("Maximum stack depth exceeded".into());
    }
    let mut de = serde_json::Deserializer::from_str(&normalized);
    de.disable_recursion_limit();
    let v = Value::deserialize(&mut de).map_err(|e| e.to_string())?;
    de.end().map_err(|e| e.to_string())?;
    Ok(v)
}

/// One pass over the JSON text, outside strings: bare `-0` -> `0`, and the
/// maximum nesting depth.
fn scan_json(s: &str) -> (String, usize) {
    let bytes = s.as_bytes();
    let mut out = String::with_capacity(s.len());
    let mut depth = 0usize;
    let mut max_depth = 0usize;
    let mut i = 0;
    let mut in_string = false;
    let mut last = 0;
    while i < bytes.len() {
        let c = bytes[i];
        if in_string {
            match c {
                b'\\' => i += 1,
                b'"' => in_string = false,
                _ => {}
            }
        } else {
            match c {
                b'"' => in_string = true,
                b'[' | b'{' => {
                    depth += 1;
                    max_depth = max_depth.max(depth);
                }
                b']' | b'}' => depth = depth.saturating_sub(1),
                b'-' if bytes.get(i + 1) == Some(&b'0')
                    && !bytes
                        .get(i + 2)
                        .is_some_and(|n| matches!(n, b'.' | b'e' | b'E' | b'0'..=b'9')) =>
                {
                    out.push_str(&s[last..i]);
                    last = i + 1;
                }
                _ => {}
            }
        }
        i += 1;
    }
    out.push_str(&s[last..]);
    (out, max_depth)
}

/// `JsonFile::parseJson` (associative decoding).
fn parse_json(contents: &str) -> Result<Value> {
    json_decode(contents).map_err(ManipulatorError::Parse)
}

/// `@json_decode($s)` then PHP truthiness test: `false` if the JSON is
/// invalid or if the decoded value is falsy (`null`, `false`, `0`, `""`,
/// `"0"`, empty array; an empty object is an empty array in associative
/// mode, but a truthy `stdClass` otherwise). In object mode, a key starting
/// with a null byte is a decoding failure
/// (`JSON_ERROR_INVALID_PROPERTY_NAME`).
fn decodes_truthy(s: &str, assoc: bool) -> bool {
    match json_decode(s) {
        Ok(v) => {
            if !assoc && has_nul_key(&v) {
                return false;
            }
            match &v {
                Value::Object(m) => !assoc || !m.is_empty(),
                other => php_truthy(other),
            }
        }
        Err(_) => false,
    }
}

fn has_nul_key(v: &Value) -> bool {
    match v {
        Value::Object(m) => m.iter().any(|(k, v)| k.starts_with('\0') || has_nul_key(v)),
        Value::Array(a) => a.iter().any(has_nul_key),
        _ => false,
    }
}

fn php_truthy(v: &Value) -> bool {
    match v {
        Value::Null => false,
        Value::Bool(b) => *b,
        Value::Number(n) => n.as_f64().is_some_and(|f| f != 0.0),
        Value::String(s) => !(s.is_empty() || s == "0"),
        Value::Array(a) => !a.is_empty(),
        Value::Object(m) => !m.is_empty(),
    }
}

/// `isset($decoded[$key])`: present and not `null`.
fn isset<'v>(decoded: &'v Value, key: &str) -> Option<&'v Value> {
    php_index(decoded, key).filter(|v| !v.is_null())
}

/// `$value[$key]` on a decoded PHP array: object by key, list by canonical
/// index.
fn php_index<'v>(value: &'v Value, key: &str) -> Option<&'v Value> {
    match value {
        Value::Object(m) => m.get(key),
        Value::Array(a) => canonical_index(key).and_then(|i| a.get(i)),
        // `isset("abc"[1])`: an integer offset into the string (negative
        // from the end) is set; the value itself stands in for the
        // character, which is enough for existence tests.
        Value::String(s) => php_int_key(key)
            .filter(|&i| {
                (0..s.len() as i64).contains(&(if i < 0 { i + s.len() as i64 } else { i }))
            })
            .map(|_| value),
        _ => None,
    }
}

/// A key PHP converts to an integer, negatives included (`"-3"`, never
/// `"-0"` nor `"03"`).
fn php_int_key(key: &str) -> Option<i64> {
    if let Some(i) = canonical_index(key) {
        return i64::try_from(i).ok();
    }
    let rest = key.strip_prefix('-')?;
    if rest.starts_with(['1', '2', '3', '4', '5', '6', '7', '8', '9'])
        && rest.bytes().all(|b| b.is_ascii_digit())
    {
        return key.parse().ok();
    }
    None
}

/// A key PHP converts to an integer (`"0"`, `"12"`, never `"012"`).
fn canonical_index(key: &str) -> Option<usize> {
    if key == "0" {
        return Some(0);
    }
    if key.starts_with(['1', '2', '3', '4', '5', '6', '7', '8', '9'])
        && key.bytes().all(|b| b.is_ascii_digit())
    {
        return key.parse().ok();
    }
    None
}

/// `array_is_list` of an array decoded from a JSON object: keys `"0"`,
/// `"1"`, ... in order.
fn php_is_list(m: &Map<String, Value>) -> bool {
    m.keys()
        .enumerate()
        .all(|(i, k)| canonical_index(k) == Some(i))
}

/// `$subName` is truthy in the PHP sense (`if ($subName && ...)`).
fn truthy_str(s: &str) -> bool {
    !(s.is_empty() || s == "0")
}

/// `strnatcmp` (ext/standard/strnatcmp.c, case-sensitive).
pub fn strnatcmp(a: &str, b: &str) -> std::cmp::Ordering {
    use std::cmp::Ordering::{Equal, Greater, Less};
    let (a, b) = (a.as_bytes(), b.as_bytes());
    if a.is_empty() || b.is_empty() {
        return a.len().cmp(&b.len());
    }
    // A PHP string is null-terminated: reading past the end yields 0.
    let at = |s: &[u8], i: usize| s.get(i).copied().unwrap_or(0);
    let (mut ap, mut bp) = (0usize, 0usize);
    let (mut ca, mut cb) = (a[0], b[0]);
    while ca == b'0' && ap + 1 < a.len() && a[ap + 1].is_ascii_digit() {
        ap += 1;
        ca = a[ap];
    }
    while cb == b'0' && bp + 1 < b.len() && b[bp + 1].is_ascii_digit() {
        bp += 1;
        cb = b[bp];
    }
    let is_space = |c: u8| matches!(c, b' ' | b'\t' | b'\n' | 0x0B | 0x0C | b'\r');
    loop {
        while is_space(ca) {
            ap += 1;
            ca = at(a, ap);
        }
        while is_space(cb) {
            bp += 1;
            cb = at(b, bp);
        }
        if ca.is_ascii_digit() && cb.is_ascii_digit() {
            let fractional = ca == b'0' || cb == b'0';
            let result = if fractional {
                compare_left(a, &mut ap, b, &mut bp)
            } else {
                compare_right(a, &mut ap, b, &mut bp)
            };
            if result != Equal {
                return result;
            }
            if ap >= a.len() && bp >= b.len() {
                return Equal;
            }
            if ap >= a.len() {
                return Less;
            }
            if bp >= b.len() {
                return Greater;
            }
            ca = a[ap];
            cb = b[bp];
        }
        match ca.cmp(&cb) {
            Less => return Less,
            Greater => return Greater,
            Equal => {}
        }
        ap += 1;
        bp += 1;
        if ap >= a.len() && bp >= b.len() {
            return Equal;
        }
        if ap >= a.len() {
            return Less;
        }
        if bp >= b.len() {
            return Greater;
        }
        ca = a[ap];
        cb = b[bp];
    }
}

/// Two left-aligned numbers (fractions): the first difference wins.
fn compare_left(a: &[u8], ap: &mut usize, b: &[u8], bp: &mut usize) -> std::cmp::Ordering {
    use std::cmp::Ordering::{Equal, Greater, Less};
    loop {
        let da = a.get(*ap).filter(|c| c.is_ascii_digit());
        let db = b.get(*bp).filter(|c| c.is_ascii_digit());
        match (da, db) {
            (None, None) => return Equal,
            (None, Some(_)) => return Less,
            (Some(_), None) => return Greater,
            (Some(x), Some(y)) => match x.cmp(y) {
                Less => return Less,
                Greater => return Greater,
                Equal => {}
            },
        }
        *ap += 1;
        *bp += 1;
    }
}

/// Two right-aligned numbers: the longer one wins, otherwise the first
/// difference (remembered in `bias`).
fn compare_right(a: &[u8], ap: &mut usize, b: &[u8], bp: &mut usize) -> std::cmp::Ordering {
    use std::cmp::Ordering::{Equal, Greater, Less};
    let mut bias = Equal;
    loop {
        let da = a.get(*ap).filter(|c| c.is_ascii_digit());
        let db = b.get(*bp).filter(|c| c.is_ascii_digit());
        match (da, db) {
            (None, None) => return bias,
            (None, Some(_)) => return Less,
            (Some(_), None) => return Greater,
            (Some(x), Some(y)) => {
                if bias == Equal {
                    bias = x.cmp(y);
                }
            }
        }
        *ap += 1;
        *bp += 1;
    }
}

/// The sort prefix of `sortPackages`: platforms first (`php`, `hhvm`,
/// `ext-*`, `lib-*`, the others), then packages.
fn sort_prefix(requirement: &str) -> String {
    if !is_platform_package(requirement) {
        return format!("5-{requirement}");
    }
    // The five replacements chain on the same string; once prefixed with a
    // digit, it is no longer touched by `^\D`.
    let mut s = requirement.to_owned();
    for (prefix, digit) in [("php", "0"), ("hhvm", "1"), ("ext", "2"), ("lib", "3")] {
        if s.starts_with(prefix) {
            s = format!("{digit}-{s}");
        }
    }
    if s.starts_with(|c: char| !c.is_ascii_digit()) {
        s = format!("4-{s}");
    }
    s
}

/// `sortPackages`: stable `uksort` by `strnatcmp` of the prefixes. As soon
/// as the comparator is called (two or more entries), an integer key makes
/// `isPlatformPackage(string $name)` fail (`strict_types`).
fn sort_packages(packages: &mut Map<String, Value>) -> Result<()> {
    if packages.len() >= 2 && packages.keys().any(|k| php_int_key(k).is_some()) {
        return Err(ManipulatorError::Logic(
            "PlatformRepository::isPlatformPackage(): Argument #1 ($name) must be of type string, int given".into(),
        ));
    }
    let mut entries: Vec<(String, Value)> = std::mem::take(packages).into_iter().collect();
    entries.sort_by(|(a, _), (b, _)| strnatcmp(&sort_prefix(a), &sort_prefix(b)));
    packages.extend(entries);
    Ok(())
}

/// An object reduced to the `stdClass` sentinel (an `ArrayObject`).
fn is_stdclass_marker(m: &Map<String, Value>) -> bool {
    m.len() == 1 && m.contains_key(STDCLASS_MARKER)
}

pub struct JsonManipulator {
    contents: String,
    newline: &'static str,
    indent: String,
}

impl JsonManipulator {
    /// Constructor: `trim`, `{}` -> `{\n}`, newline and indentation
    /// detection.
    pub fn new(contents: &str) -> Result<Self> {
        let contents = contents.trim_matches([' ', '\t', '\n', '\r', '\0', '\x0B']);
        let contents = if contents.is_empty() { "{}" } else { contents };
        if captures(&compile(r"^\{(.*)\}$", S)?, contents)?.is_none() {
            return Err(ManipulatorError::NotAnObject);
        }
        let newline = if contents.contains("\r\n") {
            "\r\n"
        } else {
            "\n"
        };
        let contents = if contents == "{}" {
            format!("{{{newline}}}")
        } else {
            contents.to_owned()
        };
        let indent = detect_indenting(&contents)?;
        Ok(Self {
            contents,
            newline,
            indent,
        })
    }

    /// `getContents`: the text followed by a newline.
    pub fn contents(&self) -> String {
        format!("{}{}", self.contents, self.newline)
    }

    /// `addLink`: adds or replaces `$package: $constraint` in the `$type`
    /// section, preserving the existing spelling of the name.
    pub fn add_link(
        &mut self,
        link_type: &str,
        package: &str,
        constraint: &str,
        sort: bool,
    ) -> Result<bool> {
        let decoded = parse_json(&self.contents)?;
        if isset(&decoded, link_type).is_none() {
            let mut m = Map::new();
            m.insert(package.to_owned(), Value::String(constraint.to_owned()));
            return self.add_main_key(link_type, &Value::Object(m));
        }

        let regex = compile(
            &format!(
                r#"{DEFINES}^(?P<start>\s*\{{\s*(?:(?&string)\s*:\s*(?&json)\s*,\s*)*?)(?P<property>{}\s*:\s*)(?P<value>(?&json))(?P<end>.*)"#,
                preg_quote(&encode_str(link_type)?)
            ),
            SX,
        )?;
        let Some(caps) = captures(&regex, &self.contents)? else {
            return Ok(false);
        };
        let start = named(&caps, "start").unwrap_or("").to_owned();
        let property = named(&caps, "property").unwrap_or("").to_owned();
        let end = named(&caps, "end").unwrap_or("").to_owned();
        let mut links = named(&caps, "value").unwrap_or("").to_owned();

        // The name may be written `vendor\/name` in the file.
        let package_regex = preg_quote(package).replace('/', "\\\\?/");
        let regex = compile(
            &format!(r#"{DEFINES}"(?P<package>{package_regex})"(\s*:\s*)(?&string)"#),
            IX,
        )?;
        if let Some(pm) = captures(&regex, &links)? {
            let existing = named(&pm, "package").unwrap_or("").to_owned();
            let package_regex = preg_quote(&existing).replace('/', "\\\\?/");
            let regex = compile(
                &format!(r#"{DEFINES}"{package_regex}"(?P<separator>\s*:\s*)(?&string)"#),
                IX,
            )?;
            let name = encode_str(&existing.replace("\\/", "/"))?;
            links = replace_all(&regex, &links, |m| {
                Ok(format!(
                    "{name}{}\"{constraint}\"",
                    named(m, "separator").unwrap_or("")
                ))
            })?
            .0;
        } else {
            let regex = compile(r"^\s*\{\s*\S+.*?(\s*\}\s*)$", S)?;
            let tail = captures(&regex, &links)?.and_then(|m| group(&m, 1).map(str::to_owned));
            if let Some(tail) = tail {
                let regex = compile(&format!("{}$", preg_quote(&tail)), NONE)?;
                let replacement = format!(
                    ",{nl}{ind}{ind}{}: {}{tail}",
                    encode_str(package)?,
                    encode_str(constraint)?,
                    nl = self.newline,
                    ind = self.indent
                );
                links = replace_literal(&regex, &links, &replacement)?.0;
            } else {
                links = format!(
                    "{{{nl}{ind}{ind}{}: {}{nl}{ind}}}",
                    encode_str(package)?,
                    encode_str(constraint)?,
                    nl = self.newline,
                    ind = self.indent
                );
            }
        }

        if sort {
            let requirements = json_decode(&links).map_err(|e| {
                ManipulatorError::Logic(format!("sortPackages(): links are not JSON: {e}"))
            })?;
            links = match requirements {
                Value::Object(mut m) => {
                    sort_packages(&mut m)?;
                    self.format(&Value::Object(m), 0, false)?
                }
                // A list is a PHP array with integer keys: `uksort` compares
                // nothing when there is at most one element.
                Value::Array(a) => {
                    let mut m = list_to_map(a);
                    sort_packages(&mut m)?;
                    self.format(&Value::Object(m), 0, false)?
                }
                _ => {
                    return Err(ManipulatorError::Logic(
                        "sortPackages(): Argument #1 ($packages) must be of type array".into(),
                    ))
                }
            };
        }

        self.contents = format!("{start}{property}{links}{end}");
        Ok(true)
    }

    /// `removeConfigSetting` (excluding `policy.*`).
    pub fn remove_config_setting(&mut self, name: &str) -> Result<bool> {
        if name.starts_with("policy.") && name.matches('.').count() >= 2 {
            return Err(ManipulatorError::Unsupported(format!(
                "removeConfigSetting({name}): policy lists"
            )));
        }
        self.remove_sub_node("config", name)
    }

    /// `addSubNode`: adds or replaces `$name` (or `$name.$subName` for
    /// `config`/`extra`/`scripts`) in the `$mainNode` object.
    pub fn add_sub_node(
        &mut self,
        main_node: &str,
        name: &str,
        value: &Value,
        append: bool,
    ) -> Result<bool> {
        let decoded = parse_json(&self.contents)?;
        let (name, sub_name) = split_sub_name(main_node, name);

        if isset(&decoded, main_node).is_none() {
            let inner = match sub_name {
                Some(sub) => {
                    let mut m = Map::new();
                    m.insert(sub.to_owned(), value.clone());
                    Value::Object(m)
                }
                None => value.clone(),
            };
            let mut m = Map::new();
            m.insert(name.to_owned(), inner);
            self.add_main_key(main_node, &Value::Object(m))?;
            return Ok(true);
        }

        let node_regex = self.node_regex(main_node)?;
        let Some(caps) = captures_or_limit(&node_regex, &self.contents)? else {
            return Ok(false);
        };
        let mut children = named(&caps, "content").unwrap_or("").to_owned();

        if !decodes_truthy(&children, false) {
            return Ok(false);
        }

        let child_regex = compile(
            &format!(
                r#"{DEFINES}(?P<start>"{}"\s*:\s*)(?P<content>(?&json))(?P<end>,?)"#,
                preg_quote(name)
            ),
            X,
        )?;
        if captures(&child_regex, &children)?.is_some() {
            children = replace_all(&child_regex, &children, |m| {
                let content = named(m, "content").unwrap_or("");
                let value = match sub_name {
                    Some(sub) => {
                        let mut cur = match json_decode(content) {
                            Ok(Value::Object(m)) => m,
                            Ok(Value::Array(a)) => list_to_map(a),
                            _ => Map::new(),
                        };
                        cur.insert(sub.to_owned(), value.clone());
                        Value::Object(cur)
                    }
                    None => value.clone(),
                };
                Ok(format!(
                    "{}{}{}",
                    named(m, "start").unwrap_or(""),
                    self.format(&value, 1, false)?,
                    named(m, "end").unwrap_or("")
                ))
            })?
            .0;
        } else {
            let regex = compile(
                r"^\{(?P<leadingspace>\s*?)(?P<content>\S+.*?)?(?P<trailingspace>\s*)\}$",
                S,
            )?;
            let Some(m) = captures(&regex, &children)? else {
                return Err(ManipulatorError::Logic(format!(
                    "Nothing matched above for: {children}"
                )));
            };
            let leading = named(&m, "leadingspace").unwrap_or("").to_owned();
            let trailing = named(&m, "trailingspace").unwrap_or("").to_owned();
            let has_content = named(&m, "content").is_some();
            let value = match sub_name {
                Some(sub) => {
                    let mut m = Map::new();
                    m.insert(sub.to_owned(), value.clone());
                    Value::Object(m)
                }
                None => value.clone(),
            };
            let entry = format!("{}: {}", encode_str(name)?, self.format(&value, 1, false)?);
            if has_content {
                if append {
                    let regex = compile(&format!("{trailing}}}$"), NONE)?;
                    let replacement = format!(
                        ",{nl}{ind}{ind}{entry}{trailing}}}",
                        nl = self.newline,
                        ind = self.indent
                    );
                    children = replace_literal(&regex, &children, &replacement)?.0;
                } else {
                    let regex = compile(&format!("^{{{leading}"), NONE)?;
                    let replacement = format!(
                        "{{{leading}{entry},{nl}{ind}{ind}",
                        nl = self.newline,
                        ind = self.indent
                    );
                    children = replace_literal(&regex, &children, &replacement)?.0;
                }
            } else {
                children = format!(
                    "{{{nl}{ind}{ind}{entry}{trailing}}}",
                    nl = self.newline,
                    ind = self.indent
                );
            }
        }

        self.contents = replace_all(&node_regex, &self.contents, |m| {
            Ok(format!(
                "{}{children}{}",
                named(m, "start").unwrap_or(""),
                named(m, "end").unwrap_or("")
            ))
        })?
        .0;
        Ok(true)
    }

    /// `removeSubNode`: removes `$name` (or `$name.$subName`) from the
    /// `$mainNode` object.
    pub fn remove_sub_node(&mut self, main_node: &str, name: &str) -> Result<bool> {
        let decoded = parse_json(&self.contents)?;
        if !php_index(&decoded, main_node).is_some_and(php_truthy) {
            return Ok(true);
        }

        let node_regex = self.node_regex(main_node)?;
        let Some(caps) = captures_or_limit(&node_regex, &self.contents)? else {
            return Ok(false);
        };
        let children = named(&caps, "content").unwrap_or("").to_owned();

        if !decodes_truthy(&children, true) {
            return Ok(false);
        }

        let (name, sub_name) = split_sub_name(main_node, name);

        let node = php_index(&decoded, main_node).unwrap_or(&Value::Null);
        let Some(entry) = isset(node, name) else {
            return Ok(true);
        };
        if let Some(sub) = sub_name {
            if truthy_str(sub) && isset(entry, sub).is_none() {
                return Ok(true);
            }
        }

        let key_regex = preg_quote(name).replace('/', "\\\\?/");
        let children_clean = if captures(&compile(&format!(r#""{key_regex}"\s*:"#), I)?, &children)?
            .is_some()
        {
            let all = compile(&format!(r#"{DEFINES}"{key_regex}"\s*:\s*(?:(?&json))"#), X)?;
            let mut best = String::new();
            let mut any = false;
            for m in all.captures_iter(children.as_bytes()) {
                let m = m.map_err(|e| ManipulatorError::Regex(e.to_string()))?;
                any = true;
                let whole = group(&m, 0).unwrap_or("");
                if best.len() < whole.len() {
                    best = whole.to_owned();
                }
            }
            if !any {
                return Err(ManipulatorError::Logic(
                    "JsonManipulator: $childrenClean is not defined. Please report at https://github.com/composer/composer/issues/new.".into(),
                ));
            }
            let (mut clean, count) = replace_literal(
                &compile(&format!(r",\s*{}", preg_quote(&best)), I)?,
                &children,
                "",
            )?;
            if count != 1 {
                let (clean2, count2) = replace_literal(
                    &compile(&format!(r"{}\s*,?\s*", preg_quote(&best)), I)?,
                    &clean,
                    "",
                )?;
                if count2 != 1 {
                    return Ok(false);
                }
                clean = clean2;
            }
            clean
        } else {
            children.clone()
        };

        let regex = compile(r"^\{\s*?(?P<content>\S+.*?)?(?P<trailingspace>\s*)\}$", S)?;
        if let Some(m) = captures(&regex, &children_clean)? {
            if named(&m, "content").is_none() {
                let empty = format!("{{{}{}}}", self.newline, self.indent);
                self.contents = replace_all(&node_regex, &self.contents, |m| {
                    Ok(format!(
                        "{}{empty}{}",
                        named(m, "start").unwrap_or(""),
                        named(m, "end").unwrap_or("")
                    ))
                })?
                .0;
                if let Some(sub) = sub_name {
                    let cur = json_decode(&children).map_err(ManipulatorError::Parse)?;
                    let inner = unset_sub(&cur, name, sub)?;
                    self.add_sub_node(main_node, name, &inner, true)?;
                }
                return Ok(true);
            }
        }

        let replacement = match sub_name {
            Some(sub) => {
                let content = named(&caps, "content").unwrap_or("");
                let mut cur = match json_decode(content) {
                    Ok(Value::Object(m)) => m,
                    Ok(Value::Array(a)) => list_to_map(a),
                    _ => Map::new(),
                };
                let inner = unset_sub(&Value::Object(cur.clone()), name, sub)?;
                cur.insert(name.to_owned(), inner);
                self.format(&Value::Object(cur), 0, true)?
            }
            None => children_clean,
        };
        self.contents = replace_all(&node_regex, &self.contents, |m| {
            Ok(format!(
                "{}{replacement}{}",
                named(m, "start").unwrap_or(""),
                named(m, "end").unwrap_or("")
            ))
        })?
        .0;
        Ok(true)
    }

    /// `addMainKey`: replaces the root key if it exists, otherwise appends
    /// it at the end of the object.
    pub fn add_main_key(&mut self, key: &str, content: &Value) -> Result<bool> {
        let decoded = parse_json(&self.contents)?;
        let content = self.format(content, 0, false)?;
        let encoded_key = encode_str(key)?;

        let regex = compile(
            &format!(
                r#"{DEFINES}^(?P<start>\s*\{{\s*(?:(?&string)\s*:\s*(?&json)\s*,\s*)*?)(?P<key>{}\s*:\s*(?&json))(?P<end>.*)"#,
                preg_quote(&encoded_key)
            ),
            SX,
        )?;
        if isset(&decoded, key).is_some() {
            if let Some(m) = captures(&regex, &self.contents)? {
                let key_text = named(&m, "key").unwrap_or("");
                if !decodes_truthy(&format!("{{{key_text}}}"), false) {
                    return Ok(false);
                }
                self.contents = format!(
                    "{}{encoded_key}: {content}{}",
                    named(&m, "start").unwrap_or(""),
                    named(&m, "end").unwrap_or("")
                );
                return Ok(true);
            }
        }

        let regex = compile(r"[^{\s](\s*)\}$", NONE)?;
        if let Some(m) = captures(&regex, &self.contents)? {
            let ws = group(&m, 1).unwrap_or("").to_owned();
            let regex = compile(&format!(r"{ws}\}}$"), NONE)?;
            let replacement = format!(
                ",{nl}{ind}{encoded_key}: {content}{nl}}}",
                nl = self.newline,
                ind = self.indent
            );
            self.contents = replace_literal(&regex, &self.contents, &replacement)?.0;
            return Ok(true);
        }

        let regex = compile(r"\}$", NONE)?;
        let replacement = format!(
            "{ind}{encoded_key}: {content}{nl}}}",
            nl = self.newline,
            ind = self.indent
        );
        self.contents = replace_literal(&regex, &self.contents, &replacement)?.0;
        Ok(true)
    }

    /// `removeMainKey`.
    pub fn remove_main_key(&mut self, key: &str) -> Result<bool> {
        let decoded = parse_json(&self.contents)?;
        if php_index(&decoded, key).is_none() {
            return Ok(true);
        }

        let regex = compile(
            &format!(
                r#"{DEFINES}^(?P<start>\s*\{{\s*(?:(?&string)\s*:\s*(?&json)\s*,\s*)*?)(?P<removal>{}\s*:\s*(?&json))\s*,?\s*(?P<end>.*)"#,
                preg_quote(&encode_str(key)?)
            ),
            SX,
        )?;
        let Some(m) = captures(&regex, &self.contents)? else {
            return Ok(false);
        };
        let mut start = named(&m, "start").unwrap_or("").to_owned();
        let removal = named(&m, "removal").unwrap_or("");
        let end = named(&m, "end").unwrap_or("").to_owned();

        if !decodes_truthy(&format!("{{{removal}}}"), false) {
            return Ok(false);
        }

        // Last key removed: the comma preceding it goes with it.
        if captures(&compile(r",\s*$", NONE)?, &start)?.is_some()
            && captures(&compile(r"^\}$", NONE)?, &end)?.is_some()
        {
            let regex = compile(r",(\s*)$", NONE)?;
            let stripped =
                replace_all(&regex, &start, |m| Ok(group(m, 1).unwrap_or("").to_owned()))?.0;
            let chars: Vec<char> = self.indent.chars().collect();
            start = stripped.trim_end_matches(chars.as_slice()).to_owned();
        }

        self.contents = format!("{start}{end}");
        if captures(&compile(r"^\{\s*\}\s*$", NONE)?, &self.contents)?.is_some() {
            self.contents = "{\n}".to_owned();
        }
        Ok(true)
    }

    /// `removeMainKeyIfEmpty`: removes the key if it is an empty array.
    pub fn remove_main_key_if_empty(&mut self, key: &str) -> Result<bool> {
        let decoded = parse_json(&self.contents)?;
        let Some(v) = php_index(&decoded, key) else {
            return Ok(true);
        };
        let empty = match v {
            Value::Array(a) => a.is_empty(),
            Value::Object(m) => m.is_empty(),
            _ => false,
        };
        if empty {
            return self.remove_main_key(key);
        }
        Ok(true)
    }

    /// `format`: formats a value at the given depth, with the file's
    /// indentation and newline.
    pub fn format(&self, data: &Value, depth: usize, was_object: bool) -> Result<String> {
        let indent = |n: usize| self.indent.repeat(n);
        let empty_object = |was_object: bool| {
            if was_object {
                format!("{{{}{}}}", self.newline, indent(depth + 1))
            } else {
                "[]".to_owned()
            }
        };
        match data {
            Value::Object(m) if is_stdclass_marker(m) => Ok(empty_object(true)),
            Value::Object(m) if m.is_empty() => Ok(empty_object(was_object)),
            Value::Array(a) if a.is_empty() => Ok(empty_object(was_object)),
            Value::Array(a) => {
                let items = a
                    .iter()
                    .map(|v| self.format(v, depth + 1, false))
                    .collect::<Result<Vec<_>>>()?;
                Ok(format!("[{}]", items.join(", ")))
            }
            Value::Object(m) if php_is_list(m) => {
                let items = m
                    .values()
                    .map(|v| self.format(v, depth + 1, false))
                    .collect::<Result<Vec<_>>>()?;
                Ok(format!("[{}]", items.join(", ")))
            }
            Value::Object(m) => {
                let mut elems = Vec::with_capacity(m.len());
                for (k, v) in m {
                    elems.push(format!(
                        "{}{}: {}",
                        indent(depth + 2),
                        encode_str(k)?,
                        self.format(v, depth + 1, false)?
                    ));
                }
                Ok(format!(
                    "{{{nl}{}{nl}{}}}",
                    elems.join(&format!(",{}", self.newline)),
                    indent(depth + 1),
                    nl = self.newline
                ))
            }
            scalar => encode(scalar),
        }
    }

    /// The pattern shared by `addSubNode`/`removeSubNode`: everything up to
    /// the root key, then its object, then the rest.
    fn node_regex(&self, main_node: &str) -> Result<Regex> {
        compile(
            &format!(
                r#"{DEFINES}^(?P<start> \s* \{{ \s* (?: (?&string) \s* : (?&json) \s* , \s* )*?{}\s*:\s*)(?P<content>(?&object))(?P<end>.*)"#,
                preg_quote(&encode_str(main_node)?)
            ),
            SX,
        )
    }
}

/// `config`/`extra`/`scripts` accept `name.sub` to target a sub-key.
fn split_sub_name<'a>(main_node: &str, name: &'a str) -> (&'a str, Option<&'a str>) {
    if matches!(main_node, "config" | "extra" | "scripts") {
        if let Some((n, s)) = name.split_once('.') {
            return (n, Some(s));
        }
    }
    (name, None)
}

/// A decoded list seen as a PHP array with integer keys.
fn list_to_map(a: Vec<Value>) -> Map<String, Value> {
    a.into_iter()
        .enumerate()
        .map(|(i, v)| (i.to_string(), v))
        .collect()
}

/// `unset($cur[$name][$sub]); if ($cur[$name] === []) new ArrayObject`.
fn unset_sub(cur: &Value, name: &str, sub: &str) -> Result<Value> {
    let inner = php_index(cur, name).cloned().unwrap_or(Value::Null);
    let mut m = match inner {
        Value::Object(m) => m,
        Value::Array(a) => list_to_map(a),
        // `unset` on `null` or `false` does nothing (and `=== []` is false).
        Value::Null => return Ok(Value::Null),
        Value::Bool(false) => return Ok(Value::Bool(false)),
        other => {
            return Err(ManipulatorError::Logic(format!(
                "Cannot unset offset in a non-array variable ({other})"
            )))
        }
    };
    m.shift_remove(sub);
    if m.is_empty() {
        return Ok(vivacity_core::phpjson::empty_stdclass());
    }
    Ok(Value::Object(m))
}

/// `JsonFile::detectIndenting`: the first line matching `^[ \t]+"`.
pub fn detect_indenting(json: &str) -> Result<String> {
    let re = compile(r#"^([ \t]+)""#, Flags { m: true, ..NONE })?;
    Ok(captures(&re, json)?
        .and_then(|c| group(&c, 1).map(str::to_owned))
        .unwrap_or_else(|| INDENT_DEFAULT.to_owned()))
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::cmp::Ordering::{Equal, Greater, Less};

    #[test]
    fn strnatcmp_matches_php() {
        assert_eq!(strnatcmp("0-php", "1-hhvm"), Less);
        assert_eq!(strnatcmp("ext-json", "ext-Json"), Greater);
        assert_eq!(strnatcmp("a10", "a9"), Greater);
        assert_eq!(strnatcmp("a 1", "a1"), Equal);
        assert_eq!(strnatcmp("a01", "a1"), Less);
        assert_eq!(strnatcmp("1.10", "1.9"), Greater);
        assert_eq!(strnatcmp("01", "1"), Equal);
        assert_eq!(strnatcmp("", "a"), Less);
        assert_eq!(strnatcmp("abc", "ab"), Greater);
    }

    #[test]
    fn sort_prefixes() {
        assert_eq!(sort_prefix("php"), "0-php");
        assert_eq!(sort_prefix("php-64bit"), "0-php-64bit");
        assert_eq!(sort_prefix("hhvm"), "1-hhvm");
        assert_eq!(sort_prefix("ext-json"), "2-ext-json");
        assert_eq!(sort_prefix("lib-icu"), "3-lib-icu");
        assert_eq!(sort_prefix("composer-plugin-api"), "4-composer-plugin-api");
        assert_eq!(sort_prefix("acme/lib"), "5-acme/lib");
    }

    #[test]
    fn add_link_to_existing_section() {
        let mut m =
            JsonManipulator::new("{\n    \"require\": {\n        \"a/b\": \"^1\"\n    }\n}\n")
                .expect("new");
        assert!(m.add_link("require", "c/d", "^2", false).expect("add"));
        assert_eq!(
            m.contents(),
            "{\n    \"require\": {\n        \"a/b\": \"^1\",\n        \"c/d\": \"^2\"\n    }\n}\n"
        );
        assert!(m.add_link("require", "A/B", "^3", true).expect("replace"));
        assert_eq!(
            m.contents(),
            "{\n    \"require\": {\n        \"a/b\": \"^3\",\n        \"c/d\": \"^2\"\n    }\n}\n"
        );
    }

    #[test]
    fn add_link_creates_section_and_remove_drops_it() {
        let mut m = JsonManipulator::new("{}").expect("new");
        assert!(m.add_link("require", "a/b", "^1", false).expect("add"));
        assert_eq!(
            m.contents(),
            "{\n    \"require\": {\n        \"a/b\": \"^1\"\n    }\n}\n"
        );
        assert!(m.remove_sub_node("require", "a/b").expect("remove"));
        assert_eq!(m.contents(), "{\n    \"require\": {\n    }\n}\n");
        assert!(m.remove_main_key_if_empty("require").expect("drop"));
        assert_eq!(m.contents(), "{\n}\n");
    }

    #[test]
    fn crlf_and_tabs_are_kept() {
        let mut m =
            JsonManipulator::new("{\r\n\t\"require\": {\r\n\t\t\"a/b\": \"^1\"\r\n\t}\r\n}")
                .expect("new");
        assert!(m.add_link("require", "c/d", "^2", false).expect("add"));
        assert_eq!(
            m.contents(),
            "{\r\n\t\"require\": {\r\n\t\t\"a/b\": \"^1\",\r\n\t\t\"c/d\": \"^2\"\r\n\t}\r\n}\r\n"
        );
    }

    #[test]
    fn config_sub_key_removal() {
        let mut m = JsonManipulator::new(
            "{\n    \"config\": {\n        \"allow-plugins\": {\n            \"a/b\": true,\n            \"c/d\": false\n        }\n    }\n}",
        )
        .expect("new");
        assert!(m
            .remove_config_setting("allow-plugins.a/b")
            .expect("remove"));
        assert_eq!(
            m.contents(),
            "{\n    \"config\": {\n        \"allow-plugins\": {\n            \"c/d\": false\n        }\n    }\n}\n"
        );
        assert!(m
            .remove_config_setting("allow-plugins.c/d")
            .expect("remove"));
        assert_eq!(
            m.contents(),
            "{\n    \"config\": {\n        \"allow-plugins\": {\n        }\n    }\n}\n"
        );
    }
}