From 58674dc3c415142dbdd93b990d9f5b4fe10bef69 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Sat, 15 Sep 2018 20:29:22 +0300
Subject: ModuleTreeDescriptor

---
 crates/libanalysis/src/db/mod.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'crates/libanalysis/src/db')

diff --git a/crates/libanalysis/src/db/mod.rs b/crates/libanalysis/src/db/mod.rs
index a775b5f75..1111a4f87 100644
--- a/crates/libanalysis/src/db/mod.rs
+++ b/crates/libanalysis/src/db/mod.rs
@@ -68,22 +68,32 @@ pub(crate) fn file_set(ctx: QueryCtx) -> Arc<(Vec<FileId>, FileResolverImp)> {
 pub(crate) use self::queries::file_syntax;
 
 mod queries {
+    use std::sync::Arc;
     use libsyntax2::File;
+    use libeditor::LineIndex;
     use {FileId};
     use super::{Query, QueryCtx, QueryRegistry, file_text};
 
     pub(crate) fn register_queries(reg: &mut QueryRegistry) {
-        reg.add(FILE_SYNTAX, "FILE_SYNTAX")
+        reg.add(FILE_SYNTAX, "FILE_SYNTAX");
+        reg.add(FILE_LINES, "FILE_LINES");
     }
 
     pub(crate) fn file_syntax(ctx: QueryCtx, file_id: FileId) -> File {
         (&*ctx.get(FILE_SYNTAX, file_id)).clone()
     }
+    pub(crate) fn file_lines(ctx: QueryCtx, file_id: FileId) -> Arc<LineIndex> {
+        ctx.get(FILE_LINES, file_id)
+    }
 
     pub(super) const FILE_SYNTAX: Query<FileId, File> = Query(16, |ctx, file_id: &FileId| {
         let text = file_text(ctx, *file_id);
         File::parse(&*text)
     });
+    pub(super) const FILE_LINES: Query<FileId, LineIndex> = Query(17, |ctx, file_id: &FileId| {
+        let text = file_text(ctx, *file_id);
+        LineIndex::new(&*text)
+    });
 }
 
 impl QueryRegistry {
-- 
cgit v1.2.3