Function ensureAdminUserYaml

config_ensure.go:233–249

(data []byte, username, pubKey string)

}

func ensureAdminUserYaml(data []byte, username, pubKey string) (*yaml.Node, bool, error) {
	rootNode, targetNode, err := yaml.EnsureDocument(data)
	if err != nil {
		return nil, false, err
	}

	usersNode, _ := targetNode.EnsureKey("users", yaml.NewMappingNode(), nil)
	userNode, _ := usersNode.EnsureKey(username, yaml.NewMappingNode(), nil)
	keysNode, _ := userNode.EnsureKey("keys", yaml.NewSequenceNode(), nil)
	modified := keysNode.AppendUniqueScalar(yaml.NewScalarNode(pubKey, ""))

	// Ensure this user is an admin and isn't disabled
	_, adminModified := userNode.EnsureKey("is_admin", yaml.NewScalarNode("true", yaml.ScalarTagBool), nil)
	disabledModified := userNode.RemoveKey("disabled")

	return rootNode, modified || disabledModified || adminModified, nil
}