use ruff_python_ast::{HasNodeIndex, NodeIndex};

use crate::ast_node_ref::AstNodeRef;

/// Compact key for a node for use in a hash map.
#[derive(Copy, Clone, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, get_size2::GetSize)]
pub struct NodeKey(NodeIndex);

impl NodeKey {
    pub fn from_node<N>(node: N) -> Self
    where
        N: HasNodeIndex,
    {
        NodeKey(node.node_index().load())
    }

    pub fn from_node_ref<T>(node_ref: &AstNodeRef<T>) -> Self {
        NodeKey(node_ref.index())
    }
}
