Merge: doc: fixed some typos and other misc. corrections
[nit.git] / src / location.nit
index e392ff0..448d17d 100644 (file)
@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# This module is used to model Nit source-file and locations in source-file
-package location
+# Nit source-file and locations in source-file
+module location
 
 # A raw text Nit source file
 class SourceFile
@@ -23,26 +23,44 @@ class SourceFile
        var filename: String
 
        # The content of the source
-       var string: String
+       var string: String is noinit
 
-       # Create a new sourcefile using a filename and a stream
-       init(filename: String, stream: IStream)
+       # The original stream used to initialize `string`
+       var stream: Reader
+
+       init
        do
-               self.filename = filename
                string = stream.read_all
                line_starts[0] = 0
        end
 
        # Create a new sourcefile using a dummy filename and a given content
-       init from_string(filename: String, string: String)
+       init from_string(filename: String, string: String) is
+               nosuper
        do
                self.filename = filename
                self.string = string
                line_starts[0] = 0
        end
 
-       # Position of each line start
-       var line_starts: Array[Int] = new Array[Int]
+       # Offset of each line start in the content `string`.
+       #
+       # Used for fast access to each line when rendering parts of the `string`.
+       var line_starts = new Array[Int]
+
+       # Extract a given line excluding the line-terminators characters.
+       #
+       # `line_number` starts at 1 for the first line.
+       fun get_line(line_number: Int): String do
+               if line_number > line_starts.length then return ""
+               var line_start = line_starts[line_number-1]
+               var line_end = line_start
+               var string = self.string
+               while line_end+1 < string.length and string.chars[line_end+1] != '\n' and string.chars[line_end+1] != '\r' do
+                       line_end += 1
+               end
+               return string.substring(line_start, line_end-line_start+1)
+       end
 end
 
 # A location inside a source file
@@ -50,36 +68,121 @@ class Location
        super Comparable
        redef type OTHER: Location
 
-       readable var _file: nullable SourceFile
-       readable var _line_start: Int
-       readable var _line_end: Int
-       readable var _column_start: Int
-       readable var _column_end: Int
-
-       init(f: nullable SourceFile, line_s: Int, line_e: Int, column_s: Int, column_e: Int) do
-               _file = f
-               _line_start = line_s
-               _line_end = line_e
-               _column_start = column_s
-               _column_end = column_e
+       # The associated source-file
+       var file: nullable SourceFile
+
+       # The starting line number (starting from 1)
+       #
+       # If `line_start==0` then the whole file is considered
+       var line_start: Int
+
+       # The stopping line number (starting from 1)
+       var line_end: Int
+
+       # Start of this location on `line_start`
+       #
+       # A `column_start` of 1 means the first column or character.
+       #
+       # If `column_start == 0` this location concerns the whole line.
+       #
+       # Require: `column_start >= 0`
+       var column_start: Int
+
+       # End of this location on `line_end`
+       var column_end: Int
+
+       # Builds a location instance from its string representation.
+       #
+       # Examples:
+       #
+       # ~~~
+       # var loc = new Location.from_string("location.nit:82,2--105,8")
+       # assert loc.to_s == "location.nit:82,2--105,8"
+       #
+       # loc = new Location.from_string("location.nit")
+       # assert loc.to_s == "location.nit"
+       #
+       # loc = new Location.from_string("location.nit:82,2")
+       # assert loc.to_s == "location.nit:82,2--0,0"
+       #
+       # loc = new Location.from_string("location.nit:82--105")
+       # assert loc.to_s == "location.nit:82,0--105,0"
+       #
+       # loc = new Location.from_string("location.nit:82,2--105")
+       # assert loc.to_s == "location.nit:82,2--105,0"
+       #
+       # loc = new Location.from_string("location.nit:82--105,8")
+       # assert loc.to_s == "location.nit:82,0--105,8"
+       # ~~~
+       init from_string(string: String) is
+               nosuper
+       do
+               self.line_start = 0
+               self.line_end = 0
+               self.column_start = 0
+               self.column_end = 0
+               # parses the location string and init position vars
+               var parts = string.split_with(":")
+               var filename = parts.shift
+               self.file = new SourceFile(filename, new FileReader.open(filename))
+               # split position
+               if parts.is_empty then return
+               var pos = parts.first.split_with("--")
+               # split start position
+               if pos.first.has(",") then
+                       var pos1 = pos.first.split_with(",")
+                       self.line_start = pos1[0].to_i
+                       if pos1.length > 1 then
+                               self.column_start = pos1[1].to_i
+                       end
+               else
+                       self.line_start = pos.first.to_i
+               end
+               # split end position
+               if pos.length <= 1 then return
+               if pos[1].has(",") then
+                       var pos2 = pos[1].split_with(",")
+                       if pos2.length > 1 then
+                               self.line_end = pos2[0].to_i
+                               self.column_end = pos2[1].to_i
+                       else
+                               self.line_end = self.line_start
+                               self.column_end = pos2[0].to_i
+                       end
+               else
+                       self.line_end = pos[1].to_i
+               end
+       end
+
+       # Initialize a location corresponding to an opaque file.
+       #
+       # The path is used as is and is not open nor read.
+       init opaque_file(path: String)
+       do
+               var source = new SourceFile.from_string(path, "")
+               init(source, 0, 0, 0, 0)
        end
 
+       # The index in the start character in the source
+       fun pstart: Int do return file.line_starts[line_start-1] + column_start-1
+
+       # The index on the end character in the source
+       fun pend: Int do return file.line_starts[line_end-1] + column_end-1
+
        # The verbatim associated text in the source-file
        fun text: String
        do
                var res = self.text_cache
                if res != null then return res
                var l = self
-               var pstart = l.file.line_starts[l.line_start-1] + l.column_start-1
-               var pend = l.file.line_starts[l.line_end-1] + l.column_end-1
+               var pstart = self.pstart
+               var pend = self.pend
                res = l.file.string.substring(pstart, pend-pstart+1)
                self.text_cache = res
                return res
        end
 
-       private var text_cache: nullable String
-
-       init with_file(f: SourceFile) do init(f,0,0,0,0)
+       private var text_cache: nullable String = null
 
        redef fun ==(other: nullable Object): Bool do
                if other == null then return false
@@ -94,6 +197,7 @@ class Location
                return true
        end
 
+       # Is `self` included (or equals) to `loc`?
        fun located_in(loc: nullable Location): Bool do
                if loc == null then return false
 
@@ -104,7 +208,7 @@ class Location
 
                if line_start == loc.line_start then
                        if column_start < loc.column_start then return false
-                       if column_start > loc.column_end then return false
+                       if line_start == loc.line_end and column_start > loc.column_end then return false
                end
 
                if line_end == loc.line_end and column_end > loc.column_end then return false
@@ -116,9 +220,12 @@ class Location
                var file_part = ""
                if file != null then
                        file_part = file.filename
-                       if file.filename.length > 0 then file_part += ":"
                end
 
+               if line_start <= 0 then return file_part
+
+               if file != null and file.filename.length > 0 then file_part += ":"
+
                if line_start == line_end then
                        if column_start == column_end then
                                return "{file_part}{line_start},{column_start}"
@@ -130,6 +237,9 @@ class Location
                end
        end
 
+       # Return a location message according to an observer.
+       #
+       # Currently, if both are in the same file, the file information is not present in the result.
        fun relative_to(loc: nullable Location): String do
                var relative: Location
                if loc != null and loc.file == self.file then
@@ -152,26 +262,33 @@ class Location
                return column_end < other.column_end
        end
 
-       # Return the associated line with the location highlihted with color and a carret under the starting position
-       # `color' must be and terminal escape sequence used as "{escape}[{color}m;"
-       # "0;31" for red
-       # "1;31" for bright red
-       # "0;32" for green
+       # Return the associated line with the location highlighted with color and a caret under the starting position
+       # `color` must be and terminal escape sequence used as `"{escape}[{color}m;"`
+       # * `"0;31"` for red
+       # * `"1;31"` for bright red
+       # * `"0;32"` for green
        fun colored_line(color: String): String
        do
-               var esc = 27.ascii
+               var esc = 27.code_point
                var def = "{esc}[0m"
                var col = "{esc}[{color}m"
 
                var l = self
                var i = l.line_start
+               if i <= 0 then return ""
+
                var line_start = l.file.line_starts[i-1]
                var line_end = line_start
                var string = l.file.string
-               while line_end+1 < string.length and string[line_end+1] != '\n' and string[line_end+1] != '\r' do
+               while line_end+1 < string.length and string.chars[line_end+1] != '\n' and string.chars[line_end+1] != '\r' do
                        line_end += 1
                end
-               var lstart = string.substring(line_start, l.column_start - 1)
+               var lstart
+               if l.column_start > 0 then
+                       lstart = string.substring(line_start, l.column_start - 1)
+               else
+                       lstart = ""
+               end
                var cend
                if i != l.line_end then
                        cend = line_end - line_start + 1
@@ -187,9 +304,9 @@ class Location
                        lmid = ""
                        lend = ""
                end
-               var indent = new Buffer
+               var indent = new FlatBuffer
                for j in [line_start..line_start+l.column_start-1[ do
-                       if string[j] == '\t' then
+                       if string.chars[j] == '\t' then
                                indent.add '\t'
                        else
                                indent.add ' '
@@ -198,4 +315,3 @@ class Location
                return "\t{lstart}{col}{lmid}{def}{lend}\n\t{indent}^"
        end
 end
-