Function EnsureIncludes

pkg/configfile/configfile.go:168–203

EnsureIncludes makes sure the passed PostgreSQL configuration file has an include directive to every filesToInclude.

(fileName string, filesToInclude ...string)

// EnsureIncludes makes sure the passed PostgreSQL configuration file has an include directive
// to every filesToInclude.
func EnsureIncludes(fileName string, filesToInclude ...string) (changed bool, err error) {
	includeLinesToAdd := make(map[string]string, len(filesToInclude))
	for _, fileToInclude := range filesToInclude {
		includeLinesToAdd[fileToInclude] = fmt.Sprintf("include '%v'", fileToInclude)
	}

	lines, err := fileutils.ReadFileLines(fileName)
	if err != nil {
		return false, fmt.Errorf("error while reading lines of %v: %w", fileName, err)
	}

	for _, line := range lines {
		trimLine := strings.TrimSpace(line)
		for targetFile, includeLine := range includeLinesToAdd {
			if trimLine == includeLine {
				delete(includeLinesToAdd, targetFile)
			}
		}
	}

	if len(includeLinesToAdd) == 0 {
		return false, nil
	}

	for _, fileToInclude := range filesToInclude {
		if includeLine, present := includeLinesToAdd[fileToInclude]; present {
			lines = append(lines,
				"",
				fmt.Sprintf("# load CloudNativePG %s configuration", fileToInclude),
				includeLine,
			)
		}
	}

	return fileutils.WriteLinesToFile(fileName, lines)
}