diff --git a/src/main.rs b/src/main.rs index 3fa6497..9de4651 100644 --- a/src/main.rs +++ b/src/main.rs @@ -224,7 +224,12 @@ fn main() { None => link_content, }.trim().to_lowercase(); - let incoming_pages = incoming_map.get_mut(&page_id).unwrap(); + let incoming_pages = match incoming_map.get_mut(&page_id) { + Some(v) => v, + + // The page being referenced doesn't exist (a red link) + None => continue, + }; incoming_pages.push(page); } @@ -239,11 +244,13 @@ fn main() { None => (link_content.trim().to_lowercase(), link_content) }; - let permalink = page_map.get(&page_id) - .unwrap() - .permalink.as_str(); - - format!("{}", permalink, link_text) + return match page_map.get(&page_id) { + Some(linked_page) => { + let permalink = linked_page.permalink.as_str(); + format!("{}", permalink, link_text) + }, + None => format!("{}", link_text), + } }).to_string()); }