ruby_prism/

lib.rs

//! # ruby-prism
//!
//! Rustified version of Ruby's prism parser.
//!
#![warn(clippy::all, clippy::nursery, clippy::pedantic, future_incompatible, missing_docs, nonstandard_style, rust_2018_idioms, trivial_casts, trivial_numeric_casts, unreachable_pub, unused_qualifications)]

// Most of the code in this file is generated, so sometimes it generates code
// that doesn't follow the clippy rules. We don't want to see those warnings.
#[allow(clippy::too_many_lines, clippy::use_self)]
mod bindings {
    use std::ptr::NonNull;

    // In `build.rs`, we generate bindings based on the config.yml file. Here is
    // where we pull in those bindings and make them part of our library.
    include!(concat!(env!("OUT_DIR"), "/bindings.rs"));
}

mod node;
mod node_ext;
mod parse_result;

use std::ffi::CString;
use std::ptr::NonNull;

pub use self::bindings::*;
pub use self::node::{ConstantId, ConstantList, ConstantListIter, Integer, NodeList, NodeListIter};
pub use self::node_ext::{ConstantPathError, FullName};
pub use self::parse_result::{Comment, CommentType, Comments, Diagnostic, Diagnostics, Location, MagicComment, MagicComments, ParseResult};

use ruby_prism_sys::{
    pm_arena_new, pm_options_command_line_set, pm_options_encoding_locked_set, pm_options_encoding_set, pm_options_filepath_set, pm_options_free, pm_options_frozen_string_literal_set, pm_options_line_set, pm_options_main_script_set, pm_options_new, pm_options_partial_script_set,
    pm_options_scope_forwarding_set, pm_options_scope_init, pm_options_scope_local_mut, pm_options_scope_mut, pm_options_scopes_init, pm_options_t, pm_options_version_set, pm_parse, pm_parser_new, pm_string_constant_init,
};

/// The version of Ruby syntax to parse with.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Version {
    /// Use the latest version of prism.
    Latest,
    /// The vendored version of prism in `CRuby` 3.3.x.
    CRuby3_3,
    /// The vendored version of prism in `CRuby` 3.4.x.
    CRuby3_4,
    /// The vendored version of prism in `CRuby` 3.5.x / 4.0.x.
    CRuby3_5,
    /// The vendored version of prism in `CRuby` 4.1.x.
    CRuby4_1,
}

impl Version {
    /// Calls `pm_options_version_set` with the appropriate version string.
    /// `Latest` passes `NULL` to get the default behavior.
    unsafe fn set_on(self, opts: *mut pm_options_t) {
        match self {
            Self::Latest => {
                pm_options_version_set(opts, std::ptr::null(), 0);
            },
            Self::CRuby3_3 => {
                pm_options_version_set(opts, c"3.3".as_ptr(), 3);
            },
            Self::CRuby3_4 => {
                pm_options_version_set(opts, c"3.4".as_ptr(), 3);
            },
            Self::CRuby3_5 => {
                pm_options_version_set(opts, c"3.5".as_ptr(), 3);
            },
            Self::CRuby4_1 => {
                pm_options_version_set(opts, c"4.1".as_ptr(), 3);
            },
        }
    }
}

/// A command line option that affects parsing behavior.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum CommandLineFlag {
    /// `-a`: splits the input line `$_` into `$F`.
    A,
    /// `-e`: specifies a script to be executed.
    E,
    /// `-l`: chomps the input line by default.
    L,
    /// `-n`: wraps the script in a `while gets` loop.
    N,
    /// `-p`: prints the value of `$_` at the end of each loop.
    P,
    /// `-x`: searches the input file for a shebang.
    X,
}

impl From<CommandLineFlag> for u8 {
    fn from(flag: CommandLineFlag) -> Self {
        match flag {
            CommandLineFlag::A => ruby_prism_sys::PM_OPTIONS_COMMAND_LINE_A,
            CommandLineFlag::E => ruby_prism_sys::PM_OPTIONS_COMMAND_LINE_E,
            CommandLineFlag::L => ruby_prism_sys::PM_OPTIONS_COMMAND_LINE_L,
            CommandLineFlag::N => ruby_prism_sys::PM_OPTIONS_COMMAND_LINE_N,
            CommandLineFlag::P => ruby_prism_sys::PM_OPTIONS_COMMAND_LINE_P,
            CommandLineFlag::X => ruby_prism_sys::PM_OPTIONS_COMMAND_LINE_X,
        }
    }
}

/// A forwarding parameter for a scope.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum ScopeForwardingFlag {
    /// Forwarding with the `*` parameter.
    Positionals,
    /// Forwarding with the `**` parameter.
    Keywords,
    /// Forwarding with the `&` parameter.
    Block,
    /// Forwarding with the `...` parameter.
    All,
}

impl From<ScopeForwardingFlag> for u8 {
    fn from(flag: ScopeForwardingFlag) -> Self {
        match flag {
            ScopeForwardingFlag::Positionals => ruby_prism_sys::PM_OPTIONS_SCOPE_FORWARDING_POSITIONALS,
            ScopeForwardingFlag::Keywords => ruby_prism_sys::PM_OPTIONS_SCOPE_FORWARDING_KEYWORDS,
            ScopeForwardingFlag::Block => ruby_prism_sys::PM_OPTIONS_SCOPE_FORWARDING_BLOCK,
            ScopeForwardingFlag::All => ruby_prism_sys::PM_OPTIONS_SCOPE_FORWARDING_ALL,
        }
    }
}

/// A scope of locals surrounding the code that is being parsed.
#[derive(Debug, Clone, Default)]
pub struct Scope {
    locals: Vec<Vec<u8>>,
    forwarding: Vec<ScopeForwardingFlag>,
}

impl Scope {
    /// Sets the local variable names in this scope.
    #[must_use]
    pub fn locals(mut self, locals: Vec<Vec<u8>>) -> Self {
        self.locals = locals;
        self
    }

    /// Sets the forwarding parameters in this scope.
    #[must_use]
    pub fn forwarding(mut self, forwarding: Vec<ScopeForwardingFlag>) -> Self {
        self.forwarding = forwarding;
        self
    }
}

/// Options that can be passed to the parser.
#[derive(Debug, Clone, Default)]
pub struct Options {
    filepath: Option<String>,
    line: Option<i32>,
    encoding: Option<String>,
    encoding_locked: bool,
    frozen_string_literal: Option<bool>,
    command_line: Vec<CommandLineFlag>,
    version: Option<Version>,
    main_script: bool,
    partial_script: bool,
    scopes: Vec<Scope>,
}

impl Options {
    /// Sets the filepath option.
    #[must_use]
    pub fn filepath(mut self, filepath: &str) -> Self {
        self.filepath = Some(filepath.to_string());
        self
    }

    /// Sets the line option.
    #[must_use]
    pub const fn line(mut self, line: i32) -> Self {
        self.line = Some(line);
        self
    }

    /// Sets the encoding option.
    #[must_use]
    pub fn encoding(mut self, encoding: &str) -> Self {
        self.encoding = Some(encoding.to_string());
        self
    }

    /// Sets the encoding locked option.
    #[must_use]
    pub const fn encoding_locked(mut self, locked: bool) -> Self {
        self.encoding_locked = locked;
        self
    }

    /// Sets the frozen string literal option. `Some(true)` freezes string
    /// literals, `Some(false)` keeps them mutable, and `None` leaves the
    /// option unset.
    #[must_use]
    pub const fn frozen_string_literal(mut self, frozen: Option<bool>) -> Self {
        self.frozen_string_literal = frozen;
        self
    }

    /// Sets the command line flags.
    #[must_use]
    pub fn command_line(mut self, command_line: Vec<CommandLineFlag>) -> Self {
        self.command_line = command_line;
        self
    }

    /// Sets the version option.
    #[must_use]
    pub const fn version(mut self, version: Version) -> Self {
        self.version = Some(version);
        self
    }

    /// Sets the main script option.
    #[must_use]
    pub const fn main_script(mut self, main_script: bool) -> Self {
        self.main_script = main_script;
        self
    }

    /// Sets the partial script option.
    #[must_use]
    pub const fn partial_script(mut self, partial_script: bool) -> Self {
        self.partial_script = partial_script;
        self
    }

    /// Adds a scope to the options.
    #[must_use]
    pub fn scope(mut self, scope: Scope) -> Self {
        self.scopes.push(scope);
        self
    }

    /// Sets the scopes, replacing any previously added scopes.
    #[must_use]
    pub fn scopes(mut self, scopes: Vec<Scope>) -> Self {
        self.scopes = scopes;
        self
    }

    /// Builds the C-level parse options from these options. The returned
    /// `ParseOptions` must outlive any `ParseResult` created from it.
    ///
    /// # Panics
    ///
    /// Panics if `filepath` or `encoding` contain interior null bytes.
    #[must_use]
    pub fn build(self) -> ParseOptions {
        let opts = unsafe { pm_options_new() };

        let c_filepath = self.filepath.map(|filepath| {
            let cstring = CString::new(filepath).unwrap();
            unsafe { pm_options_filepath_set(opts, cstring.as_ptr()) };
            cstring
        });

        if let Some(line) = self.line {
            unsafe { pm_options_line_set(opts, line) };
        }

        let c_encoding = self.encoding.map(|encoding| {
            let cstring = CString::new(encoding).unwrap();
            unsafe { pm_options_encoding_set(opts, cstring.as_ptr()) };
            cstring
        });

        if self.encoding_locked {
            unsafe { pm_options_encoding_locked_set(opts, true) };
        }

        if let Some(frozen) = self.frozen_string_literal {
            unsafe { pm_options_frozen_string_literal_set(opts, frozen) };
        }

        let command_line = self.command_line.iter().fold(0u8, |acc, &flag| acc | u8::from(flag));
        if command_line != 0 {
            unsafe { pm_options_command_line_set(opts, command_line) };
        }

        if let Some(version) = self.version {
            unsafe { version.set_on(opts) };
        }

        if self.main_script {
            unsafe { pm_options_main_script_set(opts, true) };
        }

        if self.partial_script {
            unsafe { pm_options_partial_script_set(opts, true) };
        }

        if !self.scopes.is_empty() {
            unsafe { pm_options_scopes_init(opts, self.scopes.len()) };

            for (scope_index, scope) in self.scopes.iter().enumerate() {
                let pm_scope = unsafe { pm_options_scope_mut(opts, scope_index) };
                unsafe { pm_options_scope_init(pm_scope, scope.locals.len()) };

                for (local_index, local) in scope.locals.iter().enumerate() {
                    let pm_local = unsafe { pm_options_scope_local_mut(pm_scope, local_index) };
                    unsafe { pm_string_constant_init(pm_local, local.as_ptr().cast::<i8>(), local.len()) };
                }

                let forwarding = scope.forwarding.iter().fold(0u8, |acc, &flag| acc | u8::from(flag));
                if forwarding != 0 {
                    unsafe { pm_options_scope_forwarding_set(pm_scope, forwarding) };
                }
            }
        }

        ParseOptions {
            options: opts,
            _filepath: c_filepath,
            _encoding: c_encoding,
            _scopes: self.scopes,
        }
    }
}

/// The C-level parse options. Created from [`Options::build`]. Must outlive
/// any [`ParseResult`] created with [`parse_with_options`].
pub struct ParseOptions {
    options: *mut pm_options_t,
    // These CStrings back the constant pm_string_t values inside `options`.
    // They must not be dropped before `options` is freed.
    _filepath: Option<CString>,
    _encoding: Option<CString>,
    // The scopes data that pm_string_t values point into.
    _scopes: Vec<Scope>,
}

impl Drop for ParseOptions {
    fn drop(&mut self) {
        unsafe { pm_options_free(self.options) };
    }
}

/// Initializes a parser, parses the source, and returns the result.
///
/// # Safety
///
/// `options` must be a valid pointer to a `pm_options_t` or null.
unsafe fn parse_impl(source: &[u8], options: *const pm_options_t) -> ParseResult<'_> {
    let arena = pm_arena_new();
    let parser = pm_parser_new(arena, source.as_ptr(), source.len(), options);
    let node = NonNull::new_unchecked(pm_parse(parser));
    ParseResult::new(source, arena, parser, node)
}

/// Parses the given source string and returns a parse result.
///
/// # Panics
///
/// Panics if the parser fails to initialize.
///
#[must_use]
pub fn parse(source: &[u8]) -> ParseResult<'_> {
    unsafe { parse_impl(source, std::ptr::null()) }
}

/// Parses the given source string with the given options and returns a parse
/// result. The `options` must outlive the returned `ParseResult`.
///
/// # Panics
///
/// Panics if the parser fails to initialize.
///
#[must_use]
pub fn parse_with_options<'a>(source: &'a [u8], options: &'a ParseOptions) -> ParseResult<'a> {
    unsafe { parse_impl(source, options.options) }
}

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

    #[test]
    fn comments_test() {
        let source = "# comment 1\n# comment 2\n# comment 3\n";
        let result = parse(source.as_ref());

        for comment in result.comments() {
            assert_eq!(super::CommentType::InlineComment, comment.type_());
            let text = std::str::from_utf8(comment.text()).unwrap();
            assert!(text.starts_with("# comment"));
        }
    }

    #[test]
    fn line_offsets_test() {
        let source = "";
        let result = parse(source.as_ref());

        let expected: [u32; 1] = [0];
        assert_eq!(expected, result.line_offsets());

        let source = "1 + 1";
        let result = parse(source.as_ref());

        let expected: [u32; 1] = [0];
        assert_eq!(expected, result.line_offsets());

        let source = "begin\n1 + 1\n2 + 2\nend";
        let result = parse(source.as_ref());

        let expected: [u32; 4] = [0, 6, 12, 18];
        assert_eq!(expected, result.line_offsets());
    }

    #[test]
    fn magic_comments_test() {
        use crate::MagicComment;

        let source = "# typed: ignore\n# typed:true\n#typed: strict\n";
        let result = parse(source.as_ref());

        let comments: Vec<MagicComment<'_>> = result.magic_comments().collect();
        assert_eq!(3, comments.len());

        assert_eq!(b"typed", comments[0].key());
        assert_eq!(b"ignore", comments[0].value());

        assert_eq!(b"typed", comments[1].key());
        assert_eq!(b"true", comments[1].value());

        assert_eq!(b"typed", comments[2].key());
        assert_eq!(b"strict", comments[2].value());
    }

    #[test]
    fn data_loc_test() {
        let source = "1";
        let result = parse(source.as_ref());
        let data_loc = result.data_loc();
        assert!(data_loc.is_none());

        let source = "__END__\nabc\n";
        let result = parse(source.as_ref());
        let data_loc = result.data_loc().unwrap();
        let slice = std::str::from_utf8(result.as_slice(&data_loc)).unwrap();
        assert_eq!(slice, "__END__\nabc\n");

        let source = "1\n2\n3\n__END__\nabc\ndef\n";
        let result = parse(source.as_ref());
        let data_loc = result.data_loc().unwrap();
        let slice = std::str::from_utf8(result.as_slice(&data_loc)).unwrap();
        assert_eq!(slice, "__END__\nabc\ndef\n");
    }

    #[test]
    fn location_slice_test() {
        let source = "111 + 222 + 333";
        let result = parse(source.as_ref());

        let node = result.node();
        let node = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let node = node.as_call_node().unwrap().receiver().unwrap();
        let plus = node.as_call_node().unwrap();
        let node = plus.arguments().unwrap().arguments().iter().next().unwrap();

        let location = node.as_integer_node().unwrap().location();
        let slice = std::str::from_utf8(result.as_slice(&location)).unwrap();

        assert_eq!(slice, "222");
        assert_eq!(6, location.start);
        assert_eq!(location.start, location.start());
        assert_eq!(9, location.end());

        let recv_loc = plus.receiver().unwrap().location();
        assert_eq!(recv_loc.as_slice(), b"111");
        assert_eq!(0, recv_loc.start);
        assert_eq!(3, recv_loc.end());

        let joined = recv_loc.join(&location).unwrap();
        assert_eq!(joined.as_slice(), b"111 + 222");

        let not_joined = location.join(&recv_loc);
        assert!(not_joined.is_none());

        {
            let result = parse(source.as_ref());
            let node = result.node();
            let node = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
            let node = node.as_call_node().unwrap().receiver().unwrap();
            let plus = node.as_call_node().unwrap();
            let node = plus.arguments().unwrap().arguments().iter().next().unwrap();

            let location = node.as_integer_node().unwrap().location();
            let not_joined = recv_loc.join(&location);
            assert!(not_joined.is_none());

            let not_joined = location.join(&recv_loc);
            assert!(not_joined.is_none());
        }

        let location = node.location();
        let slice = std::str::from_utf8(result.as_slice(&location)).unwrap();

        assert_eq!(slice, "222");

        let slice = std::str::from_utf8(location.as_slice()).unwrap();

        assert_eq!(slice, "222");
    }

    #[test]
    fn location_line_column_test() {
        let source = "first\nsecond\nthird";
        let result = parse(source.as_ref());

        let node = result.node();
        let program = node.as_program_node().unwrap();
        let statements = program.statements().body();
        let mut iter = statements.iter();

        let _first = iter.next().unwrap();
        let second = iter.next().unwrap();
        let third = iter.next().unwrap();

        let second_loc = second.location();
        assert_eq!(second_loc.start_line(), 2);
        assert_eq!(second_loc.end_line(), 2);
        assert_eq!(second_loc.start_column(), 0);
        assert_eq!(second_loc.end_column(), 6);

        let third_loc = third.location();
        assert_eq!(third_loc.start_line(), 3);
        assert_eq!(third_loc.end_line(), 3);
        assert_eq!(third_loc.start_column(), 0);
        assert_eq!(third_loc.end_column(), 5);
    }

    #[test]
    fn location_chop_test() {
        let result = parse(b"foo");
        let mut location = result.node().as_program_node().unwrap().location();

        assert_eq!(location.chop().as_slice(), b"fo");
        assert_eq!(location.chop().chop().chop().as_slice(), b"");

        // Check that we don't go negative.
        for _ in 0..10 {
            location = location.chop();
        }
        assert_eq!(location.as_slice(), b"");
    }

    #[test]
    fn visitor_test() {
        use super::{visit_interpolated_regular_expression_node, visit_regular_expression_node, InterpolatedRegularExpressionNode, RegularExpressionNode, Visit};

        struct RegularExpressionVisitor {
            count: usize,
        }

        impl Visit<'_> for RegularExpressionVisitor {
            fn visit_interpolated_regular_expression_node(&mut self, node: &InterpolatedRegularExpressionNode<'_>) {
                self.count += 1;
                visit_interpolated_regular_expression_node(self, node);
            }

            fn visit_regular_expression_node(&mut self, node: &RegularExpressionNode<'_>) {
                self.count += 1;
                visit_regular_expression_node(self, node);
            }
        }

        let source = "# comment 1\n# comment 2\nmodule Foo; class Bar; /abc #{/def/}/; end; end";
        let result = parse(source.as_ref());

        let mut visitor = RegularExpressionVisitor { count: 0 };
        visitor.visit(&result.node());

        assert_eq!(visitor.count, 2);
    }

    #[test]
    fn node_upcast_test() {
        use super::Node;

        let source = "module Foo; end";
        let result = parse(source.as_ref());

        let node = result.node();
        let upcast_node = node.as_program_node().unwrap().as_node();
        assert!(matches!(upcast_node, Node::ProgramNode { .. }));

        let node = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let upcast_node = node.as_module_node().unwrap().as_node();
        assert!(matches!(upcast_node, Node::ModuleNode { .. }));
    }

    #[test]
    fn constant_id_test() {
        let source = "module Foo; x = 1; y = 2; end";
        let result = parse(source.as_ref());

        let node = result.node();
        assert_eq!(node.as_program_node().unwrap().statements().body().len(), 1);
        assert!(!node.as_program_node().unwrap().statements().body().is_empty());
        let module = node.as_program_node().and_then(|pn| pn.statements().body().first()).unwrap();
        let module = module.as_module_node().unwrap();

        assert_eq!(module.locals().len(), 2);
        assert!(!module.locals().is_empty());

        assert_eq!(module.locals().first().unwrap().as_slice(), b"x");
        assert_eq!(module.locals().last().unwrap().as_slice(), b"y");

        let source = "module Foo; end";
        let result = parse(source.as_ref());

        let node = result.node();
        assert_eq!(node.as_program_node().unwrap().statements().body().len(), 1);
        assert!(!node.as_program_node().unwrap().statements().body().is_empty());
        let module = node.as_program_node().and_then(|pn| pn.statements().body().first()).unwrap();
        let module = module.as_module_node().unwrap();

        assert_eq!(module.locals().len(), 0);
        assert!(module.locals().is_empty());
    }

    #[test]
    fn optional_loc_test() {
        let source = r"
module Example
  x = call_func(3, 4)
  y = x.call_func 5, 6
end
";
        let result = parse(source.as_ref());

        let node = result.node();
        let module = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let module = module.as_module_node().unwrap();
        let body = module.body();
        let writes = body.iter().next().unwrap().as_statements_node().unwrap().body().iter().collect::<Vec<_>>();
        assert_eq!(writes.len(), 2);

        let asgn = &writes[0];
        let call = asgn.as_local_variable_write_node().unwrap().value();
        let call = call.as_call_node().unwrap();

        let call_operator_loc = call.call_operator_loc();
        assert!(call_operator_loc.is_none());
        let closing_loc = call.closing_loc();
        assert!(closing_loc.is_some());

        let asgn = &writes[1];
        let call = asgn.as_local_variable_write_node().unwrap().value();
        let call = call.as_call_node().unwrap();

        let call_operator_loc = call.call_operator_loc();
        assert!(call_operator_loc.is_some());
        let closing_loc = call.closing_loc();
        assert!(closing_loc.is_none());
    }

    #[test]
    fn frozen_strings_test() {
        let source = r#"
# frozen_string_literal: true
"foo"
"#;
        let result = parse(source.as_ref());
        assert!(result.frozen_string_literals());

        let source = "3";
        let result = parse(source.as_ref());
        assert!(!result.frozen_string_literals());
    }

    #[test]
    fn string_flags_test() {
        let source = r#"
# frozen_string_literal: true
"foo"
"#;
        let result = parse(source.as_ref());

        let node = result.node();
        let string = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let string = string.as_string_node().unwrap();
        assert!(string.is_frozen());

        let source = r#"
"foo"
"#;
        let result = parse(source.as_ref());

        let node = result.node();
        let string = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let string = string.as_string_node().unwrap();
        assert!(!string.is_frozen());
    }

    #[test]
    fn call_flags_test() {
        let source = r"
x
";
        let result = parse(source.as_ref());

        let node = result.node();
        let call = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let call = call.as_call_node().unwrap();
        assert!(call.is_variable_call());

        let source = r"
x&.foo
";
        let result = parse(source.as_ref());

        let node = result.node();
        let call = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let call = call.as_call_node().unwrap();
        assert!(call.is_safe_navigation());
    }

    #[test]
    fn integer_flags_test() {
        let source = r"
0b1
";
        let result = parse(source.as_ref());

        let node = result.node();
        let i = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let i = i.as_integer_node().unwrap();
        assert!(i.is_binary());
        assert!(!i.is_decimal());
        assert!(!i.is_octal());
        assert!(!i.is_hexadecimal());

        let source = r"
1
";
        let result = parse(source.as_ref());

        let node = result.node();
        let i = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let i = i.as_integer_node().unwrap();
        assert!(!i.is_binary());
        assert!(i.is_decimal());
        assert!(!i.is_octal());
        assert!(!i.is_hexadecimal());

        let source = r"
0o1
";
        let result = parse(source.as_ref());

        let node = result.node();
        let i = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let i = i.as_integer_node().unwrap();
        assert!(!i.is_binary());
        assert!(!i.is_decimal());
        assert!(i.is_octal());
        assert!(!i.is_hexadecimal());

        let source = r"
0x1
";
        let result = parse(source.as_ref());

        let node = result.node();
        let i = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let i = i.as_integer_node().unwrap();
        assert!(!i.is_binary());
        assert!(!i.is_decimal());
        assert!(!i.is_octal());
        assert!(i.is_hexadecimal());
    }

    #[test]
    fn range_flags_test() {
        let source = r"
0..1
";
        let result = parse(source.as_ref());

        let node = result.node();
        let range = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let range = range.as_range_node().unwrap();
        assert!(!range.is_exclude_end());

        let source = r"
0...1
";
        let result = parse(source.as_ref());

        let node = result.node();
        let range = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let range = range.as_range_node().unwrap();
        assert!(range.is_exclude_end());
    }

    #[allow(clippy::too_many_lines, clippy::cognitive_complexity)]
    #[test]
    fn regex_flags_test() {
        let source = r"
/a/i
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(regex.is_ignore_case());
        assert!(!regex.is_extended());
        assert!(!regex.is_multi_line());
        assert!(!regex.is_euc_jp());
        assert!(!regex.is_ascii_8bit());
        assert!(!regex.is_windows_31j());
        assert!(!regex.is_utf_8());
        assert!(!regex.is_once());

        let source = r"
/a/x
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(!regex.is_ignore_case());
        assert!(regex.is_extended());
        assert!(!regex.is_multi_line());
        assert!(!regex.is_euc_jp());
        assert!(!regex.is_ascii_8bit());
        assert!(!regex.is_windows_31j());
        assert!(!regex.is_utf_8());
        assert!(!regex.is_once());

        let source = r"
/a/m
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(!regex.is_ignore_case());
        assert!(!regex.is_extended());
        assert!(regex.is_multi_line());
        assert!(!regex.is_euc_jp());
        assert!(!regex.is_ascii_8bit());
        assert!(!regex.is_windows_31j());
        assert!(!regex.is_utf_8());
        assert!(!regex.is_once());

        let source = r"
/a/e
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(!regex.is_ignore_case());
        assert!(!regex.is_extended());
        assert!(!regex.is_multi_line());
        assert!(regex.is_euc_jp());
        assert!(!regex.is_ascii_8bit());
        assert!(!regex.is_windows_31j());
        assert!(!regex.is_utf_8());
        assert!(!regex.is_once());

        let source = r"
/a/n
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(!regex.is_ignore_case());
        assert!(!regex.is_extended());
        assert!(!regex.is_multi_line());
        assert!(!regex.is_euc_jp());
        assert!(regex.is_ascii_8bit());
        assert!(!regex.is_windows_31j());
        assert!(!regex.is_utf_8());
        assert!(!regex.is_once());

        let source = r"
/a/s
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(!regex.is_ignore_case());
        assert!(!regex.is_extended());
        assert!(!regex.is_multi_line());
        assert!(!regex.is_euc_jp());
        assert!(!regex.is_ascii_8bit());
        assert!(regex.is_windows_31j());
        assert!(!regex.is_utf_8());
        assert!(!regex.is_once());

        let source = r"
/a/u
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(!regex.is_ignore_case());
        assert!(!regex.is_extended());
        assert!(!regex.is_multi_line());
        assert!(!regex.is_euc_jp());
        assert!(!regex.is_ascii_8bit());
        assert!(!regex.is_windows_31j());
        assert!(regex.is_utf_8());
        assert!(!regex.is_once());

        let source = r"
/a/o
";
        let result = parse(source.as_ref());

        let node = result.node();
        let regex = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let regex = regex.as_regular_expression_node().unwrap();
        assert!(!regex.is_ignore_case());
        assert!(!regex.is_extended());
        assert!(!regex.is_multi_line());
        assert!(!regex.is_euc_jp());
        assert!(!regex.is_ascii_8bit());
        assert!(!regex.is_windows_31j());
        assert!(!regex.is_utf_8());
        assert!(regex.is_once());
    }

    #[test]
    fn visitor_traversal_test() {
        use crate::{Node, Visit};

        #[derive(Default)]
        struct NodeCounts {
            pre_parent: usize,
            post_parent: usize,
            pre_leaf: usize,
            post_leaf: usize,
        }

        #[derive(Default)]
        struct CountingVisitor {
            counts: NodeCounts,
        }

        impl Visit<'_> for CountingVisitor {
            fn visit_branch_node_enter(&mut self, _node: Node<'_>) {
                self.counts.pre_parent += 1;
            }

            fn visit_branch_node_leave(&mut self) {
                self.counts.post_parent += 1;
            }

            fn visit_leaf_node_enter(&mut self, _node: Node<'_>) {
                self.counts.pre_leaf += 1;
            }

            fn visit_leaf_node_leave(&mut self) {
                self.counts.post_leaf += 1;
            }
        }

        let source = r"
module Example
  x = call_func(3, 4)
  y = x.call_func 5, 6
end
";
        let result = parse(source.as_ref());
        let node = result.node();
        let mut visitor = CountingVisitor::default();
        visitor.visit(&node);

        assert_eq!(7, visitor.counts.pre_parent);
        assert_eq!(7, visitor.counts.post_parent);
        assert_eq!(6, visitor.counts.pre_leaf);
        assert_eq!(6, visitor.counts.post_leaf);
    }

    #[test]
    fn visitor_lifetime_test() {
        use crate::{Node, Visit};

        #[derive(Default)]
        struct StackingNodeVisitor<'a> {
            stack: Vec<Node<'a>>,
            max_depth: usize,
        }

        impl<'pr> Visit<'pr> for StackingNodeVisitor<'pr> {
            fn visit_branch_node_enter(&mut self, node: Node<'pr>) {
                self.stack.push(node);
            }

            fn visit_branch_node_leave(&mut self) {
                self.stack.pop();
            }

            fn visit_leaf_node_leave(&mut self) {
                self.max_depth = self.max_depth.max(self.stack.len());
            }
        }

        let source = r"
module Example
  x = call_func(3, 4)
  y = x.call_func 5, 6
end
";
        let result = parse(source.as_ref());
        let node = result.node();
        let mut visitor = StackingNodeVisitor::default();
        visitor.visit(&node);

        assert_eq!(0, visitor.stack.len());
        assert_eq!(5, visitor.max_depth);
    }

    #[test]
    fn integer_value_test() {
        let result = parse("0xA".as_ref());
        let integer = result.node().as_program_node().unwrap().statements().body().iter().next().unwrap().as_integer_node().unwrap().value();
        let value: i32 = integer.try_into().unwrap();

        assert_eq!(value, 10);
    }

    #[test]
    fn integer_small_value_to_u32_digits_test() {
        let result = parse("0xA".as_ref());
        let integer = result.node().as_program_node().unwrap().statements().body().iter().next().unwrap().as_integer_node().unwrap().value();
        let (negative, digits) = integer.to_u32_digits();

        assert!(!negative);
        assert_eq!(digits, &[10]);
    }

    #[test]
    fn integer_large_value_to_u32_digits_test() {
        let result = parse("0x7FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF".as_ref());
        let integer = result.node().as_program_node().unwrap().statements().body().iter().next().unwrap().as_integer_node().unwrap().value();
        let (negative, digits) = integer.to_u32_digits();

        assert!(!negative);
        assert_eq!(digits, &[4_294_967_295, 4_294_967_295, 4_294_967_295, 2_147_483_647]);
    }

    #[test]
    fn float_value_test() {
        let result = parse("1.0".as_ref());
        let value: f64 = result.node().as_program_node().unwrap().statements().body().iter().next().unwrap().as_float_node().unwrap().value();

        assert!((value - 1.0).abs() < f64::EPSILON);
    }

    #[test]
    fn regex_value_test() {
        let result = parse(b"//");
        let node = result.node().as_program_node().unwrap().statements().body().iter().next().unwrap().as_regular_expression_node().unwrap();
        assert_eq!(node.unescaped(), b"");
    }

    #[test]
    fn node_field_lifetime_test() {
        // The code below wouldn't typecheck prior to https://github.com/ruby/prism/pull/2519,
        // but we need to stop clippy from complaining about it.
        #![allow(clippy::needless_pass_by_value)]

        use crate::Node;

        #[derive(Default)]
        struct Extract<'pr> {
            scopes: Vec<crate::ConstantId<'pr>>,
        }

        impl<'pr> Extract<'pr> {
            fn push_scope(&mut self, path: Node<'pr>) {
                if let Some(cread) = path.as_constant_read_node() {
                    self.scopes.push(cread.name());
                } else if let Some(cpath) = path.as_constant_path_node() {
                    if let Some(parent) = cpath.parent() {
                        self.push_scope(parent);
                    }
                    self.scopes.push(cpath.name().unwrap());
                } else {
                    panic!("Wrong node kind!");
                }
            }
        }

        let source = "Some::Random::Constant";
        let result = parse(source.as_ref());
        let node = result.node();
        let mut extractor = Extract::default();
        extractor.push_scope(node.as_program_node().unwrap().statements().body().iter().next().unwrap());
        assert_eq!(3, extractor.scopes.len());
    }

    #[test]
    fn parse_with_options_frozen_string_literal_test() {
        use super::{parse_with_options, Options};

        let source = b"\"foo\"";
        let options = Options::default().frozen_string_literal(Some(true)).build();
        let result = parse_with_options(source, &options);

        let node = result.node();
        let string = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        let string = string.as_string_node().unwrap();
        assert!(string.is_frozen());
    }

    #[test]
    fn parse_with_options_filepath_test() {
        use super::{parse_with_options, Options};

        let source = b"__FILE__";
        let options = Options::default().filepath("test.rb").build();
        let result = parse_with_options(source, &options);
        assert!(result.errors().next().is_none());
    }

    #[test]
    fn parse_with_options_line_test() {
        use super::{parse_with_options, Options};

        let source = b"1 + 2";
        let options = Options::default().line(10).build();
        let result = parse_with_options(source, &options);
        assert!(result.errors().next().is_none());
    }

    #[test]
    fn parse_with_options_scopes_test() {
        use super::{parse_with_options, Options, Scope};

        let source = b"x";
        let scope = Scope::default().locals(vec![b"x".to_vec()]);
        let options = Options::default().scope(scope).build();
        let result = parse_with_options(source, &options);
        assert!(result.errors().next().is_none());

        let node = result.node();
        let stmt = node.as_program_node().unwrap().statements().body().iter().next().unwrap();
        assert!(stmt.as_local_variable_read_node().is_some());
    }

    #[test]
    fn malformed_shebang() {
        let source = "#!\x00";
        let result = parse(source.as_ref());
        assert!(result.errors().next().is_none());
        assert!(result.warnings().next().is_none());
    }
}