oxidize_pdf/writer/

pdf_writer.rs

use crate::document::Document;
use crate::error::Result;
use crate::objects::{Dictionary, Object, ObjectId};
use crate::text::fonts::embedding::CjkFontType;
use crate::writer::XRefStreamWriter;
use chrono::{DateTime, Utc};
use std::collections::HashMap;
use std::io::{BufWriter, Write};
use std::path::Path;

/// Configuration for PDF writer
#[derive(Debug, Clone)]
pub struct WriterConfig {
    /// Use XRef streams instead of traditional XRef tables (PDF 1.5+)
    pub use_xref_streams: bool,
    /// PDF version to write (default: 1.7)
    pub pdf_version: String,
    /// Enable compression for streams (default: true)
    pub compress_streams: bool,
}

impl Default for WriterConfig {
    fn default() -> Self {
        Self {
            use_xref_streams: false,
            pdf_version: "1.7".to_string(),
            compress_streams: true,
        }
    }
}

pub struct PdfWriter<W: Write> {
    writer: W,
    xref_positions: HashMap<ObjectId, u64>,
    current_position: u64,
    next_object_id: u32,
    // Maps for tracking object IDs during writing
    catalog_id: Option<ObjectId>,
    pages_id: Option<ObjectId>,
    info_id: Option<ObjectId>,
    // Maps for tracking form fields and their widgets
    #[allow(dead_code)]
    field_widget_map: HashMap<String, Vec<ObjectId>>, // field name -> widget IDs
    #[allow(dead_code)]
    field_id_map: HashMap<String, ObjectId>, // field name -> field ID
    form_field_ids: Vec<ObjectId>, // form field IDs to add to page annotations
    page_ids: Vec<ObjectId>,       // page IDs for form field references
    // Configuration
    config: WriterConfig,
    // Characters used in document (for font subsetting)
    document_used_chars: Option<std::collections::HashSet<char>>,
}

impl<W: Write> PdfWriter<W> {
    pub fn new_with_writer(writer: W) -> Self {
        Self::with_config(writer, WriterConfig::default())
    }

    pub fn with_config(writer: W, config: WriterConfig) -> Self {
        Self {
            writer,
            xref_positions: HashMap::new(),
            current_position: 0,
            next_object_id: 1, // Start at 1 for sequential numbering
            catalog_id: None,
            pages_id: None,
            info_id: None,
            field_widget_map: HashMap::new(),
            field_id_map: HashMap::new(),
            form_field_ids: Vec::new(),
            page_ids: Vec::new(),
            config,
            document_used_chars: None,
        }
    }

    pub fn write_document(&mut self, document: &mut Document) -> Result<()> {
        // Store used characters for font subsetting
        if !document.used_characters.is_empty() {
            self.document_used_chars = Some(document.used_characters.clone());
        }

        self.write_header()?;

        // Reserve object IDs for fixed objects (written in order)
        self.catalog_id = Some(self.allocate_object_id());
        self.pages_id = Some(self.allocate_object_id());
        self.info_id = Some(self.allocate_object_id());

        // Write custom fonts first (so pages can reference them)
        let font_refs = self.write_fonts(document)?;

        // Write pages (they contain widget annotations and font references)
        self.write_pages(document, &font_refs)?;

        // Write form fields (must be after pages so we can track widgets)
        self.write_form_fields(document)?;

        // Write catalog (must be after forms so AcroForm has correct field references)
        self.write_catalog(document)?;

        // Write document info
        self.write_info(document)?;

        // Write xref table or stream
        let xref_position = self.current_position;
        if self.config.use_xref_streams {
            self.write_xref_stream()?;
        } else {
            self.write_xref()?;
        }

        // Write trailer (only for traditional xref)
        if !self.config.use_xref_streams {
            self.write_trailer(xref_position)?;
        }

        if let Ok(()) = self.writer.flush() {
            // Flush succeeded
        }
        Ok(())
    }

    fn write_header(&mut self) -> Result<()> {
        let header = format!("%PDF-{}\n", self.config.pdf_version);
        self.write_bytes(header.as_bytes())?;
        // Binary comment to ensure file is treated as binary
        self.write_bytes(&[b'%', 0xE2, 0xE3, 0xCF, 0xD3, b'\n'])?;
        Ok(())
    }

    fn write_catalog(&mut self, document: &mut Document) -> Result<()> {
        let catalog_id = self.catalog_id.expect("catalog_id must be set");
        let pages_id = self.pages_id.expect("pages_id must be set");

        let mut catalog = Dictionary::new();
        catalog.set("Type", Object::Name("Catalog".to_string()));
        catalog.set("Pages", Object::Reference(pages_id));

        // Process FormManager if present to update AcroForm
        // We'll write the actual fields after pages are written
        if let Some(_form_manager) = &document.form_manager {
            // Ensure AcroForm exists
            if document.acro_form.is_none() {
                document.acro_form = Some(crate::forms::AcroForm::new());
            }
        }

        // Add AcroForm if present
        if let Some(acro_form) = &document.acro_form {
            // Reserve object ID for AcroForm
            let acro_form_id = self.allocate_object_id();

            // Write AcroForm object
            self.write_object(acro_form_id, Object::Dictionary(acro_form.to_dict()))?;

            // Reference it in catalog
            catalog.set("AcroForm", Object::Reference(acro_form_id));
        }

        // Add Outlines if present
        if let Some(outline_tree) = &document.outline {
            if !outline_tree.items.is_empty() {
                let outline_root_id = self.write_outline_tree(outline_tree)?;
                catalog.set("Outlines", Object::Reference(outline_root_id));
            }
        }

        self.write_object(catalog_id, Object::Dictionary(catalog))?;
        Ok(())
    }

    fn write_page_content(&mut self, content_id: ObjectId, page: &crate::page::Page) -> Result<()> {
        let mut page_copy = page.clone();
        let content = page_copy.generate_content()?;

        // Create stream with compression if enabled
        #[cfg(feature = "compression")]
        {
            use crate::objects::Stream;
            let mut stream = Stream::new(content);
            // Only compress if config allows it
            if self.config.compress_streams {
                stream.compress_flate()?;
            }

            self.write_object(
                content_id,
                Object::Stream(stream.dictionary().clone(), stream.data().to_vec()),
            )?;
        }

        #[cfg(not(feature = "compression"))]
        {
            let mut stream_dict = Dictionary::new();
            stream_dict.set("Length", Object::Integer(content.len() as i64));

            self.write_object(content_id, Object::Stream(stream_dict, content))?;
        }

        Ok(())
    }

    fn write_outline_tree(
        &mut self,
        outline_tree: &crate::structure::OutlineTree,
    ) -> Result<ObjectId> {
        // Create root outline dictionary
        let outline_root_id = self.allocate_object_id();

        let mut outline_root = Dictionary::new();
        outline_root.set("Type", Object::Name("Outlines".to_string()));

        if !outline_tree.items.is_empty() {
            // Reserve IDs for all outline items
            let mut item_ids = Vec::new();

            // Count all items and assign IDs
            fn count_items(items: &[crate::structure::OutlineItem]) -> usize {
                let mut count = items.len();
                for item in items {
                    count += count_items(&item.children);
                }
                count
            }

            let total_items = count_items(&outline_tree.items);

            // Reserve IDs for all items
            for _ in 0..total_items {
                item_ids.push(self.allocate_object_id());
            }

            let mut id_index = 0;

            // Write root items
            let first_id = item_ids[0];
            let last_id = item_ids[outline_tree.items.len() - 1];

            outline_root.set("First", Object::Reference(first_id));
            outline_root.set("Last", Object::Reference(last_id));

            // Visible count
            let visible_count = outline_tree.visible_count();
            outline_root.set("Count", Object::Integer(visible_count));

            // Write all items recursively
            let mut written_items = Vec::new();

            for (i, item) in outline_tree.items.iter().enumerate() {
                let item_id = item_ids[id_index];
                id_index += 1;

                let prev_id = if i > 0 { Some(item_ids[i - 1]) } else { None };
                let next_id = if i < outline_tree.items.len() - 1 {
                    Some(item_ids[i + 1])
                } else {
                    None
                };

                // Write this item and its children
                let children_ids = self.write_outline_item(
                    item,
                    item_id,
                    outline_root_id,
                    prev_id,
                    next_id,
                    &mut item_ids,
                    &mut id_index,
                )?;

                written_items.extend(children_ids);
            }
        }

        self.write_object(outline_root_id, Object::Dictionary(outline_root))?;
        Ok(outline_root_id)
    }

    #[allow(clippy::too_many_arguments)]
    fn write_outline_item(
        &mut self,
        item: &crate::structure::OutlineItem,
        item_id: ObjectId,
        parent_id: ObjectId,
        prev_id: Option<ObjectId>,
        next_id: Option<ObjectId>,
        all_ids: &mut Vec<ObjectId>,
        id_index: &mut usize,
    ) -> Result<Vec<ObjectId>> {
        let mut written_ids = vec![item_id];

        // Handle children if any
        let (first_child_id, last_child_id) = if !item.children.is_empty() {
            let first_idx = *id_index;
            let first_id = all_ids[first_idx];
            let last_idx = first_idx + item.children.len() - 1;
            let last_id = all_ids[last_idx];

            // Write children
            for (i, child) in item.children.iter().enumerate() {
                let child_id = all_ids[*id_index];
                *id_index += 1;

                let child_prev = if i > 0 {
                    Some(all_ids[first_idx + i - 1])
                } else {
                    None
                };
                let child_next = if i < item.children.len() - 1 {
                    Some(all_ids[first_idx + i + 1])
                } else {
                    None
                };

                let child_ids = self.write_outline_item(
                    child, child_id, item_id, // This item is the parent
                    child_prev, child_next, all_ids, id_index,
                )?;

                written_ids.extend(child_ids);
            }

            (Some(first_id), Some(last_id))
        } else {
            (None, None)
        };

        // Create item dictionary
        let item_dict = crate::structure::outline_item_to_dict(
            item,
            parent_id,
            first_child_id,
            last_child_id,
            prev_id,
            next_id,
        );

        self.write_object(item_id, Object::Dictionary(item_dict))?;

        Ok(written_ids)
    }

    fn write_form_fields(&mut self, document: &mut Document) -> Result<()> {
        // Add collected form field IDs to AcroForm
        if !self.form_field_ids.is_empty() {
            if let Some(acro_form) = &mut document.acro_form {
                // Clear any existing fields and add the ones we found
                acro_form.fields.clear();
                for field_id in &self.form_field_ids {
                    acro_form.add_field(*field_id);
                }

                // Ensure AcroForm has the right properties
                acro_form.need_appearances = true;
                if acro_form.da.is_none() {
                    acro_form.da = Some("/Helv 12 Tf 0 g".to_string());
                }
            }
        }
        Ok(())
    }

    fn write_info(&mut self, document: &Document) -> Result<()> {
        let info_id = self.info_id.expect("info_id must be set");
        let mut info_dict = Dictionary::new();

        if let Some(ref title) = document.metadata.title {
            info_dict.set("Title", Object::String(title.clone()));
        }
        if let Some(ref author) = document.metadata.author {
            info_dict.set("Author", Object::String(author.clone()));
        }
        if let Some(ref subject) = document.metadata.subject {
            info_dict.set("Subject", Object::String(subject.clone()));
        }
        if let Some(ref keywords) = document.metadata.keywords {
            info_dict.set("Keywords", Object::String(keywords.clone()));
        }
        if let Some(ref creator) = document.metadata.creator {
            info_dict.set("Creator", Object::String(creator.clone()));
        }
        if let Some(ref producer) = document.metadata.producer {
            info_dict.set("Producer", Object::String(producer.clone()));
        }

        // Add creation date
        if let Some(creation_date) = document.metadata.creation_date {
            let date_string = format_pdf_date(creation_date);
            info_dict.set("CreationDate", Object::String(date_string));
        }

        // Add modification date
        if let Some(mod_date) = document.metadata.modification_date {
            let date_string = format_pdf_date(mod_date);
            info_dict.set("ModDate", Object::String(date_string));
        }

        self.write_object(info_id, Object::Dictionary(info_dict))?;
        Ok(())
    }

    fn write_fonts(&mut self, document: &Document) -> Result<HashMap<String, ObjectId>> {
        let mut font_refs = HashMap::new();

        // Write custom fonts from the document
        for font_name in document.custom_font_names() {
            if let Some(font) = document.get_custom_font(&font_name) {
                // For now, write all custom fonts as TrueType with Identity-H for Unicode support
                // The font from document is Arc<fonts::Font>, not text::font_manager::CustomFont
                let font_id = self.write_font_with_unicode_support(&font_name, &font)?;
                font_refs.insert(font_name.clone(), font_id);
            }
        }

        Ok(font_refs)
    }

    /// Write font with automatic Unicode support detection
    fn write_font_with_unicode_support(
        &mut self,
        font_name: &str,
        font: &crate::fonts::Font,
    ) -> Result<ObjectId> {
        // Check if any text in the document needs Unicode
        // For simplicity, always use Type0 for full Unicode support
        self.write_type0_font_from_font(font_name, font)
    }

    /// Write a Type0 font with CID support from fonts::Font
    fn write_type0_font_from_font(
        &mut self,
        font_name: &str,
        font: &crate::fonts::Font,
    ) -> Result<ObjectId> {
        // Get used characters from document for subsetting
        let used_chars = self.document_used_chars.clone().unwrap_or_else(|| {
            // If no tracking, include common characters as fallback
            let mut chars = std::collections::HashSet::new();
            for ch in "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789 .,!?".chars()
            {
                chars.insert(ch);
            }
            chars
        });
        // Allocate IDs for all font objects
        let font_id = self.allocate_object_id();
        let descendant_font_id = self.allocate_object_id();
        let descriptor_id = self.allocate_object_id();
        let font_file_id = self.allocate_object_id();
        let to_unicode_id = self.allocate_object_id();

        // Write font file (embedded TTF data with subsetting for large fonts)
        // Keep track of the glyph mapping if we subset the font
        // IMPORTANT: We need the ORIGINAL font for width calculations, not the subset
        let (font_data_to_embed, subset_glyph_mapping, original_font_for_widths) =
            if font.data.len() > 100_000 && !used_chars.is_empty() {
                // Large font - try to subset it
                match crate::text::fonts::truetype_subsetter::subset_font(
                    font.data.clone(),
                    &used_chars,
                ) {
                    Ok(subset_result) => {
                        // Successfully subsetted - keep both font data and mapping
                        // Also keep reference to original font for width calculations
                        (
                            subset_result.font_data,
                            Some(subset_result.glyph_mapping),
                            font.clone(),
                        )
                    }
                    Err(_) => {
                        // Subsetting failed, use original if under 25MB
                        if font.data.len() < 25_000_000 {
                            (font.data.clone(), None, font.clone())
                        } else {
                            // Too large even for fallback
                            (Vec::new(), None, font.clone())
                        }
                    }
                }
            } else {
                // Small font or no character tracking - use as-is
                (font.data.clone(), None, font.clone())
            };

        if !font_data_to_embed.is_empty() {
            let mut font_file_dict = Dictionary::new();
            // Add appropriate properties based on font format
            match font.format {
                crate::fonts::FontFormat::OpenType => {
                    // CFF/OpenType fonts use FontFile3 with OpenType subtype
                    font_file_dict.set("Subtype", Object::Name("OpenType".to_string()));
                    font_file_dict.set("Length1", Object::Integer(font_data_to_embed.len() as i64));
                }
                crate::fonts::FontFormat::TrueType => {
                    // TrueType fonts use FontFile2 with Length1
                    font_file_dict.set("Length1", Object::Integer(font_data_to_embed.len() as i64));
                }
            }
            let font_stream_obj = Object::Stream(font_file_dict, font_data_to_embed);
            self.write_object(font_file_id, font_stream_obj)?;
        } else {
            // No font data to embed
            let font_file_dict = Dictionary::new();
            let font_stream_obj = Object::Stream(font_file_dict, Vec::new());
            self.write_object(font_file_id, font_stream_obj)?;
        }

        // Write font descriptor
        let mut descriptor = Dictionary::new();
        descriptor.set("Type", Object::Name("FontDescriptor".to_string()));
        descriptor.set("FontName", Object::Name(font_name.to_string()));
        descriptor.set("Flags", Object::Integer(4)); // Symbolic font
        descriptor.set(
            "FontBBox",
            Object::Array(vec![
                Object::Integer(font.descriptor.font_bbox[0] as i64),
                Object::Integer(font.descriptor.font_bbox[1] as i64),
                Object::Integer(font.descriptor.font_bbox[2] as i64),
                Object::Integer(font.descriptor.font_bbox[3] as i64),
            ]),
        );
        descriptor.set(
            "ItalicAngle",
            Object::Real(font.descriptor.italic_angle as f64),
        );
        descriptor.set("Ascent", Object::Real(font.descriptor.ascent as f64));
        descriptor.set("Descent", Object::Real(font.descriptor.descent as f64));
        descriptor.set("CapHeight", Object::Real(font.descriptor.cap_height as f64));
        descriptor.set("StemV", Object::Real(font.descriptor.stem_v as f64));
        // Use appropriate FontFile type based on font format
        let font_file_key = match font.format {
            crate::fonts::FontFormat::OpenType => "FontFile3", // CFF/OpenType fonts
            crate::fonts::FontFormat::TrueType => "FontFile2", // TrueType fonts
        };
        descriptor.set(font_file_key, Object::Reference(font_file_id));
        self.write_object(descriptor_id, Object::Dictionary(descriptor))?;

        // Write CIDFont (descendant font)
        let mut cid_font = Dictionary::new();
        cid_font.set("Type", Object::Name("Font".to_string()));
        // Use appropriate CIDFont subtype based on font format
        let cid_font_subtype =
            if CjkFontType::should_use_cidfonttype2_for_preview_compatibility(font_name) {
                "CIDFontType2" // Force CIDFontType2 for CJK fonts to fix Preview.app rendering
            } else {
                match font.format {
                    crate::fonts::FontFormat::OpenType => "CIDFontType0", // CFF/OpenType fonts
                    crate::fonts::FontFormat::TrueType => "CIDFontType2", // TrueType fonts
                }
            };
        cid_font.set("Subtype", Object::Name(cid_font_subtype.to_string()));
        cid_font.set("BaseFont", Object::Name(font_name.to_string()));

        // CIDSystemInfo - Use appropriate values for CJK fonts
        let mut cid_system_info = Dictionary::new();
        let (registry, ordering, supplement) =
            if let Some(cjk_type) = CjkFontType::detect_from_name(font_name) {
                cjk_type.cid_system_info()
            } else {
                ("Adobe", "Identity", 0)
            };

        cid_system_info.set("Registry", Object::String(registry.to_string()));
        cid_system_info.set("Ordering", Object::String(ordering.to_string()));
        cid_system_info.set("Supplement", Object::Integer(supplement as i64));
        cid_font.set("CIDSystemInfo", Object::Dictionary(cid_system_info));

        cid_font.set("FontDescriptor", Object::Reference(descriptor_id));

        // Calculate a better default width based on font metrics
        let default_width = self.calculate_default_width(font);
        cid_font.set("DW", Object::Integer(default_width));

        // Generate proper width array from font metrics
        // IMPORTANT: Use the ORIGINAL font for width calculations, not the subset
        // But pass the subset mapping to know which characters we're using
        let w_array = self.generate_width_array(
            &original_font_for_widths,
            default_width,
            subset_glyph_mapping.as_ref(),
        );
        cid_font.set("W", Object::Array(w_array));

        // CIDToGIDMap - Generate proper mapping from CID (Unicode) to GlyphID
        // This is critical for Type0 fonts to work correctly
        // If we subsetted the font, use the new glyph mapping
        let cid_to_gid_map = self.generate_cid_to_gid_map(font, subset_glyph_mapping.as_ref())?;
        if !cid_to_gid_map.is_empty() {
            // Write the CIDToGIDMap as a stream
            let cid_to_gid_map_id = self.allocate_object_id();
            let mut map_dict = Dictionary::new();
            map_dict.set("Length", Object::Integer(cid_to_gid_map.len() as i64));
            let map_stream = Object::Stream(map_dict, cid_to_gid_map);
            self.write_object(cid_to_gid_map_id, map_stream)?;
            cid_font.set("CIDToGIDMap", Object::Reference(cid_to_gid_map_id));
        } else {
            cid_font.set("CIDToGIDMap", Object::Name("Identity".to_string()));
        }

        self.write_object(descendant_font_id, Object::Dictionary(cid_font))?;

        // Write ToUnicode CMap
        let cmap_data = self.generate_tounicode_cmap_from_font(font);
        let cmap_dict = Dictionary::new();
        let cmap_stream = Object::Stream(cmap_dict, cmap_data);
        self.write_object(to_unicode_id, cmap_stream)?;

        // Write Type0 font (main font)
        let mut type0_font = Dictionary::new();
        type0_font.set("Type", Object::Name("Font".to_string()));
        type0_font.set("Subtype", Object::Name("Type0".to_string()));
        type0_font.set("BaseFont", Object::Name(font_name.to_string()));
        type0_font.set("Encoding", Object::Name("Identity-H".to_string()));
        type0_font.set(
            "DescendantFonts",
            Object::Array(vec![Object::Reference(descendant_font_id)]),
        );
        type0_font.set("ToUnicode", Object::Reference(to_unicode_id));

        self.write_object(font_id, Object::Dictionary(type0_font))?;

        Ok(font_id)
    }

    /// Calculate default width based on common characters
    fn calculate_default_width(&self, font: &crate::fonts::Font) -> i64 {
        use crate::text::fonts::truetype::TrueTypeFont;

        // Try to calculate from actual font metrics
        if let Ok(tt_font) = TrueTypeFont::parse(font.data.clone()) {
            if let Ok(cmap_tables) = tt_font.parse_cmap() {
                if let Some(cmap) = cmap_tables
                    .iter()
                    .find(|t| t.platform_id == 3 && t.encoding_id == 1)
                    .or_else(|| cmap_tables.iter().find(|t| t.platform_id == 0))
                {
                    if let Ok(widths) = tt_font.get_glyph_widths(&cmap.mappings) {
                        // NOTE: get_glyph_widths already returns widths in PDF units (1000 per em)

                        // Calculate average width of common Latin characters
                        let common_chars =
                            "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789 ";
                        let mut total_width = 0;
                        let mut count = 0;

                        for ch in common_chars.chars() {
                            let unicode = ch as u32;
                            if let Some(&pdf_width) = widths.get(&unicode) {
                                total_width += pdf_width as i64;
                                count += 1;
                            }
                        }

                        if count > 0 {
                            return total_width / count;
                        }
                    }
                }
            }
        }

        // Fallback default if we can't calculate
        500
    }

    /// Generate width array for CID font
    fn generate_width_array(
        &self,
        font: &crate::fonts::Font,
        _default_width: i64,
        subset_mapping: Option<&HashMap<u32, u16>>,
    ) -> Vec<Object> {
        use crate::text::fonts::truetype::TrueTypeFont;

        let mut w_array = Vec::new();

        // Try to get actual glyph widths from the font
        if let Ok(tt_font) = TrueTypeFont::parse(font.data.clone()) {
            // IMPORTANT: Always use ORIGINAL mappings for width calculation
            // The subset_mapping has NEW GlyphIDs which don't correspond to the right glyphs
            // in the original font's width table
            let char_to_glyph = {
                // Parse cmap to get original mappings
                if let Ok(cmap_tables) = tt_font.parse_cmap() {
                    if let Some(cmap) = cmap_tables
                        .iter()
                        .find(|t| t.platform_id == 3 && t.encoding_id == 1)
                        .or_else(|| cmap_tables.iter().find(|t| t.platform_id == 0))
                    {
                        // If we have subset_mapping, filter to only include used characters
                        if let Some(subset_map) = subset_mapping {
                            let mut filtered = HashMap::new();
                            for unicode in subset_map.keys() {
                                // Get the ORIGINAL GlyphID for this Unicode
                                if let Some(&orig_glyph) = cmap.mappings.get(unicode) {
                                    filtered.insert(*unicode, orig_glyph);
                                }
                            }
                            filtered
                        } else {
                            cmap.mappings.clone()
                        }
                    } else {
                        HashMap::new()
                    }
                } else {
                    HashMap::new()
                }
            };

            if !char_to_glyph.is_empty() {
                // Get actual widths from the font
                if let Ok(widths) = tt_font.get_glyph_widths(&char_to_glyph) {
                    // NOTE: get_glyph_widths already returns widths scaled to PDF units (1000 per em)
                    // So we DON'T need to scale them again here

                    // Group consecutive characters with same width for efficiency
                    let mut sorted_chars: Vec<_> = widths.iter().collect();
                    sorted_chars.sort_by_key(|(unicode, _)| *unicode);

                    let mut i = 0;
                    while i < sorted_chars.len() {
                        let start_unicode = *sorted_chars[i].0;
                        // Width is already in PDF units from get_glyph_widths
                        let pdf_width = *sorted_chars[i].1 as i64;

                        // Find consecutive characters with same width
                        let mut end_unicode = start_unicode;
                        let mut j = i + 1;
                        while j < sorted_chars.len() && *sorted_chars[j].0 == end_unicode + 1 {
                            let next_pdf_width = *sorted_chars[j].1 as i64;
                            if next_pdf_width == pdf_width {
                                end_unicode = *sorted_chars[j].0;
                                j += 1;
                            } else {
                                break;
                            }
                        }

                        // Add to W array
                        if start_unicode == end_unicode {
                            // Single character
                            w_array.push(Object::Integer(start_unicode as i64));
                            w_array.push(Object::Array(vec![Object::Integer(pdf_width)]));
                        } else {
                            // Range of characters
                            w_array.push(Object::Integer(start_unicode as i64));
                            w_array.push(Object::Integer(end_unicode as i64));
                            w_array.push(Object::Integer(pdf_width));
                        }

                        i = j;
                    }

                    return w_array;
                }
            }
        }

        // Fallback to reasonable default widths if we can't parse the font
        let ranges = vec![
            // Space character should be narrower
            (0x20, 0x20, 250), // Space
            (0x21, 0x2F, 333), // Punctuation
            (0x30, 0x39, 500), // Numbers (0-9)
            (0x3A, 0x40, 333), // More punctuation
            (0x41, 0x5A, 667), // Uppercase letters (A-Z)
            (0x5B, 0x60, 333), // Brackets
            (0x61, 0x7A, 500), // Lowercase letters (a-z)
            (0x7B, 0x7E, 333), // More brackets
            // Extended Latin
            (0xA0, 0xA0, 250), // Non-breaking space
            (0xA1, 0xBF, 333), // Latin-1 punctuation
            (0xC0, 0xD6, 667), // Latin-1 uppercase
            (0xD7, 0xD7, 564), // Multiplication sign
            (0xD8, 0xDE, 667), // More Latin-1 uppercase
            (0xDF, 0xF6, 500), // Latin-1 lowercase
            (0xF7, 0xF7, 564), // Division sign
            (0xF8, 0xFF, 500), // More Latin-1 lowercase
            // Latin Extended-A
            (0x100, 0x17F, 500), // Latin Extended-A
            // Symbols and special characters
            (0x2000, 0x200F, 250), // Various spaces
            (0x2010, 0x2027, 333), // Hyphens and dashes
            (0x2028, 0x202F, 250), // More spaces
            (0x2030, 0x206F, 500), // General Punctuation
            (0x2070, 0x209F, 400), // Superscripts
            (0x20A0, 0x20CF, 600), // Currency symbols
            (0x2100, 0x214F, 700), // Letterlike symbols
            (0x2190, 0x21FF, 600), // Arrows
            (0x2200, 0x22FF, 600), // Mathematical operators
            (0x2300, 0x23FF, 600), // Miscellaneous technical
            (0x2500, 0x257F, 500), // Box drawing
            (0x2580, 0x259F, 500), // Block elements
            (0x25A0, 0x25FF, 600), // Geometric shapes
            (0x2600, 0x26FF, 600), // Miscellaneous symbols
            (0x2700, 0x27BF, 600), // Dingbats
        ];

        // Convert ranges to W array format
        for (start, end, width) in ranges {
            if start == end {
                // Single character
                w_array.push(Object::Integer(start));
                w_array.push(Object::Array(vec![Object::Integer(width)]));
            } else {
                // Range of characters
                w_array.push(Object::Integer(start));
                w_array.push(Object::Integer(end));
                w_array.push(Object::Integer(width));
            }
        }

        w_array
    }

    /// Generate CIDToGIDMap for Type0 font
    fn generate_cid_to_gid_map(
        &mut self,
        font: &crate::fonts::Font,
        subset_mapping: Option<&HashMap<u32, u16>>,
    ) -> Result<Vec<u8>> {
        use crate::text::fonts::truetype::TrueTypeFont;

        // If we have a subset mapping, use it directly
        // Otherwise, parse the font to get the original cmap table
        let cmap_mappings = if let Some(subset_map) = subset_mapping {
            // Use the subset mapping directly
            subset_map.clone()
        } else {
            // Parse the font to get the original cmap table
            let tt_font = TrueTypeFont::parse(font.data.clone())?;
            let cmap_tables = tt_font.parse_cmap()?;

            // Find the best cmap table (Unicode)
            let cmap = cmap_tables
                .iter()
                .find(|t| t.platform_id == 3 && t.encoding_id == 1) // Windows Unicode
                .or_else(|| cmap_tables.iter().find(|t| t.platform_id == 0)) // Unicode
                .ok_or_else(|| {
                    crate::error::PdfError::FontError("No Unicode cmap table found".to_string())
                })?;

            cmap.mappings.clone()
        };

        // Build the CIDToGIDMap
        // Since we use Unicode code points as CIDs, we need to map Unicode → GlyphID
        // The map is a binary array where index = CID (Unicode) * 2, value = GlyphID (big-endian)

        // OPTIMIZATION: Only create map for characters actually used in the document
        // Get used characters from document tracking
        let used_chars = self.document_used_chars.clone().unwrap_or_default();

        // Find the maximum Unicode value from used characters or full font
        let max_unicode = if !used_chars.is_empty() {
            // If we have used chars tracking, only map up to the highest used character
            used_chars
                .iter()
                .map(|ch| *ch as u32)
                .max()
                .unwrap_or(0x00FF) // At least Basic Latin
                .min(0xFFFF) as usize
        } else {
            // Fallback to original behavior if no tracking
            cmap_mappings
                .keys()
                .max()
                .copied()
                .unwrap_or(0xFFFF)
                .min(0xFFFF) as usize
        };

        // Create the map: 2 bytes per entry
        let mut map = vec![0u8; (max_unicode + 1) * 2];

        // Fill in the mappings
        let mut sample_mappings = Vec::new();
        for (&unicode, &glyph_id) in &cmap_mappings {
            if unicode <= max_unicode as u32 {
                let idx = (unicode as usize) * 2;
                // Write glyph_id in big-endian format
                map[idx] = (glyph_id >> 8) as u8;
                map[idx + 1] = (glyph_id & 0xFF) as u8;

                // Collect some sample mappings for debugging
                if unicode == 0x0041 || unicode == 0x0061 || unicode == 0x00E1 || unicode == 0x00F1
                {
                    sample_mappings.push((unicode, glyph_id));
                }
            }
        }

        Ok(map)
    }

    /// Generate ToUnicode CMap for Type0 font from fonts::Font
    fn generate_tounicode_cmap_from_font(&self, font: &crate::fonts::Font) -> Vec<u8> {
        use crate::text::fonts::truetype::TrueTypeFont;

        let mut cmap = String::new();

        // CMap header
        cmap.push_str("/CIDInit /ProcSet findresource begin\n");
        cmap.push_str("12 dict begin\n");
        cmap.push_str("begincmap\n");
        cmap.push_str("/CIDSystemInfo\n");
        cmap.push_str("<< /Registry (Adobe)\n");
        cmap.push_str("   /Ordering (UCS)\n");
        cmap.push_str("   /Supplement 0\n");
        cmap.push_str(">> def\n");
        cmap.push_str("/CMapName /Adobe-Identity-UCS def\n");
        cmap.push_str("/CMapType 2 def\n");
        cmap.push_str("1 begincodespacerange\n");
        cmap.push_str("<0000> <FFFF>\n");
        cmap.push_str("endcodespacerange\n");

        // Try to get actual mappings from the font
        let mut mappings = Vec::new();
        let mut has_font_mappings = false;

        if let Ok(tt_font) = TrueTypeFont::parse(font.data.clone()) {
            if let Ok(cmap_tables) = tt_font.parse_cmap() {
                // Find the best cmap table (Unicode)
                if let Some(cmap_table) = cmap_tables
                    .iter()
                    .find(|t| t.platform_id == 3 && t.encoding_id == 1) // Windows Unicode
                    .or_else(|| cmap_tables.iter().find(|t| t.platform_id == 0))
                // Unicode
                {
                    // For Identity-H encoding, we use Unicode code points as CIDs
                    // So the ToUnicode CMap should map CID (=Unicode) → Unicode
                    for (&unicode, &glyph_id) in &cmap_table.mappings {
                        if glyph_id > 0 && unicode <= 0xFFFF {
                            // Only non-.notdef glyphs
                            // Map CID (which is Unicode value) to Unicode
                            mappings.push((unicode, unicode));
                        }
                    }
                    has_font_mappings = true;
                }
            }
        }

        // If we couldn't get font mappings, use identity mapping for common ranges
        if !has_font_mappings {
            // Basic Latin and Latin-1 Supplement (0x0020-0x00FF)
            for i in 0x0020..=0x00FF {
                mappings.push((i, i));
            }

            // Latin Extended-A (0x0100-0x017F)
            for i in 0x0100..=0x017F {
                mappings.push((i, i));
            }

            // CJK Unicode ranges - CRITICAL for CJK font support
            // Hiragana (Japanese)
            for i in 0x3040..=0x309F {
                mappings.push((i, i));
            }

            // Katakana (Japanese)
            for i in 0x30A0..=0x30FF {
                mappings.push((i, i));
            }

            // CJK Unified Ideographs (Chinese, Japanese, Korean)
            for i in 0x4E00..=0x9FFF {
                mappings.push((i, i));
            }

            // Hangul Syllables (Korean)
            for i in 0xAC00..=0xD7AF {
                mappings.push((i, i));
            }

            // Common symbols and punctuation
            for i in 0x2000..=0x206F {
                mappings.push((i, i));
            }

            // Mathematical symbols
            for i in 0x2200..=0x22FF {
                mappings.push((i, i));
            }

            // Arrows
            for i in 0x2190..=0x21FF {
                mappings.push((i, i));
            }

            // Box drawing
            for i in 0x2500..=0x259F {
                mappings.push((i, i));
            }

            // Geometric shapes
            for i in 0x25A0..=0x25FF {
                mappings.push((i, i));
            }

            // Miscellaneous symbols
            for i in 0x2600..=0x26FF {
                mappings.push((i, i));
            }
        }

        // Sort mappings by CID for better organization
        mappings.sort_by_key(|&(cid, _)| cid);

        // Use more efficient bfrange where possible
        let mut i = 0;
        while i < mappings.len() {
            // Check if we can use a range
            let start_cid = mappings[i].0;
            let start_unicode = mappings[i].1;
            let mut end_idx = i;

            // Find consecutive mappings
            while end_idx + 1 < mappings.len()
                && mappings[end_idx + 1].0 == mappings[end_idx].0 + 1
                && mappings[end_idx + 1].1 == mappings[end_idx].1 + 1
                && end_idx - i < 99
            // Max 100 per block
            {
                end_idx += 1;
            }

            if end_idx > i {
                // Use bfrange for consecutive mappings
                cmap.push_str("1 beginbfrange\n");
                cmap.push_str(&format!(
                    "<{:04X}> <{:04X}> <{:04X}>\n",
                    start_cid, mappings[end_idx].0, start_unicode
                ));
                cmap.push_str("endbfrange\n");
                i = end_idx + 1;
            } else {
                // Use bfchar for individual mappings
                let mut chars = Vec::new();
                let chunk_end = (i + 100).min(mappings.len());

                for item in &mappings[i..chunk_end] {
                    chars.push(*item);
                }

                if !chars.is_empty() {
                    cmap.push_str(&format!("{} beginbfchar\n", chars.len()));
                    for (cid, unicode) in chars {
                        cmap.push_str(&format!("<{:04X}> <{:04X}>\n", cid, unicode));
                    }
                    cmap.push_str("endbfchar\n");
                }

                i = chunk_end;
            }
        }

        // CMap footer
        cmap.push_str("endcmap\n");
        cmap.push_str("CMapName currentdict /CMap defineresource pop\n");
        cmap.push_str("end\n");
        cmap.push_str("end\n");

        cmap.into_bytes()
    }

    /// Write a regular TrueType font
    #[allow(dead_code)]
    fn write_truetype_font(
        &mut self,
        font_name: &str,
        font: &crate::text::font_manager::CustomFont,
    ) -> Result<ObjectId> {
        // Allocate IDs for font objects
        let font_id = self.allocate_object_id();
        let descriptor_id = self.allocate_object_id();
        let font_file_id = self.allocate_object_id();

        // Write font file (embedded TTF data)
        if let Some(ref data) = font.font_data {
            let mut font_file_dict = Dictionary::new();
            font_file_dict.set("Length1", Object::Integer(data.len() as i64));
            let font_stream_obj = Object::Stream(font_file_dict, data.clone());
            self.write_object(font_file_id, font_stream_obj)?;
        }

        // Write font descriptor
        let mut descriptor = Dictionary::new();
        descriptor.set("Type", Object::Name("FontDescriptor".to_string()));
        descriptor.set("FontName", Object::Name(font_name.to_string()));
        descriptor.set("Flags", Object::Integer(32)); // Non-symbolic font
        descriptor.set(
            "FontBBox",
            Object::Array(vec![
                Object::Integer(-1000),
                Object::Integer(-1000),
                Object::Integer(2000),
                Object::Integer(2000),
            ]),
        );
        descriptor.set("ItalicAngle", Object::Integer(0));
        descriptor.set("Ascent", Object::Integer(font.descriptor.ascent as i64));
        descriptor.set("Descent", Object::Integer(font.descriptor.descent as i64));
        descriptor.set(
            "CapHeight",
            Object::Integer(font.descriptor.cap_height as i64),
        );
        descriptor.set("StemV", Object::Integer(font.descriptor.stem_v as i64));
        descriptor.set("FontFile2", Object::Reference(font_file_id));
        self.write_object(descriptor_id, Object::Dictionary(descriptor))?;

        // Write font dictionary
        let mut font_dict = Dictionary::new();
        font_dict.set("Type", Object::Name("Font".to_string()));
        font_dict.set("Subtype", Object::Name("TrueType".to_string()));
        font_dict.set("BaseFont", Object::Name(font_name.to_string()));
        font_dict.set("FirstChar", Object::Integer(0));
        font_dict.set("LastChar", Object::Integer(255));

        // Create widths array (simplified - all 600)
        let widths: Vec<Object> = (0..256).map(|_| Object::Integer(600)).collect();
        font_dict.set("Widths", Object::Array(widths));
        font_dict.set("FontDescriptor", Object::Reference(descriptor_id));

        // Use WinAnsiEncoding for regular TrueType
        font_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));

        self.write_object(font_id, Object::Dictionary(font_dict))?;

        Ok(font_id)
    }

    fn write_pages(
        &mut self,
        document: &Document,
        font_refs: &HashMap<String, ObjectId>,
    ) -> Result<()> {
        let pages_id = self.pages_id.expect("pages_id must be set");
        let mut pages_dict = Dictionary::new();
        pages_dict.set("Type", Object::Name("Pages".to_string()));
        pages_dict.set("Count", Object::Integer(document.pages.len() as i64));

        let mut kids = Vec::new();

        // Allocate page object IDs sequentially
        let mut page_ids = Vec::new();
        let mut content_ids = Vec::new();
        for _ in 0..document.pages.len() {
            page_ids.push(self.allocate_object_id());
            content_ids.push(self.allocate_object_id());
        }

        for page_id in &page_ids {
            kids.push(Object::Reference(*page_id));
        }

        pages_dict.set("Kids", Object::Array(kids));

        self.write_object(pages_id, Object::Dictionary(pages_dict))?;

        // Store page IDs for form field references
        self.page_ids = page_ids.clone();

        // Write individual pages with font references
        for (i, page) in document.pages.iter().enumerate() {
            let page_id = page_ids[i];
            let content_id = content_ids[i];

            self.write_page_with_fonts(page_id, pages_id, content_id, page, document, font_refs)?;
            self.write_page_content(content_id, page)?;
        }

        Ok(())
    }

    /// Compatibility alias for `write_pages` to maintain backwards compatibility
    #[allow(dead_code)]
    fn write_pages_with_fonts(
        &mut self,
        document: &Document,
        font_refs: &HashMap<String, ObjectId>,
    ) -> Result<()> {
        self.write_pages(document, font_refs)
    }

    fn write_page_with_fonts(
        &mut self,
        page_id: ObjectId,
        parent_id: ObjectId,
        content_id: ObjectId,
        page: &crate::page::Page,
        _document: &Document,
        font_refs: &HashMap<String, ObjectId>,
    ) -> Result<()> {
        // Start with the page's dictionary which includes annotations
        let mut page_dict = page.to_dict();

        page_dict.set("Type", Object::Name("Page".to_string()));
        page_dict.set("Parent", Object::Reference(parent_id));
        page_dict.set("Contents", Object::Reference(content_id));

        // Get resources dictionary or create new one
        let mut resources = if let Some(Object::Dictionary(res)) = page_dict.get("Resources") {
            res.clone()
        } else {
            Dictionary::new()
        };

        // Add font resources
        let mut font_dict = Dictionary::new();

        // Add ALL standard PDF fonts (Type1) with WinAnsiEncoding
        // This fixes the text rendering issue in dashboards where HelveticaBold was missing

        // Helvetica family
        let mut helvetica_dict = Dictionary::new();
        helvetica_dict.set("Type", Object::Name("Font".to_string()));
        helvetica_dict.set("Subtype", Object::Name("Type1".to_string()));
        helvetica_dict.set("BaseFont", Object::Name("Helvetica".to_string()));
        helvetica_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Helvetica", Object::Dictionary(helvetica_dict));

        let mut helvetica_bold_dict = Dictionary::new();
        helvetica_bold_dict.set("Type", Object::Name("Font".to_string()));
        helvetica_bold_dict.set("Subtype", Object::Name("Type1".to_string()));
        helvetica_bold_dict.set("BaseFont", Object::Name("Helvetica-Bold".to_string()));
        helvetica_bold_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Helvetica-Bold", Object::Dictionary(helvetica_bold_dict));

        let mut helvetica_oblique_dict = Dictionary::new();
        helvetica_oblique_dict.set("Type", Object::Name("Font".to_string()));
        helvetica_oblique_dict.set("Subtype", Object::Name("Type1".to_string()));
        helvetica_oblique_dict.set("BaseFont", Object::Name("Helvetica-Oblique".to_string()));
        helvetica_oblique_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set(
            "Helvetica-Oblique",
            Object::Dictionary(helvetica_oblique_dict),
        );

        let mut helvetica_bold_oblique_dict = Dictionary::new();
        helvetica_bold_oblique_dict.set("Type", Object::Name("Font".to_string()));
        helvetica_bold_oblique_dict.set("Subtype", Object::Name("Type1".to_string()));
        helvetica_bold_oblique_dict.set(
            "BaseFont",
            Object::Name("Helvetica-BoldOblique".to_string()),
        );
        helvetica_bold_oblique_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set(
            "Helvetica-BoldOblique",
            Object::Dictionary(helvetica_bold_oblique_dict),
        );

        // Times family
        let mut times_dict = Dictionary::new();
        times_dict.set("Type", Object::Name("Font".to_string()));
        times_dict.set("Subtype", Object::Name("Type1".to_string()));
        times_dict.set("BaseFont", Object::Name("Times-Roman".to_string()));
        times_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Times-Roman", Object::Dictionary(times_dict));

        let mut times_bold_dict = Dictionary::new();
        times_bold_dict.set("Type", Object::Name("Font".to_string()));
        times_bold_dict.set("Subtype", Object::Name("Type1".to_string()));
        times_bold_dict.set("BaseFont", Object::Name("Times-Bold".to_string()));
        times_bold_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Times-Bold", Object::Dictionary(times_bold_dict));

        let mut times_italic_dict = Dictionary::new();
        times_italic_dict.set("Type", Object::Name("Font".to_string()));
        times_italic_dict.set("Subtype", Object::Name("Type1".to_string()));
        times_italic_dict.set("BaseFont", Object::Name("Times-Italic".to_string()));
        times_italic_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Times-Italic", Object::Dictionary(times_italic_dict));

        let mut times_bold_italic_dict = Dictionary::new();
        times_bold_italic_dict.set("Type", Object::Name("Font".to_string()));
        times_bold_italic_dict.set("Subtype", Object::Name("Type1".to_string()));
        times_bold_italic_dict.set("BaseFont", Object::Name("Times-BoldItalic".to_string()));
        times_bold_italic_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set(
            "Times-BoldItalic",
            Object::Dictionary(times_bold_italic_dict),
        );

        // Courier family
        let mut courier_dict = Dictionary::new();
        courier_dict.set("Type", Object::Name("Font".to_string()));
        courier_dict.set("Subtype", Object::Name("Type1".to_string()));
        courier_dict.set("BaseFont", Object::Name("Courier".to_string()));
        courier_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Courier", Object::Dictionary(courier_dict));

        let mut courier_bold_dict = Dictionary::new();
        courier_bold_dict.set("Type", Object::Name("Font".to_string()));
        courier_bold_dict.set("Subtype", Object::Name("Type1".to_string()));
        courier_bold_dict.set("BaseFont", Object::Name("Courier-Bold".to_string()));
        courier_bold_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Courier-Bold", Object::Dictionary(courier_bold_dict));

        let mut courier_oblique_dict = Dictionary::new();
        courier_oblique_dict.set("Type", Object::Name("Font".to_string()));
        courier_oblique_dict.set("Subtype", Object::Name("Type1".to_string()));
        courier_oblique_dict.set("BaseFont", Object::Name("Courier-Oblique".to_string()));
        courier_oblique_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set("Courier-Oblique", Object::Dictionary(courier_oblique_dict));

        let mut courier_bold_oblique_dict = Dictionary::new();
        courier_bold_oblique_dict.set("Type", Object::Name("Font".to_string()));
        courier_bold_oblique_dict.set("Subtype", Object::Name("Type1".to_string()));
        courier_bold_oblique_dict.set("BaseFont", Object::Name("Courier-BoldOblique".to_string()));
        courier_bold_oblique_dict.set("Encoding", Object::Name("WinAnsiEncoding".to_string()));
        font_dict.set(
            "Courier-BoldOblique",
            Object::Dictionary(courier_bold_oblique_dict),
        );

        // Add custom fonts (Type0 fonts for Unicode support)
        for (font_name, font_id) in font_refs {
            font_dict.set(font_name, Object::Reference(*font_id));
        }

        resources.set("Font", Object::Dictionary(font_dict));

        // Add images as XObjects
        if !page.images().is_empty() {
            let mut xobject_dict = Dictionary::new();

            for (name, image) in page.images() {
                // Use sequential ObjectId allocation to avoid conflicts
                let image_id = self.allocate_object_id();

                // Write the image XObject
                self.write_object(image_id, image.to_pdf_object())?;

                // Add reference to XObject dictionary
                xobject_dict.set(name, Object::Reference(image_id));
            }

            resources.set("XObject", Object::Dictionary(xobject_dict));
        }

        // Add ExtGState resources for transparency
        if let Some(extgstate_states) = page.get_extgstate_resources() {
            let mut extgstate_dict = Dictionary::new();
            for (name, state) in extgstate_states {
                let mut state_dict = Dictionary::new();
                state_dict.set("Type", Object::Name("ExtGState".to_string()));

                // Add transparency parameters
                if let Some(alpha_stroke) = state.alpha_stroke {
                    state_dict.set("CA", Object::Real(alpha_stroke));
                }
                if let Some(alpha_fill) = state.alpha_fill {
                    state_dict.set("ca", Object::Real(alpha_fill));
                }

                // Add other parameters as needed
                if let Some(line_width) = state.line_width {
                    state_dict.set("LW", Object::Real(line_width));
                }
                if let Some(line_cap) = state.line_cap {
                    state_dict.set("LC", Object::Integer(line_cap as i64));
                }
                if let Some(line_join) = state.line_join {
                    state_dict.set("LJ", Object::Integer(line_join as i64));
                }
                if let Some(dash_pattern) = &state.dash_pattern {
                    let dash_objects: Vec<Object> = dash_pattern
                        .array
                        .iter()
                        .map(|&d| Object::Real(d))
                        .collect();
                    state_dict.set(
                        "D",
                        Object::Array(vec![
                            Object::Array(dash_objects),
                            Object::Real(dash_pattern.phase),
                        ]),
                    );
                }

                extgstate_dict.set(name, Object::Dictionary(state_dict));
            }
            if !extgstate_dict.is_empty() {
                resources.set("ExtGState", Object::Dictionary(extgstate_dict));
            }
        }

        page_dict.set("Resources", Object::Dictionary(resources));

        // Handle form widget annotations
        if let Some(Object::Array(annots)) = page_dict.get("Annots") {
            let mut new_annots = Vec::new();

            for annot in annots {
                if let Object::Dictionary(ref annot_dict) = annot {
                    if let Some(Object::Name(subtype)) = annot_dict.get("Subtype") {
                        if subtype == "Widget" {
                            // Process widget annotation
                            let widget_id = self.allocate_object_id();
                            self.write_object(widget_id, annot.clone())?;
                            new_annots.push(Object::Reference(widget_id));

                            // Track widget for form fields
                            if let Some(Object::Name(_ft)) = annot_dict.get("FT") {
                                if let Some(Object::String(field_name)) = annot_dict.get("T") {
                                    self.field_widget_map
                                        .entry(field_name.clone())
                                        .or_default()
                                        .push(widget_id);
                                    self.field_id_map.insert(field_name.clone(), widget_id);
                                    self.form_field_ids.push(widget_id);
                                }
                            }
                            continue;
                        }
                    }
                }
                new_annots.push(annot.clone());
            }

            if !new_annots.is_empty() {
                page_dict.set("Annots", Object::Array(new_annots));
            }
        }

        self.write_object(page_id, Object::Dictionary(page_dict))?;
        Ok(())
    }
}

impl PdfWriter<BufWriter<std::fs::File>> {
    pub fn new(path: impl AsRef<Path>) -> Result<Self> {
        let file = std::fs::File::create(path)?;
        let writer = BufWriter::new(file);

        Ok(Self {
            writer,
            xref_positions: HashMap::new(),
            current_position: 0,
            next_object_id: 1,
            catalog_id: None,
            pages_id: None,
            info_id: None,
            field_widget_map: HashMap::new(),
            field_id_map: HashMap::new(),
            form_field_ids: Vec::new(),
            page_ids: Vec::new(),
            config: WriterConfig::default(),
            document_used_chars: None,
        })
    }
}

impl<W: Write> PdfWriter<W> {
    fn allocate_object_id(&mut self) -> ObjectId {
        let id = ObjectId::new(self.next_object_id, 0);
        self.next_object_id += 1;
        id
    }

    fn write_object(&mut self, id: ObjectId, object: Object) -> Result<()> {
        self.xref_positions.insert(id, self.current_position);

        let header = format!("{} {} obj\n", id.number(), id.generation());
        self.write_bytes(header.as_bytes())?;

        self.write_object_value(&object)?;

        self.write_bytes(b"\nendobj\n")?;
        Ok(())
    }

    fn write_object_value(&mut self, object: &Object) -> Result<()> {
        match object {
            Object::Null => self.write_bytes(b"null")?,
            Object::Boolean(b) => self.write_bytes(if *b { b"true" } else { b"false" })?,
            Object::Integer(i) => self.write_bytes(i.to_string().as_bytes())?,
            Object::Real(f) => self.write_bytes(
                format!("{f:.6}")
                    .trim_end_matches('0')
                    .trim_end_matches('.')
                    .as_bytes(),
            )?,
            Object::String(s) => {
                self.write_bytes(b"(")?;
                self.write_bytes(s.as_bytes())?;
                self.write_bytes(b")")?;
            }
            Object::Name(n) => {
                self.write_bytes(b"/")?;
                self.write_bytes(n.as_bytes())?;
            }
            Object::Array(arr) => {
                self.write_bytes(b"[")?;
                for (i, obj) in arr.iter().enumerate() {
                    if i > 0 {
                        self.write_bytes(b" ")?;
                    }
                    self.write_object_value(obj)?;
                }
                self.write_bytes(b"]")?;
            }
            Object::Dictionary(dict) => {
                self.write_bytes(b"<<")?;
                for (key, value) in dict.entries() {
                    self.write_bytes(b"\n/")?;
                    self.write_bytes(key.as_bytes())?;
                    self.write_bytes(b" ")?;
                    self.write_object_value(value)?;
                }
                self.write_bytes(b"\n>>")?;
            }
            Object::Stream(dict, data) => {
                self.write_object_value(&Object::Dictionary(dict.clone()))?;
                self.write_bytes(b"\nstream\n")?;
                self.write_bytes(data)?;
                self.write_bytes(b"\nendstream")?;
            }
            Object::Reference(id) => {
                let ref_str = format!("{} {} R", id.number(), id.generation());
                self.write_bytes(ref_str.as_bytes())?;
            }
        }
        Ok(())
    }

    fn write_xref(&mut self) -> Result<()> {
        self.write_bytes(b"xref\n")?;

        // Sort by object number and write entries
        let mut entries: Vec<_> = self
            .xref_positions
            .iter()
            .map(|(id, pos)| (*id, *pos))
            .collect();
        entries.sort_by_key(|(id, _)| id.number());

        // Find the highest object number to determine size
        let max_obj_num = entries.iter().map(|(id, _)| id.number()).max().unwrap_or(0);

        // Write subsection header - PDF 1.7 spec allows multiple subsections
        // For simplicity, write one subsection from 0 to max
        self.write_bytes(b"0 ")?;
        self.write_bytes((max_obj_num + 1).to_string().as_bytes())?;
        self.write_bytes(b"\n")?;

        // Write free object entry
        self.write_bytes(b"0000000000 65535 f \n")?;

        // Write entries for all object numbers from 1 to max
        // Fill in gaps with free entries
        for obj_num in 1..=max_obj_num {
            let _obj_id = ObjectId::new(obj_num, 0);
            if let Some((_, position)) = entries.iter().find(|(id, _)| id.number() == obj_num) {
                let entry = format!("{:010} {:05} n \n", position, 0);
                self.write_bytes(entry.as_bytes())?;
            } else {
                // Free entry for gap
                self.write_bytes(b"0000000000 00000 f \n")?;
            }
        }

        Ok(())
    }

    fn write_xref_stream(&mut self) -> Result<()> {
        let catalog_id = self.catalog_id.expect("catalog_id must be set");
        let info_id = self.info_id.expect("info_id must be set");

        // Allocate object ID for the xref stream
        let xref_stream_id = self.allocate_object_id();
        let xref_position = self.current_position;

        // Create XRef stream writer with trailer information
        let mut xref_writer = XRefStreamWriter::new(xref_stream_id);
        xref_writer.set_trailer_info(catalog_id, info_id);

        // Add free entry for object 0
        xref_writer.add_free_entry(0, 65535);

        // Sort entries by object number
        let mut entries: Vec<_> = self
            .xref_positions
            .iter()
            .map(|(id, pos)| (*id, *pos))
            .collect();
        entries.sort_by_key(|(id, _)| id.number());

        // Find the highest object number (including the xref stream itself)
        let max_obj_num = entries
            .iter()
            .map(|(id, _)| id.number())
            .max()
            .unwrap_or(0)
            .max(xref_stream_id.number());

        // Add entries for all objects
        for obj_num in 1..=max_obj_num {
            if obj_num == xref_stream_id.number() {
                // The xref stream entry will be added with the correct position
                xref_writer.add_in_use_entry(xref_position, 0);
            } else if let Some((id, position)) =
                entries.iter().find(|(id, _)| id.number() == obj_num)
            {
                xref_writer.add_in_use_entry(*position, id.generation());
            } else {
                // Free entry for gap
                xref_writer.add_free_entry(0, 0);
            }
        }

        // Mark position for xref stream object
        self.xref_positions.insert(xref_stream_id, xref_position);

        // Write object header
        self.write_bytes(
            format!(
                "{} {} obj\n",
                xref_stream_id.number(),
                xref_stream_id.generation()
            )
            .as_bytes(),
        )?;

        // Get the encoded data
        let uncompressed_data = xref_writer.encode_entries();
        let final_data = if self.config.compress_streams {
            crate::compression::compress(&uncompressed_data)?
        } else {
            uncompressed_data
        };

        // Create and write dictionary
        let mut dict = xref_writer.create_dictionary(None);
        dict.set("Length", Object::Integer(final_data.len() as i64));

        // Add filter if compression is enabled
        if self.config.compress_streams {
            dict.set("Filter", Object::Name("FlateDecode".to_string()));
        }
        self.write_bytes(b"<<")?;
        for (key, value) in dict.iter() {
            self.write_bytes(b"\n/")?;
            self.write_bytes(key.as_bytes())?;
            self.write_bytes(b" ")?;
            self.write_object_value(value)?;
        }
        self.write_bytes(b"\n>>\n")?;

        // Write stream
        self.write_bytes(b"stream\n")?;
        self.write_bytes(&final_data)?;
        self.write_bytes(b"\nendstream\n")?;
        self.write_bytes(b"endobj\n")?;

        // Write startxref and EOF
        self.write_bytes(b"\nstartxref\n")?;
        self.write_bytes(xref_position.to_string().as_bytes())?;
        self.write_bytes(b"\n%%EOF\n")?;

        Ok(())
    }

    fn write_trailer(&mut self, xref_position: u64) -> Result<()> {
        let catalog_id = self.catalog_id.expect("catalog_id must be set");
        let info_id = self.info_id.expect("info_id must be set");
        // Find the highest object number to determine size
        let max_obj_num = self
            .xref_positions
            .keys()
            .map(|id| id.number())
            .max()
            .unwrap_or(0);

        let mut trailer = Dictionary::new();
        trailer.set("Size", Object::Integer((max_obj_num + 1) as i64));
        trailer.set("Root", Object::Reference(catalog_id));
        trailer.set("Info", Object::Reference(info_id));

        self.write_bytes(b"trailer\n")?;
        self.write_object_value(&Object::Dictionary(trailer))?;
        self.write_bytes(b"\nstartxref\n")?;
        self.write_bytes(xref_position.to_string().as_bytes())?;
        self.write_bytes(b"\n%%EOF\n")?;

        Ok(())
    }

    fn write_bytes(&mut self, data: &[u8]) -> Result<()> {
        self.writer.write_all(data)?;
        self.current_position += data.len() as u64;
        Ok(())
    }

    #[allow(dead_code)]
    fn create_widget_appearance_stream(&mut self, widget_dict: &Dictionary) -> Result<ObjectId> {
        // Get widget rectangle
        let rect = if let Some(Object::Array(rect_array)) = widget_dict.get("Rect") {
            if rect_array.len() >= 4 {
                if let (
                    Some(Object::Real(x1)),
                    Some(Object::Real(y1)),
                    Some(Object::Real(x2)),
                    Some(Object::Real(y2)),
                ) = (
                    rect_array.first(),
                    rect_array.get(1),
                    rect_array.get(2),
                    rect_array.get(3),
                ) {
                    (*x1, *y1, *x2, *y2)
                } else {
                    (0.0, 0.0, 100.0, 20.0) // Default
                }
            } else {
                (0.0, 0.0, 100.0, 20.0) // Default
            }
        } else {
            (0.0, 0.0, 100.0, 20.0) // Default
        };

        let width = rect.2 - rect.0;
        let height = rect.3 - rect.1;

        // Create appearance stream content
        let mut content = String::new();

        // Set graphics state
        content.push_str("q\n");

        // Draw border (black)
        content.push_str("0 0 0 RG\n"); // Black stroke color
        content.push_str("1 w\n"); // 1pt line width

        // Draw rectangle border
        content.push_str(&format!("0 0 {width} {height} re\n"));
        content.push_str("S\n"); // Stroke

        // Fill with white background
        content.push_str("1 1 1 rg\n"); // White fill color
        content.push_str(&format!("0.5 0.5 {} {} re\n", width - 1.0, height - 1.0));
        content.push_str("f\n"); // Fill

        // Restore graphics state
        content.push_str("Q\n");

        // Create stream dictionary
        let mut stream_dict = Dictionary::new();
        stream_dict.set("Type", Object::Name("XObject".to_string()));
        stream_dict.set("Subtype", Object::Name("Form".to_string()));
        stream_dict.set(
            "BBox",
            Object::Array(vec![
                Object::Real(0.0),
                Object::Real(0.0),
                Object::Real(width),
                Object::Real(height),
            ]),
        );
        stream_dict.set("Resources", Object::Dictionary(Dictionary::new()));
        stream_dict.set("Length", Object::Integer(content.len() as i64));

        // Write the appearance stream
        let stream_id = self.allocate_object_id();
        self.write_object(stream_id, Object::Stream(stream_dict, content.into_bytes()))?;

        Ok(stream_id)
    }

    #[allow(dead_code)]
    fn create_field_appearance_stream(
        &mut self,
        field_dict: &Dictionary,
        widget: &crate::forms::Widget,
    ) -> Result<ObjectId> {
        let width = widget.rect.upper_right.x - widget.rect.lower_left.x;
        let height = widget.rect.upper_right.y - widget.rect.lower_left.y;

        // Create appearance stream content
        let mut content = String::new();

        // Set graphics state
        content.push_str("q\n");

        // Draw background if specified
        if let Some(bg_color) = &widget.appearance.background_color {
            match bg_color {
                crate::graphics::Color::Gray(g) => {
                    content.push_str(&format!("{g} g\n"));
                }
                crate::graphics::Color::Rgb(r, g, b) => {
                    content.push_str(&format!("{r} {g} {b} rg\n"));
                }
                crate::graphics::Color::Cmyk(c, m, y, k) => {
                    content.push_str(&format!("{c} {m} {y} {k} k\n"));
                }
            }
            content.push_str(&format!("0 0 {width} {height} re\n"));
            content.push_str("f\n");
        }

        // Draw border
        if let Some(border_color) = &widget.appearance.border_color {
            match border_color {
                crate::graphics::Color::Gray(g) => {
                    content.push_str(&format!("{g} G\n"));
                }
                crate::graphics::Color::Rgb(r, g, b) => {
                    content.push_str(&format!("{r} {g} {b} RG\n"));
                }
                crate::graphics::Color::Cmyk(c, m, y, k) => {
                    content.push_str(&format!("{c} {m} {y} {k} K\n"));
                }
            }
            content.push_str(&format!("{} w\n", widget.appearance.border_width));
            content.push_str(&format!("0 0 {width} {height} re\n"));
            content.push_str("S\n");
        }

        // For checkboxes, add a checkmark if checked
        if let Some(Object::Name(ft)) = field_dict.get("FT") {
            if ft == "Btn" {
                if let Some(Object::Name(v)) = field_dict.get("V") {
                    if v == "Yes" {
                        // Draw checkmark
                        content.push_str("0 0 0 RG\n"); // Black
                        content.push_str("2 w\n");
                        let margin = width * 0.2;
                        content.push_str(&format!("{} {} m\n", margin, height / 2.0));
                        content.push_str(&format!("{} {} l\n", width / 2.0, margin));
                        content.push_str(&format!("{} {} l\n", width - margin, height - margin));
                        content.push_str("S\n");
                    }
                }
            }
        }

        // Restore graphics state
        content.push_str("Q\n");

        // Create stream dictionary
        let mut stream_dict = Dictionary::new();
        stream_dict.set("Type", Object::Name("XObject".to_string()));
        stream_dict.set("Subtype", Object::Name("Form".to_string()));
        stream_dict.set(
            "BBox",
            Object::Array(vec![
                Object::Real(0.0),
                Object::Real(0.0),
                Object::Real(width),
                Object::Real(height),
            ]),
        );
        stream_dict.set("Resources", Object::Dictionary(Dictionary::new()));
        stream_dict.set("Length", Object::Integer(content.len() as i64));

        // Write the appearance stream
        let stream_id = self.allocate_object_id();
        self.write_object(stream_id, Object::Stream(stream_dict, content.into_bytes()))?;

        Ok(stream_id)
    }
}

/// Format a DateTime as a PDF date string (D:YYYYMMDDHHmmSSOHH'mm)
fn format_pdf_date(date: DateTime<Utc>) -> String {
    // Format the UTC date according to PDF specification
    // D:YYYYMMDDHHmmSSOHH'mm where O is the relationship of local time to UTC (+ or -)
    let formatted = date.format("D:%Y%m%d%H%M%S");

    // For UTC, the offset is always +00'00
    format!("{formatted}+00'00")
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::objects::{Object, ObjectId};
    use crate::page::Page;

    #[test]
    fn test_pdf_writer_new_with_writer() {
        let buffer = Vec::new();
        let writer = PdfWriter::new_with_writer(buffer);
        assert_eq!(writer.current_position, 0);
        assert!(writer.xref_positions.is_empty());
    }

    #[test]
    fn test_write_header() {
        let mut buffer = Vec::new();
        let mut writer = PdfWriter::new_with_writer(&mut buffer);

        writer.write_header().unwrap();

        // Check PDF version
        assert!(buffer.starts_with(b"%PDF-1.7\n"));
        // Check binary comment
        assert_eq!(buffer.len(), 15); // 9 bytes for header + 6 bytes for binary comment
        assert_eq!(buffer[9], b'%');
        assert_eq!(buffer[10], 0xE2);
        assert_eq!(buffer[11], 0xE3);
        assert_eq!(buffer[12], 0xCF);
        assert_eq!(buffer[13], 0xD3);
        assert_eq!(buffer[14], b'\n');
    }

    #[test]
    fn test_write_catalog() {
        let mut buffer = Vec::new();
        let mut writer = PdfWriter::new_with_writer(&mut buffer);

        let mut document = Document::new();
        // Set required IDs before calling write_catalog
        writer.catalog_id = Some(writer.allocate_object_id());
        writer.pages_id = Some(writer.allocate_object_id());
        writer.info_id = Some(writer.allocate_object_id());
        writer.write_catalog(&mut document).unwrap();

        let catalog_id = writer.catalog_id.unwrap();
        assert_eq!(catalog_id.number(), 1);
        assert_eq!(catalog_id.generation(), 0);
        assert!(!buffer.is_empty());

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("1 0 obj"));
        assert!(content.contains("/Type /Catalog"));
        assert!(content.contains("/Pages 2 0 R"));
        assert!(content.contains("endobj"));
    }

    #[test]
    fn test_write_empty_document() {
        let mut buffer = Vec::new();
        let mut document = Document::new();

        {
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();
        }

        // Verify PDF structure
        let content = String::from_utf8_lossy(&buffer);
        assert!(content.starts_with("%PDF-1.7\n"));
        assert!(content.contains("trailer"));
        assert!(content.contains("%%EOF"));
    }

    #[test]
    fn test_write_document_with_pages() {
        let mut buffer = Vec::new();
        let mut document = Document::new();
        document.add_page(Page::a4());
        document.add_page(Page::letter());

        {
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();
        }

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("/Type /Pages"));
        assert!(content.contains("/Count 2"));
        assert!(content.contains("/MediaBox"));
    }

    #[test]
    fn test_write_info() {
        let mut buffer = Vec::new();
        let mut document = Document::new();
        document.set_title("Test Title");
        document.set_author("Test Author");
        document.set_subject("Test Subject");
        document.set_keywords("test, keywords");

        {
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            // Set required info_id before calling write_info
            writer.info_id = Some(writer.allocate_object_id());
            writer.write_info(&document).unwrap();
            let info_id = writer.info_id.unwrap();
            assert!(info_id.number() > 0);
        }

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("/Title (Test Title)"));
        assert!(content.contains("/Author (Test Author)"));
        assert!(content.contains("/Subject (Test Subject)"));
        assert!(content.contains("/Keywords (test, keywords)"));
        assert!(content.contains("/Producer (oxidize_pdf v"));
        assert!(content.contains("/Creator (oxidize_pdf)"));
        assert!(content.contains("/CreationDate"));
        assert!(content.contains("/ModDate"));
    }

    #[test]
    fn test_write_info_with_dates() {
        use chrono::{TimeZone, Utc};

        let mut buffer = Vec::new();
        let mut document = Document::new();

        // Set specific dates
        let creation_date = Utc.with_ymd_and_hms(2023, 1, 1, 12, 0, 0).unwrap();
        let mod_date = Utc.with_ymd_and_hms(2023, 6, 15, 18, 30, 0).unwrap();

        document.set_creation_date(creation_date);
        document.set_modification_date(mod_date);
        document.set_creator("Test Creator");
        document.set_producer("Test Producer");

        {
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            // Set required info_id before calling write_info
            writer.info_id = Some(writer.allocate_object_id());
            writer.write_info(&document).unwrap();
        }

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("/CreationDate (D:20230101"));
        assert!(content.contains("/ModDate (D:20230615"));
        assert!(content.contains("/Creator (Test Creator)"));
        assert!(content.contains("/Producer (Test Producer)"));
    }

    #[test]
    fn test_format_pdf_date() {
        use chrono::{TimeZone, Utc};

        let date = Utc.with_ymd_and_hms(2023, 12, 25, 15, 30, 45).unwrap();
        let formatted = format_pdf_date(date);

        // Should start with D: and contain date/time components
        assert!(formatted.starts_with("D:"));
        assert!(formatted.contains("20231225"));
        assert!(formatted.contains("153045"));

        // Should contain timezone offset
        assert!(formatted.contains("+") || formatted.contains("-"));
    }

    #[test]
    fn test_write_object() {
        let mut buffer = Vec::new();
        let obj_id = ObjectId::new(5, 0);
        let obj = Object::String("Hello PDF".to_string());

        {
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_object(obj_id, obj).unwrap();
            assert!(writer.xref_positions.contains_key(&obj_id));
        }

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("5 0 obj"));
        assert!(content.contains("(Hello PDF)"));
        assert!(content.contains("endobj"));
    }

    #[test]
    fn test_write_xref() {
        let mut buffer = Vec::new();
        let mut writer = PdfWriter::new_with_writer(&mut buffer);

        // Add some objects to xref
        writer.xref_positions.insert(ObjectId::new(1, 0), 15);
        writer.xref_positions.insert(ObjectId::new(2, 0), 94);
        writer.xref_positions.insert(ObjectId::new(3, 0), 152);

        writer.write_xref().unwrap();

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("xref"));
        assert!(content.contains("0 4")); // 0 to 3
        assert!(content.contains("0000000000 65535 f "));
        assert!(content.contains("0000000015 00000 n "));
        assert!(content.contains("0000000094 00000 n "));
        assert!(content.contains("0000000152 00000 n "));
    }

    #[test]
    fn test_write_trailer() {
        let mut buffer = Vec::new();
        let mut writer = PdfWriter::new_with_writer(&mut buffer);

        writer.xref_positions.insert(ObjectId::new(1, 0), 15);
        writer.xref_positions.insert(ObjectId::new(2, 0), 94);

        let catalog_id = ObjectId::new(1, 0);
        let info_id = ObjectId::new(2, 0);

        writer.catalog_id = Some(catalog_id);
        writer.info_id = Some(info_id);
        writer.write_trailer(1234).unwrap();

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("trailer"));
        assert!(content.contains("/Size 3"));
        assert!(content.contains("/Root 1 0 R"));
        assert!(content.contains("/Info 2 0 R"));
        assert!(content.contains("startxref"));
        assert!(content.contains("1234"));
        assert!(content.contains("%%EOF"));
    }

    #[test]
    fn test_write_bytes() {
        let mut buffer = Vec::new();

        {
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            assert_eq!(writer.current_position, 0);

            writer.write_bytes(b"Hello").unwrap();
            assert_eq!(writer.current_position, 5);

            writer.write_bytes(b" World").unwrap();
            assert_eq!(writer.current_position, 11);
        }

        assert_eq!(buffer, b"Hello World");
    }

    #[test]
    fn test_complete_pdf_generation() {
        let mut buffer = Vec::new();
        let mut document = Document::new();
        document.set_title("Complete Test");
        document.add_page(Page::a4());

        {
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();
        }

        // Verify complete PDF structure
        assert!(buffer.starts_with(b"%PDF-1.7\n"));
        assert!(buffer.ends_with(b"%%EOF\n"));

        let content = String::from_utf8_lossy(&buffer);
        assert!(content.contains("obj"));
        assert!(content.contains("endobj"));
        assert!(content.contains("xref"));
        assert!(content.contains("trailer"));
        assert!(content.contains("/Type /Catalog"));
        assert!(content.contains("/Type /Pages"));
        assert!(content.contains("/Type /Page"));
    }

    // Integration tests for Writer ↔ Document ↔ Page interactions
    mod integration_tests {
        use super::*;
        use crate::graphics::Color;
        use crate::graphics::Image;
        use crate::text::Font;
        use std::fs;
        use tempfile::TempDir;

        #[test]
        fn test_writer_document_integration() {
            let temp_dir = TempDir::new().unwrap();
            let file_path = temp_dir.path().join("writer_document_integration.pdf");

            let mut document = Document::new();
            document.set_title("Writer Document Integration Test");
            document.set_author("Integration Test Suite");
            document.set_subject("Testing writer-document integration");
            document.set_keywords("writer, document, integration, test");

            // Add multiple pages with different content
            let mut page1 = Page::a4();
            page1
                .text()
                .set_font(Font::Helvetica, 16.0)
                .at(100.0, 750.0)
                .write("Page 1 Content")
                .unwrap();

            let mut page2 = Page::letter();
            page2
                .text()
                .set_font(Font::TimesRoman, 14.0)
                .at(100.0, 750.0)
                .write("Page 2 Content")
                .unwrap();

            document.add_page(page1);
            document.add_page(page2);

            // Write document
            let mut writer = PdfWriter::new(&file_path).unwrap();
            writer.write_document(&mut document).unwrap();

            // Verify file creation and structure
            assert!(file_path.exists());
            let metadata = fs::metadata(&file_path).unwrap();
            assert!(metadata.len() > 1000);

            // Verify PDF structure
            let content = fs::read(&file_path).unwrap();
            let content_str = String::from_utf8_lossy(&content);
            assert!(content_str.contains("/Type /Catalog"));
            assert!(content_str.contains("/Type /Pages"));
            assert!(content_str.contains("/Count 2"));
            assert!(content_str.contains("/Title (Writer Document Integration Test)"));
            assert!(content_str.contains("/Author (Integration Test Suite)"));
        }

        #[test]
        fn test_writer_page_content_integration() {
            let temp_dir = TempDir::new().unwrap();
            let file_path = temp_dir.path().join("writer_page_content.pdf");

            let mut document = Document::new();
            document.set_title("Writer Page Content Test");

            let mut page = Page::a4();
            page.set_margins(50.0, 50.0, 50.0, 50.0);

            // Add complex content to page
            page.text()
                .set_font(Font::HelveticaBold, 18.0)
                .at(100.0, 750.0)
                .write("Complex Page Content")
                .unwrap();

            page.graphics()
                .set_fill_color(Color::rgb(0.2, 0.4, 0.8))
                .rect(100.0, 600.0, 200.0, 100.0)
                .fill();

            page.graphics()
                .set_stroke_color(Color::rgb(0.8, 0.2, 0.2))
                .set_line_width(3.0)
                .circle(400.0, 650.0, 50.0)
                .stroke();

            // Add multiple text elements
            for i in 0..5 {
                let y = 550.0 - (i as f64 * 20.0);
                page.text()
                    .set_font(Font::TimesRoman, 12.0)
                    .at(100.0, y)
                    .write(&format!("Text line {line}", line = i + 1))
                    .unwrap();
            }

            document.add_page(page);

            // Write and verify
            let mut writer = PdfWriter::new(&file_path).unwrap();
            writer.write_document(&mut document).unwrap();

            assert!(file_path.exists());
            let metadata = fs::metadata(&file_path).unwrap();
            assert!(metadata.len() > 800);

            // Verify content streams are present
            let content = fs::read(&file_path).unwrap();
            let content_str = String::from_utf8_lossy(&content);
            assert!(content_str.contains("stream"));
            assert!(content_str.contains("endstream"));
            assert!(content_str.contains("/Length"));
        }

        #[test]
        fn test_writer_image_integration() {
            let temp_dir = TempDir::new().unwrap();
            let file_path = temp_dir.path().join("writer_image_integration.pdf");

            let mut document = Document::new();
            document.set_title("Writer Image Integration Test");

            let mut page = Page::a4();

            // Create test images
            let jpeg_data1 = vec![
                0xFF, 0xD8, 0xFF, 0xC0, 0x00, 0x11, 0x08, 0x00, 0x64, 0x00, 0xC8, 0x03, 0xFF, 0xD9,
            ];
            let image1 = Image::from_jpeg_data(jpeg_data1).unwrap();

            let jpeg_data2 = vec![
                0xFF, 0xD8, 0xFF, 0xC0, 0x00, 0x11, 0x08, 0x00, 0x32, 0x00, 0x32, 0x01, 0xFF, 0xD9,
            ];
            let image2 = Image::from_jpeg_data(jpeg_data2).unwrap();

            // Add images to page
            page.add_image("test_image1", image1);
            page.add_image("test_image2", image2);

            // Draw images
            page.draw_image("test_image1", 100.0, 600.0, 200.0, 100.0)
                .unwrap();
            page.draw_image("test_image2", 350.0, 600.0, 100.0, 100.0)
                .unwrap();

            // Add text labels
            page.text()
                .set_font(Font::Helvetica, 14.0)
                .at(100.0, 750.0)
                .write("Image Integration Test")
                .unwrap();

            document.add_page(page);

            // Write and verify
            let mut writer = PdfWriter::new(&file_path).unwrap();
            writer.write_document(&mut document).unwrap();

            assert!(file_path.exists());
            let metadata = fs::metadata(&file_path).unwrap();
            assert!(metadata.len() > 1000);

            // Verify XObject and image resources
            let content = fs::read(&file_path).unwrap();
            let content_str = String::from_utf8_lossy(&content);

            // Debug output
            println!("PDF size: {} bytes", content.len());
            println!("Contains 'XObject': {}", content_str.contains("XObject"));

            // Verify XObject is properly written
            assert!(content_str.contains("XObject"));
            assert!(content_str.contains("test_image1"));
            assert!(content_str.contains("test_image2"));
            assert!(content_str.contains("/Type /XObject"));
            assert!(content_str.contains("/Subtype /Image"));
        }

        #[test]
        fn test_writer_buffer_vs_file_output() {
            let temp_dir = TempDir::new().unwrap();
            let file_path = temp_dir.path().join("buffer_vs_file_output.pdf");

            let mut document = Document::new();
            document.set_title("Buffer vs File Output Test");

            let mut page = Page::a4();
            page.text()
                .set_font(Font::Helvetica, 12.0)
                .at(100.0, 700.0)
                .write("Testing buffer vs file output")
                .unwrap();

            document.add_page(page);

            // Write to buffer
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();
            }

            // Write to file
            {
                let mut writer = PdfWriter::new(&file_path).unwrap();
                writer.write_document(&mut document).unwrap();
            }

            // Read file content
            let file_content = fs::read(&file_path).unwrap();

            // Both should be valid PDFs
            assert!(buffer.starts_with(b"%PDF-1.7"));
            assert!(file_content.starts_with(b"%PDF-1.7"));
            assert!(buffer.ends_with(b"%%EOF\n"));
            assert!(file_content.ends_with(b"%%EOF\n"));

            // Both should contain the same structural elements
            let buffer_str = String::from_utf8_lossy(&buffer);
            let file_str = String::from_utf8_lossy(&file_content);

            assert!(buffer_str.contains("obj"));
            assert!(file_str.contains("obj"));
            assert!(buffer_str.contains("xref"));
            assert!(file_str.contains("xref"));
            assert!(buffer_str.contains("trailer"));
            assert!(file_str.contains("trailer"));
        }

        #[test]
        fn test_writer_large_document_performance() {
            let temp_dir = TempDir::new().unwrap();
            let file_path = temp_dir.path().join("large_document_performance.pdf");

            let mut document = Document::new();
            document.set_title("Large Document Performance Test");

            // Create many pages with content
            for i in 0..20 {
                let mut page = Page::a4();

                // Add title
                page.text()
                    .set_font(Font::HelveticaBold, 16.0)
                    .at(100.0, 750.0)
                    .write(&format!("Page {page}", page = i + 1))
                    .unwrap();

                // Add content lines
                for j in 0..30 {
                    let y = 700.0 - (j as f64 * 20.0);
                    if y > 100.0 {
                        page.text()
                            .set_font(Font::TimesRoman, 10.0)
                            .at(100.0, y)
                            .write(&format!(
                                "Line {line} on page {page}",
                                line = j + 1,
                                page = i + 1
                            ))
                            .unwrap();
                    }
                }

                // Add some graphics
                page.graphics()
                    .set_fill_color(Color::rgb(0.8, 0.8, 0.9))
                    .rect(50.0, 50.0, 100.0, 50.0)
                    .fill();

                document.add_page(page);
            }

            // Write document and measure performance
            let start = std::time::Instant::now();
            let mut writer = PdfWriter::new(&file_path).unwrap();
            writer.write_document(&mut document).unwrap();
            let duration = start.elapsed();

            // Verify file creation and reasonable performance
            assert!(file_path.exists());
            let metadata = fs::metadata(&file_path).unwrap();
            assert!(metadata.len() > 10000); // Should be substantial
            assert!(duration.as_secs() < 5); // Should complete within 5 seconds

            // Verify PDF structure
            let content = fs::read(&file_path).unwrap();
            let content_str = String::from_utf8_lossy(&content);
            assert!(content_str.contains("/Count 20"));
        }

        #[test]
        fn test_writer_metadata_handling() {
            let temp_dir = TempDir::new().unwrap();
            let file_path = temp_dir.path().join("metadata_handling.pdf");

            let mut document = Document::new();
            document.set_title("Metadata Handling Test");
            document.set_author("Test Author");
            document.set_subject("Testing metadata handling in writer");
            document.set_keywords("metadata, writer, test, integration");

            let mut page = Page::a4();
            page.text()
                .set_font(Font::Helvetica, 14.0)
                .at(100.0, 700.0)
                .write("Metadata Test Document")
                .unwrap();

            document.add_page(page);

            // Write document
            let mut writer = PdfWriter::new(&file_path).unwrap();
            writer.write_document(&mut document).unwrap();

            // Verify metadata in PDF
            let content = fs::read(&file_path).unwrap();
            let content_str = String::from_utf8_lossy(&content);

            assert!(content_str.contains("/Title (Metadata Handling Test)"));
            assert!(content_str.contains("/Author (Test Author)"));
            assert!(content_str.contains("/Subject (Testing metadata handling in writer)"));
            assert!(content_str.contains("/Keywords (metadata, writer, test, integration)"));
            assert!(content_str.contains("/Creator (oxidize_pdf)"));
            assert!(content_str.contains("/Producer (oxidize_pdf v"));
            assert!(content_str.contains("/CreationDate"));
            assert!(content_str.contains("/ModDate"));
        }

        #[test]
        fn test_writer_empty_document() {
            let temp_dir = TempDir::new().unwrap();
            let file_path = temp_dir.path().join("empty_document.pdf");

            let mut document = Document::new();
            document.set_title("Empty Document Test");

            // Write empty document (no pages)
            let mut writer = PdfWriter::new(&file_path).unwrap();
            writer.write_document(&mut document).unwrap();

            // Verify valid PDF structure even with no pages
            assert!(file_path.exists());
            let metadata = fs::metadata(&file_path).unwrap();
            assert!(metadata.len() > 200); // Should have basic structure

            let content = fs::read(&file_path).unwrap();
            let content_str = String::from_utf8_lossy(&content);
            assert!(content_str.contains("%PDF-1.7"));
            assert!(content_str.contains("/Type /Catalog"));
            assert!(content_str.contains("/Type /Pages"));
            assert!(content_str.contains("/Count 0"));
            assert!(content_str.contains("%%EOF"));
        }

        #[test]
        fn test_writer_error_handling() {
            let mut document = Document::new();
            document.set_title("Error Handling Test");
            document.add_page(Page::a4());

            // Test invalid path
            let result = PdfWriter::new("/invalid/path/that/does/not/exist.pdf");
            assert!(result.is_err());

            // Test writing to buffer should work
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            let result = writer.write_document(&mut document);
            assert!(result.is_ok());
            assert!(!buffer.is_empty());
        }

        #[test]
        fn test_writer_object_id_management() {
            let mut buffer = Vec::new();
            let mut document = Document::new();
            document.set_title("Object ID Management Test");

            // Add multiple pages to test object ID generation
            for i in 0..5 {
                let mut page = Page::a4();
                page.text()
                    .set_font(Font::Helvetica, 12.0)
                    .at(100.0, 700.0)
                    .write(&format!("Page {page}", page = i + 1))
                    .unwrap();
                document.add_page(page);
            }

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            // Verify object numbering in PDF
            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj")); // Catalog
            assert!(content.contains("2 0 obj")); // Pages
            assert!(content.contains("3 0 obj")); // First page
            assert!(content.contains("4 0 obj")); // First page content
            assert!(content.contains("5 0 obj")); // Second page
            assert!(content.contains("6 0 obj")); // Second page content

            // Verify xref table
            assert!(content.contains("xref"));
            assert!(content.contains("0 ")); // Subsection start
            assert!(content.contains("0000000000 65535 f")); // Free object entry
        }

        #[test]
        fn test_writer_content_stream_handling() {
            let mut buffer = Vec::new();
            let mut document = Document::new();
            document.set_title("Content Stream Test");

            let mut page = Page::a4();

            // Add content that will generate a content stream
            page.text()
                .set_font(Font::Helvetica, 12.0)
                .at(100.0, 700.0)
                .write("Content Stream Test")
                .unwrap();

            page.graphics()
                .set_fill_color(Color::rgb(0.5, 0.5, 0.5))
                .rect(100.0, 600.0, 200.0, 50.0)
                .fill();

            document.add_page(page);

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            // Verify content stream structure
            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("stream"));
            assert!(content.contains("endstream"));
            assert!(content.contains("/Length"));

            // Should contain content stream operations (may be compressed)
            assert!(content.contains("stream\n")); // Should have at least one stream
            assert!(content.contains("endstream")); // Should have matching endstream
        }

        #[test]
        fn test_writer_font_resource_handling() {
            let mut buffer = Vec::new();
            let mut document = Document::new();
            document.set_title("Font Resource Test");

            let mut page = Page::a4();

            // Use different fonts to test font resource generation
            page.text()
                .set_font(Font::Helvetica, 12.0)
                .at(100.0, 700.0)
                .write("Helvetica Font")
                .unwrap();

            page.text()
                .set_font(Font::TimesRoman, 14.0)
                .at(100.0, 650.0)
                .write("Times Roman Font")
                .unwrap();

            page.text()
                .set_font(Font::Courier, 10.0)
                .at(100.0, 600.0)
                .write("Courier Font")
                .unwrap();

            document.add_page(page);

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            // Verify font resources in PDF
            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Font"));
            assert!(content.contains("/Helvetica"));
            assert!(content.contains("/Times-Roman"));
            assert!(content.contains("/Courier"));
            assert!(content.contains("/Type /Font"));
            assert!(content.contains("/Subtype /Type1"));
        }

        #[test]
        fn test_writer_cross_reference_table() {
            let mut buffer = Vec::new();
            let mut document = Document::new();
            document.set_title("Cross Reference Test");

            // Add content to generate multiple objects
            for i in 0..3 {
                let mut page = Page::a4();
                page.text()
                    .set_font(Font::Helvetica, 12.0)
                    .at(100.0, 700.0)
                    .write(&format!("Page {page}", page = i + 1))
                    .unwrap();
                document.add_page(page);
            }

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            // Verify cross-reference table structure
            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("xref"));
            assert!(content.contains("trailer"));
            assert!(content.contains("startxref"));
            assert!(content.contains("%%EOF"));

            // Verify xref entries format
            let xref_start = content.find("xref").unwrap();
            let xref_section = &content[xref_start..];
            assert!(xref_section.contains("0000000000 65535 f")); // Free object entry

            // Should contain 'n' entries for used objects
            let n_count = xref_section.matches(" n ").count();
            assert!(n_count > 0); // Should have some object entries

            // Verify trailer dictionary
            assert!(content.contains("/Size"));
            assert!(content.contains("/Root"));
            assert!(content.contains("/Info"));
        }
    }

    // Comprehensive tests for writer.rs
    #[cfg(test)]
    mod comprehensive_tests {
        use super::*;
        use crate::page::Page;
        use crate::text::Font;
        use std::io::{self, ErrorKind, Write};

        // Mock writer that simulates IO errors
        struct FailingWriter {
            fail_after: usize,
            written: usize,
            error_kind: ErrorKind,
        }

        impl FailingWriter {
            fn new(fail_after: usize, error_kind: ErrorKind) -> Self {
                Self {
                    fail_after,
                    written: 0,
                    error_kind,
                }
            }
        }

        impl Write for FailingWriter {
            fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
                if self.written >= self.fail_after {
                    return Err(io::Error::new(self.error_kind, "Simulated write error"));
                }
                self.written += buf.len();
                Ok(buf.len())
            }

            fn flush(&mut self) -> io::Result<()> {
                if self.written >= self.fail_after {
                    return Err(io::Error::new(self.error_kind, "Simulated flush error"));
                }
                Ok(())
            }
        }

        // Test 1: Write failure during header
        #[test]
        fn test_write_failure_during_header() {
            let failing_writer = FailingWriter::new(5, ErrorKind::PermissionDenied);
            let mut writer = PdfWriter::new_with_writer(failing_writer);
            let mut document = Document::new();

            let result = writer.write_document(&mut document);
            assert!(result.is_err());
        }

        // Test 2: Empty arrays and dictionaries
        #[test]
        fn test_write_empty_collections() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Empty array
            writer
                .write_object(ObjectId::new(1, 0), Object::Array(vec![]))
                .unwrap();

            // Empty dictionary
            let empty_dict = Dictionary::new();
            writer
                .write_object(ObjectId::new(2, 0), Object::Dictionary(empty_dict))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("[]")); // Empty array
            assert!(content.contains("<<\n>>")); // Empty dictionary
        }

        // Test 3: Deeply nested structures
        #[test]
        fn test_write_deeply_nested_structures() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create deeply nested array
            let mut nested = Object::Array(vec![Object::Integer(1)]);
            for _ in 0..10 {
                nested = Object::Array(vec![nested]);
            }

            writer.write_object(ObjectId::new(1, 0), nested).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("[[[[[[[[[["));
            assert!(content.contains("]]]]]]]]]]"));
        }

        // Test 4: Large integers
        #[test]
        fn test_write_large_integers() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_cases = vec![i64::MAX, i64::MIN, 0, -1, 1, 999999999999999];

            for (i, &value) in test_cases.iter().enumerate() {
                writer
                    .write_object(ObjectId::new(i as u32 + 1, 0), Object::Integer(value))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            for value in test_cases {
                assert!(content.contains(&value.to_string()));
            }
        }

        // Test 5: Floating point edge cases
        #[test]
        fn test_write_float_edge_cases() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_cases = [
                0.0, -0.0, 1.0, -1.0, 0.123456, -0.123456, 1234.56789, 0.000001, 1000000.0,
            ];

            for (i, &value) in test_cases.iter().enumerate() {
                writer
                    .write_object(ObjectId::new(i as u32 + 1, 0), Object::Real(value))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);

            // Check formatting rules
            assert!(content.contains("0")); // 0.0 should be "0"
            assert!(content.contains("1")); // 1.0 should be "1"
            assert!(content.contains("0.123456"));
            assert!(content.contains("1234.567")); // Should be rounded
        }

        // Test 6: Special characters in strings
        #[test]
        fn test_write_special_characters_in_strings() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_strings = vec![
                "Simple string",
                "String with (parentheses)",
                "String with \\backslash",
                "String with \nnewline",
                "String with \ttab",
                "String with \rcarriage return",
                "Unicode: café",
                "Emoji: 🎯",
                "", // Empty string
            ];

            for (i, s) in test_strings.iter().enumerate() {
                writer
                    .write_object(
                        ObjectId::new(i as u32 + 1, 0),
                        Object::String(s.to_string()),
                    )
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);

            // Verify strings are properly enclosed
            assert!(content.contains("(Simple string)"));
            assert!(content.contains("()")); // Empty string
        }

        // Test 7: Escape sequences in names
        #[test]
        fn test_write_names_with_special_chars() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_names = vec![
                "SimpleName",
                "Name With Spaces",
                "Name#With#Hash",
                "Name/With/Slash",
                "Name(With)Parens",
                "Name[With]Brackets",
                "", // Empty name
            ];

            for (i, name) in test_names.iter().enumerate() {
                writer
                    .write_object(
                        ObjectId::new(i as u32 + 1, 0),
                        Object::Name(name.to_string()),
                    )
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);

            // Names should be prefixed with /
            assert!(content.contains("/SimpleName"));
            assert!(content.contains("/")); // Empty name should be just /
        }

        // Test 8: Binary data in streams
        #[test]
        fn test_write_binary_streams() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create stream with binary data
            let mut dict = Dictionary::new();
            let binary_data: Vec<u8> = (0..=255).collect();
            dict.set("Length", Object::Integer(binary_data.len() as i64));

            writer
                .write_object(ObjectId::new(1, 0), Object::Stream(dict, binary_data))
                .unwrap();

            let content = buffer;

            // Verify stream structure
            assert!(content.windows(6).any(|w| w == b"stream"));
            assert!(content.windows(9).any(|w| w == b"endstream"));

            // Verify binary data is present
            let stream_start = content.windows(6).position(|w| w == b"stream").unwrap() + 7; // "stream\n"
            let stream_end = content.windows(9).position(|w| w == b"endstream").unwrap();

            assert!(stream_end > stream_start);
            // Allow for line ending differences
            let data_length = stream_end - stream_start;
            assert!((256..=257).contains(&data_length));
        }

        // Test 9: Zero-length streams
        #[test]
        fn test_write_zero_length_stream() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let mut dict = Dictionary::new();
            dict.set("Length", Object::Integer(0));

            writer
                .write_object(ObjectId::new(1, 0), Object::Stream(dict, vec![]))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Length 0"));
            assert!(content.contains("stream\n\nendstream"));
        }

        // Test 10: Duplicate dictionary keys
        #[test]
        fn test_write_duplicate_dictionary_keys() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let mut dict = Dictionary::new();
            dict.set("Key", Object::Integer(1));
            dict.set("Key", Object::Integer(2)); // Overwrite

            writer
                .write_object(ObjectId::new(1, 0), Object::Dictionary(dict))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Should only have the last value
            assert!(content.contains("/Key 2"));
            assert!(!content.contains("/Key 1"));
        }

        // Test 11: Unicode in metadata
        #[test]
        fn test_write_unicode_metadata() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            document.set_title("Título en Español");
            document.set_author("作者");
            document.set_subject("Тема документа");
            document.set_keywords("מילות מפתח");

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            let content = buffer;

            // Verify metadata is present in some form
            let content_str = String::from_utf8_lossy(&content);
            assert!(content_str.contains("Title") || content_str.contains("Título"));
            assert!(content_str.contains("Author") || content_str.contains("作者"));
        }

        // Test 12: Very long strings
        #[test]
        fn test_write_very_long_strings() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let long_string = "A".repeat(10000);
            writer
                .write_object(ObjectId::new(1, 0), Object::String(long_string.clone()))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains(&format!("({long_string})")));
        }

        // Test 13: Maximum object ID
        #[test]
        fn test_write_maximum_object_id() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let max_id = ObjectId::new(u32::MAX, 65535);
            writer.write_object(max_id, Object::Null).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains(&format!("{} 65535 obj", u32::MAX)));
        }

        // Test 14: Complex page with multiple resources
        #[test]
        fn test_write_complex_page() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            let mut page = Page::a4();

            // Add various content
            page.text()
                .set_font(Font::Helvetica, 12.0)
                .at(100.0, 700.0)
                .write("Text with Helvetica")
                .unwrap();

            page.text()
                .set_font(Font::TimesRoman, 14.0)
                .at(100.0, 650.0)
                .write("Text with Times")
                .unwrap();

            page.graphics()
                .set_fill_color(crate::graphics::Color::Rgb(1.0, 0.0, 0.0))
                .rect(50.0, 50.0, 100.0, 100.0)
                .fill();

            page.graphics()
                .set_stroke_color(crate::graphics::Color::Rgb(0.0, 0.0, 1.0))
                .move_to(200.0, 200.0)
                .line_to(300.0, 300.0)
                .stroke();

            document.add_page(page);

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Verify multiple fonts
            assert!(content.contains("/Helvetica"));
            assert!(content.contains("/Times-Roman"));

            // Verify graphics operations (content is compressed, so check for stream presence)
            assert!(content.contains("stream"));
            assert!(content.contains("endstream"));
            assert!(content.contains("/FlateDecode")); // Compression filter
        }

        // Test 15: Document with 100 pages
        #[test]
        fn test_write_many_pages_document() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            for i in 0..100 {
                let mut page = Page::a4();
                page.text()
                    .set_font(Font::Helvetica, 12.0)
                    .at(100.0, 700.0)
                    .write(&format!("Page {}", i + 1))
                    .unwrap();
                document.add_page(page);
            }

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Verify page count
            assert!(content.contains("/Count 100"));

            // Verify that we have page objects (100 pages + 1 pages tree = 101 total)
            let page_type_count = content.matches("/Type /Page").count();
            assert!(page_type_count >= 100);

            // Verify content streams exist (compressed)
            assert!(content.contains("/FlateDecode"));
        }

        // Test 16: Write failure during xref
        #[test]
        fn test_write_failure_during_xref() {
            let failing_writer = FailingWriter::new(1000, ErrorKind::Other);
            let mut writer = PdfWriter::new_with_writer(failing_writer);
            let mut document = Document::new();

            // Add some content to ensure we get past header
            for _ in 0..5 {
                document.add_page(Page::a4());
            }

            let result = writer.write_document(&mut document);
            assert!(result.is_err());
        }

        // Test 17: Position tracking accuracy
        #[test]
        fn test_position_tracking_accuracy() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Write several objects and verify positions
            let ids = vec![
                ObjectId::new(1, 0),
                ObjectId::new(2, 0),
                ObjectId::new(3, 0),
            ];

            for id in &ids {
                writer.write_object(*id, Object::Null).unwrap();
            }

            // Verify positions were tracked
            for id in &ids {
                assert!(writer.xref_positions.contains_key(id));
                let pos = writer.xref_positions[id];
                assert!(pos < writer.current_position);
            }
        }

        // Test 18: Object reference cycles
        #[test]
        fn test_write_object_reference_cycles() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create dictionary with self-reference
            let mut dict = Dictionary::new();
            dict.set("Self", Object::Reference(ObjectId::new(1, 0)));
            dict.set("Other", Object::Reference(ObjectId::new(2, 0)));

            writer
                .write_object(ObjectId::new(1, 0), Object::Dictionary(dict))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Self 1 0 R"));
            assert!(content.contains("/Other 2 0 R"));
        }

        // Test 19: Different page sizes
        #[test]
        fn test_write_different_page_sizes() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            // Add pages with different sizes
            document.add_page(Page::a4());
            document.add_page(Page::letter());
            document.add_page(Page::new(200.0, 300.0)); // Custom size

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Verify different MediaBox values
            assert!(content.contains("[0 0 595")); // A4 width
            assert!(content.contains("[0 0 612")); // Letter width
            assert!(content.contains("[0 0 200 300]")); // Custom size
        }

        // Test 20: Empty metadata fields
        #[test]
        fn test_write_empty_metadata() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            // Set empty strings
            document.set_title("");
            document.set_author("");

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Should have empty strings
            assert!(content.contains("/Title ()"));
            assert!(content.contains("/Author ()"));
        }

        // Test 21: Write to read-only location (simulated)
        #[test]
        fn test_write_permission_error() {
            let failing_writer = FailingWriter::new(0, ErrorKind::PermissionDenied);
            let mut writer = PdfWriter::new_with_writer(failing_writer);
            let mut document = Document::new();

            let result = writer.write_document(&mut document);
            assert!(result.is_err());
        }

        // Test 22: Xref with many objects
        #[test]
        fn test_write_xref_many_objects() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create many objects
            for i in 1..=1000 {
                writer
                    .xref_positions
                    .insert(ObjectId::new(i, 0), (i * 100) as u64);
            }

            writer.write_xref().unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Verify xref structure
            assert!(content.contains("xref"));
            assert!(content.contains("0 1001")); // 0 + 1000 objects

            // Verify proper formatting of positions
            assert!(content.contains("0000000000 65535 f"));
            assert!(content.contains(" n "));
        }

        // Test 23: Stream with compression markers
        #[test]
        fn test_write_stream_with_filter() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let mut dict = Dictionary::new();
            dict.set("Length", Object::Integer(100));
            dict.set("Filter", Object::Name("FlateDecode".to_string()));

            let data = vec![0u8; 100];
            writer
                .write_object(ObjectId::new(1, 0), Object::Stream(dict, data))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Filter /FlateDecode"));
            assert!(content.contains("/Length 100"));
        }

        // Test 24: Arrays with mixed types
        #[test]
        fn test_write_mixed_type_arrays() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let array = vec![
                Object::Integer(42),
                Object::Real(3.14),
                Object::String("Hello".to_string()),
                Object::Name("World".to_string()),
                Object::Boolean(true),
                Object::Null,
                Object::Reference(ObjectId::new(5, 0)),
            ];

            writer
                .write_object(ObjectId::new(1, 0), Object::Array(array))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("[42 3.14 (Hello) /World true null 5 0 R]"));
        }

        // Test 25: Dictionary with nested structures
        #[test]
        fn test_write_nested_dictionaries() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let mut inner = Dictionary::new();
            inner.set("Inner", Object::Integer(1));

            let mut middle = Dictionary::new();
            middle.set("Middle", Object::Dictionary(inner));

            let mut outer = Dictionary::new();
            outer.set("Outer", Object::Dictionary(middle));

            writer
                .write_object(ObjectId::new(1, 0), Object::Dictionary(outer))
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Outer <<"));
            assert!(content.contains("/Middle <<"));
            assert!(content.contains("/Inner 1"));
        }

        // Test 26: Maximum generation number
        #[test]
        fn test_write_max_generation_number() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let id = ObjectId::new(1, 65535);
            writer.write_object(id, Object::Null).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 65535 obj"));
        }

        // Test 27: Cross-platform line endings
        #[test]
        fn test_write_consistent_line_endings() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            writer.write_header().unwrap();

            let content = buffer;

            // PDF should use \n consistently
            assert!(content.windows(2).filter(|w| w == b"\r\n").count() == 0);
            assert!(content.windows(1).filter(|w| w == b"\n").count() > 0);
        }

        // Test 28: Flush behavior
        #[test]
        fn test_writer_flush_behavior() {
            struct FlushCounter {
                buffer: Vec<u8>,
                flush_count: std::cell::RefCell<usize>,
            }

            impl Write for FlushCounter {
                fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
                    self.buffer.extend_from_slice(buf);
                    Ok(buf.len())
                }

                fn flush(&mut self) -> io::Result<()> {
                    *self.flush_count.borrow_mut() += 1;
                    Ok(())
                }
            }

            let flush_counter = FlushCounter {
                buffer: Vec::new(),
                flush_count: std::cell::RefCell::new(0),
            };

            let mut writer = PdfWriter::new_with_writer(flush_counter);
            let mut document = Document::new();

            writer.write_document(&mut document).unwrap();

            // Verify flush was called
            assert!(*writer.writer.flush_count.borrow() > 0);
        }

        // Test 29: Special PDF characters in content
        #[test]
        fn test_write_pdf_special_characters() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test parentheses in strings
            writer
                .write_object(
                    ObjectId::new(1, 0),
                    Object::String("Text with ) and ( parentheses".to_string()),
                )
                .unwrap();

            // Test backslash
            writer
                .write_object(
                    ObjectId::new(2, 0),
                    Object::String("Text with \\ backslash".to_string()),
                )
                .unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Should properly handle special characters
            assert!(content.contains("(Text with ) and ( parentheses)"));
            assert!(content.contains("(Text with \\ backslash)"));
        }

        // Test 30: Resource dictionary structure
        #[test]
        fn test_write_resource_dictionary() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            let mut page = Page::a4();

            // Add multiple resources
            page.text()
                .set_font(Font::Helvetica, 12.0)
                .at(100.0, 700.0)
                .write("Test")
                .unwrap();

            page.graphics()
                .set_fill_color(crate::graphics::Color::Rgb(1.0, 0.0, 0.0))
                .rect(50.0, 50.0, 100.0, 100.0)
                .fill();

            document.add_page(page);

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Verify resource dictionary structure
            assert!(content.contains("/Resources"));
            assert!(content.contains("/Font"));
            // Basic structure verification
            assert!(content.contains("stream") && content.contains("endstream"));
        }

        // Test 31: Error recovery after failed write
        #[test]
        fn test_error_recovery_after_failed_write() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Attempt to write an object
            writer
                .write_object(ObjectId::new(1, 0), Object::Null)
                .unwrap();

            // Verify state is still consistent
            assert!(writer.xref_positions.contains_key(&ObjectId::new(1, 0)));
            assert!(writer.current_position > 0);

            // Should be able to continue writing
            writer
                .write_object(ObjectId::new(2, 0), Object::Null)
                .unwrap();
            assert!(writer.xref_positions.contains_key(&ObjectId::new(2, 0)));
        }

        // Test 32: Memory efficiency with large document
        #[test]
        fn test_memory_efficiency_large_document() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            // Create document with repetitive content
            for i in 0..50 {
                let mut page = Page::a4();

                // Add lots of text
                for j in 0..20 {
                    page.text()
                        .set_font(Font::Helvetica, 10.0)
                        .at(50.0, 700.0 - (j as f64 * 30.0))
                        .write(&format!("Line {j} on page {i}"))
                        .unwrap();
                }

                document.add_page(page);
            }

            let _initial_capacity = buffer.capacity();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            // Verify reasonable memory usage
            assert!(!buffer.is_empty());
            assert!(buffer.capacity() <= buffer.len() * 2); // No excessive allocation
        }

        // Test 33: Trailer dictionary validation
        #[test]
        fn test_trailer_dictionary_content() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Set required IDs before calling write_trailer
            writer.catalog_id = Some(ObjectId::new(1, 0));
            writer.info_id = Some(ObjectId::new(2, 0));
            writer.xref_positions.insert(ObjectId::new(1, 0), 0);
            writer.xref_positions.insert(ObjectId::new(2, 0), 0);

            // Write minimal content
            writer.write_trailer(1000).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Verify trailer structure
            assert!(content.contains("trailer"));
            assert!(content.contains("/Size"));
            assert!(content.contains("/Root 1 0 R"));
            assert!(content.contains("/Info 2 0 R"));
            assert!(content.contains("startxref"));
            assert!(content.contains("1000"));
            assert!(content.contains("%%EOF"));
        }

        // Test 34: Write bytes handles partial writes
        #[test]
        fn test_write_bytes_partial_writes() {
            struct PartialWriter {
                buffer: Vec<u8>,
                max_per_write: usize,
            }

            impl Write for PartialWriter {
                fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
                    let to_write = buf.len().min(self.max_per_write);
                    self.buffer.extend_from_slice(&buf[..to_write]);
                    Ok(to_write)
                }

                fn flush(&mut self) -> io::Result<()> {
                    Ok(())
                }
            }

            let partial_writer = PartialWriter {
                buffer: Vec::new(),
                max_per_write: 10,
            };

            let mut writer = PdfWriter::new_with_writer(partial_writer);

            // Write large data
            let large_data = vec![b'A'; 100];
            writer.write_bytes(&large_data).unwrap();

            // Verify all data was written
            assert_eq!(writer.writer.buffer.len(), 100);
            assert!(writer.writer.buffer.iter().all(|&b| b == b'A'));
        }

        // Test 35: Object ID conflicts
        #[test]
        fn test_object_id_conflict_handling() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let id = ObjectId::new(1, 0);

            // Write same ID twice
            writer.write_object(id, Object::Integer(1)).unwrap();
            writer.write_object(id, Object::Integer(2)).unwrap();

            // Position should be updated
            assert!(writer.xref_positions.contains_key(&id));

            let content = String::from_utf8_lossy(&buffer);

            // Both objects should be written
            assert!(content.matches("1 0 obj").count() == 2);
        }

        // Test 36: Content stream encoding
        #[test]
        fn test_content_stream_encoding() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            let mut page = Page::a4();

            // Add text with special characters
            page.text()
                .set_font(Font::Helvetica, 12.0)
                .at(100.0, 700.0)
                .write("Special: €£¥")
                .unwrap();

            document.add_page(page);

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            // Content should be written (exact encoding depends on implementation)
            assert!(!buffer.is_empty());
        }

        // Test 37: PDF version in header
        #[test]
        fn test_pdf_version_header() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            writer.write_header().unwrap();

            let content = &buffer;

            // Verify PDF version
            assert!(content.starts_with(b"%PDF-1.7\n"));

            // Verify binary marker
            assert_eq!(content[9], b'%');
            assert_eq!(content[10], 0xE2);
            assert_eq!(content[11], 0xE3);
            assert_eq!(content[12], 0xCF);
            assert_eq!(content[13], 0xD3);
            assert_eq!(content[14], b'\n');
        }

        // Test 38: Page content operations order
        #[test]
        fn test_page_content_operations_order() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            let mut page = Page::a4();

            // Add operations in specific order
            page.graphics()
                .save_state()
                .set_fill_color(crate::graphics::Color::Rgb(1.0, 0.0, 0.0))
                .rect(50.0, 50.0, 100.0, 100.0)
                .fill()
                .restore_state();

            document.add_page(page);

            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Operations should maintain order
            // Note: Exact content depends on compression
            assert!(content.contains("stream"));
            assert!(content.contains("endstream"));
        }

        // Test 39: Invalid UTF-8 handling
        #[test]
        fn test_invalid_utf8_handling() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create string with invalid UTF-8
            let invalid_utf8 = vec![0xFF, 0xFE, 0xFD];
            let string = String::from_utf8_lossy(&invalid_utf8).to_string();

            writer
                .write_object(ObjectId::new(1, 0), Object::String(string))
                .unwrap();

            // Should not panic and should write something
            assert!(!buffer.is_empty());
        }

        // Test 40: Round-trip write and parse
        #[test]
        fn test_roundtrip_write_parse() {
            use crate::parser::PdfReader;
            use std::io::Cursor;

            let mut buffer = Vec::new();
            let mut document = Document::new();

            document.set_title("Round-trip Test");
            document.add_page(Page::a4());

            // Write document
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();
            }

            // Try to parse what we wrote
            let cursor = Cursor::new(buffer);
            let result = PdfReader::new(cursor);

            // Even if parsing fails (due to simplified writer),
            // we should have written valid PDF structure
            assert!(result.is_ok() || result.is_err()); // Either outcome is acceptable for this test
        }

        // Test to validate that all referenced ObjectIds exist in xref table
        #[test]
        fn test_pdf_object_references_are_valid() {
            let mut buffer = Vec::new();
            let mut document = Document::new();
            document.set_title("Object Reference Validation Test");

            // Create a page with form fields (the problematic case)
            let mut page = Page::a4();

            // Add some text content
            page.text()
                .set_font(Font::Helvetica, 12.0)
                .at(50.0, 700.0)
                .write("Form with validation:")
                .unwrap();

            // Add form widgets that previously caused invalid references
            use crate::forms::{BorderStyle, TextField, Widget, WidgetAppearance};
            use crate::geometry::{Point, Rectangle};
            use crate::graphics::Color;

            let text_appearance = WidgetAppearance {
                border_color: Some(Color::rgb(0.0, 0.0, 0.5)),
                background_color: Some(Color::rgb(0.95, 0.95, 1.0)),
                border_width: 1.0,
                border_style: BorderStyle::Solid,
            };

            let name_widget = Widget::new(Rectangle::new(
                Point::new(150.0, 640.0),
                Point::new(400.0, 660.0),
            ))
            .with_appearance(text_appearance);

            page.add_form_widget(name_widget.clone());
            document.add_page(page);

            // Enable forms and add field
            let form_manager = document.enable_forms();
            let name_field = TextField::new("name_field").with_default_value("");
            form_manager
                .add_text_field(name_field, name_widget, None)
                .unwrap();

            // Write the document
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_document(&mut document).unwrap();

            // Parse the generated PDF to validate structure
            let content = String::from_utf8_lossy(&buffer);

            // Extract xref section to find max object ID
            if let Some(xref_start) = content.find("xref\n") {
                let xref_section = &content[xref_start..];
                let lines: Vec<&str> = xref_section.lines().collect();
                if lines.len() > 1 {
                    let first_line = lines[1]; // Second line after "xref"
                    if let Some(space_pos) = first_line.find(' ') {
                        let (start_str, count_str) = first_line.split_at(space_pos);
                        let start_id: u32 = start_str.parse().unwrap_or(0);
                        let count: u32 = count_str.trim().parse().unwrap_or(0);
                        let max_valid_id = start_id + count - 1;

                        // Check that no references exceed the xref table size
                        // Look for patterns like "1000 0 R" that shouldn't exist
                        assert!(
                            !content.contains("1000 0 R"),
                            "Found invalid ObjectId reference 1000 0 R - max valid ID is {max_valid_id}"
                        );
                        assert!(
                            !content.contains("1001 0 R"),
                            "Found invalid ObjectId reference 1001 0 R - max valid ID is {max_valid_id}"
                        );
                        assert!(
                            !content.contains("1002 0 R"),
                            "Found invalid ObjectId reference 1002 0 R - max valid ID is {max_valid_id}"
                        );
                        assert!(
                            !content.contains("1003 0 R"),
                            "Found invalid ObjectId reference 1003 0 R - max valid ID is {max_valid_id}"
                        );

                        // Verify all object references are within valid range
                        for line in content.lines() {
                            if line.contains(" 0 R") {
                                // Extract object IDs from references
                                let words: Vec<&str> = line.split_whitespace().collect();
                                for i in 0..words.len().saturating_sub(2) {
                                    if words[i + 1] == "0" && words[i + 2] == "R" {
                                        if let Ok(obj_id) = words[i].parse::<u32>() {
                                            assert!(obj_id <= max_valid_id,
                                                   "Object reference {obj_id} 0 R exceeds xref table size (max: {max_valid_id})");
                                        }
                                    }
                                }
                            }
                        }

                        println!("✅ PDF structure validation passed: all {count} object references are valid (max ID: {max_valid_id})");
                    }
                }
            } else {
                panic!("Could not find xref section in generated PDF");
            }
        }

        #[test]
        fn test_xref_stream_generation() {
            let mut buffer = Vec::new();
            let mut document = Document::new();
            document.set_title("XRef Stream Test");

            let page = Page::a4();
            document.add_page(page);

            // Create writer with XRef stream configuration
            let config = WriterConfig {
                use_xref_streams: true,
                pdf_version: "1.5".to_string(),
                compress_streams: true,
            };
            let mut writer = PdfWriter::with_config(&mut buffer, config);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Should have PDF 1.5 header
            assert!(content.starts_with("%PDF-1.5\n"));

            // Should NOT have traditional xref table
            assert!(!content.contains("\nxref\n"));
            assert!(!content.contains("\ntrailer\n"));

            // Should have XRef stream object
            assert!(content.contains("/Type /XRef"));
            assert!(content.contains("/Filter /FlateDecode"));
            assert!(content.contains("/W ["));
            assert!(content.contains("/Root "));
            assert!(content.contains("/Info "));

            // Should have startxref pointing to XRef stream
            assert!(content.contains("\nstartxref\n"));
            assert!(content.contains("\n%%EOF\n"));
        }

        #[test]
        fn test_writer_config_default() {
            let config = WriterConfig::default();
            assert!(!config.use_xref_streams);
            assert_eq!(config.pdf_version, "1.7");
        }

        #[test]
        fn test_pdf_version_in_header() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            let page = Page::a4();
            document.add_page(page);

            // Test with custom version
            let config = WriterConfig {
                use_xref_streams: false,
                pdf_version: "1.4".to_string(),
                compress_streams: true,
            };
            let mut writer = PdfWriter::with_config(&mut buffer, config);
            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.starts_with("%PDF-1.4\n"));
        }

        #[test]
        fn test_xref_stream_with_multiple_objects() {
            let mut buffer = Vec::new();
            let mut document = Document::new();
            document.set_title("Multi Object XRef Stream Test");

            // Add multiple pages to create more objects
            for i in 0..3 {
                let mut page = Page::a4();
                page.text()
                    .set_font(Font::Helvetica, 12.0)
                    .at(100.0, 700.0)
                    .write(&format!("Page {page}", page = i + 1))
                    .unwrap();
                document.add_page(page);
            }

            let config = WriterConfig {
                use_xref_streams: true,
                pdf_version: "1.5".to_string(),
                compress_streams: true,
            };
            let mut writer = PdfWriter::with_config(&mut buffer, config);
            writer.write_document(&mut document).unwrap();
        }

        #[test]
        fn test_write_pdf_header() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            writer.write_header().unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.starts_with("%PDF-"));
            assert!(content.contains("\n%"));
        }

        #[test]
        fn test_write_empty_document() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            // Empty document should still generate valid PDF
            let mut writer = PdfWriter::new_with_writer(&mut buffer);
            let result = writer.write_document(&mut document);
            assert!(result.is_ok());

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.starts_with("%PDF-"));
            assert!(content.contains("%%EOF"));
        }

        // Note: The following tests were removed as they use methods that don't exist
        // in the current PdfWriter API (write_string, write_name, write_real, etc.)
        // These would need to be reimplemented using the actual available methods.

        /*
            #[test]
            fn test_write_string_escaping() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                // Test various string escaping scenarios
                writer.write_string(b"Normal text").unwrap();
                assert!(buffer.contains(&b'('[0]));

                buffer.clear();
                writer.write_string(b"Text with (parentheses)").unwrap();
                let content = String::from_utf8_lossy(&buffer);
                assert!(content.contains("\\(") || content.contains("\\)"));

                buffer.clear();
                writer.write_string(b"Text with \\backslash").unwrap();
                let content = String::from_utf8_lossy(&buffer);
                assert!(content.contains("\\\\"));
            }

            #[test]
            fn test_write_name_escaping() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                // Normal name
                writer.write_name("Type").unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "/Type");

                buffer.clear();
                writer.write_name("Name With Spaces").unwrap();
                let content = String::from_utf8_lossy(&buffer);
                assert!(content.starts_with("/"));
                assert!(content.contains("#20")); // Space encoded as #20

                buffer.clear();
                writer.write_name("Special#Characters").unwrap();
                let content = String::from_utf8_lossy(&buffer);
                assert!(content.contains("#23")); // # encoded as #23
            }

            #[test]
            fn test_write_real_number() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                writer.write_real(3.14159).unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "3.14159");

                buffer.clear();
                writer.write_real(0.0).unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "0");

                buffer.clear();
                writer.write_real(-123.456).unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "-123.456");

                buffer.clear();
                writer.write_real(1000.0).unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "1000");
            }

            #[test]
            fn test_write_array() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let array = vec![
                    PdfObject::Integer(1),
                    PdfObject::Real(2.5),
                    PdfObject::Name(PdfName::new("Test".to_string())),
                    PdfObject::Boolean(true),
                    PdfObject::Null,
                ];

                writer.write_array(&array).unwrap();
                let content = String::from_utf8_lossy(&buffer);

                assert!(content.starts_with("["));
                assert!(content.ends_with("]"));
                assert!(content.contains("1"));
                assert!(content.contains("2.5"));
                assert!(content.contains("/Test"));
                assert!(content.contains("true"));
                assert!(content.contains("null"));
            }

            #[test]
            fn test_write_dictionary() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let mut dict = HashMap::new();
                dict.insert(PdfName::new("Type".to_string()),
                           PdfObject::Name(PdfName::new("Page".to_string())));
                dict.insert(PdfName::new("Count".to_string()),
                           PdfObject::Integer(10));
                dict.insert(PdfName::new("Kids".to_string()),
                           PdfObject::Array(vec![PdfObject::Reference(1, 0)]));

                writer.write_dictionary(&dict).unwrap();
                let content = String::from_utf8_lossy(&buffer);

                assert!(content.starts_with("<<"));
                assert!(content.ends_with(">>"));
                assert!(content.contains("/Type /Page"));
                assert!(content.contains("/Count 10"));
                assert!(content.contains("/Kids [1 0 R]"));
            }

            #[test]
            fn test_write_stream() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let mut dict = HashMap::new();
                dict.insert(PdfName::new("Length".to_string()),
                           PdfObject::Integer(20));

                let data = b"This is stream data.";
                writer.write_stream(&dict, data).unwrap();

                let content = String::from_utf8_lossy(&buffer);
                assert!(content.contains("<<"));
                assert!(content.contains("/Length 20"));
                assert!(content.contains(">>"));
                assert!(content.contains("stream\n"));
                assert!(content.contains("This is stream data."));
                assert!(content.contains("\nendstream"));
            }

            #[test]
            fn test_write_indirect_object() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let obj = PdfObject::Dictionary({
                    let mut dict = HashMap::new();
                    dict.insert(PdfName::new("Type".to_string()),
                               PdfObject::Name(PdfName::new("Catalog".to_string())));
                    dict
                });

                writer.write_indirect_object(1, 0, &obj).unwrap();
                let content = String::from_utf8_lossy(&buffer);

                assert!(content.starts_with("1 0 obj"));
                assert!(content.contains("<<"));
                assert!(content.contains("/Type /Catalog"));
                assert!(content.contains(">>"));
                assert!(content.ends_with("endobj\n"));
            }

            #[test]
            fn test_write_xref_entry() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                writer.write_xref_entry(0, 65535, 'f').unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "0000000000 65535 f \n");

                buffer.clear();
                writer.write_xref_entry(123456, 0, 'n').unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "0000123456 00000 n \n");

                buffer.clear();
                writer.write_xref_entry(9999999999, 99, 'n').unwrap();
                assert_eq!(String::from_utf8_lossy(&buffer), "9999999999 00099 n \n");
            }

            #[test]
            fn test_write_trailer() {
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let mut trailer_dict = HashMap::new();
                trailer_dict.insert(PdfName::new("Size".to_string()),
                                  PdfObject::Integer(10));
                trailer_dict.insert(PdfName::new("Root".to_string()),
                                  PdfObject::Reference(1, 0));
                trailer_dict.insert(PdfName::new("Info".to_string()),
                                  PdfObject::Reference(2, 0));

                writer.write_trailer(&trailer_dict, 12345).unwrap();
                let content = String::from_utf8_lossy(&buffer);

                assert!(content.starts_with("trailer\n"));
                assert!(content.contains("<<"));
                assert!(content.contains("/Size 10"));
                assert!(content.contains("/Root 1 0 R"));
                assert!(content.contains("/Info 2 0 R"));
                assert!(content.contains(">>"));
                assert!(content.contains("startxref\n12345\n%%EOF"));
            }

            #[test]
            fn test_compress_stream_data() {
                let mut writer = PdfWriter::new(&mut Vec::new());

                let data = b"This is some text that should be compressed. It contains repeated patterns patterns patterns.";
                let compressed = writer.compress_stream(data).unwrap();

                // Compressed data should have compression header
                assert!(compressed.len() > 0);

                // Decompress to verify
                use flate2::read::ZlibDecoder;
                use std::io::Read;
                let mut decoder = ZlibDecoder::new(&compressed[..]);
                let mut decompressed = Vec::new();
                decoder.read_to_end(&mut decompressed).unwrap();

                assert_eq!(decompressed, data);
            }

            #[test]
            fn test_write_pages_tree() {
                let mut buffer = Vec::new();
                let mut document = Document::new();

                // Add multiple pages with different sizes
                document.add_page(Page::a4());
                document.add_page(Page::a3());
                document.add_page(Page::letter());

                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();

                let content = String::from_utf8_lossy(&buffer);

                // Should have pages object
                assert!(content.contains("/Type /Pages"));
                assert!(content.contains("/Count 3"));
                assert!(content.contains("/Kids ["));

                // Should have individual page objects
                assert!(content.contains("/Type /Page"));
                assert!(content.contains("/Parent "));
                assert!(content.contains("/MediaBox ["));
            }

            #[test]
            fn test_write_font_resources() {
                let mut buffer = Vec::new();
                let mut document = Document::new();

                let mut page = Page::a4();
                page.text()
                    .set_font(Font::Helvetica, 12.0)
                    .at(100.0, 700.0)
                    .write("Helvetica")
                    .unwrap();
                page.text()
                    .set_font(Font::Times, 14.0)
                    .at(100.0, 680.0)
                    .write("Times")
                    .unwrap();
                page.text()
                    .set_font(Font::Courier, 10.0)
                    .at(100.0, 660.0)
                    .write("Courier")
                    .unwrap();

                document.add_page(page);

                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();

                let content = String::from_utf8_lossy(&buffer);

                // Should have font resources
                assert!(content.contains("/Font <<"));
                assert!(content.contains("/Type /Font"));
                assert!(content.contains("/Subtype /Type1"));
                assert!(content.contains("/BaseFont /Helvetica"));
                assert!(content.contains("/BaseFont /Times-Roman"));
                assert!(content.contains("/BaseFont /Courier"));
            }

            #[test]
            fn test_write_image_xobject() {
                let mut buffer = Vec::new();
                let mut document = Document::new();

                let mut page = Page::a4();
                // Simulate adding an image (would need actual image data in real usage)
                // This test verifies the structure is written correctly

                document.add_page(page);

                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();

                let content = String::from_utf8_lossy(&buffer);

                // Basic structure should be present
                assert!(content.contains("/Resources"));
            }

            #[test]
            fn test_write_document_with_metadata() {
                let mut buffer = Vec::new();
                let mut document = Document::new();

                document.set_title("Test Document");
                document.set_author("Test Author");
                document.set_subject("Test Subject");
                document.set_keywords(vec!["test".to_string(), "pdf".to_string()]);
                document.set_creator("Test Creator");
                document.set_producer("oxidize-pdf");

                document.add_page(Page::a4());

                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();

                let content = String::from_utf8_lossy(&buffer);

                // Should have info dictionary
                assert!(content.contains("/Title (Test Document)"));
                assert!(content.contains("/Author (Test Author)"));
                assert!(content.contains("/Subject (Test Subject)"));
                assert!(content.contains("/Keywords (test, pdf)"));
                assert!(content.contains("/Creator (Test Creator)"));
                assert!(content.contains("/Producer (oxidize-pdf)"));
                assert!(content.contains("/CreationDate"));
                assert!(content.contains("/ModDate"));
            }

            #[test]
            fn test_write_cross_reference_stream() {
                let mut buffer = Vec::new();
                let config = WriterConfig {
                    use_xref_streams: true,
                    pdf_version: "1.5".to_string(),
                    compress_streams: true,
                };

                let mut writer = PdfWriter::with_config(&mut buffer, config);
                let mut document = Document::new();
                document.add_page(Page::a4());

                writer.write_document(&mut document).unwrap();

                let content = buffer.clone();

                // Should contain compressed xref stream
                let content_str = String::from_utf8_lossy(&content);
                assert!(content_str.contains("/Type /XRef"));
                assert!(content_str.contains("/Filter /FlateDecode"));
                assert!(content_str.contains("/W ["));
                assert!(content_str.contains("/Index ["));
            }

            #[test]
            fn test_write_linearized_hint() {
                // Test placeholder for linearized PDF support
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let mut document = Document::new();

                document.add_page(Page::a4());
                writer.write_document(&mut document).unwrap();

                // Linearization would add specific markers
                let content = String::from_utf8_lossy(&buffer);
                assert!(content.starts_with("%PDF-"));
            }

            #[test]
            fn test_write_encrypted_document() {
                // Test placeholder for encryption support
                let mut buffer = Vec::new();
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let mut document = Document::new();

                document.add_page(Page::a4());
                writer.write_document(&mut document).unwrap();

                let content = String::from_utf8_lossy(&buffer);
                // Would contain /Encrypt dictionary if implemented
                assert!(!content.contains("/Encrypt"));
            }

            #[test]
            fn test_object_number_allocation() {
                let mut writer = PdfWriter::new(&mut Vec::new());

                let obj1 = writer.allocate_object_number();
                let obj2 = writer.allocate_object_number();
                let obj3 = writer.allocate_object_number();

                assert_eq!(obj1, 1);
                assert_eq!(obj2, 2);
                assert_eq!(obj3, 3);

                // Object numbers should be sequential
                assert_eq!(obj2 - obj1, 1);
                assert_eq!(obj3 - obj2, 1);
            }

            #[test]
            fn test_write_page_content_stream() {
                let mut buffer = Vec::new();
                let mut document = Document::new();

                let mut page = Page::a4();
                page.text()
                    .set_font(Font::Helvetica, 24.0)
                    .at(100.0, 700.0)
                    .write("Hello, PDF!")
                    .unwrap();

                page.graphics()
                    .move_to(100.0, 600.0)
                    .line_to(500.0, 600.0)
                    .stroke();

                document.add_page(page);

                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();

                let content = String::from_utf8_lossy(&buffer);

                // Should have content stream with text and graphics operations
                assert!(content.contains("BT")); // Begin text
                assert!(content.contains("ET")); // End text
                assert!(content.contains("Tf")); // Set font
                assert!(content.contains("Td")); // Position text
                assert!(content.contains("Tj")); // Show text
                assert!(content.contains(" m ")); // Move to
                assert!(content.contains(" l ")); // Line to
                assert!(content.contains(" S")); // Stroke
            }
        }

        #[test]
        fn test_writer_config_default() {
            let config = WriterConfig::default();
            assert!(!config.use_xref_streams);
            assert_eq!(config.pdf_version, "1.7");
            assert!(config.compress_streams);
        }

        #[test]
        fn test_writer_config_custom() {
            let config = WriterConfig {
                use_xref_streams: true,
                pdf_version: "2.0".to_string(),
                compress_streams: false,
            };
            assert!(config.use_xref_streams);
            assert_eq!(config.pdf_version, "2.0");
            assert!(!config.compress_streams);
        }

        #[test]
        fn test_pdf_writer_new() {
            let buffer = Vec::new();
            let writer = PdfWriter::new_with_writer(buffer);
            assert_eq!(writer.current_position, 0);
            assert_eq!(writer.next_object_id, 1);
            assert!(writer.catalog_id.is_none());
            assert!(writer.pages_id.is_none());
            assert!(writer.info_id.is_none());
        }

        #[test]
        fn test_pdf_writer_with_config() {
            let config = WriterConfig {
                use_xref_streams: true,
                pdf_version: "1.5".to_string(),
                compress_streams: false,
            };
            let buffer = Vec::new();
            let writer = PdfWriter::with_config(buffer, config.clone());
            assert_eq!(writer.config.pdf_version, "1.5");
            assert!(writer.config.use_xref_streams);
            assert!(!writer.config.compress_streams);
        }

        #[test]
        fn test_allocate_object_id() {
            let buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(buffer);

            let id1 = writer.allocate_object_id();
            assert_eq!(id1, ObjectId::new(1, 0));

            let id2 = writer.allocate_object_id();
            assert_eq!(id2, ObjectId::new(2, 0));

            let id3 = writer.allocate_object_id();
            assert_eq!(id3, ObjectId::new(3, 0));

            assert_eq!(writer.next_object_id, 4);
        }

        #[test]
        fn test_write_header_version() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_header().unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.starts_with("%PDF-1.7\n"));
            // Binary comment should be present
            assert!(buffer.len() > 10);
            assert_eq!(buffer[9], b'%');
        }

        #[test]
        fn test_write_header_custom_version() {
            let mut buffer = Vec::new();
            {
                let config = WriterConfig {
                    pdf_version: "2.0".to_string(),
                    ..Default::default()
                };
                let mut writer = PdfWriter::with_config(&mut buffer, config);
                writer.write_header().unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.starts_with("%PDF-2.0\n"));
        }

        #[test]
        fn test_write_object_integer() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let obj_id = ObjectId::new(1, 0);
                let obj = Object::Integer(42);
                writer.write_object(obj_id, obj).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("42"));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_write_dictionary_object() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let obj_id = ObjectId::new(1, 0);

                let mut dict = Dictionary::new();
                dict.set("Type", Object::Name("Test".to_string()));
                dict.set("Count", Object::Integer(5));

                writer
                    .write_object(obj_id, Object::Dictionary(dict))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("/Type /Test"));
            assert!(content.contains("/Count 5"));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_write_array_object() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let obj_id = ObjectId::new(1, 0);

                let array = vec![Object::Integer(1), Object::Integer(2), Object::Integer(3)];

                writer.write_object(obj_id, Object::Array(array)).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("[1 2 3]"));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_write_string_object() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let obj_id = ObjectId::new(1, 0);

                writer
                    .write_object(obj_id, Object::String("Hello PDF".to_string()))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("(Hello PDF)"));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_write_reference_object() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let mut dict = Dictionary::new();
                dict.set("Parent", Object::Reference(ObjectId::new(2, 0)));

                writer
                    .write_object(ObjectId::new(1, 0), Object::Dictionary(dict))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Parent 2 0 R"));
        }

        // test_write_stream_object removed due to API differences

        #[test]
        fn test_write_boolean_objects() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                writer
                    .write_object(ObjectId::new(1, 0), Object::Boolean(true))
                    .unwrap();
                writer
                    .write_object(ObjectId::new(2, 0), Object::Boolean(false))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("true"));
            assert!(content.contains("2 0 obj"));
            assert!(content.contains("false"));
        }

        #[test]
        fn test_write_real_object() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                writer
                    .write_object(ObjectId::new(1, 0), Object::Real(3.14159))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("3.14159"));
        }

        #[test]
        fn test_write_null_object() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                writer
                    .write_object(ObjectId::new(1, 0), Object::Null)
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("null"));
        }

        #[test]
        fn test_write_nested_structures() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let mut inner_dict = Dictionary::new();
                inner_dict.set("Key", Object::String("Value".to_string()));

                let mut outer_dict = Dictionary::new();
                outer_dict.set("Inner", Object::Dictionary(inner_dict));
                outer_dict.set(
                    "Array",
                    Object::Array(vec![Object::Integer(1), Object::Name("Test".to_string())]),
                );

                writer
                    .write_object(ObjectId::new(1, 0), Object::Dictionary(outer_dict))
                    .unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Inner <<"));
            assert!(content.contains("/Key (Value)"));
            assert!(content.contains("/Array [1 /Test]"));
        }

        #[test]
        fn test_xref_positions_tracking() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let id1 = ObjectId::new(1, 0);
                let id2 = ObjectId::new(2, 0);

                writer.write_object(id1, Object::Integer(1)).unwrap();
                let pos1 = writer.xref_positions.get(&id1).copied();
                assert!(pos1.is_some());

                writer.write_object(id2, Object::Integer(2)).unwrap();
                let pos2 = writer.xref_positions.get(&id2).copied();
                assert!(pos2.is_some());

                // Position 2 should be after position 1
                assert!(pos2.unwrap() > pos1.unwrap());
            }
        }

        #[test]
        fn test_write_info_basic() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.info_id = Some(ObjectId::new(3, 0));

                let mut document = Document::new();
                document.set_title("Test Document");
                document.set_author("Test Author");

                writer.write_info(&document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("3 0 obj"));
            assert!(content.contains("/Title (Test Document)"));
            assert!(content.contains("/Author (Test Author)"));
            assert!(content.contains("/Producer"));
            assert!(content.contains("/CreationDate"));
        }

        #[test]
        fn test_write_info_with_all_metadata() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.info_id = Some(ObjectId::new(3, 0));

                let mut document = Document::new();
                document.set_title("Title");
                document.set_author("Author");
                document.set_subject("Subject");
                document.set_keywords("keyword1, keyword2");
                document.set_creator("Creator");

                writer.write_info(&document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Title (Title)"));
            assert!(content.contains("/Author (Author)"));
            assert!(content.contains("/Subject (Subject)"));
            assert!(content.contains("/Keywords (keyword1, keyword2)"));
            assert!(content.contains("/Creator (Creator)"));
        }

        #[test]
        fn test_write_catalog_basic() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.catalog_id = Some(ObjectId::new(1, 0));
                writer.pages_id = Some(ObjectId::new(2, 0));

                let mut document = Document::new();
                writer.write_catalog(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("/Type /Catalog"));
            assert!(content.contains("/Pages 2 0 R"));
        }

        #[test]
        fn test_write_catalog_with_outline() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.catalog_id = Some(ObjectId::new(1, 0));
                writer.pages_id = Some(ObjectId::new(2, 0));

                let mut document = Document::new();
                let mut outline = crate::structure::OutlineTree::new();
                outline.add_item(crate::structure::OutlineItem::new("Chapter 1"));
                document.outline = Some(outline);

                writer.write_catalog(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Type /Catalog"));
            assert!(content.contains("/Outlines"));
        }

        #[test]
        fn test_write_xref_basic() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                // Add some objects to xref
                writer.xref_positions.insert(ObjectId::new(0, 65535), 0);
                writer.xref_positions.insert(ObjectId::new(1, 0), 15);
                writer.xref_positions.insert(ObjectId::new(2, 0), 100);

                writer.write_xref().unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("xref"));
            assert!(content.contains("0 3")); // 3 objects starting at 0
            assert!(content.contains("0000000000 65535 f"));
            assert!(content.contains("0000000015 00000 n"));
            assert!(content.contains("0000000100 00000 n"));
        }

        #[test]
        fn test_write_trailer_complete() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.catalog_id = Some(ObjectId::new(1, 0));
                writer.info_id = Some(ObjectId::new(2, 0));

                // Add some objects
                writer.xref_positions.insert(ObjectId::new(0, 65535), 0);
                writer.xref_positions.insert(ObjectId::new(1, 0), 15);
                writer.xref_positions.insert(ObjectId::new(2, 0), 100);

                writer.write_trailer(1000).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("trailer"));
            assert!(content.contains("/Size 3"));
            assert!(content.contains("/Root 1 0 R"));
            assert!(content.contains("/Info 2 0 R"));
            assert!(content.contains("startxref"));
            assert!(content.contains("1000"));
            assert!(content.contains("%%EOF"));
        }

        // escape_string test removed - method is private

        // format_date test removed - method is private

        #[test]
        fn test_write_bytes_tracking() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                let data = b"Test data";
                writer.write_bytes(data).unwrap();
                assert_eq!(writer.current_position, data.len() as u64);

                writer.write_bytes(b" more").unwrap();
                assert_eq!(writer.current_position, (data.len() + 5) as u64);
            }

            assert_eq!(buffer, b"Test data more");
        }

        #[test]
        fn test_complete_document_write() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let mut document = Document::new();

                // Add a page
                let page = crate::page::Page::new(612.0, 792.0);
                document.add_page(page);

                // Set metadata
                document.set_title("Test PDF");
                document.set_author("Test Suite");

                // Write the document
                writer.write_document(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);

            // Check PDF structure
            assert!(content.starts_with("%PDF-"));
            assert!(content.contains("/Type /Catalog"));
            assert!(content.contains("/Type /Pages"));
            assert!(content.contains("/Type /Page"));
            assert!(content.contains("/Title (Test PDF)"));
            assert!(content.contains("/Author (Test Suite)"));
            assert!(content.contains("xref") || content.contains("/Type /XRef"));
            assert!(content.ends_with("%%EOF\n"));
        }

        // ========== NEW COMPREHENSIVE TESTS ==========

        #[test]
        fn test_writer_resource_cleanup() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                // Allocate many object IDs to test cleanup
                let ids: Vec<_> = (0..100).map(|_| writer.allocate_object_id()).collect();

                // Verify all IDs are unique and sequential
                for (i, &id) in ids.iter().enumerate() {
                    assert_eq!(id, (i + 1) as u32);
                }

                // Test that we can still allocate after cleanup
                let next_id = writer.allocate_object_id();
                assert_eq!(next_id, 101);
            }
            // Writer should be properly dropped here
        }

        #[test]
        fn test_writer_concurrent_safety() {
            use std::sync::{Arc, Mutex};
            use std::thread;

            let buffer = Arc::new(Mutex::new(Vec::new()));
            let buffer_clone = Arc::clone(&buffer);

            let handle = thread::spawn(move || {
                let mut buf = buffer_clone.lock().unwrap();
                let mut writer = PdfWriter::new_with_writer(&mut *buf);

                // Simulate concurrent operations
                for i in 0..10 {
                    let id = writer.allocate_object_id();
                    assert_eq!(id, (i + 1) as u32);
                }

                // Write some data
                writer.write_bytes(b"Thread test").unwrap();
            });

            handle.join().unwrap();

            let buffer = buffer.lock().unwrap();
            assert_eq!(&*buffer, b"Thread test");
        }

        #[test]
        fn test_writer_memory_efficiency() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                // Test that large objects don't cause excessive memory usage
                let large_data = vec![b'X'; 10_000];
                writer.write_bytes(&large_data).unwrap();

                // Verify position tracking is accurate
                assert_eq!(writer.current_position, 10_000);

                // Write more data
                writer.write_bytes(b"END").unwrap();
                assert_eq!(writer.current_position, 10_003);
            }

            // Verify buffer contents
            assert_eq!(buffer.len(), 10_003);
            assert_eq!(&buffer[10_000..], b"END");
        }

        #[test]
        fn test_writer_edge_case_handling() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);

                // Test empty writes
                writer.write_bytes(b"").unwrap();
                assert_eq!(writer.current_position, 0);

                // Test single byte writes
                writer.write_bytes(b"A").unwrap();
                assert_eq!(writer.current_position, 1);

                // Test null bytes
                writer.write_bytes(b"\0").unwrap();
                assert_eq!(writer.current_position, 2);

                // Test high ASCII values
                writer.write_bytes(b"\xFF\xFE").unwrap();
                assert_eq!(writer.current_position, 4);
            }

            assert_eq!(buffer, vec![b'A', 0, 0xFF, 0xFE]);
        }

        #[test]
        fn test_writer_cross_reference_consistency() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                let mut document = Document::new();

                // Create a document with multiple objects
                for i in 0..5 {
                    let page = crate::page::Page::new(612.0, 792.0);
                    document.add_page(page);
                }

                document.set_title(&format!("Test Document {}", std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap().as_secs()));

                writer.write_document(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);

            // Verify cross-reference structure
            if content.contains("xref") {
                // Traditional xref table
                assert!(content.contains("0000000000 65535 f"));
                assert!(content.contains("0000000000 00000 n") || content.contains("trailer"));
            } else {
                // XRef stream
                assert!(content.contains("/Type /XRef"));
            }

            // Should have proper trailer
            assert!(content.contains("/Size"));
            assert!(content.contains("/Root"));
        }

        #[test]
        fn test_writer_config_validation() {
            let mut config = WriterConfig::default();
            assert_eq!(config.pdf_version, "1.7");
            assert!(!config.use_xref_streams);
            assert!(config.compress_streams);

            // Test custom configuration
            config.pdf_version = "1.4".to_string();
            config.use_xref_streams = true;
            config.compress_streams = false;

            let buffer = Vec::new();
            let writer = PdfWriter::with_config(buffer, config.clone());
            assert_eq!(writer.config.pdf_version, "1.4");
            assert!(writer.config.use_xref_streams);
            assert!(!writer.config.compress_streams);
        }

        #[test]
        fn test_pdf_version_validation() {
            let test_versions = ["1.0", "1.1", "1.2", "1.3", "1.4", "1.5", "1.6", "1.7", "2.0"];

            for version in &test_versions {
                let mut config = WriterConfig::default();
                config.pdf_version = version.to_string();

                let mut buffer = Vec::new();
                {
                    let mut writer = PdfWriter::with_config(&mut buffer, config);
                    writer.write_header().unwrap();
                }

                let content = String::from_utf8_lossy(&buffer);
                assert!(content.starts_with(&format!("%PDF-{}", version)));
            }
        }

        #[test]
        fn test_object_id_allocation_sequence() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test sequential allocation
            let id1 = writer.allocate_object_id();
            let id2 = writer.allocate_object_id();
            let id3 = writer.allocate_object_id();

            assert_eq!(id1.number(), 1);
            assert_eq!(id2.number(), 2);
            assert_eq!(id3.number(), 3);
            assert_eq!(id1.generation(), 0);
            assert_eq!(id2.generation(), 0);
            assert_eq!(id3.generation(), 0);
        }

        #[test]
        fn test_xref_position_tracking() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let id1 = ObjectId::new(1, 0);
            let id2 = ObjectId::new(2, 0);

            // Write first object
            writer.write_header().unwrap();
            let pos1 = writer.current_position;
            writer.write_object(id1, Object::Integer(42)).unwrap();

            // Write second object
            let pos2 = writer.current_position;
            writer.write_object(id2, Object::String("test".to_string())).unwrap();

            // Verify positions are tracked
            assert_eq!(writer.xref_positions.get(&id1), Some(&pos1));
            assert_eq!(writer.xref_positions.get(&id2), Some(&pos2));
            assert!(pos2 > pos1);
        }

        #[test]
        fn test_binary_header_generation() {
            let mut buffer = Vec::new();
            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_header().unwrap();
            }

            // Check binary comment is present
            assert!(buffer.len() > 10);
            assert_eq!(&buffer[0..5], b"%PDF-");

            // Find the binary comment line
            let content = buffer.as_slice();
            let mut found_binary = false;
            for i in 0..content.len() - 5 {
                if content[i] == b'%' && content[i + 1] == 0xE2 {
                    found_binary = true;
                    break;
                }
            }
            assert!(found_binary, "Binary comment marker not found");
        }

        #[test]
        fn test_large_object_handling() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create a large string object
            let large_string = "A".repeat(10000);
            let id = ObjectId::new(1, 0);

            writer.write_object(id, Object::String(large_string.clone())).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains(&large_string));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_unicode_string_encoding() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let unicode_strings = vec![
                "Hello 世界",
                "café",
                "🎯 emoji test",
                "Ω α β γ δ",
                "\u{FEFF}BOM test",
            ];

            for (i, s) in unicode_strings.iter().enumerate() {
                let id = ObjectId::new((i + 1) as u32, 0);
                writer.write_object(id, Object::String(s.to_string())).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            // Verify objects are written properly
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("2 0 obj"));
        }

        #[test]
        fn test_special_characters_in_names() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let special_names = vec![
                "Name With Spaces",
                "Name#With#Hash",
                "Name/With/Slash",
                "Name(With)Parens",
                "Name[With]Brackets",
                "",
            ];

            for (i, name) in special_names.iter().enumerate() {
                let id = ObjectId::new((i + 1) as u32, 0);
                writer.write_object(id, Object::Name(name.to_string())).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            // Names should be properly escaped
            assert!(content.contains("Name#20With#20Spaces") || content.contains("Name With Spaces"));
        }

        #[test]
        fn test_deep_nested_structures() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create deeply nested dictionary
            let mut current = Dictionary::new();
            current.set("Level", Object::Integer(0));

            for i in 1..=10 {
                let mut next = Dictionary::new();
                next.set("Level", Object::Integer(i));
                next.set("Parent", Object::Dictionary(current));
                current = next;
            }

            let id = ObjectId::new(1, 0);
            writer.write_object(id, Object::Dictionary(current)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("/Level"));
        }

        #[test]
        fn test_xref_stream_vs_table_consistency() {
            let mut document = Document::new();
            document.add_page(crate::page::Page::new(612.0, 792.0));

            // Test with traditional xref table
            let mut buffer_table = Vec::new();
            {
                let config = WriterConfig {
                    use_xref_streams: false,
                    ..Default::default()
                };
                let mut writer = PdfWriter::with_config(&mut buffer_table, config);
                writer.write_document(&mut document.clone()).unwrap();
            }

            // Test with xref stream
            let mut buffer_stream = Vec::new();
            {
                let config = WriterConfig {
                    use_xref_streams: true,
                    ..Default::default()
                };
                let mut writer = PdfWriter::with_config(&mut buffer_stream, config);
                writer.write_document(&mut document.clone()).unwrap();
            }

            let content_table = String::from_utf8_lossy(&buffer_table);
            let content_stream = String::from_utf8_lossy(&buffer_stream);

            // Both should be valid PDFs
            assert!(content_table.starts_with("%PDF-"));
            assert!(content_stream.starts_with("%PDF-"));

            // Traditional should have xref table
            assert!(content_table.contains("xref"));
            assert!(content_table.contains("trailer"));

            // Stream version should have XRef object
            assert!(content_stream.contains("/Type /XRef") || content_stream.contains("xref"));
        }

        #[test]
        fn test_compression_flag_effects() {
            let mut document = Document::new();
            let mut page = crate::page::Page::new(612.0, 792.0);
            let mut gc = page.graphics();
            gc.show_text("Test content with compression").unwrap();
            document.add_page(page);

            // Test with compression enabled
            let mut buffer_compressed = Vec::new();
            {
                let config = WriterConfig {
                    compress_streams: true,
                    ..Default::default()
                };
                let mut writer = PdfWriter::with_config(&mut buffer_compressed, config);
                writer.write_document(&mut document.clone()).unwrap();
            }

            // Test with compression disabled
            let mut buffer_uncompressed = Vec::new();
            {
                let config = WriterConfig {
                    compress_streams: false,
                    ..Default::default()
                };
                let mut writer = PdfWriter::with_config(&mut buffer_uncompressed, config);
                writer.write_document(&mut document.clone()).unwrap();
            }

            // Compressed version should be smaller (usually)
            // Note: For small content, overhead might make it larger
            assert!(buffer_compressed.len() > 0);
            assert!(buffer_uncompressed.len() > 0);
        }

        #[test]
        fn test_empty_document_handling() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.starts_with("%PDF-"));
            assert!(content.contains("/Type /Catalog"));
            assert!(content.contains("/Type /Pages"));
            assert!(content.contains("/Count 0"));
            assert!(content.ends_with("%%EOF\n"));
        }

        #[test]
        fn test_object_reference_resolution() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let id1 = ObjectId::new(1, 0);
            let id2 = ObjectId::new(2, 0);

            // Create objects that reference each other
            let mut dict1 = Dictionary::new();
            dict1.set("Type", Object::Name("Test".to_string()));
            dict1.set("Reference", Object::Reference(id2));

            let mut dict2 = Dictionary::new();
            dict2.set("Type", Object::Name("Test2".to_string()));
            dict2.set("BackRef", Object::Reference(id1));

            writer.write_object(id1, Object::Dictionary(dict1)).unwrap();
            writer.write_object(id2, Object::Dictionary(dict2)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("2 0 obj"));
            assert!(content.contains("2 0 R"));
            assert!(content.contains("1 0 R"));
        }

        #[test]
        fn test_metadata_field_encoding() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let mut document = Document::new();
            document.set_title("Test Title with Ümlauts");
            document.set_author("Authör Name");
            document.set_subject("Subject with 中文");
            document.set_keywords("keyword1, keyword2, ключевые слова");

            writer.write_document(&mut document).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Title"));
            assert!(content.contains("/Author"));
            assert!(content.contains("/Subject"));
            assert!(content.contains("/Keywords"));
        }

        #[test]
        fn test_object_generation_numbers() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test different generation numbers
            let id_gen0 = ObjectId::new(1, 0);
            let id_gen1 = ObjectId::new(1, 1);
            let id_gen5 = ObjectId::new(2, 5);

            writer.write_object(id_gen0, Object::Integer(0)).unwrap();
            writer.write_object(id_gen1, Object::Integer(1)).unwrap();
            writer.write_object(id_gen5, Object::Integer(5)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("1 1 obj"));
            assert!(content.contains("2 5 obj"));
        }

        #[test]
        fn test_array_serialization_edge_cases() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_arrays = vec![
                // Empty array
                vec![],
                // Single element
                vec![Object::Integer(42)],
                // Mixed types
                vec![
                    Object::Integer(1),
                    Object::Real(3.14),
                    Object::String("test".to_string()),
                    Object::Name("TestName".to_string()),
                    Object::Boolean(true),
                    Object::Null,
                ],
                // Nested arrays
                vec![
                    Object::Array(vec![Object::Integer(1), Object::Integer(2)]),
                    Object::Array(vec![Object::String("a".to_string()), Object::String("b".to_string())]),
                ],
            ];

            for (i, array) in test_arrays.iter().enumerate() {
                let id = ObjectId::new((i + 1) as u32, 0);
                writer.write_object(id, Object::Array(array.clone())).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("[]")); // Empty array
            assert!(content.contains("[42]")); // Single element
            assert!(content.contains("true")); // Boolean
            assert!(content.contains("null")); // Null
        }

        #[test]
        fn test_real_number_precision() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_reals = vec![
                0.0,
                1.0,
                -1.0,
                3.14159265359,
                0.000001,
                1000000.5,
                -0.123456789,
                std::f64::consts::E,
                std::f64::consts::PI,
            ];

            for (i, real) in test_reals.iter().enumerate() {
                let id = ObjectId::new((i + 1) as u32, 0);
                writer.write_object(id, Object::Real(*real)).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("3.14159"));
            assert!(content.contains("0.000001"));
            assert!(content.contains("1000000.5"));
        }

        #[test]
        fn test_circular_reference_detection() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let id1 = ObjectId::new(1, 0);
            let id2 = ObjectId::new(2, 0);

            // Create circular reference (should not cause infinite loop)
            let mut dict1 = Dictionary::new();
            dict1.set("Ref", Object::Reference(id2));

            let mut dict2 = Dictionary::new();
            dict2.set("Ref", Object::Reference(id1));

            writer.write_object(id1, Object::Dictionary(dict1)).unwrap();
            writer.write_object(id2, Object::Dictionary(dict2)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("2 0 obj"));
        }

        #[test]
        fn test_document_structure_integrity() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            // Add multiple pages with different sizes
            document.add_page(crate::page::Page::new(612.0, 792.0)); // Letter
            document.add_page(crate::page::Page::new(595.0, 842.0)); // A4
            document.add_page(crate::page::Page::new(720.0, 1008.0)); // Legal

            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);

            // Verify structure
            assert!(content.contains("/Count 3"));
            assert!(content.contains("/MediaBox [0 0 612 792]"));
            assert!(content.contains("/MediaBox [0 0 595 842]"));
            assert!(content.contains("/MediaBox [0 0 720 1008]"));
        }

        #[test]
        fn test_xref_table_boundary_conditions() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test with object 0 (free object)
            writer.xref_positions.insert(ObjectId::new(0, 65535), 0);

            // Test with high object numbers
            writer.xref_positions.insert(ObjectId::new(999999, 0), 1234567890);

            // Test with high generation numbers
            writer.xref_positions.insert(ObjectId::new(1, 65534), 100);

            writer.write_xref().unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("0000000000 65535 f"));
            assert!(content.contains("1234567890 00000 n"));
            assert!(content.contains("0000000100 65534 n"));
        }

        #[test]
        fn test_trailer_completeness() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            writer.catalog_id = Some(ObjectId::new(1, 0));
            writer.info_id = Some(ObjectId::new(2, 0));

            // Add multiple objects to ensure proper size calculation
            for i in 0..10 {
                writer.xref_positions.insert(ObjectId::new(i, 0), (i * 100) as u64);
            }

            writer.write_trailer(5000).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("trailer"));
            assert!(content.contains("/Size 10"));
            assert!(content.contains("/Root 1 0 R"));
            assert!(content.contains("/Info 2 0 R"));
            assert!(content.contains("startxref"));
            assert!(content.contains("5000"));
            assert!(content.contains("%%EOF"));
        }

        #[test]
        fn test_position_tracking_accuracy() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let initial_pos = writer.current_position;
            assert_eq!(initial_pos, 0);

            writer.write_bytes(b"Hello").unwrap();
            assert_eq!(writer.current_position, 5);

            writer.write_bytes(b" World").unwrap();
            assert_eq!(writer.current_position, 11);

            writer.write_bytes(b"!").unwrap();
            assert_eq!(writer.current_position, 12);

            assert_eq!(buffer, b"Hello World!");
        }

        #[test]
        fn test_error_handling_write_failures() {
            // Test with a mock writer that fails
            struct FailingWriter {
                fail_after: usize,
                written: usize,
            }

            impl Write for FailingWriter {
                fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
                    if self.written + buf.len() > self.fail_after {
                        Err(std::io::Error::new(std::io::ErrorKind::Other, "Mock failure"))
                    } else {
                        self.written += buf.len();
                        Ok(buf.len())
                    }
                }

                fn flush(&mut self) -> std::io::Result<()> {
                    Ok(())
                }
            }

            let failing_writer = FailingWriter { fail_after: 10, written: 0 };
            let mut writer = PdfWriter::new_with_writer(failing_writer);

            // This should fail when trying to write more than 10 bytes
            let result = writer.write_bytes(b"This is a long string that will fail");
            assert!(result.is_err());
        }

        #[test]
        fn test_object_serialization_consistency() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test consistent serialization of the same object
            let test_obj = Object::Dictionary({
                let mut dict = Dictionary::new();
                dict.set("Type", Object::Name("Test".to_string()));
                dict.set("Value", Object::Integer(42));
                dict
            });

            let id1 = ObjectId::new(1, 0);
            let id2 = ObjectId::new(2, 0);

            writer.write_object(id1, test_obj.clone()).unwrap();
            writer.write_object(id2, test_obj.clone()).unwrap();

            let content = String::from_utf8_lossy(&buffer);

            // Both objects should have identical content except for object ID
            let lines: Vec<&str> = content.lines().collect();
            let obj1_content: Vec<&str> = lines.iter()
                .skip_while(|line| !line.contains("1 0 obj"))
                .take_while(|line| !line.contains("endobj"))
                .skip(1) // Skip the "1 0 obj" line
                .copied()
                .collect();

            let obj2_content: Vec<&str> = lines.iter()
                .skip_while(|line| !line.contains("2 0 obj"))
                .take_while(|line| !line.contains("endobj"))
                .skip(1) // Skip the "2 0 obj" line
                .copied()
                .collect();

            assert_eq!(obj1_content, obj2_content);
        }

        #[test]
        fn test_font_subsetting_integration() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Simulate used characters for font subsetting
            let mut used_chars = std::collections::HashSet::new();
            used_chars.insert('A');
            used_chars.insert('B');
            used_chars.insert('C');
            used_chars.insert(' ');

            writer.document_used_chars = Some(used_chars.clone());

            // Verify the used characters are stored
            assert!(writer.document_used_chars.is_some());
            let stored_chars = writer.document_used_chars.as_ref().unwrap();
            assert!(stored_chars.contains(&'A'));
            assert!(stored_chars.contains(&'B'));
            assert!(stored_chars.contains(&'C'));
            assert!(stored_chars.contains(&' '));
            assert!(!stored_chars.contains(&'Z'));
        }

        #[test]
        fn test_form_field_tracking() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test form field ID tracking
            let field_id = ObjectId::new(10, 0);
            let widget_id1 = ObjectId::new(11, 0);
            let widget_id2 = ObjectId::new(12, 0);

            writer.field_id_map.insert("test_field".to_string(), field_id);
            writer.field_widget_map.insert(
                "test_field".to_string(),
                vec![widget_id1, widget_id2]
            );
            writer.form_field_ids.push(field_id);

            // Verify tracking
            assert_eq!(writer.field_id_map.get("test_field"), Some(&field_id));
            assert_eq!(writer.field_widget_map.get("test_field"), Some(&vec![widget_id1, widget_id2]));
            assert!(writer.form_field_ids.contains(&field_id));
        }

        #[test]
        fn test_page_id_tracking() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let page_ids = vec![
                ObjectId::new(5, 0),
                ObjectId::new(6, 0),
                ObjectId::new(7, 0),
            ];

            writer.page_ids = page_ids.clone();

            assert_eq!(writer.page_ids.len(), 3);
            assert_eq!(writer.page_ids[0].number(), 5);
            assert_eq!(writer.page_ids[1].number(), 6);
            assert_eq!(writer.page_ids[2].number(), 7);
        }

        #[test]
        fn test_catalog_pages_info_id_allocation() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test that required IDs are properly allocated
            writer.catalog_id = Some(writer.allocate_object_id());
            writer.pages_id = Some(writer.allocate_object_id());
            writer.info_id = Some(writer.allocate_object_id());

            assert!(writer.catalog_id.is_some());
            assert!(writer.pages_id.is_some());
            assert!(writer.info_id.is_some());

            // IDs should be sequential
            assert_eq!(writer.catalog_id.unwrap().number(), 1);
            assert_eq!(writer.pages_id.unwrap().number(), 2);
            assert_eq!(writer.info_id.unwrap().number(), 3);
        }

        #[test]
        fn test_boolean_object_serialization() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            writer.write_object(ObjectId::new(1, 0), Object::Boolean(true)).unwrap();
            writer.write_object(ObjectId::new(2, 0), Object::Boolean(false)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("true"));
            assert!(content.contains("false"));
        }

        #[test]
        fn test_null_object_serialization() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            writer.write_object(ObjectId::new(1, 0), Object::Null).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("null"));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_stream_object_handling() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let stream_data = b"This is stream content";
            let mut stream_dict = Dictionary::new();
            stream_dict.set("Length", Object::Integer(stream_data.len() as i64));

            let stream = crate::objects::Stream {
                dict: stream_dict,
                data: stream_data.to_vec(),
            };

            writer.write_object(ObjectId::new(1, 0), Object::Stream(stream)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("/Length"));
            assert!(content.contains("stream"));
            assert!(content.contains("This is stream content"));
            assert!(content.contains("endstream"));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_integer_boundary_values() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_integers = vec![
                i64::MIN,
                -1000000,
                -1,
                0,
                1,
                1000000,
                i64::MAX,
            ];

            for (i, int_val) in test_integers.iter().enumerate() {
                let id = ObjectId::new((i + 1) as u32, 0);
                writer.write_object(id, Object::Integer(*int_val)).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains(&i64::MIN.to_string()));
            assert!(content.contains(&i64::MAX.to_string()));
        }

        #[test]
        fn test_real_number_special_values() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_reals = vec![
                0.0,
                -0.0,
                f64::MIN,
                f64::MAX,
                1.0 / 3.0, // Repeating decimal
                f64::EPSILON,
            ];

            for (i, real_val) in test_reals.iter().enumerate() {
                if real_val.is_finite() {
                    let id = ObjectId::new((i + 1) as u32, 0);
                    writer.write_object(id, Object::Real(*real_val)).unwrap();
                }
            }

            let content = String::from_utf8_lossy(&buffer);
            // Should contain some real numbers
            assert!(content.contains("0.33333") || content.contains("0.3"));
        }

        #[test]
        fn test_empty_containers() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Empty array
            writer.write_object(ObjectId::new(1, 0), Object::Array(vec![])).unwrap();

            // Empty dictionary
            writer.write_object(ObjectId::new(2, 0), Object::Dictionary(Dictionary::new())).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("[]"));
            assert!(content.contains("<<>>") || content.contains("<< >>"));
        }

        #[test]
        fn test_write_document_with_forms() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            // Add a page
            document.add_page(crate::page::Page::new(612.0, 792.0));

            // Add form manager to trigger AcroForm creation
            document.form_manager = Some(crate::forms::FormManager::new());

            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/AcroForm") || content.contains("AcroForm"));
        }

        #[test]
        fn test_write_document_with_outlines() {
            let mut buffer = Vec::new();
            let mut document = Document::new();

            // Add a page
            document.add_page(crate::page::Page::new(612.0, 792.0));

            // Add outline tree
            let mut outline_tree = crate::document::OutlineTree::new();
            outline_tree.add_item(crate::document::OutlineItem {
                title: "Chapter 1".to_string(),
                ..Default::default()
            });
            document.outline = Some(outline_tree);

            {
                let mut writer = PdfWriter::new_with_writer(&mut buffer);
                writer.write_document(&mut document).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Outlines") || content.contains("Chapter 1"));
        }

        #[test]
        fn test_string_escaping_edge_cases() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_strings = vec![
                "Simple string",
                "String with \\backslash",
                "String with (parentheses)",
                "String with \nnewline",
                "String with \ttab",
                "String with \rcarriage return",
                "Unicode: café",
                "Emoji: 🎯",
                "", // Empty string
            ];

            for (i, s) in test_strings.iter().enumerate() {
                let id = ObjectId::new((i + 1) as u32, 0);
                writer.write_object(id, Object::String(s.to_string())).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            // Should contain escaped or encoded strings
            assert!(content.contains("Simple string"));
        }

        #[test]
        fn test_name_escaping_edge_cases() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            let test_names = vec![
                "SimpleName",
                "Name With Spaces",
                "Name#With#Hash",
                "Name/With/Slash",
                "Name(With)Parens",
                "Name[With]Brackets",
                "", // Empty name
            ];

            for (i, name) in test_names.iter().enumerate() {
                let id = ObjectId::new((i + 1) as u32, 0);
                writer.write_object(id, Object::Name(name.to_string())).unwrap();
            }

            let content = String::from_utf8_lossy(&buffer);
            // Names should be properly escaped or handled
            assert!(content.contains("/SimpleName"));
        }

        #[test]
        fn test_maximum_nesting_depth() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create maximum reasonable nesting
            let mut current = Object::Integer(0);
            for i in 1..=100 {
                let mut dict = Dictionary::new();
                dict.set(&format!("Level{}", i), current);
                current = Object::Dictionary(dict);
            }

            writer.write_object(ObjectId::new(1, 0), current).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj"));
            assert!(content.contains("/Level"));
        }

        #[test]
        fn test_writer_state_isolation() {
            // Test that different writers don't interfere with each other
            let mut buffer1 = Vec::new();
            let mut buffer2 = Vec::new();

            let mut writer1 = PdfWriter::new_with_writer(&mut buffer1);
            let mut writer2 = PdfWriter::new_with_writer(&mut buffer2);

            // Write different objects to each writer
            writer1.write_object(ObjectId::new(1, 0), Object::Integer(111)).unwrap();
            writer2.write_object(ObjectId::new(1, 0), Object::Integer(222)).unwrap();

            let content1 = String::from_utf8_lossy(&buffer1);
            let content2 = String::from_utf8_lossy(&buffer2);

            assert!(content1.contains("111"));
            assert!(content2.contains("222"));
            assert!(!content1.contains("222"));
            assert!(!content2.contains("111"));
        }
        */

        /* Temporarily disabled for coverage measurement
        #[test]
        fn test_font_embedding() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test font dictionary creation
            let mut font_dict = Dictionary::new();
            font_dict.insert("Type".to_string(), PdfObject::Name(PdfName::new("Font")));
            font_dict.insert("Subtype".to_string(), PdfObject::Name(PdfName::new("Type1")));
            font_dict.insert("BaseFont".to_string(), PdfObject::Name(PdfName::new("Helvetica")));

            writer.write_object(ObjectId::new(1, 0), Object::Dictionary(font_dict)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Type /Font"));
            assert!(content.contains("/Subtype /Type1"));
            assert!(content.contains("/BaseFont /Helvetica"));
        }

        #[test]
        fn test_form_field_writing() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create a form field dictionary
            let field_dict = Dictionary::new()
                .set("FT", Name::new("Tx")) // Text field
                .set("T", String::from("Name".as_bytes().to_vec()))
                .set("V", String::from("John Doe".as_bytes().to_vec()));

            writer.write_object(ObjectId::new(1, 0), Object::Dictionary(field_dict)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/FT /Tx"));
            assert!(content.contains("(Name)"));
            assert!(content.contains("(John Doe)"));
        }

        #[test]
        fn test_write_binary_data() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test binary stream data
            let binary_data = vec![0xFF, 0xD8, 0xFF, 0xE0, 0x00, 0x10]; // JPEG header
            let stream = Object::Stream(
                Dictionary::new()
                    .set("Length", Object::Integer(binary_data.len() as i64))
                    .set("Filter", Object::Name("DCTDecode".to_string())),
                binary_data.clone(),
            );

            writer.write_object(ObjectId::new(1, 0), stream).unwrap();

            let content = buffer.clone();
            // Verify stream structure
            let content_str = String::from_utf8_lossy(&content);
            assert!(content_str.contains("/Length 6"));
            assert!(content_str.contains("/Filter /DCTDecode"));
            // Binary data should be present
            assert!(content.windows(6).any(|window| window == &binary_data[..]));
        }

        #[test]
        fn test_write_large_dictionary() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create a dictionary with many entries
            let mut dict = Dictionary::new();
            for i in 0..50 {
                dict = dict.set(format!("Key{}", i), Object::Integer(i));
            }

            writer.write_object(ObjectId::new(1, 0), Object::Dictionary(dict)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("/Key0 0"));
            assert!(content.contains("/Key49 49"));
            assert!(content.contains("<<") && content.contains(">>"));
        }

        #[test]
        fn test_write_nested_arrays() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Create nested arrays
            let inner_array = Object::Array(vec![Object::Integer(1), Object::Integer(2), Object::Integer(3)]);
            let outer_array = Object::Array(vec![
                Object::Integer(0),
                inner_array,
                Object::String("test".to_string()),
            ]);

            writer.write_object(ObjectId::new(1, 0), outer_array).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("[0 [1 2 3] (test)]"));
        }

        #[test]
        fn test_write_object_with_generation() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test non-zero generation number
            writer.write_object(ObjectId::new(5, 3), Object::Boolean(true)).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("5 3 obj"));
            assert!(content.contains("true"));
            assert!(content.contains("endobj"));
        }

        #[test]
        fn test_write_empty_objects() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test empty dictionary
            writer.write_object(ObjectId::new(1, 0), Object::Dictionary(Dictionary::new())).unwrap();
            // Test empty array
            writer.write_object(ObjectId::new(2, 0), Object::Array(vec![])).unwrap();
            // Test empty string
            writer.write_object(ObjectId::new(3, 0), Object::String(String::new())).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj\n<<>>"));
            assert!(content.contains("2 0 obj\n[]"));
            assert!(content.contains("3 0 obj\n()"));
        }

        #[test]
        fn test_escape_special_chars_in_strings() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            // Test string with special characters
            let special_string = String::from("Test (with) \\backslash\\ and )parens(".as_bytes().to_vec());
            writer.write_object(ObjectId::new(1, 0), special_string).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            // Should escape parentheses and backslashes
            assert!(content.contains("(Test \\(with\\) \\\\backslash\\\\ and \\)parens\\()"));
        }

        // #[test]
        // fn test_write_hex_string() {
        //     let mut buffer = Vec::new();
        //     let mut writer = PdfWriter::new_with_writer(&mut buffer);
        //
        //     // Create hex string (high bit bytes)
        //     let hex_data = vec![0xFF, 0xAB, 0xCD, 0xEF];
        //     let hex_string = Object::String(format!("{:02X}", hex_data.iter().map(|b| format!("{:02X}", b)).collect::<String>()));
        //
        //     writer.write_object(ObjectId::new(1, 0), hex_string).unwrap();
        //
        //     let content = String::from_utf8_lossy(&buffer);
        //     assert!(content.contains("FFABCDEF"));
        // }

        #[test]
        fn test_null_object() {
            let mut buffer = Vec::new();
            let mut writer = PdfWriter::new_with_writer(&mut buffer);

            writer.write_object(ObjectId::new(1, 0), Object::Null).unwrap();

            let content = String::from_utf8_lossy(&buffer);
            assert!(content.contains("1 0 obj\nnull\nendobj"));
        }
        */
    }
}