57 lines
2.1 KiB
Rust
57 lines
2.1 KiB
Rust
// TODO these functions are so shit
|
|
/// Finds the index of the first occurrence of `search_bytes` in `bytes`.
|
|
fn find_pattern(bytes: &[u8], search_bytes: &[u8]) -> Option<usize>
|
|
{
|
|
bytes.windows(search_bytes.len())
|
|
.position(|window| window == search_bytes)
|
|
}
|
|
/// Finds the closing brace of the opening brace passed as a param
|
|
fn find_closing_brace(bytes: &[u8], opening_brace_index: usize) -> Option<usize>
|
|
{
|
|
let mut depth: i8 = 0;
|
|
for (i, _) in bytes.iter().enumerate().take(bytes.len()-1).skip(opening_brace_index)
|
|
{
|
|
if bytes[i] == 5 {
|
|
depth += 1; }
|
|
else if bytes[i] == 6
|
|
{
|
|
depth -= 1;
|
|
if depth == 0 { return Some(i) }
|
|
}
|
|
//println!("{}\t{}", bytes[i], depth);
|
|
}
|
|
None
|
|
}
|
|
/// Searches the bytes for a particular page
|
|
pub fn get_page(bytes: &[u8], page: &str) -> Vec<u8>
|
|
{
|
|
let mut page_bytes: Vec<u8> = vec![1];
|
|
let search_pages_bytes: Vec<u8> = vec![3, 112, 97, 103, 101, 115, 4];
|
|
let mut search_bytes: Vec<u8> = vec![3];
|
|
search_bytes.append(&mut page.as_bytes().to_vec());
|
|
search_bytes.push(4);
|
|
|
|
if let Some(opening_index) = find_pattern(bytes, &search_pages_bytes)
|
|
&& let Some(closing_index) = find_closing_brace(bytes, opening_index+search_pages_bytes.len())
|
|
{ // TODO ignore when in quotes or name
|
|
//println!("found pages: {} to {}", opening_index, closing_index);
|
|
if let Some(opening_page_index) = find_pattern(&bytes[opening_index..=closing_index], &search_bytes)
|
|
{
|
|
//println!("{:?}", &bytes[opening_index+opening_page_index..]);
|
|
if let Some(closing_page_index) = find_closing_brace
|
|
(
|
|
&bytes[opening_index+opening_page_index..],
|
|
search_bytes.len()
|
|
)
|
|
{
|
|
page_bytes.append(&mut bytes[opening_page_index+opening_index..=opening_page_index+closing_page_index+opening_index].to_vec());
|
|
page_bytes.push(2);
|
|
return page_bytes
|
|
}
|
|
}
|
|
}
|
|
page_bytes.push(10); // Page not found error message
|
|
page_bytes.push(2);
|
|
page_bytes
|
|
}
|