numscript/internal/analysis/document_symbols.go

package analysis

import (
	"slices"

	"github.com/formancehq/numscript/internal/parser"
)

type DocumentSymbolKind = float64

// !important! keep in sync with
// https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#textDocument_documentSymbol
const (
	DocumentSymbolVariable DocumentSymbolKind = 13
)

type DocumentSymbol struct {
	Name           string
	Detail         string
	Range          parser.Range
	SelectionRange parser.Range
	Kind           DocumentSymbolKind
}

// results are sorted by start position
func (r *CheckResult) GetSymbols() []DocumentSymbol {
	var symbols []DocumentSymbol
	for k, v := range r.DeclaredVars {
		symbols = append(symbols, DocumentSymbol{
			Name:           k,
			Kind:           DocumentSymbolVariable,
			Detail:         v.Type.Name,
			Range:          v.Name.Range,
			SelectionRange: v.Name.Range,
		})
	}

	slices.SortStableFunc(symbols, func(a, b DocumentSymbol) int {
		if a.Range.Start.GtEq(b.Range.Start) {
			return 1
		} else {
			return -1
		}
	})

	return symbols
}