config-lsp/handlers/sshd_config/ast/sshd_config_fields.go

package ast

func (o SSHDOption) GetType() SSHDEntryType {
	return SSHDEntryTypeOption
}

func (o SSHDOption) GetOption() SSHDOption {
	return o
}

func (m SSHDMatchBlock) GetType() SSHDEntryType {
	return SSHDEntryTypeMatchBlock
}

func (m SSHDMatchBlock) GetOption() SSHDOption {
	return *m.MatchOption
}

func (c SSHDConfig) FindMatchBlock(line uint32) *SSHDMatchBlock {
	for currentLine := line; currentLine > 0; currentLine-- {
		rawEntry, found := c.Options.Get(currentLine)

		if !found {
			continue
		}

		switch entry := rawEntry.(type) {
		case *SSHDMatchBlock:
			return entry
		}
	}

	return nil
}

func (c SSHDConfig) FindOption(line uint32) (*SSHDOption, *SSHDMatchBlock) {
	matchBlock := c.FindMatchBlock(line)

	if matchBlock != nil {
		if line == matchBlock.MatchOption.Start.Line {
			return matchBlock.MatchOption, matchBlock
		}

		rawEntry, found := matchBlock.Options.Get(line)

		if found {
			return rawEntry.(*SSHDOption), matchBlock
		} else {
			return nil, matchBlock
		}
	}

	rawEntry, found := c.Options.Get(line)

	if found {
		switch rawEntry.(type) {
		case *SSHDMatchBlock:
			return rawEntry.(*SSHDMatchBlock).MatchOption, rawEntry.(*SSHDMatchBlock)
		case *SSHDOption:
			return rawEntry.(*SSHDOption), nil
		}
	}

	return nil, nil
}

func (c SSHDConfig) GetAllOptions() []*SSHDOption {
	options := make(
		[]*SSHDOption,
		0,
		// Approximation, this does not need to be exact
		c.Options.Size()+10,
	)

	for _, rawEntry := range c.Options.Values() {
		switch entry := rawEntry.(type) {
		case *SSHDOption:
			options = append(options, entry)
		case *SSHDMatchBlock:
			options = append(options, entry.MatchOption)

			for _, rawOption := range entry.Options.Values() {
				options = append(options, rawOption.(*SSHDOption))
			}
		}
	}

	return options
}