config-lsp/handlers/wireguard/code-actions.go

package wireguard

import (
	protocol "github.com/tliron/glsp/protocol_3_16"
)

type codeActionName string

const (
	codeActionGeneratePrivateKey   codeActionName = "generatePrivateKey"
	codeActionGeneratePresharedKey codeActionName = "generatePresharedKey"
	codeActionAddKeepalive         codeActionName = "addKeepalive"
)

type codeActionGeneratePrivateKeyArgs struct {
	URI  protocol.DocumentUri
	Line uint32
}

func codeActionGeneratePrivateKeyArgsFromArguments(arguments map[string]any) codeActionGeneratePrivateKeyArgs {
	return codeActionGeneratePrivateKeyArgs{
		URI:  arguments["URI"].(protocol.DocumentUri),
		Line: uint32(arguments["Line"].(float64)),
	}
}

type codeActionGeneratePresharedKeyArgs struct {
	URI  protocol.DocumentUri
	Line uint32
}

func codeActionGeneratePresharedKeyArgsFromArguments(arguments map[string]any) codeActionGeneratePresharedKeyArgs {
	return codeActionGeneratePresharedKeyArgs{
		URI:  arguments["URI"].(protocol.DocumentUri),
		Line: uint32(arguments["Line"].(float64)),
	}
}

type codeActionAddKeepaliveArgs struct {
	URI          protocol.DocumentUri
	SectionIndex uint32
}

func codeActionAddKeepaliveArgsFromArguments(arguments map[string]any) codeActionAddKeepaliveArgs {
	return codeActionAddKeepaliveArgs{
		URI:          arguments["URI"].(protocol.DocumentUri),
		SectionIndex: uint32(arguments["SectionIndex"].(float64)),
	}
}

func (p wireguardProperty) getInsertRange(line uint32) protocol.Range {
	var insertPosition uint32 = p.Separator.Location.End
	var length uint32 = 0

	if p.Value != nil {
		insertPosition = p.Value.Location.Start - 1
		// Length of the value; +1 because of the starting space
		length = (p.Value.Location.End - p.Value.Location.Start) + 1
	}

	return protocol.Range{
		Start: protocol.Position{
			Line:      line,
			Character: insertPosition,
		},
		End: protocol.Position{
			Line:      line,
			Character: insertPosition + length,
		},
	}
}

func (p *wireguardParser) runGeneratePrivateKey(args codeActionGeneratePrivateKeyArgs) (*protocol.ApplyWorkspaceEditParams, error) {
	privateKey, err := createNewPrivateKey()

	if err != nil {
		return &protocol.ApplyWorkspaceEditParams{}, err
	}

	section, property := p.getPropertyByLine(args.Line)

	if section == nil || property == nil {
		return nil, nil
	}

	label := "Generate Private Key"
	return &protocol.ApplyWorkspaceEditParams{
		Label: &label,
		Edit: protocol.WorkspaceEdit{
			Changes: map[protocol.DocumentUri][]protocol.TextEdit{
				args.URI: {
					{
						NewText: " " + privateKey,
						Range:   property.getInsertRange(args.Line),
					},
				},
			},
		},
	}, nil
}

func (p *wireguardParser) runGeneratePresharedKey(args codeActionGeneratePresharedKeyArgs) (*protocol.ApplyWorkspaceEditParams, error) {
	presharedKey, err := createPresharedKey()

	if err != nil {
		return &protocol.ApplyWorkspaceEditParams{}, err
	}

	section, property := p.getPropertyByLine(args.Line)

	if section == nil || property == nil {
		return nil, nil
	}

	label := "Generate Preshared Key"
	return &protocol.ApplyWorkspaceEditParams{
		Label: &label,
		Edit: protocol.WorkspaceEdit{
			Changes: map[protocol.DocumentUri][]protocol.TextEdit{
				args.URI: {
					{
						NewText: " " + presharedKey,
						Range:   property.getInsertRange(args.Line),
					},
				},
			},
		},
	}, nil
}

func (p *wireguardParser) runAddKeepalive(args codeActionAddKeepaliveArgs) (*protocol.ApplyWorkspaceEditParams, error) {
	section := p.Sections[args.SectionIndex]

	label := "Add PersistentKeepalive"
	return &protocol.ApplyWorkspaceEditParams{
		Label: &label,
		Edit: protocol.WorkspaceEdit{
			Changes: map[protocol.DocumentUri][]protocol.TextEdit{
				args.URI: {
					{
						NewText: "PersistentKeepalive = 25\n",
						Range: protocol.Range{
							Start: protocol.Position{
								Line:      section.EndLine + 1,
								Character: 0,
							},
							End: protocol.Position{
								Line:      section.EndLine + 1,
								Character: 0,
							},
						},
					},
				},
			},
		},
	}, nil
}