1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
use std::{fmt, hash, mem::ManuallyDrop, ptr};
use crate::{green::SyntaxKind, interning::Resolver, TextSize};
use lasso::Spur;
use triomphe::Arc;
#[repr(align(2))]
#[derive(Debug, PartialEq, Eq, Hash, Copy, Clone)]
pub(super) struct GreenTokenData {
pub(super) kind: SyntaxKind,
pub(super) text: Spur,
pub(super) text_len: TextSize,
}
pub struct GreenToken {
ptr: ptr::NonNull<GreenTokenData>,
}
unsafe impl Send for GreenToken {}
unsafe impl Sync for GreenToken {}
impl GreenToken {
fn add_tag(ptr: ptr::NonNull<GreenTokenData>) -> ptr::NonNull<GreenTokenData> {
unsafe {
let ptr = ((ptr.as_ptr() as usize) | 1) as *mut GreenTokenData;
ptr::NonNull::new_unchecked(ptr)
}
}
fn remove_tag(ptr: ptr::NonNull<GreenTokenData>) -> ptr::NonNull<GreenTokenData> {
unsafe {
let ptr = ((ptr.as_ptr() as usize) & !1) as *mut GreenTokenData;
ptr::NonNull::new_unchecked(ptr)
}
}
fn data(&self) -> &GreenTokenData {
unsafe { &*Self::remove_tag(self.ptr).as_ptr() }
}
#[inline]
pub(super) fn new(data: GreenTokenData) -> GreenToken {
let ptr = Arc::into_raw(Arc::new(data));
let ptr = ptr::NonNull::new(ptr as *mut _).unwrap();
GreenToken {
ptr: Self::add_tag(ptr),
}
}
#[inline]
pub fn kind(&self) -> SyntaxKind {
self.data().kind
}
#[inline]
pub fn text<'i, I>(&self, resolver: &'i I) -> &'i str
where
I: Resolver + ?Sized,
{
resolver.resolve(&self.data().text)
}
#[inline]
pub fn text_len(&self) -> TextSize {
self.data().text_len
}
}
impl fmt::Debug for GreenToken {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let data = self.data();
f.debug_struct("GreenToken")
.field("kind", &data.kind)
.field("text", &data.text)
.finish()
}
}
impl Clone for GreenToken {
fn clone(&self) -> Self {
let ptr = Self::remove_tag(self.ptr);
let ptr = unsafe {
let arc = ManuallyDrop::new(Arc::from_raw(ptr.as_ptr()));
Arc::into_raw(Arc::clone(&arc))
};
let ptr = unsafe { ptr::NonNull::new_unchecked(ptr as *mut _) };
GreenToken {
ptr: Self::add_tag(ptr),
}
}
}
impl Eq for GreenToken {}
impl PartialEq for GreenToken {
fn eq(&self, other: &Self) -> bool {
self.data() == other.data()
}
}
impl hash::Hash for GreenToken {
fn hash<H>(&self, state: &mut H)
where
H: hash::Hasher,
{
self.data().hash(state)
}
}
impl Drop for GreenToken {
fn drop(&mut self) {
unsafe {
Arc::from_raw(Self::remove_tag(self.ptr).as_ptr());
}
}
}