hadris-iso 2.4.0

#[cfg(feature = "unstable-streaming")]
use alloc::boxed::Box;
use core::{fmt, ops::Deref};
use std::{fs::FileType, path::PathBuf};

use alloc::{
    collections::{BTreeMap, BTreeSet, VecDeque},
    string::{String, ToString},
    sync::Arc,
    vec::Vec,
};

use super::FileConversionError;
use crate::{
    directory::{DirectoryRef, FileFlags},
    file::{ConvertedName, EntryType},
    io::{self},
    read::PathSeparator,
    rrip::RripOptions,
    susp::SplitSu,
    write::{utils::*, writer::*},
};

/// Represents a file in the write order: (directory ID, file index).
pub type FileOrder = (DirectoryId, usize);

/// Maps (directory ID, entry type) to directory reference for relocation.
pub type RelocationMap = BTreeMap<(usize, EntryType), DirectoryRef>;

/// RRIP timestamp (7 bytes: year, month, day, hour, minute, second, offset).
pub type RripTime = [u8; 7];

/// A compact input tree for callers that do not need per-entry metadata.
///
/// [`InputTree`] is the richer model for Rock Ridge metadata and host
/// filesystem imports. Both models are accepted by
/// [`IsoImageWriter::create`](crate::write::IsoImageWriter::create).
pub struct InputFiles {
    /// Separator used by paths referenced from writer options.
    pub path_separator: PathSeparator,
    /// Root-level files and directories.
    pub files: Vec<File>,
}

#[derive(Clone, PartialEq, Eq)]
/// A file or directory in the compact [`InputFiles`] model.
pub enum File {
    /// The `File` variant.
    File {
        /// The `name` field.
        name: Arc<String>,
        /// The `contents` field.
        contents: Vec<u8>,
    },
    /// The `Directory` variant.
    Directory {
        /// The `name` field.
        name: Arc<String>,
        /// The `children` field.
        children: Vec<File>,
    },
}

impl core::fmt::Debug for File {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let mut dbg = f.debug_struct("File");
        match self {
            Self::Directory { name, children } => {
                dbg.field("name", name);
                dbg.field("children", children);
            }
            Self::File { name, contents } => {
                dbg.field("name", name);
                dbg.field("data_len", &contents.len());
            }
        }
        dbg.finish()
    }
}

impl File {
    /// Performs the `name` operation.
    pub fn name(&self) -> Arc<String> {
        match self {
            File::File { name, .. } => name.clone(),
            File::Directory { name, .. } => name.clone(),
        }
    }
}

/// A metadata-aware tree used to create an ISO image.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct InputTree {
    /// The `path_separator` field.
    pub path_separator: PathSeparator,
    /// The `entries` field.
    pub entries: Vec<InputEntry>,
}

impl InputTree {
    /// Recursive validation.
    ///
    /// Checks:
    /// - Depth: max 8 directories (unless RRIP deep dirs enabled)
    /// - Path length: max 255 bytes (unless RRIP deep dirs enabled)
    /// - Symlinks: require RRIP preserve_symlinks
    /// - Devices: require RRIP preserve_devices
    /// - File size: max 4 GiB
    pub(crate) fn validate(&self, rrip: Option<&RripOptions>) -> crate::io::Result<()> {
        Self::visit(&self.entries, rrip, 1, 0)
    }

    fn visit(
        entries: &[InputEntry],
        rrip: Option<&RripOptions>,
        depth: usize,
        path_len: usize,
    ) -> io::Result<()> {
        for entry in entries {
            match &entry.kind {
                InputEntryKind::Directory(children) => {
                    let child_path_len = if path_len == 0 {
                        entry.name.len()
                    } else {
                        path_len + 1 + entry.name.len()
                    };
                    if (depth >= 8 || child_path_len > 255)
                        && !rrip
                            .is_some_and(|options| options.enabled && options.relocate_deep_dirs)
                    {
                        return Err(io::Error::new(
                            io::ErrorKind::InvalidInput,
                            "directory depth or path length exceeds ISO 9660 limits and RRIP relocation is disabled",
                        ));
                    }

                    Self::visit(children, rrip, depth + 1, child_path_len)?;
                }
                InputEntryKind::Symlink(_) => {
                    if !rrip.is_some_and(|options| options.enabled && options.preserve_symlinks) {
                        return Err(io::Error::new(
                            io::ErrorKind::InvalidInput,
                            "symbolic links require RRIP preserve_symlinks",
                        ));
                    }
                }
                InputEntryKind::CharacterDevice { .. } | InputEntryKind::BlockDevice { .. } => {
                    if !rrip.is_some_and(|options| options.enabled && options.preserve_devices) {
                        return Err(io::Error::new(
                            io::ErrorKind::InvalidInput,
                            "device entries require RRIP preserve_devices",
                        ));
                    }
                }
                InputEntryKind::File(_contents) => {
                    // if contents.len() as u64 > Self::MAX_SINGLE_EXTENT_FILE_LEN {
                    //     return Err(io::Error::new(
                    //         io::ErrorKind::InvalidInput,
                    //         "file exceeds 4 GiB; the ISO writer stores each file in a single \
                    //         extent and cannot yet emit multi-extent records",
                    //     ));
                    // }
                }
                #[cfg(feature = "unstable-streaming")]
                InputEntryKind::Source(_) => {}
                #[cfg(test)]
                InputEntryKind::TestFile { .. } => {}
            }
        }
        Ok(())
    }
}

/// Optional POSIX metadata for an input entry.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
pub struct InputMetadata {
    /// The `mode` field.
    pub mode: Option<u32>,
    /// The `uid` field.
    pub uid: Option<u32>,
    /// The `gid` field.
    pub gid: Option<u32>,
    /// Creation time as seconds since the Unix epoch.
    pub created: Option<i64>,
    /// Modification time as seconds since the Unix epoch.
    pub modified: Option<i64>,
    /// Access time as seconds since the Unix epoch.
    pub accessed: Option<i64>,
}

impl InputMetadata {
    /// Creates metadata from filesystem metadata.
    pub(crate) fn from_fs(fs_metadata: std::fs::Metadata) -> Self {
        #[allow(unused_mut)]
        let mut metadata = Self {
            created: system_time_seconds(fs_metadata.created()),
            modified: system_time_seconds(fs_metadata.modified()),
            accessed: system_time_seconds(fs_metadata.accessed()),
            ..Self::default()
        };

        #[cfg(unix)]
        {
            use std::os::unix::fs::MetadataExt;
            metadata.mode = Some(fs_metadata.mode() & 0o7777);
            metadata.uid = Some(fs_metadata.uid());
            metadata.gid = Some(fs_metadata.gid());
        }

        metadata
    }
}

/// File contents that are read from a reader while the image is written.
///
/// An [`InputEntryKind::File`] holds its contents in memory. A `FileSource` holds
/// only the length and a way to open a reader, so a large file is streamed into the
/// image in fixed-size chunks and never has to fit in memory. The writer opens the
/// reader once per file, in image order, and reads exactly `len` bytes from it: a
/// reader that ends early fails the write with [`ErrorKind::UnexpectedEof`], and
/// bytes beyond `len` are never read.
///
/// The reader is a blocking [`std::io::Read`] in both the sync and the async writer;
/// the async writer awaits only its output.
///
/// Requires the `unstable-streaming` feature, which is outside the V2 API
/// stability promise: this type and [`InputEntryKind::Source`] may change in any
/// release.
///
/// [`ErrorKind::UnexpectedEof`]: crate::io::ErrorKind::UnexpectedEof
#[cfg(feature = "unstable-streaming")]
#[cfg_attr(docsrs, doc(cfg(feature = "unstable-streaming")))]
#[derive(Clone)]
pub struct FileSource {
    len: u64,
    open:
        alloc::sync::Arc<dyn Fn() -> std::io::Result<Box<dyn std::io::Read + Send>> + Send + Sync>,
}

#[cfg(feature = "unstable-streaming")]
impl FileSource {
    /// A source of `len` bytes that `open` produces a reader for.
    pub fn new<F>(len: u64, open: F) -> Self
    where
        F: Fn() -> std::io::Result<Box<dyn std::io::Read + Send>> + Send + Sync + 'static,
    {
        Self {
            len,
            open: alloc::sync::Arc::new(open),
        }
    }

    /// A source that streams the file at `path`, whose length is taken now.
    ///
    /// The file is opened when the image is written; its length must not change
    /// in between.
    ///
    /// ```
    /// # #[cfg(feature = "unstable-streaming")] {
    /// use hadris_iso::write::FileSource;
    ///
    /// let path = std::env::temp_dir().join("hadris-iso-file-source-doc.bin");
    /// std::fs::write(&path, b"streamed")?;
    /// let source = FileSource::from_path(&path)?;
    /// assert_eq!(source.len(), 8);
    /// let mut contents = Vec::new();
    /// source.open()?.read_to_end(&mut contents)?;
    /// assert_eq!(contents, b"streamed");
    /// # std::fs::remove_file(&path)?;
    /// # }
    /// # Ok::<(), std::io::Error>(())
    /// ```
    pub fn from_path(path: impl Into<PathBuf>) -> std::io::Result<Self> {
        let path = path.into();
        let len = std::fs::metadata(&path)?.len();
        Ok(Self::new(len, move || {
            std::fs::File::open(&path).map(|file| Box::new(file) as Box<dyn std::io::Read + Send>)
        }))
    }

    /// The number of bytes the source provides.
    pub const fn len(&self) -> u64 {
        self.len
    }

    /// Whether the source provides no bytes.
    pub const fn is_empty(&self) -> bool {
        self.len == 0
    }

    /// Opens a fresh reader positioned at the start of the contents.
    pub fn open(&self) -> std::io::Result<Box<dyn std::io::Read + Send>> {
        (self.open)()
    }
}

#[cfg(feature = "unstable-streaming")]
impl fmt::Debug for FileSource {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("FileSource")
            .field("len", &self.len)
            .finish_non_exhaustive()
    }
}

#[cfg(feature = "unstable-streaming")]
impl PartialEq for FileSource {
    /// Two sources are equal when they share the same opener and length.
    fn eq(&self, other: &Self) -> bool {
        self.len == other.len && alloc::sync::Arc::ptr_eq(&self.open, &other.open)
    }
}

#[cfg(feature = "unstable-streaming")]
impl Eq for FileSource {}

/// The data represented by an [`InputEntry`].
///
/// With the `unstable-streaming` feature enabled this enum gains the
/// `Source` variant, so exhaustive matches on it must account for
/// that variant when the feature is on.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum InputEntryKind {
    /// The `File` variant.
    File(Vec<u8>),
    /// A file whose contents are streamed from a reader while the image is written.
    ///
    /// Requires the `unstable-streaming` feature and is outside the V2 API
    /// stability promise.
    #[cfg(feature = "unstable-streaming")]
    #[cfg_attr(docsrs, doc(cfg(feature = "unstable-streaming")))]
    Source(FileSource),
    /// A virtual sparse file used by large-file regression tests.
    #[cfg(test)]
    TestFile {
        /// Virtual file size in bytes.
        size: u64,
    },
    /// The `Directory` variant.
    Directory(Vec<InputEntry>),
    /// The `Symlink` variant.
    Symlink(String),
    /// A POSIX character device.
    CharacterDevice {
        /// Device-class identifier.
        major: u32,
        /// Device identifier within the class.
        minor: u32,
    },
    /// A POSIX block device.
    BlockDevice {
        /// Device-class identifier.
        major: u32,
        /// Device identifier within the class.
        minor: u32,
    },
}

impl InputEntryKind {
    pub(crate) fn file_len(&self) -> Option<u64> {
        match self {
            Self::File(contents) => Some(contents.len() as u64),
            #[cfg(feature = "unstable-streaming")]
            Self::Source(source) => Some(source.len()),
            #[cfg(test)]
            Self::TestFile { size } => Some(*size),
            _ => None,
        }
    }

    /// Creates an `InputEntryKind` from a filesystem file type and path.
    ///
    /// This function inspects the file type and constructs the appropriate variant:
    /// - Regular files → `File` (contents are read into memory).
    /// - Directories → `Directory` (recursively reads all children).
    /// - Symlinks → `Symlink` (reads the link target as a string).
    /// - Character/block devices → `CharacterDevice` or `BlockDevice` (Unix only).
    pub(crate) fn new(
        file_type: FileType,
        path: PathBuf,
    ) -> core::result::Result<Self, FileConversionError> {
        if file_type.is_file() {
            let content = std::fs::read(&path)?;
            Ok(InputEntryKind::File(content))
        } else if file_type.is_dir() {
            let dir = read_input_directory_recursively(&path)?;
            Ok(InputEntryKind::Directory(dir))
        } else if file_type.is_symlink() {
            let target = std::fs::read_link(&path)?;
            let symlink = target
                .to_str()
                .ok_or_else(|| FileConversionError::InvalidUtf8Path(target.clone()))?
                .to_string();
            Ok(InputEntryKind::Symlink(symlink))
        } else {
            #[cfg(unix)]
            {
                use std::os::unix::fs::{FileTypeExt, MetadataExt};
                const DEV_MAJOR_MASK_LOW: u64 = 0xfff;
                const DEV_MAJOR_MASK_HIGH: u64 = 0xfffff000;
                const DEV_MINOR_MASK_LOW: u64 = 0xff;
                const DEV_MINOR_MASK_HIGH: u64 = 0xffffff00;

                let fs_metadata = std::fs::symlink_metadata(&path)?;
                let device = fs_metadata.rdev();
                let major =
                    ((device >> 8) & DEV_MAJOR_MASK_LOW) | ((device >> 32) & DEV_MAJOR_MASK_HIGH);
                let minor = (device & DEV_MINOR_MASK_LOW) | ((device >> 12) & DEV_MINOR_MASK_HIGH);
                if file_type.is_char_device() {
                    Ok(InputEntryKind::CharacterDevice {
                        major: major as u32,
                        minor: minor as u32,
                    })
                } else if file_type.is_block_device() {
                    Ok(InputEntryKind::BlockDevice {
                        major: major as u32,
                        minor: minor as u32,
                    })
                } else {
                    Err(FileConversionError::UnsupportedFileType(path))
                }
            }
            #[cfg(not(unix))]
            return Err(FileConversionError::UnsupportedFileType(path));
        }
    }
}

/// A named entry in the input tree, representing a file, directory, symlink, or device.
///
/// This struct is the building block of [`InputTree`]. Each entry has a name,
/// a kind (file, directory, symlink, or device), and optional POSIX metadata.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct InputEntry {
    /// The name of the entry (e.g., `"README.txt"` or `"src"`).
    pub name: Arc<String>,
    /// The kind of entry (file, directory, symlink, or device).
    pub kind: InputEntryKind,
    /// Optional POSIX metadata (permissions, ownership, timestamps).
    pub metadata: InputMetadata,
}

impl InputEntry {
    /// Creates a regular file entry with content.
    pub fn file(name: impl Into<String>, contents: impl Into<Vec<u8>>) -> Self {
        Self::new(name, InputEntryKind::File(contents.into()))
    }

    /// Creates a directory entry with children.
    pub fn directory(name: impl Into<String>, children: Vec<Self>) -> Self {
        Self::new(name, InputEntryKind::Directory(children))
    }

    /// Creates a symbolic link entry.
    pub fn symlink(name: impl Into<String>, target: impl Into<String>) -> Self {
        Self::new(name, InputEntryKind::Symlink(target.into()))
    }

    /// Creates a character device entry (Unix only).
    pub fn character_device(name: impl Into<String>, major: u32, minor: u32) -> Self {
        Self::new(name, InputEntryKind::CharacterDevice { major, minor })
    }

    /// Creates a block device entry (Unix only).
    pub fn block_device(name: impl Into<String>, major: u32, minor: u32) -> Self {
        Self::new(name, InputEntryKind::BlockDevice { major, minor })
    }

    /// Sets the metadata for this entry.
    pub fn with_metadata(mut self, metadata: InputMetadata) -> Self {
        self.metadata = metadata;
        self
    }

    /// Returns the name of the entry.
    pub fn name(&self) -> Arc<String> {
        self.name.clone()
    }

    fn new(name: impl Into<String>, kind: InputEntryKind) -> Self {
        Self {
            name: Arc::new(name.into()),
            kind,
            metadata: InputMetadata::default(),
        }
    }
}

impl InputTree {
    /// Performs the `new` operation.
    pub fn new(path_separator: PathSeparator, entries: Vec<InputEntry>) -> Self {
        Self {
            path_separator,
            entries,
        }
    }

    /// Performs the `from_fs` operation.
    pub fn from_fs(
        root_path: &std::path::Path,
        path_separator: PathSeparator,
    ) -> core::result::Result<Self, FileConversionError> {
        if !root_path.is_dir() {
            return Err(FileConversionError::Io(std::io::Error::new(
                std::io::ErrorKind::InvalidInput,
                alloc::format!("Root path '{root_path:?}' is not a directory"),
            )));
        }
        Ok(Self::new(
            path_separator,
            read_input_directory_recursively(root_path)?,
        ))
    }
}

impl From<InputFiles> for InputTree {
    fn from(value: InputFiles) -> Self {
        fn convert(file: File) -> InputEntry {
            match file {
                File::File { name, contents } => InputEntry {
                    name,
                    kind: InputEntryKind::File(contents),
                    metadata: InputMetadata::default(),
                },
                File::Directory { name, children } => InputEntry {
                    name,
                    kind: InputEntryKind::Directory(children.into_iter().map(convert).collect()),
                    metadata: InputMetadata::default(),
                },
            }
        }
        Self::new(
            value.path_separator,
            value.files.into_iter().map(convert).collect(),
        )
    }
}

/// Depth-first tree walker.
pub struct FileTreeWalker<'a>(VecDeque<StackFrame<'a>>);

/// Walker internal state.
enum StackFrame<'a> {
    Node(&'a InputEntry),
    DirExit(&'a InputEntry),
}

/// Walker events.
#[derive(Debug, PartialEq, Eq)]
pub enum TreeWalkerItem<'a> {
    /// Entering a directory.
    EnterDirectory(&'a InputEntry),
    /// A file.
    File(&'a InputEntry),
    /// Exiting a directory.
    ExitDirectory(&'a InputEntry),
}

impl<'a> FileTreeWalker<'a> {
    /// New walker.
    pub fn new(input: &'a InputTree) -> Self {
        let mut stack = VecDeque::new();
        for file in input.entries.iter().rev() {
            stack.push_back(StackFrame::Node(file));
        }
        FileTreeWalker(stack)
    }

    /// Walk the tree and build WrittenFiles.
    pub fn walk(self, written_files: &mut WrittenFiles) {
        let mut next_directory_id = 1usize;
        let mut current_dir = written_files.root_dir();

        for file in self {
            match file {
                TreeWalkerItem::EnterDirectory(dir) => {
                    let name = dir.name();
                    let metadata = dir.metadata;
                    let written_dir = written_files.get_mut(&current_dir);
                    let index = written_dir.push_dir(name, metadata);
                    written_dir.dirs[index].id = next_directory_id;
                    next_directory_id += 1;
                    current_dir.push(index);
                }
                TreeWalkerItem::ExitDirectory(_dir) => {
                    current_dir.pop();
                }
                TreeWalkerItem::File(file) => {
                    // Extents are assigned in the planning pass below.
                    // Empty files keep extent 0 (per ISO 9660 they have no
                    // data to reference).
                    let dir = written_files.get_mut(&current_dir);
                    dir.push_file(WrittenFile::new(
                        file.name.clone(),
                        file.kind.clone(),
                        file.metadata,
                    ));
                }
            };
        }
    }
}

impl<'a> Iterator for FileTreeWalker<'a> {
    type Item = TreeWalkerItem<'a>;

    fn next(&mut self) -> Option<Self::Item> {
        let frame = self.0.pop_back()?;
        match frame {
            StackFrame::Node(file) => match &file.kind {
                InputEntryKind::Directory(children) => {
                    // Yield that we are entering this directory (pre-order event)
                    let current_dir = file;

                    // Push an Exit frame to signal leaving this directory later
                    self.0.push_back(StackFrame::DirExit(current_dir));

                    // Push children in reverse order for DFS
                    for child in children.iter().rev() {
                        self.0.push_back(StackFrame::Node(child));
                    }

                    Some(TreeWalkerItem::EnterDirectory(current_dir))
                }
                _ => Some(TreeWalkerItem::File(file)),
            },
            StackFrame::DirExit(dir) => Some(TreeWalkerItem::ExitDirectory(dir)),
        }
    }
}

/// Selects which RRIP fields to emit for a directory entry.
pub enum RripEntryKind<'a> {
    /// Root "." entry: SP + ER + PX + NM(CURRENT)
    RootDot {
        /// File metadata.
        metadata: InputMetadata,
        /// Number of hard links.
        nlink: u32,
    },
    /// Root ".." entry: PX + NM(PARENT)
    RootDotDot {
        /// File metadata.
        metadata: InputMetadata,
        /// Number of hard links.
        nlink: u32,
    },
    /// Non-root "." entry: PX + NM(CURRENT)
    Dot {
        /// File metadata.
        metadata: InputMetadata,
        /// Number of hard links.
        nlink: u32,
    },
    /// Non-root ".." entry: PX + NM(PARENT)
    DotDot {
        /// File metadata.
        metadata: InputMetadata,
        /// Number of hard links.
        nlink: u32,
    },
    /// Named directory entry.
    Directory {
        /// Original filename.
        original_name: &'a str,
        /// File metadata.
        metadata: InputMetadata,
        /// Number of hard links.
        nlink: u32,
    },
    /// Named file entry.
    Entry {
        /// Original filename.
        original_name: &'a str,
        /// File metadata.
        metadata: InputMetadata,
        /// File contents.
        kind: &'a InputEntryKind,
    },
}

/// Pending directory records ready for layout planning and writing.
///
/// # Overview
/// These records represent a single directory's contents, including:
/// - "." and ".." entries
/// - Child directories
/// - Files (including multi-extent files)
///
/// # RRIP System Use
/// Each record can have inline System Use data (RRIP extensions).
/// If the data exceeds the available inline space, overflow is tracked
/// and written to a separate continuation area.
///
/// # Multi-Extent Files
/// Files larger than 4 GiB are split into multiple extents.
/// Each extent gets its own directory record with the same name.
/// The first record contains the RRIP data; subsequent extents
/// have empty System Use.
///
/// # Processing Phases
/// 1. Build all records with RRIP entries split against inline space
/// 2. If any record has overflow: allocate continuation area, patch CE entries
/// 3. Write directory records with inline SU bytes
///
/// # Ordering
/// Records are ordered by File Identifier (ECMA-119 9.3):
/// - "." (0x00) first
/// - ".." (0x01) second
/// - All other entries sorted by name
pub struct PendingRecords(Vec<PendingRecord>);

impl PendingRecords {
    /// Build the pending records for one directory: dot/dotdot, child
    /// directories, and files, with RRIP system-use areas split against the
    /// inline budget, names deduplicated, and records ordered by File
    /// Identifier (ECMA-119 9.3). Sizes never depend on extent values, so
    /// the planning pass calls this with placeholder refs and the write pass
    /// rebuilds with the real ones.
    #[allow(clippy::too_many_arguments)]
    pub fn new(
        ty: EntryType,
        dir: &WrittenDirectory,
        is_root: bool,
        inode_counter: &mut u32,
        rrip_options: Option<&RripOptions>,
        fallback_time: &RripTime,
        relocation_refs: &BTreeMap<(usize, EntryType), DirectoryRef>,
        sector_size: u64,
    ) -> io::Result<Self> {
        let rrip_options = rrip_options.filter(|options| options.enabled);
        let has_rrip = ty.supports_rrip() && rrip_options.is_some();
        let options = rrip_options.copied().unwrap_or_else(RripOptions::disabled);
        let directory_nlink = 2 + dir.dirs.len() as u32;

        let mut records: Vec<PendingRecord> = Vec::new();

        records.push(Self::build_dot_record(
            has_rrip,
            is_root,
            &dir.metadata,
            directory_nlink,
            &options,
            fallback_time,
        ));
        records.push(Self::build_dotdot_record(
            has_rrip,
            is_root,
            dir,
            ty,
            &options,
            fallback_time,
            relocation_refs,
        )?);

        Self::build_directory_records(
            &mut records,
            &dir.dirs,
            ty,
            has_rrip,
            &options,
            inode_counter,
            fallback_time,
            relocation_refs,
        )?;

        Self::build_file_records(
            &mut records,
            &dir.files,
            ty,
            has_rrip,
            &options,
            inode_counter,
            fallback_time,
            sector_size,
        )?;

        Self::deduplicate_names(&mut records, ty);
        Self::sort_records(&mut records);

        Ok(Self(records))
    }

    fn build_dot_record(
        has_rrip: bool,
        is_root: bool,
        metadata: &InputMetadata,
        nlink: u32,
        options: &RripOptions,
        fallback_time: &RripTime,
    ) -> PendingRecord {
        let split = if has_rrip {
            let kind = if is_root {
                RripEntryKind::RootDot {
                    metadata: *metadata,
                    nlink,
                }
            } else {
                RripEntryKind::Dot {
                    metadata: *metadata,
                    nlink,
                }
            };
            let max = available_su_space(1); // name is b"\x00"
            build_rrip_entries(kind, 0, options, fallback_time).build_split(max)
        } else {
            SplitSu::empty()
        };
        PendingRecord::current_dir(split)
    }

    fn build_dotdot_record(
        has_rrip: bool,
        is_root: bool,
        dir: &WrittenDirectory,
        ty: EntryType,
        options: &RripOptions,
        fallback_time: &RripTime,
        relocation_refs: &BTreeMap<(usize, EntryType), DirectoryRef>,
    ) -> io::Result<PendingRecord> {
        let split = if has_rrip {
            let kind = if is_root {
                RripEntryKind::RootDotDot {
                    metadata: dir.metadata,
                    nlink: 2 + dir.dirs.len() as u32,
                }
            } else {
                RripEntryKind::DotDot {
                    metadata: dir.metadata,
                    nlink: 2 + dir.dirs.len() as u32,
                }
            };
            let max = available_su_space(1); // name is b"\x01"
            let mut builder = build_rrip_entries(kind, 0, options, fallback_time);
            if let DirectoryRelocation::Moved { logical_parent, .. } = dir.relocation {
                let parent = relocation_refs
                    .get(&(logical_parent, ty))
                    .copied()
                    .unwrap_or_default();
                builder.add_pl(parent.extent.0 as u32);
            }
            builder.build_split(max)
        } else {
            SplitSu::empty()
        };
        Ok(PendingRecord::parent_dir(split))
    }

    #[allow(clippy::too_many_arguments)]
    fn build_directory_records(
        records: &mut Vec<PendingRecord>,
        dirs: &[WrittenDirectory],
        ty: EntryType,
        has_rrip: bool,
        options: &RripOptions,
        inode_counter: &mut u32,
        fallback_time: &RripTime,
        relocation_refs: &BTreeMap<(usize, EntryType), DirectoryRef>,
    ) -> io::Result<()> {
        for directory in dirs {
            let converted_name = ty.convert_directory_name(&directory.name);
            let split = if has_rrip {
                let inode = *inode_counter;
                *inode_counter += 1;
                let max = available_su_space(converted_name.as_bytes().len());
                let mut builder = build_rrip_entries(
                    RripEntryKind::Directory {
                        original_name: &directory.rrip_name,
                        metadata: directory.metadata,
                        nlink: 2 + directory.dirs.len() as u32,
                    },
                    inode,
                    options,
                    fallback_time,
                );
                match directory.relocation {
                    DirectoryRelocation::Placeholder { target } => {
                        let target = relocation_refs.get(&(target, ty)).ok_or_else(|| {
                            io::Error::new(
                                io::ErrorKind::InvalidData,
                                "relocated directory extent was not written",
                            )
                        })?;
                        builder.add_cl(target.extent.0 as u32);
                    }
                    DirectoryRelocation::Moved { .. } => {
                        builder.add_re();
                    }
                    DirectoryRelocation::None => {}
                }
                builder.build_split(max)
            } else {
                SplitSu::empty()
            };

            let dir_ref = directory.get_dir_ref(ty, relocation_refs);
            let record = PendingRecord::new(&converted_name, split, dir_ref, FileFlags::DIRECTORY);

            records.push(record);
        }
        Ok(())
    }

    #[allow(clippy::too_many_arguments)]
    fn build_file_records(
        records: &mut Vec<PendingRecord>,
        files: &[WrittenFile],
        ty: EntryType,
        has_rrip: bool,
        options: &RripOptions,
        inode_counter: &mut u32,
        fallback_time: &RripTime,
        sector_size: u64,
    ) -> io::Result<()> {
        for file in files {
            let converted_name = ty.convert_name(&file.name);

            // Main file entry (with RRIP if enabled)
            let split = if has_rrip {
                let inode = *inode_counter;
                *inode_counter += 1;
                let max = available_su_space(converted_name.as_bytes().len());
                let kind = RripEntryKind::Entry {
                    original_name: &file.name,
                    metadata: file.metadata,
                    kind: &file.kind,
                };
                build_rrip_entries(kind, inode, options, fallback_time).build_split(max)
            } else {
                SplitSu::empty()
            };

            let len = file.kind.file_len().unwrap_or(file.entry.size as u64);
            let cursor = file.entry.extent.0 as u64 * sector_size;
            let mut additional_extents =
                ExtentIter::new(len, cursor, sector_size).skip(1).peekable();
            let first_flags = if additional_extents.peek().is_none() {
                FileFlags::empty()
            } else {
                FileFlags::NOT_FINAL
            };

            let first = PendingRecord::new(&converted_name, split, file.entry, first_flags);
            records.push(first);

            while let Some((dir_ref, _)) = additional_extents.next() {
                let flags = if additional_extents.peek().is_none() {
                    FileFlags::empty()
                } else {
                    FileFlags::NOT_FINAL
                };

                let record = PendingRecord::new(&converted_name, SplitSu::empty(), dir_ref, flags);
                records.push(record);
            }
        }
        Ok(())
    }

    fn deduplicate_names(records: &mut [PendingRecord], ty: EntryType) {
        let mut seen: BTreeSet<PendingRecordName> = BTreeSet::new();
        let mut start = 0;

        while start < records.len() {
            if records[start].name.len() == 1
                && (records[start].name[0] == 0x00 || records[start].name[0] == 0x01)
            {
                start += 1;
                continue;
            }

            let mut end = start + 1;
            while end < records.len() && records[end - 1].flags.contains(FileFlags::NOT_FINAL) {
                end += 1;
            }

            let original_name = records[start].name.clone();
            let unique_name = if seen.contains(&original_name) {
                let mut suffix = 1;
                loop {
                    let candidate = apply_dedup_suffix(&original_name, suffix, ty).into();
                    suffix += 1;
                    if !seen.contains(&candidate) {
                        break candidate;
                    }
                }
            } else {
                original_name.clone()
            };

            seen.insert(unique_name.clone());

            for record in &mut records[start..end] {
                record.name = unique_name.clone();
            }

            start = end;
        }
    }

    fn sort_records(records: &mut [PendingRecord]) {
        records.sort_by(|a, b| {
            let rank = |name: &[u8]| match name {
                [0x00] => 0,
                [0x01] => 1,
                _ => 2,
            };
            rank(&a.name)
                .cmp(&rank(&b.name))
                .then_with(|| a.name.cmp(&b.name))
        });
    }

    /// Total length of all overflow (continuation area) data.
    pub fn overflow_len(&self) -> u64 {
        self.0
            .iter()
            .filter(|r| r.split.has_overflow())
            .map(|r| r.split.overflow.len() as u64)
            .sum()
    }

    /// Returns true if any record has RRIP overflow (needs continuation area).
    pub fn has_overflow(&self) -> bool {
        self.0.iter().any(|r| r.split.has_overflow())
    }

    /// Iterates over pending records.
    pub fn iter(&self) -> impl Iterator<Item = &PendingRecord> {
        self.0.iter()
    }

    /// Mutably iterates over pending records.
    pub fn iter_mut(&mut self) -> impl Iterator<Item = &mut PendingRecord> {
        self.0.iter_mut()
    }
}

/// A reference-counted, cloneable name for a pending directory record.
///
/// This type wraps an `Arc<[u8]>`, allowing efficient sharing of the name
/// across multiple records (e.g., multi-extent files where all extents share
/// the same name). It is cheap to clone and can be compared and hashed.
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub struct PendingRecordName(Arc<[u8]>);

impl From<Vec<u8>> for PendingRecordName {
    fn from(value: Vec<u8>) -> Self {
        Self(value.into())
    }
}

impl Deref for PendingRecordName {
    type Target = [u8];

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

/// A pending directory record, built in phase 1 and written in phases 2-3.
#[derive(Debug, Clone)]
pub struct PendingRecord {
    /// File or directory name.
    pub name: PendingRecordName,
    /// System Use data (split into inline and overflow).
    pub split: SplitSu,
    /// Location and size of the file/dir.
    pub dir_ref: DirectoryRef,
    /// File flags (directory, hidden, etc.).
    pub flags: FileFlags,
}

impl PendingRecord {
    /// Creates a new pending directory record.
    pub fn new(
        name: &ConvertedName,
        split: SplitSu,
        dir_ref: DirectoryRef,
        flags: FileFlags,
    ) -> Self {
        Self {
            name: PendingRecordName(name.as_bytes().into()),
            split,
            dir_ref,
            flags,
        }
    }

    /// Current directory "." entry.
    pub fn current_dir(dot_split: SplitSu) -> Self {
        Self {
            name: PendingRecordName([0x00].into()),
            split: dot_split,
            dir_ref: DirectoryRef::default(),
            flags: FileFlags::DIRECTORY,
        }
    }

    /// Parent directory ".." entry.
    pub fn parent_dir(dotdot_split: SplitSu) -> Self {
        Self {
            name: PendingRecordName([0x01].into()),
            split: dotdot_split,
            dir_ref: DirectoryRef::default(),
            flags: FileFlags::DIRECTORY,
        }
    }
}

/// An iterator that splits a file into ISO 9660 extents.
pub struct ExtentIter {
    remaining: u64,
    cursor: u64,
    sector_size: u64,
}

impl ExtentIter {
    /// Creates a new extent iterator.
    pub fn new(len: u64, cursor: u64, sector_size: u64) -> Self {
        Self {
            remaining: len,
            cursor,
            sector_size,
        }
    }

    // ECMA-119 6.5.4: Non-final extents must be multiples of the logical block size.
    fn max_extent_size(&self) -> u64 {
        (u32::MAX as u64 / self.sector_size) * self.sector_size
    }

    /// Collects all extents into a `Vec<DirectoryRef>` and returns the final cursor.
    #[cfg(test)]
    pub fn collect_with_cursor(self) -> (Vec<DirectoryRef>, u64) {
        let mut extents = Vec::new();
        let mut cursor = self.cursor;

        for (extent, new_cursor) in self {
            extents.push(extent);
            cursor = new_cursor;
        }

        (extents, cursor)
    }
}

impl Iterator for ExtentIter {
    type Item = (DirectoryRef, u64);

    fn next(&mut self) -> Option<Self::Item> {
        if self.remaining == 0 {
            return None;
        }

        let chunk = self.remaining.min(self.max_extent_size());

        let aligned = (self.cursor + self.sector_size - 1) & !(self.sector_size - 1);
        let extent = DirectoryRef::new((aligned / self.sector_size) as usize, chunk as usize);

        self.cursor = aligned + chunk;
        self.remaining -= chunk;

        Some((extent, self.cursor))
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn extent_iter_splits_correctly() {
        let sector_size = 2048;
        let cursor = 0;

        let len = u32::MAX as u64 + 1;
        let iter = ExtentIter::new(len, cursor, sector_size);
        let (extents, final_cursor) = iter.collect_with_cursor();

        assert_eq!(extents.len(), 2);

        assert_eq!(extents[0].size, 4_294_965_248);
        assert_eq!(extents[0].extent.0, 0);

        assert_eq!(extents[1].size, 2048);
        let expected_sector: u64 = 4_294_965_248 / 2048;
        assert_eq!(extents[1].extent.0, expected_sector as usize);

        assert_eq!(final_cursor, len);
    }

    #[test]
    fn extent_iter_aligns_non_final_extents_to_sector_size() {
        let sector_size = 4096;
        let len = u32::MAX as u64 + 1;
        let (extents, final_cursor) = ExtentIter::new(len, 0, sector_size).collect_with_cursor();

        assert_eq!(extents.len(), 2);
        assert_eq!(extents[0].size as u64, 4_294_963_200);
        assert_eq!(extents[0].size as u64 % sector_size, 0);
        assert_eq!(extents[1].size as u64, len - 4_294_963_200);
        assert_eq!(extents[1].extent.0 as u64, 4_294_963_200 / sector_size);
        assert_eq!(final_cursor, len);
    }

    #[test]
    fn extent_iter_small_file() {
        let sector_size = 2048;
        let len = 1024; // < 4 GiB
        let iter = ExtentIter::new(len, 0, sector_size);
        let (extents, final_cursor) = iter.collect_with_cursor();

        assert_eq!(extents.len(), 1);
        assert_eq!(extents[0].size, 1024);
        assert_eq!(extents[0].extent.0, 0);
        assert_eq!(final_cursor, 1024);
    }

    #[test]
    fn deduplicates_logical_files_without_splitting_extent_names() {
        let record = |flags| PendingRecord {
            name: b"README.TXT;1".to_vec().into(),
            split: SplitSu::empty(),
            dir_ref: DirectoryRef::default(),
            flags,
        };
        let mut records = vec![
            record(FileFlags::NOT_FINAL),
            record(FileFlags::empty()),
            record(FileFlags::empty()),
        ];

        PendingRecords::deduplicate_names(&mut records, EntryType::default());

        assert_eq!(records[0].name, records[1].name);
        assert_ne!(records[1].name, records[2].name);
    }
}