1package protocol
  2
  3import "fmt"
  4
  5// TextEditResult is an interface for types that represent workspace symbols
  6type WorkspaceSymbolResult interface {
  7	GetName() string
  8	GetLocation() Location
  9	isWorkspaceSymbol() // marker method
 10}
 11
 12func (ws *WorkspaceSymbol) GetName() string { return ws.Name }
 13func (ws *WorkspaceSymbol) GetLocation() Location {
 14	switch v := ws.Location.Value.(type) {
 15	case Location:
 16		return v
 17	case LocationUriOnly:
 18		return Location{URI: v.URI}
 19	}
 20	return Location{}
 21}
 22func (ws *WorkspaceSymbol) isWorkspaceSymbol() {}
 23
 24func (si *SymbolInformation) GetName() string       { return si.Name }
 25func (si *SymbolInformation) GetLocation() Location { return si.Location }
 26func (si *SymbolInformation) isWorkspaceSymbol()    {}
 27
 28// Results converts the Value to a slice of WorkspaceSymbolResult
 29func (r Or_Result_workspace_symbol) Results() ([]WorkspaceSymbolResult, error) {
 30	if r.Value == nil {
 31		return make([]WorkspaceSymbolResult, 0), nil
 32	}
 33	switch v := r.Value.(type) {
 34	case []WorkspaceSymbol:
 35		results := make([]WorkspaceSymbolResult, len(v))
 36		for i := range v {
 37			results[i] = &v[i]
 38		}
 39		return results, nil
 40	case []SymbolInformation:
 41		results := make([]WorkspaceSymbolResult, len(v))
 42		for i := range v {
 43			results[i] = &v[i]
 44		}
 45		return results, nil
 46	default:
 47		return nil, fmt.Errorf("unknown symbol type: %T", r.Value)
 48	}
 49}
 50
 51// TextEditResult is an interface for types that represent document symbols
 52type DocumentSymbolResult interface {
 53	GetRange() Range
 54	GetName() string
 55	isDocumentSymbol() // marker method
 56}
 57
 58func (ds *DocumentSymbol) GetRange() Range   { return ds.Range }
 59func (ds *DocumentSymbol) GetName() string   { return ds.Name }
 60func (ds *DocumentSymbol) isDocumentSymbol() {}
 61
 62func (si *SymbolInformation) GetRange() Range { return si.Location.Range }
 63
 64// Note: SymbolInformation already has GetName() implemented above
 65func (si *SymbolInformation) isDocumentSymbol() {}
 66
 67// Results converts the Value to a slice of DocumentSymbolResult
 68func (r Or_Result_textDocument_documentSymbol) Results() ([]DocumentSymbolResult, error) {
 69	if r.Value == nil {
 70		return make([]DocumentSymbolResult, 0), nil
 71	}
 72	switch v := r.Value.(type) {
 73	case []DocumentSymbol:
 74		results := make([]DocumentSymbolResult, len(v))
 75		for i := range v {
 76			results[i] = &v[i]
 77		}
 78		return results, nil
 79	case []SymbolInformation:
 80		results := make([]DocumentSymbolResult, len(v))
 81		for i := range v {
 82			results[i] = &v[i]
 83		}
 84		return results, nil
 85	default:
 86		return nil, fmt.Errorf("unknown document symbol type: %T", v)
 87	}
 88}
 89
 90// TextEditResult is an interface for types that can be used as text edits
 91type TextEditResult interface {
 92	GetRange() Range
 93	GetNewText() string
 94	isTextEdit() // marker method
 95}
 96
 97func (te *TextEdit) GetRange() Range    { return te.Range }
 98func (te *TextEdit) GetNewText() string { return te.NewText }
 99func (te *TextEdit) isTextEdit()        {}
100
101// Convert Or_TextDocumentEdit_edits_Elem to TextEdit
102func (e Or_TextDocumentEdit_edits_Elem) AsTextEdit() (TextEdit, error) {
103	if e.Value == nil {
104		return TextEdit{}, fmt.Errorf("nil text edit")
105	}
106	switch v := e.Value.(type) {
107	case TextEdit:
108		return v, nil
109	case AnnotatedTextEdit:
110		return TextEdit{
111			Range:   v.Range,
112			NewText: v.NewText,
113		}, nil
114	default:
115		return TextEdit{}, fmt.Errorf("unknown text edit type: %T", e.Value)
116	}
117}