934 lines
28 KiB
Rust
934 lines
28 KiB
Rust
mod render;
|
|
mod retrieval;
|
|
|
|
#[cfg(test)]
|
|
mod tests;
|
|
|
|
extern crate pdf;
|
|
|
|
use crate::pdf::object::Resolve;
|
|
use crate::render::Renderer;
|
|
use image::DynamicImage;
|
|
use lazy_static::lazy_static;
|
|
use pdf::file::{File, FileOptions, NoLog, ObjectCache, StreamCache};
|
|
use pdf::object::{Object, ObjectWrite, Page, PageRc, PlainRef, Stream, Updater};
|
|
use pdf::primitive::{Dictionary, Primitive};
|
|
use pdf::xref::XRef;
|
|
use regex::Regex;
|
|
use retrieval::{get_prim_by_path_with_file, get_stream_data_by_path_with_file};
|
|
use serde::{Deserialize, Serialize};
|
|
use std::collections::{HashMap, VecDeque};
|
|
use std::ops::Deref;
|
|
use std::path::Path;
|
|
use std::sync::{Arc, RwLock};
|
|
use tauri::ipc::{InvokeResponseBody, Response};
|
|
use tauri::{Manager, State};
|
|
use uuid::Uuid;
|
|
use pdf::backend::Backend;
|
|
|
|
type CosFile = File<Vec<u8>, ObjectCache, StreamCache, NoLog>;
|
|
#[derive(Serialize, Debug, Clone)]
|
|
pub struct XRefTableModel {
|
|
pub size: usize,
|
|
pub entries: Vec<XRefEntryModel>,
|
|
}
|
|
#[derive(Serialize, Debug, Clone)]
|
|
pub struct XRefEntryModel {
|
|
pub obj_num: u64,
|
|
pub gen_num: u64,
|
|
pub obj_type: String,
|
|
pub offset: u64,
|
|
}
|
|
#[derive(Serialize, Debug, Clone)]
|
|
pub struct PdfFile {
|
|
pub id: String,
|
|
pub name: String,
|
|
pub path: String,
|
|
pub page_count: u32,
|
|
pub xref_entries: usize,
|
|
pub pages: Vec<PageModel>,
|
|
}
|
|
|
|
#[derive(Serialize, Debug, Clone)]
|
|
pub struct PrimitiveModel {
|
|
pub key: String,
|
|
pub ptype: String,
|
|
pub sub_type: String,
|
|
pub value: String,
|
|
pub children: Vec<PrimitiveModel>,
|
|
pub trace: Vec<PathTrace>,
|
|
pub expanded: bool,
|
|
}
|
|
|
|
#[derive(Serialize, Debug, Clone)]
|
|
pub struct PrimitiveTreeView {
|
|
pub depth: usize,
|
|
pub key: String,
|
|
pub ptype: String,
|
|
pub sub_type: String,
|
|
pub value: String,
|
|
pub container: bool,
|
|
pub expanded: bool,
|
|
pub path: Vec<PathTrace>,
|
|
pub active: bool,
|
|
}
|
|
|
|
#[derive(Serialize, Debug, Clone)]
|
|
pub struct PathTrace {
|
|
pub key: String,
|
|
#[serde(skip_serializing)]
|
|
pub last_ref: Option<PlainRef>,
|
|
pub last_jump: String,
|
|
}
|
|
impl PathTrace {
|
|
fn new(key: String, last_ref: Option<PlainRef>) -> PathTrace {
|
|
PathTrace {
|
|
key,
|
|
last_jump: last_ref
|
|
.map(|r| r.id.to_string())
|
|
.unwrap_or("Trailer".to_string()),
|
|
last_ref: last_ref,
|
|
}
|
|
}
|
|
|
|
fn trailer() -> PathTrace {
|
|
PathTrace {
|
|
key: "Trailer".to_string(),
|
|
last_ref: None,
|
|
last_jump: "Trailer".to_string(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Debug, Clone)]
|
|
pub struct PageModel {
|
|
key: String,
|
|
obj_num: u64,
|
|
page_num: u64,
|
|
}
|
|
#[derive(Deserialize, Serialize, Debug, Clone)]
|
|
pub struct TreeViewRequest {
|
|
key: String,
|
|
children: Vec<TreeViewRequest>,
|
|
expand: bool,
|
|
}
|
|
|
|
impl TreeViewRequest {
|
|
fn step(&self) -> Result<Step, String> {
|
|
Step::parse_step(&self.key)
|
|
}
|
|
}
|
|
|
|
#[derive(Deserialize, Serialize, Debug, Clone)]
|
|
pub struct ContentsModel {
|
|
parts: Vec<Vec<String>>,
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn get_all_files(session: State<Session>) -> Vec<PdfFile> {
|
|
session
|
|
.get_all_files()
|
|
.iter()
|
|
.map(|s| s.read().unwrap().pdf_file.clone())
|
|
.collect()
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn get_changes(id: &str, session: State<Session>) -> Result<Vec<PrimitiveModel>, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
Ok(file.cos_file.changes().iter()
|
|
.map(|(obj, (prim, gen))| PrimitiveModel::from_primitive_until_refs(prim, PathTrace::new(obj.to_string(), Some(PlainRef{id: *obj, gen: *gen}))))
|
|
.collect())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn clear_changes(id: &str, session: State<Session>) -> Result<(), String> {
|
|
let lock = session.get_file(id)?;
|
|
let mut file = lock.write().unwrap();
|
|
file.cos_file.clear_changes();
|
|
Ok(())
|
|
}
|
|
#[tauri::command]
|
|
fn get_all_file_ids(session: State<Session>) -> Vec<String> {
|
|
session.get_all_file_ids()
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn close_file(id: &str, session: State<Session>) {
|
|
session.handle_close(id);
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn get_file_by_id(id: &str, session: State<Session>) -> Result<PdfFile, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
Ok(file.pdf_file.clone())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn upload(path: &str, session: State<Session>) -> Result<String, String> {
|
|
let file = t!(FileOptions::cached().open(path));
|
|
|
|
let pdf_file = to_pdf_file(path, &file)?;
|
|
let id = pdf_file.id.clone();
|
|
|
|
session.handle_upload(pdf_file, file)?;
|
|
|
|
Ok(id)
|
|
}
|
|
|
|
fn to_pdf_file(path: &str, file: &CosFile) -> Result<PdfFile, String> {
|
|
fn parse_title_from_path(path: &str) -> Option<String> {
|
|
Path::new(path)
|
|
.file_name()
|
|
.and_then(|f| f.to_str().map(|s| s.to_string()))
|
|
}
|
|
|
|
let file_name = if let Some(ref info) = file.trailer.info_dict {
|
|
info.title
|
|
.as_ref()
|
|
.map(|p| p.to_string_lossy())
|
|
.unwrap_or(parse_title_from_path(path).unwrap_or_else(|| "Not found".to_string()))
|
|
} else {
|
|
"Not found".to_string()
|
|
};
|
|
|
|
let pages = file
|
|
.pages()
|
|
.enumerate()
|
|
.map(|(i, page_ref)| PageModel {
|
|
key: format!("Page {}", i + 1),
|
|
obj_num: page_ref.unwrap().get_ref().get_inner().id,
|
|
page_num: (i + 1) as u64,
|
|
})
|
|
.collect();
|
|
|
|
let pdf_file = PdfFile {
|
|
id: Uuid::new_v4().to_string(),
|
|
name: file_name.to_string().into(),
|
|
path: path.to_string().into(),
|
|
page_count: file.num_pages(),
|
|
xref_entries: file.get_xref().len(),
|
|
pages: pages,
|
|
};
|
|
Ok(pdf_file)
|
|
}
|
|
|
|
fn read_contents(page: &Page, resolver: &impl Resolve) -> Result<ContentsModel, String> {
|
|
if let Some(ref contents) = page.contents {
|
|
let mut parts = vec![];
|
|
for part in &contents.parts {
|
|
let data = &t!(part.data(resolver));
|
|
let string = String::from_utf8_lossy(data);
|
|
let part = string
|
|
.split("\n")
|
|
.map(|s| s.to_string())
|
|
.collect::<Vec<String>>();
|
|
parts.push(part);
|
|
}
|
|
return Ok(ContentsModel { parts });
|
|
}
|
|
Err(String::from("Error occurred"))
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn get_contents(id: &str, path: &str, session: State<'_, Session>) -> Result<ContentsModel, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
|
|
let (_, page_prim, _) = get_prim_by_path_with_file(path, &file.cos_file)?;
|
|
let resolver = file.cos_file.resolver();
|
|
|
|
let page = t!(pdf::object::Page::from_primitive(page_prim, &resolver));
|
|
read_contents(&page, &resolver)
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn update_contents(
|
|
id: &str,
|
|
path: &str,
|
|
session: State<'_, Session>,
|
|
contents: ContentsModel,
|
|
) -> Result<(), String> {
|
|
let lock = session.get_file(id)?;
|
|
let mut file = lock.write().unwrap();
|
|
update_contents_with_file(path, contents, &mut file.cos_file)
|
|
}
|
|
|
|
fn update_contents_with_file(path: &str, contents: ContentsModel, file: &mut CosFile) -> Result<(), String> {
|
|
let mut parts = vec![];
|
|
for part in contents.parts {
|
|
let stream_ref = Stream::new((), part.join("\n").as_bytes());
|
|
parts.push(stream_ref);
|
|
}
|
|
let new_contents = pdf::content::Content{parts};
|
|
|
|
if let Some(Step::Page(num)) = Step::parse(path).pop_front() {
|
|
let page_rc: PageRc = t!(file.get_page(num - 1));
|
|
let page_ref = page_rc.get_ref().get_inner().clone();
|
|
let mut page = page_rc.deref().clone();
|
|
page.contents = Some(new_contents);
|
|
let _ = t!(file.update(page_ref, page));
|
|
return Ok(());
|
|
}
|
|
Err(String::from("Path does not lead to a page!"))
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn get_stream_data_as_string(
|
|
id: &str,
|
|
path: &str,
|
|
session: State<'_, Session>,
|
|
) -> Result<String, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
let data = get_stream_data_by_path_with_file(path, &file.cos_file)?;
|
|
Ok(String::from_utf8_lossy(&data).into_owned())
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn update_stream_data_as_string(
|
|
id: &str,
|
|
path: &str,
|
|
new_data: &str,
|
|
session: State<'_, Session>,
|
|
) -> Result<(), String> {
|
|
let lock = session.get_file(id)?;
|
|
let mut file = lock.write().unwrap();
|
|
let (_, _, trace) = get_prim_by_path_with_file(path, &file.cos_file)?;
|
|
let i = trace.len() - 1;
|
|
let plain_ref = trace[i].last_ref.expect("No trace to update");
|
|
let stream = Stream::new((), new_data.as_bytes());
|
|
let _ = t!(file.cos_file.update(plain_ref, stream));
|
|
Ok(())
|
|
}
|
|
|
|
fn format_img_as_response(img: DynamicImage) -> Result<Response, String> {
|
|
use std::mem;
|
|
|
|
let w: u32 = img.width();
|
|
let h: u32 = img.height();
|
|
let mut data: Vec<u8> = img.into_rgba8().into_raw();
|
|
|
|
let w_bytes = w.to_le_bytes(); // or to_be_bytes() depending on endianness
|
|
let h_bytes = h.to_le_bytes();
|
|
|
|
let old_capacity = data.capacity();
|
|
|
|
let ptr = data.as_mut_ptr();
|
|
let len = data.len();
|
|
mem::forget(data);
|
|
|
|
// Create a new Vec with the two u32s in front
|
|
let merged = unsafe {
|
|
let mut new_vec = Vec::from_raw_parts(ptr, len, old_capacity);
|
|
|
|
// Insert u32 bytes at the front
|
|
new_vec.splice(0..0, w_bytes.iter().copied());
|
|
new_vec.splice(4..4, h_bytes.iter().copied());
|
|
|
|
new_vec
|
|
};
|
|
Ok(Response::new(InvokeResponseBody::from(merged)))
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn get_stream_data_as_image(
|
|
id: &str,
|
|
path: &str,
|
|
session: State<'_, Session>,
|
|
) -> Result<Response, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
let img = retrieval::get_image_by_path(path, &file.cos_file)?;
|
|
format_img_as_response(img)
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn get_page_by_num(
|
|
id: &str,
|
|
num: u32,
|
|
session: State<'_, Session>,
|
|
) -> Result<Response, String> {
|
|
render_page_by_path(id, &format!("Page{}", num), session).await
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn render_page_by_path(
|
|
id: &str,
|
|
path: &str,
|
|
session: State<'_, Session>,
|
|
) -> Result<Response, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
let page = get_page_by_path(path, &file.cos_file)?;
|
|
let mut renderer = Renderer::new(&file.cos_file, 150);
|
|
let img = renderer.render(&page)?;
|
|
|
|
format_img_as_response(img)
|
|
}
|
|
|
|
fn get_page_by_path(path: &str, file: &CosFile) -> Result<pdf::object::Page, String> {
|
|
let resolver = file.resolver();
|
|
let (_, page_prim, _) = get_prim_by_path_with_file(path, file)?;
|
|
let page = t!(pdf::object::Page::from_primitive(page_prim, &resolver));
|
|
Ok(page)
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn get_prim_by_path(
|
|
id: &str,
|
|
path: &str,
|
|
session: State<Session>,
|
|
) -> Result<PrimitiveModel, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
get_prim_model_by_path_with_file(path, &file.cos_file)
|
|
}
|
|
|
|
fn get_prim_model_by_path_with_file(path: &str, file: &CosFile) -> Result<PrimitiveModel, String> {
|
|
let steps = Step::parse(path);
|
|
if steps.len() == 0 {
|
|
return Err(String::from(format!("{:?} is not a valid path!", steps)));
|
|
}
|
|
if steps.back().unwrap() == &Step::Data {
|
|
return handle_data_step(steps, file);
|
|
}
|
|
|
|
let (_, prim, trace) = retrieval::get_prim_by_steps_with_file(steps, file)?;
|
|
|
|
Ok(PrimitiveModel::from_primitive_with_children(&prim, trace))
|
|
}
|
|
|
|
fn handle_data_step(mut steps: VecDeque<Step>, file: &CosFile) -> Result<PrimitiveModel, String> {
|
|
let _data = steps.pop_back();
|
|
let (_, prim, trace) = retrieval::get_prim_by_steps_with_file(steps, file)?;
|
|
if let Primitive::Stream(stream) = prim {
|
|
let sub_type = match stream.info.get("Subtype") {
|
|
Some(element) => match element {
|
|
Primitive::Name(name) => name.as_str().to_string(),
|
|
_ => String::from("-"),
|
|
},
|
|
_ => String::from("-"),
|
|
};
|
|
return Ok(PrimitiveModel::as_data(sub_type, trace));
|
|
}
|
|
Err(String::from("Parent of Data is not a stream"))
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn get_prim_tree_by_path(
|
|
id: &str,
|
|
paths: Vec<TreeViewRequest>,
|
|
session: State<Session>,
|
|
) -> Result<Vec<PrimitiveTreeView>, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
|
|
let results = paths
|
|
.into_iter()
|
|
.map(|path| get_prim_tree_by_path_with_file(path, &file.cos_file))
|
|
.collect::<Result<Vec<_>, String>>()?
|
|
.into_iter()
|
|
.flatten()
|
|
.collect();
|
|
|
|
Ok(results)
|
|
}
|
|
|
|
fn get_prim_tree_by_path_with_file(
|
|
node: TreeViewRequest,
|
|
file: &CosFile,
|
|
) -> Result<Vec<PrimitiveTreeView>, String> {
|
|
let step = node.step()?;
|
|
let (parent, trace) = retrieval::resolve_parent(step.clone(), file)?;
|
|
|
|
let trace = vec![trace];
|
|
let mut parent_model: PrimitiveModel;
|
|
if node.expand {
|
|
parent_model = PrimitiveModel::from_primitive_with_children(&parent, trace);
|
|
|
|
for child in node.children.iter() {
|
|
expand(child, &mut parent_model, &parent, file)?;
|
|
}
|
|
} else {
|
|
parent_model = PrimitiveModel::from_primitive(step.get_key(), &parent, trace);
|
|
}
|
|
|
|
Ok(PrimitiveTreeView::flatten(0, parent_model))
|
|
}
|
|
|
|
fn expand(
|
|
node: &TreeViewRequest,
|
|
parent_model: &mut PrimitiveModel,
|
|
parent: &Primitive,
|
|
file: &CosFile,
|
|
) -> Result<(), String> {
|
|
if !node.expand {
|
|
return Ok(());
|
|
}
|
|
let step = node.step()?;
|
|
if let Ok(prim) = retrieval::resolve_step(parent, &step) {
|
|
if let Primitive::Reference(x_ref) = prim {
|
|
let jump = retrieval::resolve_xref(x_ref.id, file)?;
|
|
let mut jump_trace = parent_model.trace.clone();
|
|
jump_trace.push(PathTrace::new(step.get_key(), Some(x_ref.clone())));
|
|
let mut to_expand = parent_model.get_child(step.get_key()).unwrap();
|
|
to_expand.add_children(&jump, &jump_trace);
|
|
expand_children(node, file, &jump, &mut to_expand)?;
|
|
} else {
|
|
let mut to_expand = parent_model.get_child(step.get_key()).unwrap();
|
|
to_expand.add_children(prim, &to_expand.trace.clone());
|
|
expand_children(node, file, prim, &mut to_expand)?;
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn expand_children(
|
|
node: &TreeViewRequest,
|
|
file: &CosFile,
|
|
prim: &Primitive,
|
|
mut expanded: &mut PrimitiveModel,
|
|
) -> Result<(), String> {
|
|
for child in node.children.iter() {
|
|
expand(child, &mut expanded, prim, file)?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[derive(Debug, PartialEq, Clone)]
|
|
pub enum Step {
|
|
String(String),
|
|
Page(u32),
|
|
Number(u64),
|
|
Trailer,
|
|
Data,
|
|
}
|
|
|
|
impl Step {
|
|
fn parse_step(path: &str) -> Result<Step, String> {
|
|
lazy_static! {
|
|
static ref PAGE_RE: Regex = Regex::new(r"^Page(\d+)$").unwrap();
|
|
}
|
|
if path.len() == 0 {
|
|
return Err(String::from("Path is empty"));
|
|
}
|
|
|
|
Ok(match &path.parse::<u64>().ok() {
|
|
Some(i) => Step::Number(*i),
|
|
None => match &path[..] {
|
|
"Data" => Step::Data,
|
|
"Trailer" => Step::Trailer,
|
|
"/" => Step::Trailer,
|
|
_ => {
|
|
if let Some(caps) = PAGE_RE.captures(path) {
|
|
Step::Page(
|
|
caps[1]
|
|
.parse::<u32>()
|
|
.map_err(|_| format!("Invalid page number in {}", path))?,
|
|
)
|
|
} else {
|
|
Step::String(path.to_string())
|
|
}
|
|
}
|
|
},
|
|
})
|
|
}
|
|
|
|
fn parse(path: &str) -> VecDeque<Step> {
|
|
let mut steps = VecDeque::new();
|
|
if path.starts_with("/") {
|
|
steps.push_back(Step::Trailer);
|
|
}
|
|
let split_path = path.split("/").collect::<VecDeque<&str>>();
|
|
split_path
|
|
.iter()
|
|
.filter_map(|s| Step::parse_step(s).ok())
|
|
.collect::<VecDeque<Step>>()
|
|
}
|
|
|
|
fn get_key(&self) -> String {
|
|
match self {
|
|
Step::String(s) => s.clone(),
|
|
Step::Number(i) => i.to_string(),
|
|
Step::Trailer => "Trailer".to_string(),
|
|
Step::Page(n) => format!("Page{}", n),
|
|
Step::Data => "Data".into(),
|
|
}
|
|
}
|
|
}
|
|
|
|
fn append_path(key: String, path: &Vec<PathTrace>) -> Vec<PathTrace> {
|
|
let mut new_path = path.clone();
|
|
let last_jump = new_path.last().unwrap().last_ref.clone();
|
|
new_path.push(PathTrace::new(key, last_jump));
|
|
new_path
|
|
}
|
|
|
|
impl PrimitiveModel {
|
|
fn from_primitive(key: String, primitive: &Primitive, path: Vec<PathTrace>) -> PrimitiveModel {
|
|
let value: String = match primitive {
|
|
Primitive::Null => "Null".to_string(),
|
|
Primitive::Integer(i) => i.to_string(),
|
|
Primitive::Number(f) => f.to_string(),
|
|
Primitive::Boolean(b) => b.to_string(),
|
|
Primitive::String(s) => s.to_string_lossy(),
|
|
Primitive::Stream(_) => "-".to_string(),
|
|
Primitive::Dictionary(dict) => PrimitiveModel::format_dict_content(dict),
|
|
Primitive::Array(arr) => PrimitiveModel::format_arr_content(arr),
|
|
Primitive::Reference(pref) => {
|
|
format!("Obj Nr: {} Gen Nr: {}", pref.id, pref.gen)
|
|
}
|
|
|
|
Primitive::Name(name) => name.clone().as_str().to_string(),
|
|
};
|
|
let sub_type: String = match primitive {
|
|
Primitive::Dictionary(d) => d
|
|
.get("Type")
|
|
.and_then(|value| match value {
|
|
Primitive::Name(name) => Some(name.clone().as_str().to_string()),
|
|
_ => None,
|
|
})
|
|
.unwrap_or(String::from("-")),
|
|
_ => String::from("-"),
|
|
};
|
|
PrimitiveModel {
|
|
key: key,
|
|
ptype: primitive.get_debug_name().into(),
|
|
sub_type: sub_type,
|
|
value: value,
|
|
children: Vec::new(),
|
|
trace: path,
|
|
expanded: false,
|
|
}
|
|
}
|
|
|
|
fn as_data(sub_type: String, trace: Vec<PathTrace>) -> PrimitiveModel {
|
|
PrimitiveModel {
|
|
key: "Data".to_string(),
|
|
ptype: "Stream Data".to_string(),
|
|
sub_type: sub_type,
|
|
value: "".to_string(),
|
|
children: Vec::new(),
|
|
trace: append_path("Data".to_string(), &trace),
|
|
expanded: false,
|
|
}
|
|
}
|
|
|
|
fn format_dict_content(dict: &Dictionary) -> String {
|
|
let mut result = String::from("{");
|
|
let mut count = 0;
|
|
for (key, value) in dict.iter() {
|
|
result.push_str(&format!(
|
|
"{}: {}",
|
|
key,
|
|
PrimitiveModel::format_prim_short(value)
|
|
));
|
|
count += 1;
|
|
if count < 4 {
|
|
result.push_str(", ");
|
|
} else {
|
|
result.push_str(",...");
|
|
break;
|
|
}
|
|
}
|
|
result.push_str("}");
|
|
result
|
|
}
|
|
fn format_arr_content(arr: &Vec<Primitive>) -> String {
|
|
if arr.len() == 0 {
|
|
return "[]".to_string();
|
|
}
|
|
let mut result = String::from("[");
|
|
let contents = if arr.len() > 4 { &arr[0..4] } else { &arr[..] };
|
|
for i in 0..contents.len() {
|
|
let prim = contents.get(i).unwrap();
|
|
result.push_str(&PrimitiveModel::format_prim_short(prim));
|
|
if i != contents.len() - 1 {
|
|
result.push_str(", ");
|
|
}
|
|
}
|
|
|
|
if arr.len() > 4 {
|
|
result.push_str(",...");
|
|
}
|
|
result.push_str("]");
|
|
result
|
|
}
|
|
|
|
fn format_prim_short(prim: &Primitive) -> String {
|
|
match prim {
|
|
Primitive::Integer(i) => format!("{}", i),
|
|
Primitive::Number(n) => format!("{}", n),
|
|
Primitive::Boolean(b) => format!("{}", b),
|
|
Primitive::String(s) => s.to_string().unwrap_or(String::from("-")),
|
|
Primitive::Name(n) => n.as_str().to_string(),
|
|
_ => prim.get_debug_name().to_string(),
|
|
}
|
|
}
|
|
|
|
fn from_primitive_with_children(
|
|
primitive: &Primitive,
|
|
trace: Vec<PathTrace>,
|
|
) -> PrimitiveModel {
|
|
let mut model = PrimitiveModel::from_primitive(
|
|
trace.last().unwrap().key.clone(),
|
|
primitive,
|
|
trace.clone(),
|
|
);
|
|
model.add_children(primitive, &trace);
|
|
model
|
|
}
|
|
|
|
fn from_primitive_until_refs(primitive: &Primitive, trace: PathTrace) -> PrimitiveModel {
|
|
let mut parent_model = PrimitiveModel::from_primitive(
|
|
trace.key.clone(),
|
|
primitive,
|
|
vec![trace.clone()],
|
|
);
|
|
let mut model = parent_model;
|
|
model.add_children(primitive, &vec!(trace));
|
|
model
|
|
}
|
|
|
|
fn add_children(&mut self, primitive: &Primitive, path: &Vec<PathTrace>) {
|
|
self.expanded = true;
|
|
match primitive {
|
|
Primitive::Dictionary(dict) => dict.iter().for_each(|(name, value)| {
|
|
self.add_child(
|
|
name.clone().as_str().to_string(),
|
|
value,
|
|
append_path(name.clone().as_str().to_string(), &path),
|
|
);
|
|
}),
|
|
Primitive::Array(arr) => arr.iter().enumerate().for_each(|(i, obj)| {
|
|
self.add_child(i.to_string(), obj, append_path(i.to_string(), &path));
|
|
}),
|
|
Primitive::Stream(stream) => {
|
|
stream.info.iter().for_each(|(name, value)| {
|
|
self.add_child(
|
|
name.clone().as_str().to_string(),
|
|
value,
|
|
append_path(name.clone().as_str().to_string(), &path),
|
|
);
|
|
});
|
|
self.children.push(PrimitiveModel {
|
|
key: "Data".to_string(),
|
|
ptype: "Stream Data".to_string(),
|
|
sub_type: "-".to_string(),
|
|
value: "".to_string(),
|
|
children: vec![],
|
|
trace: append_path("Data".to_string(), &path),
|
|
expanded: false,
|
|
});
|
|
}
|
|
_ => (),
|
|
};
|
|
}
|
|
|
|
fn add_child(
|
|
&mut self,
|
|
key: String,
|
|
child: &Primitive,
|
|
path: Vec<PathTrace>,
|
|
) -> &PrimitiveModel {
|
|
let child_model = Self::from_primitive(key, child, path);
|
|
self.children.push(child_model);
|
|
&self.children[self.children.len() - 1]
|
|
}
|
|
|
|
fn get_child(&mut self, key: String) -> Option<&mut PrimitiveModel> {
|
|
self.children.iter_mut().find(|child| child.key == key)
|
|
}
|
|
|
|
fn is_container(&self) -> bool {
|
|
self.ptype == "Dictionary"
|
|
|| self.ptype == "Array"
|
|
|| self.ptype == "Stream"
|
|
|| self.ptype == "Reference"
|
|
}
|
|
fn drain_children(&mut self) -> Vec<PrimitiveModel> {
|
|
self.children.drain(..).collect()
|
|
}
|
|
}
|
|
|
|
impl PrimitiveTreeView {
|
|
fn from_primitive(depth: usize, primitive: PrimitiveModel) -> PrimitiveTreeView {
|
|
let is_container = primitive.is_container();
|
|
PrimitiveTreeView {
|
|
depth: depth,
|
|
key: primitive.key,
|
|
ptype: primitive.ptype,
|
|
sub_type: primitive.sub_type,
|
|
value: primitive.value,
|
|
container: is_container,
|
|
expanded: primitive.expanded,
|
|
path: primitive.trace,
|
|
active: true,
|
|
}
|
|
}
|
|
fn flatten(depth: usize, mut primitive: PrimitiveModel) -> Vec<PrimitiveTreeView> {
|
|
let mut views: Vec<PrimitiveTreeView> = Vec::new();
|
|
let children = primitive.drain_children();
|
|
views.push(PrimitiveTreeView::from_primitive(depth, primitive));
|
|
children.into_iter().for_each(|child| {
|
|
views.extend(PrimitiveTreeView::flatten(depth + 1, child.clone()));
|
|
});
|
|
views
|
|
}
|
|
}
|
|
#[tauri::command]
|
|
fn get_xref_table(id: &str, session: State<Session>) -> Result<XRefTableModel, String> {
|
|
let lock = session.get_file(id)?;
|
|
let file = lock.read().unwrap();
|
|
get_xref_table_model_with_file(&file.cos_file)
|
|
}
|
|
|
|
fn get_xref_table_model_with_file(file: &CosFile) -> Result<XRefTableModel, String> {
|
|
let resolver = file.resolver();
|
|
let x_ref_table = file.get_xref();
|
|
let mut models: Vec<XRefEntryModel> = Vec::new();
|
|
|
|
for (i, x_ref) in x_ref_table.iter_real().enumerate() {
|
|
models.push(match x_ref {
|
|
XRef::Raw { pos, gen_nr } => {
|
|
let prim: Primitive = resolver
|
|
.resolve(PlainRef {
|
|
id: i as u64,
|
|
gen: *gen_nr,
|
|
})
|
|
.unwrap();
|
|
XRefEntryModel {
|
|
obj_num: i as u64,
|
|
gen_num: *gen_nr,
|
|
obj_type: prim.get_debug_name().to_string().into(),
|
|
offset: *pos as u64,
|
|
}
|
|
}
|
|
XRef::Stream { stream_id, index } => XRefEntryModel {
|
|
obj_num: i as u64,
|
|
gen_num: 0,
|
|
obj_type: "Stream".into(),
|
|
offset: *index as u64,
|
|
},
|
|
XRef::Free {
|
|
next_obj_nr,
|
|
gen_nr,
|
|
} => XRefEntryModel {
|
|
obj_num: i as u64,
|
|
gen_num: *gen_nr as u64,
|
|
obj_type: "Free".into(),
|
|
offset: *next_obj_nr as u64,
|
|
},
|
|
XRef::Promised => XRefEntryModel {
|
|
obj_num: i as u64,
|
|
gen_num: 0,
|
|
obj_type: "Promised".into(),
|
|
offset: 0,
|
|
},
|
|
XRef::Invalid => XRefEntryModel {
|
|
obj_num: i as u64,
|
|
gen_num: 0,
|
|
obj_type: "Invalid".into(),
|
|
offset: 0,
|
|
},
|
|
});
|
|
}
|
|
Ok(XRefTableModel {
|
|
size: x_ref_table.len(),
|
|
entries: models,
|
|
})
|
|
}
|
|
|
|
struct Session {
|
|
files: RwLock<HashMap<String, Arc<RwLock<SessionFile>>>>,
|
|
}
|
|
|
|
struct SessionFile {
|
|
pdf_file: PdfFile,
|
|
cos_file: CosFile,
|
|
}
|
|
|
|
impl Session {
|
|
fn load() -> Session {
|
|
Session {
|
|
files: RwLock::new(HashMap::new()),
|
|
}
|
|
}
|
|
|
|
fn get_file(&self, id: &str) -> Result<Arc<RwLock<SessionFile>>, String> {
|
|
let lock = self.files.read().unwrap();
|
|
lock.get(id)
|
|
.cloned()
|
|
.ok_or(format!(" File {} not found!", id))
|
|
}
|
|
|
|
fn get_all_files(&self) -> Vec<Arc<RwLock<SessionFile>>> {
|
|
self.files.read().unwrap().values().cloned().collect()
|
|
}
|
|
|
|
fn get_all_file_ids(&self) -> Vec<String> {
|
|
self.files
|
|
.read()
|
|
.unwrap()
|
|
.keys()
|
|
.map(|f| f.clone())
|
|
.collect()
|
|
}
|
|
|
|
fn handle_upload(&self, pdf_file: PdfFile, cos_file: CosFile) -> Result<(), String> {
|
|
if let Ok(mut files) = self.files.write() {
|
|
return match files.insert(
|
|
pdf_file.id.clone(),
|
|
Arc::new(RwLock::new(SessionFile {
|
|
pdf_file: pdf_file,
|
|
cos_file: cos_file,
|
|
})),
|
|
) {
|
|
Some(_) => Err("File could not be uploaded!".to_string()),
|
|
None => Ok(()),
|
|
};
|
|
};
|
|
Err("Lock could not be acquired!".to_string())
|
|
}
|
|
|
|
fn handle_close(&self, id: &str) {
|
|
self.files.write().unwrap().remove(id);
|
|
}
|
|
}
|
|
|
|
#[cfg_attr(mobile, tauri::mobile_entry_point)]
|
|
pub fn run() {
|
|
tauri::Builder::default()
|
|
.plugin(tauri_plugin_dialog::init())
|
|
.plugin(tauri_plugin_fs::init())
|
|
.plugin(tauri_plugin_opener::init())
|
|
.setup(|app| {
|
|
app.manage(Session::load());
|
|
Ok(())
|
|
})
|
|
.invoke_handler(tauri::generate_handler![
|
|
upload,
|
|
get_all_files,
|
|
get_all_file_ids,
|
|
get_file_by_id,
|
|
close_file,
|
|
get_prim_by_path,
|
|
get_prim_tree_by_path,
|
|
get_xref_table,
|
|
get_contents,
|
|
update_contents,
|
|
get_stream_data_as_string,
|
|
update_stream_data_as_string,
|
|
get_stream_data_as_image,
|
|
get_page_by_num,
|
|
get_changes,
|
|
clear_changes
|
|
])
|
|
.run(tauri::generate_context!())
|
|
.expect("error while running tauri application");
|
|
}
|