use std::mem::MaybeUninit;
use std::sync::LazyLock;
use Default::default;
use dsb::cell::Style;
use dsb::{Cell, F};
use fimg::Image;
use itertools::Itertools;
use lsp_types::*;
use crate::FG;
use crate::text::{color, color_, set_a};
#[derive(Debug)]
pub struct Complete {
pub r: CompletionResponse,
pub start: usize,
pub selection: usize,
pub vo: usize,
}
impl Complete {
pub fn next(&mut self, f: &str) {
let n = filter(self, f).count();
self.selection += 1;
if self.selection == n {
self.vo = 0;
self.selection = 0;
}
if self.selection >= self.vo + N {
self.vo += 1;
}
}
pub fn back(&mut self, f: &str) {
let n = filter(self, f).count();
if self.selection == 0 {
self.vo = n - N;
self.selection = n - 1;
} else {
self.selection -= 1;
if self.selection < self.vo {
self.vo -= 1;
}
}
}
}
fn score<'a>(
x: impl Iterator<Item = &'a CompletionItem>,
filter: &'_ str,
) -> impl Iterator<Item = (u32, &'a CompletionItem, Vec<u32>)> {
#[thread_local]
static mut MATCHER: LazyLock<nucleo::Matcher> =
LazyLock::new(|| nucleo::Matcher::new(nucleo::Config::DEFAULT));
let p = nucleo::pattern::Pattern::parse(
filter,
nucleo::pattern::CaseMatching::Smart,
nucleo::pattern::Normalization::Smart,
);
x.map(move |y| {
let mut utf32 = vec![];
let hay = y.filter_text.as_deref().unwrap_or(&y.label);
let mut indices = vec![];
let score = p
.indices(
nucleo::Utf32Str::new(hay, &mut utf32),
unsafe { &mut *MATCHER },
&mut indices,
)
.unwrap_or(0);
indices.sort_unstable();
indices.dedup();
(score, y, indices)
})
}
fn filter<'a>(
completion: &'a Complete,
filter: &'_ str,
) -> impl Iterator<Item = &'a CompletionItem> {
let x = &completion.r;
let y = match x {
CompletionResponse::Array(x) => x,
CompletionResponse::List(x) => &x.items,
};
y.iter().filter(|y| {
filter.is_empty()
|| y.filter_text
.as_deref()
.unwrap_or(&y.label)
.chars()
.any(|x| filter.chars().contains(&x))
// .collect::<HashSet<_>>()
// .intersection(&filter.chars().collect())
// .count()
// > 0
})
}
pub fn s(completion: &Complete, c: usize, f: &str) -> Vec<Cell> {
let mut out = vec![];
let i = score(filter(completion, f), f)
.sorted_by_key(|x| x.0)
.rev()
.zip(0..)
.skip(completion.vo)
.take(N);
// let Some((s, x)) = i.next() else {
// return vec![];
// };
// let mut q = Dq::<_, 13>::new((s, x));
// for (s, x) in i {
// if q.first().0 <= s {
// q.push_front((s, x));
// }
// }
// fuzzy_aho_corasick::FuzzyAhoCorasickBuilder::new()
// .fuzzy(
// FuzzyLimits::new()
// .insertions(20)
// .deletions(2)
// .edits(4)
// .substitutions(5)
// .swaps(3),
// .penalties(FuzzyPenalties {
// )
// insertion: 0.0,
// deletion: 1.0,
// substitution: 0.5,
// swap: 0.5,
// })
// .build(
// y.iter().map(|x| x.filter_text.as_deref().unwrap_or(&x.label)),
// )
// .search(filter, 0.25)
// .into_iter()
// .map(|x| &y[x.pattern_index])
// // .take(13);
// // for x in y
// // .iter()
// // .filter(|x| {
// // x.filter_text
// // .as_deref()
// // .unwrap_or(&x.label)
// // .starts_with(filter)
// // })
// .take(13)
i.for_each(|((_, x, indices), i)| {
r(x, c, i == completion.selection, &indices, &mut out)
});
out
}
fn charc(c: &str) -> usize {
c.chars().count()
}
#[implicit_fn::implicit_fn]
fn r(
x: &CompletionItem,
c: usize,
selected: bool,
indices: &[u32],
to: &mut Vec<Cell>,
) {
let bg = if selected { color(*b"262d3b") } else { color(*b"1c212b") };
const T_BG: [u8; 3] = color(*b"11141a");
let ds: Style = Style { bg: bg, color: FG, flags: 0 };
let d: Cell = Cell { letter: None, style: ds };
let mut b = vec![d; c];
const MAP: [([u8; 3], [u8; 3], &str); 26] = {
car::map!(
amap::amap! {
const { CompletionItemKind::TEXT.0 as usize } => ("#9a9b9a", " "),
const { CompletionItemKind::METHOD.0 as usize } | const { CompletionItemKind::FUNCTION.0 as usize } => ("#FFD173", "λ "),
const { CompletionItemKind::CONSTRUCTOR.0 as usize } => ("#FFAD66", "->"),
const { CompletionItemKind::FIELD.0 as usize } => ("#E06C75", "x."),
const { CompletionItemKind::VARIABLE.0 as usize } => ("#E06C75", "x "),
const { CompletionItemKind::MODULE.0 as usize } => ("#D5FF80", "::"),
const { CompletionItemKind::PROPERTY.0 as usize } => ("#e6e1cf", "x."),
const { CompletionItemKind::VALUE.0 as usize } => ("#DFBFFF", "4 "),
const { CompletionItemKind::ENUM.0 as usize } => ("#73b9ff", "u"),
const { CompletionItemKind::ENUM_MEMBER.0 as usize } => ("#73b9ff", ":"),
const { CompletionItemKind::SNIPPET.0 as usize } => ("#9a9b9a", "! "),
const { CompletionItemKind::INTERFACE.0 as usize } => ("#E5C07B", "t "),
const { CompletionItemKind::REFERENCE.0 as usize } => ("#9a9b9a", "& "),
const { CompletionItemKind::CONSTANT.0 as usize } => ("#DFBFFF", "N "),
const { CompletionItemKind::STRUCT.0 as usize } => ("#73D0FF", "X{"),
const { CompletionItemKind::OPERATOR.0 as usize } => ("#F29E74", "+ "),
const { CompletionItemKind::TYPE_PARAMETER.0 as usize } => ("#9a9b9a", "T "),
const { CompletionItemKind::KEYWORD.0 as usize } => ("#FFAD66", "as"),
_ => ("#9a9b9a", " ")
},
|(x, y)| (set_a(color_(x), 0.5), color_(x), y)
)
};
let (bgt, col, ty) =
MAP[x.kind.unwrap_or(CompletionItemKind(50)).0 as usize];
b.iter_mut().zip(ty.chars()).for_each(|(x, c)| {
*x = Style { bg: bgt, color: col, flags: Style::BOLD }.basic(c)
});
let i = &mut b[2..];
let left = i.len() as i32 - charc(&x.label) as i32 - 2;
if let Some(details) = &x.detail {
let details = if left < charc(details) as i32 {
details
.chars()
.take(left as _)
.chain(['…'])
.collect::<Vec<_>>()
.into_iter()
.rev()
.collect::<Vec<_>>()
.into_iter()
} else {
details.chars().rev().collect::<Vec<_>>().into_iter()
};
i.iter_mut()
.rev()
.zip(details.map(|x| {
Style { bg, color: [154, 155, 154], ..default() }.basic(x)
}))
.for_each(|(a, b)| *a = b);
}
i.iter_mut()
.zip(x.label.chars().map(|x| ds.basic(x)))
.zip(0..)
.for_each(|((a, b), i)| {
*a = b;
if indices.contains(&i) {
a.style |= (Style::BOLD, color(*b"ffcc66"));
}
});
to.extend(b);
}
pub const N: usize = 13;
#[test]
fn t() {
let ppem = 20.0;
let lh = 10.0;
let (w, h) = (611, 8000);
let (c, r) = dsb::fit(&crate::FONT, ppem, lh, (w, h));
dbg!(dsb::size(&crate::FONT, ppem, lh, (c, r)));
let y = serde_json::from_str(include_str!("../complete_")).unwrap();
let cells =
s(&Complete { r: y, start: 0, selection: 0, vo: 0 }, c, "");
dbg!(c, r);
dbg!(w, h);
let mut fonts = dsb::Fonts::new(
F::FontRef(*crate::FONT, &[(2003265652, 550.0)]),
F::instance(*crate::FONT, *crate::BFONT),
F::FontRef(*crate::IFONT, &[(2003265652, 550.0)]),
F::instance(*crate::IFONT, *crate::BIFONT),
);
let mut x = Image::build(w as u32, h as u32).fill(crate::hov::BG);
unsafe {
dsb::render(
&cells,
(c, r),
ppem,
&mut fonts,
lh,
true,
x.as_mut(),
(0, 0),
)
};
// println!("{:?}", now.elapsed());
x.as_ref().save("x");
}
pub struct Dq<T, const N: usize> {
arr: [MaybeUninit<T>; N],
front: u8,
len: u8,
}
impl<T: Copy, const N: usize> Dq<T, N> {
pub fn new(first: T) -> Self {
let mut dq = Dq {
arr: [const { MaybeUninit::uninit() }; N],
front: 0,
len: 1,
};
dq.arr[0].write(first);
dq
}
pub fn first(&mut self) -> T {
unsafe {
self.arr.get_unchecked(self.front as usize).assume_init()
}
}
pub fn push_front(&mut self, elem: T) {
// sub 1
match self.front {
0 => self.front = N as u8 - 1,
n => self.front = n - 1,
}
self.len += 1;
unsafe {
self.arr.get_unchecked_mut(self.front as usize).write(elem)
};
}
pub fn iter(&self) -> impl Iterator<Item = T> + '_ {
self.arr
.iter()
.cycle()
.skip(self.front as _)
.take((self.len as usize).min(N))
.map(|x| unsafe { x.assume_init() })
}
}