mirror of
https://github.com/google/go-jsonnet.git
synced 2025-08-08 07:17:12 +02:00
We used to treat dummy paths like <stdin>, <std>, <extvar> as real import locations, which causes obvious problem for importers. After this change we consistently pass "" (an empty string) as location for non-imported paths. We exposed new functions to properly handle all paths. The first function family, EvaluateFile* which allow evaluating a Jsonnet program at a specified importable path. It should be used in most cases. The second function family, EvaluateAnonymousSnippet* allows evaluating a snippet without an importable path. It should be used for situations like the code passed as a commandline argument, stdin, etc. The old function family, EvaluateSnippet* is now deprecated, but works the same as before, i.e. the passed filenames are treated as imported paths. Changes are required to custom importers to make sure they satisfy the refined contract. Fixes #329.
975 lines
24 KiB
Go
975 lines
24 KiB
Go
/*
|
|
Copyright 2016 Google Inc. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package parser
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"unicode/utf8"
|
|
|
|
"github.com/google/go-jsonnet/ast"
|
|
"github.com/google/go-jsonnet/internal/errors"
|
|
)
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Token
|
|
|
|
type tokenKind int
|
|
|
|
const (
|
|
// Symbols
|
|
tokenBraceL tokenKind = iota
|
|
tokenBraceR
|
|
tokenBracketL
|
|
tokenBracketR
|
|
tokenComma
|
|
tokenDollar
|
|
tokenDot
|
|
tokenParenL
|
|
tokenParenR
|
|
tokenSemicolon
|
|
|
|
// Arbitrary length lexemes
|
|
tokenIdentifier
|
|
tokenNumber
|
|
tokenOperator
|
|
tokenStringBlock
|
|
tokenStringDouble
|
|
tokenStringSingle
|
|
tokenVerbatimStringDouble
|
|
tokenVerbatimStringSingle
|
|
|
|
// Keywords
|
|
tokenAssert
|
|
tokenElse
|
|
tokenError
|
|
tokenFalse
|
|
tokenFor
|
|
tokenFunction
|
|
tokenIf
|
|
tokenImport
|
|
tokenImportStr
|
|
tokenIn
|
|
tokenLocal
|
|
tokenNullLit
|
|
tokenSelf
|
|
tokenSuper
|
|
tokenTailStrict
|
|
tokenThen
|
|
tokenTrue
|
|
|
|
// A special token that holds line/column information about the end of the
|
|
// file.
|
|
tokenEndOfFile
|
|
)
|
|
|
|
var tokenKindStrings = []string{
|
|
// Symbols
|
|
tokenBraceL: `"{"`,
|
|
tokenBraceR: `"}"`,
|
|
tokenBracketL: `"["`,
|
|
tokenBracketR: `"]"`,
|
|
tokenComma: `","`,
|
|
tokenDollar: `"$"`,
|
|
tokenDot: `"."`,
|
|
tokenParenL: `"("`,
|
|
tokenParenR: `")"`,
|
|
tokenSemicolon: `";"`,
|
|
|
|
// Arbitrary length lexemes
|
|
tokenIdentifier: "IDENTIFIER",
|
|
tokenNumber: "NUMBER",
|
|
tokenOperator: "OPERATOR",
|
|
tokenStringBlock: "STRING_BLOCK",
|
|
tokenStringDouble: "STRING_DOUBLE",
|
|
tokenStringSingle: "STRING_SINGLE",
|
|
tokenVerbatimStringDouble: "VERBATIM_STRING_DOUBLE",
|
|
tokenVerbatimStringSingle: "VERBATIM_STRING_SINGLE",
|
|
|
|
// Keywords
|
|
tokenAssert: "assert",
|
|
tokenElse: "else",
|
|
tokenError: "error",
|
|
tokenFalse: "false",
|
|
tokenFor: "for",
|
|
tokenFunction: "function",
|
|
tokenIf: "if",
|
|
tokenImport: "import",
|
|
tokenImportStr: "importstr",
|
|
tokenIn: "in",
|
|
tokenLocal: "local",
|
|
tokenNullLit: "null",
|
|
tokenSelf: "self",
|
|
tokenSuper: "super",
|
|
tokenTailStrict: "tailstrict",
|
|
tokenThen: "then",
|
|
tokenTrue: "true",
|
|
|
|
// A special token that holds line/column information about the end of the
|
|
// file.
|
|
tokenEndOfFile: "end of file",
|
|
}
|
|
|
|
var tokenHasContent = map[tokenKind]bool{
|
|
tokenIdentifier: true,
|
|
tokenNumber: true,
|
|
tokenOperator: true,
|
|
tokenStringBlock: true,
|
|
tokenStringDouble: true,
|
|
tokenStringSingle: true,
|
|
tokenVerbatimStringDouble: true,
|
|
tokenVerbatimStringSingle: true,
|
|
}
|
|
|
|
func (tk tokenKind) String() string {
|
|
if tk < 0 || int(tk) >= len(tokenKindStrings) {
|
|
panic(fmt.Sprintf("INTERNAL ERROR: Unknown token kind:: %d", tk))
|
|
}
|
|
return tokenKindStrings[tk]
|
|
}
|
|
|
|
type token struct {
|
|
kind tokenKind // The type of the token
|
|
fodder ast.Fodder // Any fodder that occurs before this token
|
|
data string // Content of the token if it is not a keyword
|
|
|
|
// Extra info for when kind == tokenStringBlock
|
|
stringBlockIndent string // The sequence of whitespace that indented the block.
|
|
stringBlockTermIndent string // This is always fewer whitespace characters than in stringBlockIndent.
|
|
|
|
loc ast.LocationRange
|
|
}
|
|
|
|
// Tokens is a slice of token structs.
|
|
type Tokens []token
|
|
|
|
func (t *token) String() string {
|
|
if t.data == "" {
|
|
return t.kind.String()
|
|
} else if tokenHasContent[t.kind] {
|
|
return fmt.Sprintf("(%v, \"%v\")", t.kind, t.data)
|
|
} else {
|
|
return fmt.Sprintf("\"%v\"", t.data)
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Helpers
|
|
|
|
func isUpper(r rune) bool {
|
|
return r >= 'A' && r <= 'Z'
|
|
}
|
|
|
|
func isLower(r rune) bool {
|
|
return r >= 'a' && r <= 'z'
|
|
}
|
|
|
|
func isNumber(r rune) bool {
|
|
return r >= '0' && r <= '9'
|
|
}
|
|
|
|
func isIdentifierFirst(r rune) bool {
|
|
return isUpper(r) || isLower(r) || r == '_'
|
|
}
|
|
|
|
func isIdentifier(r rune) bool {
|
|
return isIdentifierFirst(r) || isNumber(r)
|
|
}
|
|
|
|
func isSymbol(r rune) bool {
|
|
switch r {
|
|
case '!', '$', ':', '~', '+', '-', '&', '|', '^', '=', '<', '>', '*', '/', '%':
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func isHorizontalWhitespace(r rune) bool {
|
|
return r == ' ' || r == '\t' || r == '\r'
|
|
}
|
|
|
|
func isWhitespace(r rune) bool {
|
|
return r == '\n' || isHorizontalWhitespace(r)
|
|
}
|
|
|
|
// stripWhitespace strips whitespace from both ends of a string, but only up to
|
|
// margin on the left hand side. E.g., stripWhitespace(" foo ", 1) == " foo".
|
|
func stripWhitespace(s string, margin int) string {
|
|
runes := []rune(s)
|
|
if len(s) == 0 {
|
|
return s // Avoid underflow below.
|
|
}
|
|
i := 0
|
|
for i < len(runes) && isHorizontalWhitespace(runes[i]) && i < margin {
|
|
i++
|
|
}
|
|
j := len(runes)
|
|
for j > i && isHorizontalWhitespace(runes[j-1]) {
|
|
j--
|
|
}
|
|
return string(runes[i:j])
|
|
}
|
|
|
|
// Split a string by \n and also strip left (up to margin) & right whitespace from each line. */
|
|
func lineSplit(s string, margin int) []string {
|
|
var ret []string
|
|
var buf bytes.Buffer
|
|
for _, r := range s {
|
|
if r == '\n' {
|
|
ret = append(ret, stripWhitespace(buf.String(), margin))
|
|
buf.Reset()
|
|
} else {
|
|
buf.WriteRune(r)
|
|
}
|
|
}
|
|
return append(ret, stripWhitespace(buf.String(), margin))
|
|
}
|
|
|
|
// Check that b has at least the same whitespace prefix as a and returns the
|
|
// amount of this whitespace, otherwise returns 0. If a has no whitespace
|
|
// prefix than return 0.
|
|
func checkWhitespace(a, b string) int {
|
|
i := 0
|
|
for ; i < len(a); i++ {
|
|
if a[i] != ' ' && a[i] != '\t' {
|
|
// a has run out of whitespace and b matched up to this point. Return
|
|
// result.
|
|
return i
|
|
}
|
|
if i >= len(b) {
|
|
// We ran off the edge of b while a still has whitespace. Return 0 as
|
|
// failure.
|
|
return 0
|
|
}
|
|
if a[i] != b[i] {
|
|
// a has whitespace but b does not. Return 0 as failure.
|
|
return 0
|
|
}
|
|
}
|
|
// We ran off the end of a and b kept up
|
|
return i
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Lexer
|
|
|
|
type position struct {
|
|
byteNo int // Byte position of last rune read
|
|
lineNo int // Line number
|
|
lineStart int // Rune position of the last newline
|
|
}
|
|
|
|
type lexer struct {
|
|
diagnosticFilename ast.DiagnosticFileName // The file name being lexed, only used for errors
|
|
importedFilename string // Imported filename, used for resolving relative imports
|
|
input string // The input string
|
|
source *ast.Source
|
|
|
|
pos position // Current position in input
|
|
|
|
tokens Tokens // The tokens that we've generated so far
|
|
|
|
// Information about the token we are working on right now
|
|
fodder ast.Fodder
|
|
tokenStart int
|
|
tokenStartLoc ast.Location
|
|
|
|
// Was the last rune the first rune on a line (ignoring initial whitespace).
|
|
freshLine bool
|
|
}
|
|
|
|
const lexEOF = -1
|
|
|
|
func makeLexer(diagnosticFilename ast.DiagnosticFileName, importedFilename, input string) *lexer {
|
|
return &lexer{
|
|
input: input,
|
|
diagnosticFilename: diagnosticFilename,
|
|
importedFilename: importedFilename,
|
|
source: ast.BuildSource(diagnosticFilename, input),
|
|
pos: position{byteNo: 0, lineNo: 1, lineStart: 0},
|
|
tokenStartLoc: ast.Location{Line: 1, Column: 1},
|
|
freshLine: true,
|
|
}
|
|
}
|
|
|
|
// next returns the next rune in the input.
|
|
func (l *lexer) next() rune {
|
|
if int(l.pos.byteNo) >= len(l.input) {
|
|
return lexEOF
|
|
}
|
|
r, w := utf8.DecodeRuneInString(l.input[l.pos.byteNo:])
|
|
l.pos.byteNo += w
|
|
if r == '\n' {
|
|
l.pos.lineStart = l.pos.byteNo
|
|
l.pos.lineNo++
|
|
l.freshLine = true
|
|
} else if l.freshLine {
|
|
if !isWhitespace(r) {
|
|
l.freshLine = false
|
|
}
|
|
}
|
|
return r
|
|
}
|
|
|
|
func (l *lexer) acceptN(n int) {
|
|
for i := 0; i < n; i++ {
|
|
l.next()
|
|
}
|
|
}
|
|
|
|
// peek returns but does not consume the next rune in the input.
|
|
func (l *lexer) peek() rune {
|
|
if int(l.pos.byteNo) >= len(l.input) {
|
|
return lexEOF
|
|
}
|
|
r, _ := utf8.DecodeRuneInString(l.input[l.pos.byteNo:])
|
|
return r
|
|
}
|
|
|
|
func locationFromPosition(pos position) ast.Location {
|
|
return ast.Location{Line: pos.lineNo, Column: pos.byteNo - pos.lineStart + 1}
|
|
}
|
|
|
|
func (l *lexer) location() ast.Location {
|
|
return locationFromPosition(l.pos)
|
|
}
|
|
|
|
// Reset the current working token start to the current cursor position. This
|
|
// may throw away some characters. This does not throw away any accumulated
|
|
// fodder.
|
|
func (l *lexer) resetTokenStart() {
|
|
l.tokenStart = l.pos.byteNo
|
|
l.tokenStartLoc = l.location()
|
|
}
|
|
|
|
func (l *lexer) emitFullToken(kind tokenKind, data, stringBlockIndent, stringBlockTermIndent string) {
|
|
l.tokens = append(l.tokens, token{
|
|
kind: kind,
|
|
fodder: l.fodder,
|
|
data: data,
|
|
stringBlockIndent: stringBlockIndent,
|
|
stringBlockTermIndent: stringBlockTermIndent,
|
|
loc: ast.MakeLocationRange(l.importedFilename, l.source, l.tokenStartLoc, l.location()),
|
|
})
|
|
l.fodder = ast.Fodder{}
|
|
}
|
|
|
|
func (l *lexer) emitToken(kind tokenKind) {
|
|
l.emitFullToken(kind, l.input[l.tokenStart:l.pos.byteNo], "", "")
|
|
l.resetTokenStart()
|
|
}
|
|
|
|
func (l *lexer) addFodder(kind ast.FodderKind, blanks int, indent int, comment []string) {
|
|
elem := ast.MakeFodderElement(kind, blanks, indent, comment)
|
|
l.fodder = append(l.fodder, elem)
|
|
}
|
|
|
|
func (l *lexer) addFodderSafe(kind ast.FodderKind, blanks int, indent int, comment []string) {
|
|
elem := ast.MakeFodderElement(kind, blanks, indent, comment)
|
|
ast.FodderAppend(&l.fodder, elem)
|
|
}
|
|
|
|
func (l *lexer) makeStaticErrorPoint(msg string, loc ast.Location) errors.StaticError {
|
|
return errors.MakeStaticError(msg, ast.MakeLocationRange(l.importedFilename, l.source, loc, loc))
|
|
}
|
|
|
|
// lexWhitespace consumes all whitespace and returns the number of \n and number of
|
|
// spaces after last \n. It also converts \t to spaces.
|
|
// The parameter 'r' is the rune that begins the whitespace.
|
|
func (l *lexer) lexWhitespace() (int, int) {
|
|
r := l.peek()
|
|
indent := 0
|
|
newLines := 0
|
|
for ; isWhitespace(r); r = l.peek() {
|
|
l.next()
|
|
switch r {
|
|
case '\r':
|
|
// Ignore.
|
|
|
|
case '\n':
|
|
indent = 0
|
|
newLines++
|
|
|
|
case ' ':
|
|
indent++
|
|
|
|
// This only works for \t at the beginning of lines, but we strip it everywhere else
|
|
// anyway. The only case where this will cause a problem is spaces followed by \t
|
|
// at the beginning of a line. However that is rare, ill-advised, and if re-indentation
|
|
// is enabled it will be fixed later.
|
|
case '\t':
|
|
indent += 8
|
|
}
|
|
}
|
|
return newLines, indent
|
|
}
|
|
|
|
// lexUntilNewLine consumes all text until the end of the line and returns the
|
|
// number of newlines after that as well as the next indent.
|
|
func (l *lexer) lexUntilNewline() (string, int, int) {
|
|
// Compute 'text'.
|
|
var buf bytes.Buffer
|
|
lastNonSpace := 0
|
|
for r := l.peek(); r != lexEOF && r != '\n'; r = l.peek() {
|
|
l.next()
|
|
buf.WriteRune(r)
|
|
if !isHorizontalWhitespace(r) {
|
|
lastNonSpace = buf.Len()
|
|
}
|
|
}
|
|
// Trim whitespace off the end.
|
|
buf.Truncate(lastNonSpace)
|
|
text := buf.String()
|
|
|
|
// Consume the '\n' and following indent.
|
|
var newLines int
|
|
newLines, indent := l.lexWhitespace()
|
|
blanks := 0
|
|
if newLines > 0 {
|
|
blanks = newLines - 1
|
|
}
|
|
return text, blanks, indent
|
|
}
|
|
|
|
// lexNumber will consume a number and emit a token. It is assumed
|
|
// that the next rune to be served by the lexer will be a leading digit.
|
|
func (l *lexer) lexNumber() error {
|
|
// This function should be understood with reference to the linked image:
|
|
// http://www.json.org/number.gif
|
|
|
|
// Note, we deviate from the json.org documentation as follows:
|
|
// There is no reason to lex negative numbers as atomic tokens, it is better to parse them
|
|
// as a unary operator combined with a numeric literal. This avoids x-1 being tokenized as
|
|
// <identifier> <number> instead of the intended <identifier> <binop> <number>.
|
|
|
|
type numLexState int
|
|
const (
|
|
numBegin numLexState = iota
|
|
numAfterZero
|
|
numAfterOneToNine
|
|
numAfterDot
|
|
numAfterDigit
|
|
numAfterE
|
|
numAfterExpSign
|
|
numAfterExpDigit
|
|
)
|
|
|
|
state := numBegin
|
|
|
|
outerLoop:
|
|
for {
|
|
r := l.peek()
|
|
switch state {
|
|
case numBegin:
|
|
switch {
|
|
case r == '0':
|
|
state = numAfterZero
|
|
case r >= '1' && r <= '9':
|
|
state = numAfterOneToNine
|
|
default:
|
|
// The caller should ensure the first rune is a digit.
|
|
panic("Couldn't lex number")
|
|
}
|
|
case numAfterZero:
|
|
switch r {
|
|
case '.':
|
|
state = numAfterDot
|
|
case 'e', 'E':
|
|
state = numAfterE
|
|
default:
|
|
break outerLoop
|
|
}
|
|
case numAfterOneToNine:
|
|
switch {
|
|
case r == '.':
|
|
state = numAfterDot
|
|
case r == 'e' || r == 'E':
|
|
state = numAfterE
|
|
case r >= '0' && r <= '9':
|
|
state = numAfterOneToNine
|
|
default:
|
|
break outerLoop
|
|
}
|
|
case numAfterDot:
|
|
switch {
|
|
case r >= '0' && r <= '9':
|
|
state = numAfterDigit
|
|
default:
|
|
return l.makeStaticErrorPoint(
|
|
fmt.Sprintf("Couldn't lex number, junk after decimal point: %v", strconv.QuoteRuneToASCII(r)),
|
|
l.location())
|
|
}
|
|
case numAfterDigit:
|
|
switch {
|
|
case r == 'e' || r == 'E':
|
|
state = numAfterE
|
|
case r >= '0' && r <= '9':
|
|
state = numAfterDigit
|
|
default:
|
|
break outerLoop
|
|
}
|
|
case numAfterE:
|
|
switch {
|
|
case r == '+' || r == '-':
|
|
state = numAfterExpSign
|
|
case r >= '0' && r <= '9':
|
|
state = numAfterExpDigit
|
|
default:
|
|
return l.makeStaticErrorPoint(
|
|
fmt.Sprintf("Couldn't lex number, junk after 'E': %v", strconv.QuoteRuneToASCII(r)),
|
|
l.location())
|
|
}
|
|
case numAfterExpSign:
|
|
if r >= '0' && r <= '9' {
|
|
state = numAfterExpDigit
|
|
} else {
|
|
return l.makeStaticErrorPoint(
|
|
fmt.Sprintf("Couldn't lex number, junk after exponent sign: %v", strconv.QuoteRuneToASCII(r)),
|
|
l.location())
|
|
}
|
|
|
|
case numAfterExpDigit:
|
|
if r >= '0' && r <= '9' {
|
|
state = numAfterExpDigit
|
|
} else {
|
|
break outerLoop
|
|
}
|
|
}
|
|
l.next()
|
|
}
|
|
|
|
l.emitToken(tokenNumber)
|
|
return nil
|
|
}
|
|
|
|
// getTokenKindFromID will return a keyword if the identifier string is
|
|
// recognised as one, otherwise it will return tokenIdentifier.
|
|
func getTokenKindFromID(str string) tokenKind {
|
|
switch str {
|
|
case "assert":
|
|
return tokenAssert
|
|
case "else":
|
|
return tokenElse
|
|
case "error":
|
|
return tokenError
|
|
case "false":
|
|
return tokenFalse
|
|
case "for":
|
|
return tokenFor
|
|
case "function":
|
|
return tokenFunction
|
|
case "if":
|
|
return tokenIf
|
|
case "import":
|
|
return tokenImport
|
|
case "importstr":
|
|
return tokenImportStr
|
|
case "in":
|
|
return tokenIn
|
|
case "local":
|
|
return tokenLocal
|
|
case "null":
|
|
return tokenNullLit
|
|
case "self":
|
|
return tokenSelf
|
|
case "super":
|
|
return tokenSuper
|
|
case "tailstrict":
|
|
return tokenTailStrict
|
|
case "then":
|
|
return tokenThen
|
|
case "true":
|
|
return tokenTrue
|
|
default:
|
|
// Not a keyword, assume it is an identifier
|
|
return tokenIdentifier
|
|
}
|
|
}
|
|
|
|
// IsValidIdentifier is true if the string could be a valid identifier.
|
|
func IsValidIdentifier(str string) bool {
|
|
if len(str) == 0 {
|
|
return false
|
|
}
|
|
for i, r := range str {
|
|
if i == 0 {
|
|
if !isIdentifierFirst(r) {
|
|
return false
|
|
}
|
|
} else {
|
|
if !isIdentifier(r) {
|
|
return false
|
|
}
|
|
}
|
|
}
|
|
return getTokenKindFromID(str) == tokenIdentifier
|
|
}
|
|
|
|
// lexIdentifier will consume an identifer and emit a token. It is assumed
|
|
// that the next rune to be served by the lexer will not be a leading digit.
|
|
// This may emit a keyword or an identifier.
|
|
func (l *lexer) lexIdentifier() {
|
|
r := l.peek()
|
|
if !isIdentifierFirst(r) {
|
|
panic("Unexpected character in lexIdentifier")
|
|
}
|
|
for ; r != lexEOF; r = l.peek() {
|
|
if !isIdentifier(r) {
|
|
break
|
|
}
|
|
l.next()
|
|
}
|
|
l.emitToken(getTokenKindFromID(l.input[l.tokenStart:l.pos.byteNo]))
|
|
}
|
|
|
|
// lexSymbol will lex a token that starts with a symbol. This could be a
|
|
// C or C++ comment, block quote or an operator. This function assumes that the next
|
|
// rune to be served by the lexer will be the first rune of the new token.
|
|
func (l *lexer) lexSymbol() error {
|
|
// freshLine is reset by next() so cache it here.
|
|
freshLine := l.freshLine
|
|
r := l.next()
|
|
|
|
// Single line C++ style comment
|
|
if r == '#' || (r == '/' && l.peek() == '/') {
|
|
comment, blanks, indent := l.lexUntilNewline()
|
|
var k ast.FodderKind
|
|
if freshLine {
|
|
k = ast.FodderParagraph
|
|
} else {
|
|
k = ast.FodderLineEnd
|
|
}
|
|
l.addFodder(k, blanks, indent, []string{string(r) + comment})
|
|
return nil
|
|
}
|
|
|
|
// C style comment (could be interstitial or paragraph comment)
|
|
if r == '/' && l.peek() == '*' {
|
|
margin := l.pos.byteNo - l.pos.lineStart - 1
|
|
commentStartLoc := l.tokenStartLoc
|
|
|
|
//nolint:ineffassign,staticcheck
|
|
r := l.next() // consume the initial '*'
|
|
for r = l.next(); r != '*' || l.peek() != '/'; r = l.next() {
|
|
if r == lexEOF {
|
|
return l.makeStaticErrorPoint(
|
|
"Multi-line comment has no terminating */",
|
|
commentStartLoc)
|
|
}
|
|
}
|
|
|
|
l.next() // Consume trailing '/'
|
|
// Includes the "/*" and "*/".
|
|
comment := l.input[l.tokenStart:l.pos.byteNo]
|
|
|
|
newLinesAfter, indentAfter := l.lexWhitespace()
|
|
if !strings.ContainsRune(comment, '\n') {
|
|
l.addFodder(ast.FodderInterstitial, 0, 0, []string{comment})
|
|
if newLinesAfter > 0 {
|
|
l.addFodder(ast.FodderLineEnd, newLinesAfter-1, indentAfter, []string{})
|
|
}
|
|
} else {
|
|
lines := lineSplit(comment, margin)
|
|
if lines[0][0] != '/' {
|
|
panic(fmt.Sprintf("Invalid parsing of C style comment %v", lines))
|
|
}
|
|
// Little hack to support FodderParagraphs with * down the LHS:
|
|
// Add a space to lines that start with a '*'
|
|
allStar := true
|
|
for _, l := range lines {
|
|
if len(l) == 0 || l[0] != '*' {
|
|
allStar = false
|
|
}
|
|
}
|
|
if allStar {
|
|
for i := range lines {
|
|
if lines[i][0] == '*' {
|
|
lines[i] = " " + lines[i]
|
|
}
|
|
}
|
|
}
|
|
if newLinesAfter == 0 {
|
|
// Ensure a line end after the paragraph.
|
|
newLinesAfter = 1
|
|
indentAfter = 0
|
|
}
|
|
l.addFodderSafe(ast.FodderParagraph, newLinesAfter-1, indentAfter, lines)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if r == '|' && strings.HasPrefix(l.input[l.pos.byteNo:], "||") {
|
|
commentStartLoc := l.tokenStartLoc
|
|
l.acceptN(2) // Skip "||"
|
|
var cb bytes.Buffer
|
|
|
|
// Skip whitespace
|
|
for r = l.next(); r == ' ' || r == '\t' || r == '\r'; r = l.next() {
|
|
}
|
|
|
|
// Skip \n
|
|
if r != '\n' {
|
|
return l.makeStaticErrorPoint("Text block requires new line after |||.",
|
|
commentStartLoc)
|
|
}
|
|
|
|
// Process leading blank lines before calculating stringBlockIndent
|
|
for r = l.peek(); r == '\n'; r = l.peek() {
|
|
l.next()
|
|
cb.WriteRune(r)
|
|
}
|
|
numWhiteSpace := checkWhitespace(l.input[l.pos.byteNo:], l.input[l.pos.byteNo:])
|
|
stringBlockIndent := l.input[l.pos.byteNo : l.pos.byteNo+numWhiteSpace]
|
|
if numWhiteSpace == 0 {
|
|
return l.makeStaticErrorPoint("Text block's first line must start with whitespace",
|
|
commentStartLoc)
|
|
}
|
|
|
|
for {
|
|
if numWhiteSpace <= 0 {
|
|
panic("Unexpected value for numWhiteSpace")
|
|
}
|
|
l.acceptN(numWhiteSpace)
|
|
for r = l.next(); r != '\n'; r = l.next() {
|
|
if r == lexEOF {
|
|
return l.makeStaticErrorPoint("Unexpected EOF", commentStartLoc)
|
|
}
|
|
cb.WriteRune(r)
|
|
}
|
|
cb.WriteRune('\n')
|
|
|
|
// Skip any blank lines
|
|
for r = l.peek(); r == '\n'; r = l.peek() {
|
|
l.next()
|
|
cb.WriteRune(r)
|
|
}
|
|
|
|
// Look at the next line
|
|
numWhiteSpace = checkWhitespace(stringBlockIndent, l.input[l.pos.byteNo:])
|
|
if numWhiteSpace == 0 {
|
|
// End of the text block
|
|
var stringBlockTermIndent string
|
|
for r = l.peek(); r == ' ' || r == '\t'; r = l.peek() {
|
|
l.next()
|
|
stringBlockTermIndent += string(r)
|
|
}
|
|
if !strings.HasPrefix(l.input[l.pos.byteNo:], "|||") {
|
|
return l.makeStaticErrorPoint("Text block not terminated with |||", commentStartLoc)
|
|
}
|
|
l.acceptN(3) // Skip '|||'
|
|
l.emitFullToken(tokenStringBlock, cb.String(),
|
|
stringBlockIndent, stringBlockTermIndent)
|
|
l.resetTokenStart()
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
// Assume any string of symbols is a single operator.
|
|
for r = l.peek(); isSymbol(r); r = l.peek() {
|
|
// Not allowed // in operators
|
|
if r == '/' && strings.HasPrefix(l.input[l.pos.byteNo:], "/") {
|
|
break
|
|
}
|
|
// Not allowed /* in operators
|
|
if r == '/' && strings.HasPrefix(l.input[l.pos.byteNo:], "*") {
|
|
break
|
|
}
|
|
// Not allowed ||| in operators
|
|
if r == '|' && strings.HasPrefix(l.input[l.pos.byteNo:], "||") {
|
|
break
|
|
}
|
|
l.next()
|
|
}
|
|
|
|
// Operators are not allowed to end with + - ~ ! unless they are one rune long.
|
|
// So, wind it back if we need to, but stop at the first rune.
|
|
// This relies on the hack that all operator symbols are ASCII and thus there is
|
|
// no need to treat this substring as general UTF-8.
|
|
for r = rune(l.input[l.pos.byteNo-1]); l.pos.byteNo > l.tokenStart+1; l.pos.byteNo-- {
|
|
switch r {
|
|
case '+', '-', '~', '!', '$':
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
|
|
if l.input[l.tokenStart:l.pos.byteNo] == "$" {
|
|
l.emitToken(tokenDollar)
|
|
} else {
|
|
l.emitToken(tokenOperator)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Lex returns a slice of tokens recognised in input.
|
|
func Lex(diagnosticFilename ast.DiagnosticFileName, importedFilename, input string) (Tokens, error) {
|
|
l := makeLexer(diagnosticFilename, importedFilename, input)
|
|
|
|
var err error
|
|
for {
|
|
newLines, indent := l.lexWhitespace()
|
|
// If it's the end of the file, discard final whitespace.
|
|
if l.peek() == lexEOF {
|
|
l.next()
|
|
l.resetTokenStart()
|
|
break
|
|
}
|
|
if newLines > 0 {
|
|
// Otherwise store whitespace in fodder.
|
|
blanks := newLines - 1
|
|
l.addFodder(ast.FodderLineEnd, blanks, indent, []string{})
|
|
}
|
|
l.resetTokenStart() // Don't include whitespace in actual token.
|
|
r := l.peek()
|
|
switch r {
|
|
case '{':
|
|
l.next()
|
|
l.emitToken(tokenBraceL)
|
|
case '}':
|
|
l.next()
|
|
l.emitToken(tokenBraceR)
|
|
case '[':
|
|
l.next()
|
|
l.emitToken(tokenBracketL)
|
|
case ']':
|
|
l.next()
|
|
l.emitToken(tokenBracketR)
|
|
case ',':
|
|
l.next()
|
|
l.emitToken(tokenComma)
|
|
case '.':
|
|
l.next()
|
|
l.emitToken(tokenDot)
|
|
case '(':
|
|
l.next()
|
|
l.emitToken(tokenParenL)
|
|
case ')':
|
|
l.next()
|
|
l.emitToken(tokenParenR)
|
|
case ';':
|
|
l.next()
|
|
l.emitToken(tokenSemicolon)
|
|
|
|
case '0', '1', '2', '3', '4', '5', '6', '7', '8', '9':
|
|
err = l.lexNumber()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// String literals
|
|
|
|
case '"':
|
|
stringStartLoc := l.location()
|
|
l.next()
|
|
for r = l.next(); ; r = l.next() {
|
|
if r == lexEOF {
|
|
return nil, l.makeStaticErrorPoint("Unterminated String", stringStartLoc)
|
|
}
|
|
if r == '"' {
|
|
// Don't include the quotes in the token data
|
|
l.emitFullToken(tokenStringDouble, l.input[l.tokenStart+1:l.pos.byteNo-1], "", "")
|
|
l.resetTokenStart()
|
|
break
|
|
}
|
|
if r == '\\' && l.peek() != lexEOF {
|
|
//nolint:ineffassign,staticcheck
|
|
r = l.next()
|
|
}
|
|
}
|
|
case '\'':
|
|
stringStartLoc := l.location()
|
|
l.next()
|
|
for r = l.next(); ; r = l.next() {
|
|
if r == lexEOF {
|
|
return nil, l.makeStaticErrorPoint("Unterminated String", stringStartLoc)
|
|
}
|
|
if r == '\'' {
|
|
// Don't include the quotes in the token data
|
|
l.emitFullToken(tokenStringSingle, l.input[l.tokenStart+1:l.pos.byteNo-1], "", "")
|
|
l.resetTokenStart()
|
|
break
|
|
}
|
|
if r == '\\' && l.peek() != lexEOF {
|
|
//nolint:ineffassign,staticcheck
|
|
r = l.next()
|
|
}
|
|
}
|
|
case '@':
|
|
stringStartLoc := l.location()
|
|
l.next()
|
|
// Verbatim string literals.
|
|
// ' and " quoting is interpreted here, unlike non-verbatim strings
|
|
// where it is done later by jsonnet_string_unescape. This is OK
|
|
// in this case because no information is lost by resoving the
|
|
// repeated quote into a single quote, so we can go back to the
|
|
// original form in the formatter.
|
|
var data []rune
|
|
quot := l.next()
|
|
var kind tokenKind
|
|
if quot == '"' {
|
|
kind = tokenVerbatimStringDouble
|
|
} else if quot == '\'' {
|
|
kind = tokenVerbatimStringSingle
|
|
} else {
|
|
return nil, l.makeStaticErrorPoint(
|
|
fmt.Sprintf("Couldn't lex verbatim string, junk after '@': %v", quot),
|
|
stringStartLoc,
|
|
)
|
|
}
|
|
for r = l.next(); ; r = l.next() {
|
|
if r == lexEOF {
|
|
return nil, l.makeStaticErrorPoint("Unterminated String", stringStartLoc)
|
|
} else if r == quot {
|
|
if l.peek() == quot {
|
|
l.next()
|
|
data = append(data, r)
|
|
} else {
|
|
l.emitFullToken(kind, string(data), "", "")
|
|
l.resetTokenStart()
|
|
break
|
|
}
|
|
} else {
|
|
data = append(data, r)
|
|
}
|
|
}
|
|
|
|
default:
|
|
if isIdentifierFirst(r) {
|
|
l.lexIdentifier()
|
|
} else if isSymbol(r) || r == '#' {
|
|
err = l.lexSymbol()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
return nil, l.makeStaticErrorPoint(
|
|
fmt.Sprintf("Could not lex the character %s", strconv.QuoteRuneToASCII(r)),
|
|
l.location())
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
// We are currently at the EOF. Emit a special token to capture any
|
|
// trailing fodder
|
|
l.emitToken(tokenEndOfFile)
|
|
return l.tokens, nil
|
|
}
|