#lang pyret/library

provide *
provide-types *

import global as _
import valueskeleton as VS

data Srcloc:
  | builtin(module-name) with:
    method format(self, _):
      "<builtin " + self.module-name + ">"
    end,
    method key(self): self.module-name end,
    method same-file(self, other):
      is-builtin(other) and (other.module-name == self.module-name)
    end,
    method before(self, other):
      cases(Srcloc) other:
        | builtin(module-name) => (self.module-name < other.module-name)
        | srcloc(_, _, _, _, _, _, _) => false
      end
    end,
    method contains(self, other):
      false
    end,
    method is-builtin(self): true end
  | srcloc(
        source :: String,
        start-line :: Number,
        start-column :: Number,
        start-char :: Number,
        end-line :: Number,
        end-column :: Number,
        end-char :: Number
      ) with:
    method format(self, show-file):
      doc: "Returns either 'file: line, col' or just 'line, col', depending on the show-file flag"
      # if show-file: self.source + ": " else: "" end
      #   + "line " + tostring(self.start-line)
      #   + ", column " + tostring(self.start-column)
      if show-file:
        self.source + ":" + tostring(self.start-line) + ":" + tostring(self.start-column)
          + "-" + tostring(self.end-line) + ":" + tostring(self.end-column)
      else:
        "line " + tostring(self.start-line) + ", column " + tostring(self.start-column)
      end
    end,
    method key(self): self.source + ":" + tostring(self.start-char) + "-" + tostring(self.end-char) end,
    method same-file(self, other :: Srcloc):
      is-srcloc(other) and (self.source == other.source)
    end,
    method before(self, other :: Srcloc):
      doc: "Returns true if this location comes before the other one, assuming they come from the same file"
      cases(Srcloc) other:
        | builtin(_) => true
        | else => self.start-char < other.start-char
      end
    end,
    method at-start(self):
      srcloc(self.source,
        self.start-line, self.start-column, self.start-char,
        self.start-line, self.start-column, self.start-char)
    end,
    method at-end(self):
      srcloc(self.source,
        self.end-line, self.end-column, self.end-char,
        self.end-line, self.end-column, self.end-char)
    end,
    method _plus(self, other :: Srcloc%(is-srcloc)):
      # Note: assumes that both locations are from same file
      if self.start-char <= other.start-char:
        if self.end-char >= other.end-char:
          self
        else:
          srcloc(self.source,
            self.start-line, self.start-column, self.start-char,
            other.end-line, other.end-column, other.end-char)
        end
      else:
        if self.end-char > other.end-char:
          srcloc(self.source,
            other.start-line, other.start-column, other.start-char,
            self.end-line, self.end-column, self.end-char)
        else:
          other
        end
      end
    end,
    method upto(self, other :: Srcloc%(is-srcloc)):
      # Note: assumes that both locations are from same file
      if self.start-char <= other.end-char:
        srcloc(self.source,
          self.start-line, self.start-column, self.start-char,
          other.start-line, other.start-column, other.start-char)
      else:
        self
      end
    end,
    method upto-end(self, other :: Srcloc%(is-srcloc)):
      # Note: assumes that both locations are from same file
      if self.start-char <= other.end-char:
        srcloc(self.source,
          self.start-line, self.start-column, self.start-char,
          other.end-line, other.end-column, other.end-char)
      else:
        self
      end
    end,
    method contains(self, other):
      is-srcloc(other)
        and (self.start-line <= other.start-line)
        and (self.start-char <= other.start-char)
        and (self.end-line >= other.end-line)
        and (self.end-char >= other.end-char)
    end,
    method is-builtin(self): false end
sharing:
  method after(self, other): other.before(self) end
end