Method completion

backend/plugin/parser/pg/completion.go:189–233

()

}

func (c *Completer) completion() ([]base.Candidate, error) {
	// Check if the caret token is quoted.
	if c.caretTokenIndex < len(c.tokens) {
		tok := c.tokens[c.caretTokenIndex]
		if tok.Type == pgparser.IDENT && tok.Loc < len(c.sql) && c.sql[tok.Loc] == '"' {
			c.caretTokenIsQuoted = true
		}
	}

	caretIndex := c.caretTokenIndex
	if caretIndex > 0 && !isNoSeparatorRequired(c.tokens[caretIndex-1].Type) {
		caretIndex--
	}
	c.referencesStack = append([][]base.TableReference{{}}, c.referencesStack...)

	// Use omni parser to collect grammar candidates instead of C3.
	completionOffset := c.cursorByteOffset
	candidates := pgparser.Collect(c.sql, completionOffset)

	// If Collect returned no rule candidates and the cursor is at the end of a
	// partial identifier token (prefix), retry Collect at the start of that
	// token.  This enables completion for "SELECT * FROM t|" style inputs
	// where the parser sees a complete token and doesn't know what grammar
	// rule to suggest.
	if len(candidates.Rules) == 0 {
		if prefixTok, ok := c.prefixToken(); ok {
			completionOffset = prefixTok.Loc
			candidates = pgparser.Collect(c.sql, completionOffset)
		}
	}

	for _, rc := range candidates.Rules {
		if rc.Rule == "columnref" {
			completionContext := pgparser.CollectCompletion(c.sql, completionOffset)
			c.collectLeadingTableReferences(caretIndex)
			c.takeReferencesSnapshot()
			c.collectRemainingTableReferences()
			c.takeReferencesSnapshot()
			c.collectCompletionScopeReferences(completionContext)
			break
		}
	}

	return c.convertCandidates(candidates)
}

// prefixToken returns the token immediately before (or containing) the cursor
// position when that token is an identifier-like token that the user is still