Function setConfigKey

iamlivecore/csm.go:19–65

(filename, section, line string, unset bool)

)

func setConfigKey(filename, section, line string, unset bool) error {
	fileinfo, err := os.Stat(filename)
	if err != nil {
		return err
	}

	file, err := os.OpenFile(filename, os.O_RDONLY, fileinfo.Mode().Perm())
	if err != nil {
		return err
	}
	defer file.Close()

	var newLines []string
	isCorrectSection := false

	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		scannedLine := scanner.Text()
		if scannedLine != line || !isCorrectSection { // write all but lines to be removed
			newLines = append(newLines, scannedLine)
		}
		if scannedLine == fmt.Sprintf("[%s]", section) {
			isCorrectSection = true
			if !unset {
				newLines = append(newLines, line)
			}
		} else if strings.HasPrefix(scannedLine, "[") && strings.HasSuffix(scannedLine, "]") {
			isCorrectSection = false
		}
	}
	if err := scanner.Err(); err != nil {
		return err
	}
	file.Close()

	file, err = os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC, fileinfo.Mode().Perm())
	if err != nil {
		return err
	}
	defer file.Close()

	writer := bufio.NewWriter(file)
	for _, line := range newLines {
		fmt.Fprintln(writer, line)
	}
	return writer.Flush()
}

func setINIConfigAndFileFlush() {
	cfgfilepath := "~/.aws/config"