package sshconfig

import (
	"config-lsp/handlers/ssh_config/ast"
	"config-lsp/handlers/ssh_config/fields"
	"config-lsp/utils"
)

func (d SSHDocument) FindOptionByNameAndBlock(
	option fields.NormalizedOptionName,
	block ast.SSHBlock,
) *ast.AllOptionInfo {
	for _, info := range d.FindOptionsByName(option) {
		if info.Block == block {
			return &info
		}
	}

	return nil
}

func (d SSHDocument) FindOptionsByName(
	option fields.NormalizedOptionName,
) []ast.AllOptionInfo {
	options := make([]ast.AllOptionInfo, 0, 5)

	for _, info := range d.Config.GetAllOptions() {
		if info.Option.Key.Key == option {
			options = append(options, info)
		}
	}

	return options
}

func (d SSHDocument) DoesOptionExist(
	option fields.NormalizedOptionName,
	block ast.SSHBlock,
) bool {
	return d.FindOptionByNameAndBlock(option, block) != nil
}

var matchOption = fields.CreateNormalizedName("Match")

func (d SSHDocument) GetAllMatchBlocks() []*ast.SSHMatchBlock {
	matchBlocks := make([]*ast.SSHMatchBlock, 0, 5)

	options := d.Indexes.AllOptionsPerName[matchOption]
	blocks := utils.KeysOfMap(options)

	for _, block := range blocks {
		switch block.GetBlockType() {
		case ast.SSHBlockTypeMatch:
			matchBlocks = append(matchBlocks, block.(*ast.SSHMatchBlock))
		}
	}

	return matchBlocks
}