minify-html/src/proc/checkpoint.rs

use crate::proc::Processor;
use crate::proc::range::ProcessorRange;

#[derive(Copy, Clone)]
pub struct Checkpoint {
    // Avoid implementing a read position checkpoint, as source code does get modified (e.g. normalising entities), and
    // there's no check to see if source has since been overwritten (e.g. writing over source and then restoring earlier
    // write position).
    write_next: usize,
}

impl Checkpoint {
    #[inline(always)]
    pub fn get_written_range_since(&self, amount: usize) -> ProcessorRange {
        ProcessorRange {
            start: self.write_next,
            end: self.write_next + amount,
        }
    }

    #[inline(always)]
    pub fn new(proc: &Processor) -> Checkpoint {
        Checkpoint {
            write_next: proc.write_next,
        }
    }

    #[inline(always)]
    pub fn last_written(&self, proc: &mut Processor) -> Option<u8> {
        if proc.write_next <= self.write_next {
            None
        } else {
            Some(proc.code[proc.write_next - 1])
        }
    }

    /// Discard characters written since checkpoint but keep source position.
    #[inline(always)]
    pub fn erase_written(&self, proc: &mut Processor) -> () {
        proc.write_next = self.write_next;
    }

    /// Get written characters since checkpoint as range.
    #[inline(always)]
    pub fn written_range(&self, proc: &mut Processor) -> ProcessorRange {
        ProcessorRange { start: self.write_next, end: proc.write_next }
    }

    /// Get amount of output characters written since self.
    #[inline(always)]
    pub fn written_count(&self, proc: &mut Processor) -> usize {
        proc.write_next - self.write_next
    }
}