From 8d786dc4c3ce26dbb3432023c7461bd879993bfd Mon Sep 17 00:00:00 2001
From: Lukas Wirth <lukastw97@gmail.com>
Date: Tue, 30 Mar 2021 22:26:03 +0200
Subject: [PATCH] Replace Range<usize> usage with TextRange

---
 crates/hir_def/src/attr.rs  | 42 +++++++++++++------------------------
 crates/ide/src/doc_links.rs | 17 +++++++++++----
 2 files changed, 27 insertions(+), 32 deletions(-)

diff --git a/crates/hir_def/src/attr.rs b/crates/hir_def/src/attr.rs
index 7791402c96a..74bb6de35b1 100644
--- a/crates/hir_def/src/attr.rs
+++ b/crates/hir_def/src/attr.rs
@@ -1,8 +1,8 @@
 //! A higher level attributes based on TokenTree, with also some shortcuts.
 
 use std::{
-    cmp::Ordering,
-    ops::{self, Range},
+    convert::{TryFrom, TryInto},
+    ops,
     sync::Arc,
 };
 
@@ -479,6 +479,7 @@ pub fn docs_with_rangemap(
             if !doc.is_empty() {
                 for line in doc.split('\n') {
                     let line = line.trim_end();
+                    let line_len = line.len();
                     let (offset, line) = match line.char_indices().nth(indent) {
                         Some((offset, _)) => (offset, &line[offset..]),
                         None => (0, line),
@@ -486,9 +487,9 @@ pub fn docs_with_rangemap(
                     let buf_offset = buf.len();
                     buf.push_str(line);
                     mapping.push((
-                        Range { start: buf_offset, end: buf.len() },
+                        TextRange::new(buf_offset.try_into().ok()?, buf.len().try_into().ok()?),
                         idx,
-                        Range { start: offset, end: line.len() },
+                        TextRange::new(offset.try_into().ok()?, line_len.try_into().ok()?),
                     ));
                     buf.push('\n');
                 }
@@ -565,31 +566,18 @@ pub struct DocsRangeMap {
     // (docstring-line-range, attr_index, attr-string-range)
     // a mapping from the text range of a line of the [`Documentation`] to the attribute index and
     // the original (untrimmed) syntax doc line
-    mapping: Vec<(Range<usize>, u32, Range<usize>)>,
+    mapping: Vec<(TextRange, u32, TextRange)>,
 }
 
 impl DocsRangeMap {
-    pub fn map(&self, range: Range<usize>) -> Option<InFile<TextRange>> {
-        let found = self
-            .mapping
-            .binary_search_by(|(probe, ..)| {
-                if probe.contains(&range.start) {
-                    Ordering::Equal
-                } else {
-                    probe.start.cmp(&range.end)
-                }
-            })
-            .ok()?;
+    pub fn map(&self, range: TextRange) -> Option<InFile<TextRange>> {
+        let found = self.mapping.binary_search_by(|(probe, ..)| probe.ordering(range)).ok()?;
         let (line_docs_range, idx, original_line_src_range) = self.mapping[found].clone();
-        if range.end > line_docs_range.end {
+        if !line_docs_range.contains_range(range) {
             return None;
         }
 
-        let relative_range = Range {
-            start: range.start - line_docs_range.start,
-            end: range.end - line_docs_range.start,
-        };
-        let range_len = TextSize::from((range.end - range.start) as u32);
+        let relative_range = range - line_docs_range.start();
 
         let &InFile { file_id, value: ref source } = &self.source[idx as usize];
         match source {
@@ -599,12 +587,10 @@ pub fn map(&self, range: Range<usize>) -> Option<InFile<TextRange>> {
                 let text_range = comment.syntax().text_range();
                 let range = TextRange::at(
                     text_range.start()
-                        + TextSize::from(
-                            (comment.prefix().len()
-                                + original_line_src_range.start
-                                + relative_range.start) as u32,
-                        ),
-                    text_range.len().min(range_len),
+                        + TextSize::try_from(comment.prefix().len()).ok()?
+                        + original_line_src_range.start()
+                        + relative_range.start(),
+                    text_range.len().min(range.len()),
                 );
                 Some(InFile { file_id, value: range })
             }
diff --git a/crates/ide/src/doc_links.rs b/crates/ide/src/doc_links.rs
index 9a9a4111387..2edd551cbca 100644
--- a/crates/ide/src/doc_links.rs
+++ b/crates/ide/src/doc_links.rs
@@ -1,6 +1,9 @@
 //! Extracts, resolves and rewrites links and intra-doc links in markdown documentation.
 
-use std::{convert::TryFrom, iter::once, ops::Range};
+use std::{
+    convert::{TryFrom, TryInto},
+    iter::once,
+};
 
 use itertools::Itertools;
 use pulldown_cmark::{BrokenLink, CowStr, Event, InlineStr, LinkType, Options, Parser, Tag};
@@ -15,7 +18,9 @@
     defs::{Definition, NameClass, NameRefClass},
     RootDatabase,
 };
-use syntax::{ast, match_ast, AstNode, SyntaxKind::*, SyntaxNode, SyntaxToken, TokenAtOffset, T};
+use syntax::{
+    ast, match_ast, AstNode, SyntaxKind::*, SyntaxNode, SyntaxToken, TextRange, TokenAtOffset, T,
+};
 
 use crate::{FilePosition, Semantics};
 
@@ -115,7 +120,7 @@ pub(crate) fn external_docs(
 /// Extracts all links from a given markdown text.
 pub(crate) fn extract_definitions_from_markdown(
     markdown: &str,
-) -> Vec<(Range<usize>, String, Option<hir::Namespace>)> {
+) -> Vec<(TextRange, String, Option<hir::Namespace>)> {
     Parser::new_with_broken_link_callback(
         markdown,
         Options::empty(),
@@ -126,7 +131,11 @@ pub(crate) fn extract_definitions_from_markdown(
         if let Event::Start(Tag::Link(_, target, title)) = event {
             let link = if target.is_empty() { title } else { target };
             let (link, ns) = parse_intra_doc_link(&link);
-            Some((range, link.to_string(), ns))
+            Some((
+                TextRange::new(range.start.try_into().ok()?, range.end.try_into().ok()?),
+                link.to_string(),
+                ns,
+            ))
         } else {
             None
         }