Method handleDocumentSymbol

pkg/lsp/handler.go:1190–1220

handleDocumentSymbol returns symbols in the document (SQL statements, tables, columns)

(params json.RawMessage)

// handleDocumentSymbol returns symbols in the document (SQL statements, tables, columns)
func (h *Handler) handleDocumentSymbol(params json.RawMessage) ([]DocumentSymbol, error) {
	var p DocumentSymbolParams
	if err := json.Unmarshal(params, &p); err != nil {
		return nil, err
	}

	content, ok := h.server.Documents().GetContent(p.TextDocument.URI)
	if !ok {
		return []DocumentSymbol{}, nil
	}

	// Parse the SQL to extract symbols
	ast, err := gosqlx.Parse(content)
	if err != nil {
		// Return empty symbols on parse error
		return []DocumentSymbol{}, nil
	}

	symbols := []DocumentSymbol{}
	lines := strings.Split(content, "\n")

	// Extract symbols from each statement
	for i, stmt := range ast.Statements {
		symbol := h.extractStatementSymbol(stmt, i, lines, content)
		if symbol != nil {
			symbols = append(symbols, *symbol)
		}
	}

	return symbols, nil
}

// extractStatementSymbol extracts a document symbol from a SQL statement
func (h *Handler) extractStatementSymbol(stmt interface{}, index int, lines []string, content string) *DocumentSymbol {