Function parseAndUpdateGoModFile

rivershared/cmd/update-mod-go/main.go:100–155

(checkOnly bool, modFilename, workFilename, workGoVersion, workToolchainName string)

}

func parseAndUpdateGoModFile(checkOnly bool, modFilename, workFilename, workGoVersion, workToolchainName string) (bool, error) {
	modFileData, err := os.ReadFile(modFilename)
	if err != nil {
		return false, fmt.Errorf("error reading file %q: %w", modFilename, err)
	}

	modFile, err := modfile.Parse(modFilename, modFileData, nil)
	if err != nil {
		return false, fmt.Errorf("error parsing file %q: %w", modFilename, err)
	}

	var anyMismatch bool

	fmt.Printf("%s\n", modFilename)

	if workGoVersion != modFile.Go.Version {
		if checkOnly {
			return false, fmt.Errorf("go directive of %q (%s) doesn't match %q (%s)", modFilename, modFile.Go.Version, workFilename, workGoVersion)
		}

		anyMismatch = true
		if err := modFile.AddGoStmt(workGoVersion); err != nil {
			return false, fmt.Errorf("error adding go statement: %w", err)
		}
		fmt.Printf("    set go to %s for %s\n", workGoVersion, modFilename)
	}

	if workToolchainName != modFile.Toolchain.Name {
		if checkOnly {
			return false, fmt.Errorf("toolchain directive of %q (%s) doesn't match %q (%s)", modFilename, modFile.Toolchain.Name, workFilename, workToolchainName)
		}

		anyMismatch = true
		if err := modFile.AddToolchainStmt(workToolchainName); err != nil {
			return false, fmt.Errorf("error adding toolchain statement: %w", err)
		}
		fmt.Printf("    set toolchain to %s for %s\n", workToolchainName, modFilename)
	}

	if !checkOnly {
		if anyMismatch {
			updatedFileData, err := modFile.Format()
			if err != nil {
				return false, fmt.Errorf("error formatting file %q after update: %w", modFilename, err)
			}

			if err := os.WriteFile(modFilename, updatedFileData, 0o600); err != nil { //nolint:gosec // validated by localPath
				return false, fmt.Errorf("error writing file %q after update: %w", modFilename, err)
			}
		} else {
			fmt.Printf("    no changes\n")
		}
	}

	return anyMismatch, nil
}