feat: allow replacing buildpack list atomically

Add buildpacks:set --replace so callers can replace an app's complete ordered buildpack list in one command while preserving existing single-buildpack and --index behavior.

Closes #8802
This commit is contained in:
Jose Diaz-Gonzalez
2026-07-07 18:37:38 -04:00
parent 25b1356435
commit 16f8b25bda
6 changed files with 205 additions and 4 deletions

View File

@@ -139,3 +139,79 @@ func TestGetBuildpacksAppJSONEmptyArray(t *testing.T) {
Expect(err).NotTo(HaveOccurred())
Expect(buildpacks).To(BeEmpty())
}
func TestSetBuildpackListReplacesEntireList(t *testing.T) {
RegisterTestingT(t)
tmpDir := setupTestEnvironment(t)
defer teardownTestEnvironment(tmpDir)
propDir := filepath.Join(tmpDir, "config", "buildpacks", "test-app")
Expect(os.MkdirAll(propDir, 0755)).To(Succeed())
Expect(os.WriteFile(filepath.Join(propDir, "buildpacks"), []byte("https://github.com/heroku/heroku-buildpack-nodejs.git\nhttps://github.com/heroku/heroku-buildpack-python.git\nhttps://github.com/heroku/heroku-buildpack-ruby.git\n"), 0644)).To(Succeed())
err := setBuildpackList("test-app", []string{"heroku/ruby", "heroku/nodejs"}, 0)
Expect(err).NotTo(HaveOccurred())
buildpacks, err := getBuildpacks("test-app")
Expect(err).NotTo(HaveOccurred())
Expect(buildpacks).To(HaveLen(2))
Expect(buildpacks[0]).To(Equal("https://github.com/heroku/heroku-buildpack-ruby.git"))
Expect(buildpacks[1]).To(Equal("https://github.com/heroku/heroku-buildpack-nodejs.git"))
}
func TestSetBuildpackListSingleBuildpack(t *testing.T) {
RegisterTestingT(t)
tmpDir := setupTestEnvironment(t)
defer teardownTestEnvironment(tmpDir)
propDir := filepath.Join(tmpDir, "config", "buildpacks", "test-app")
Expect(os.MkdirAll(propDir, 0755)).To(Succeed())
Expect(os.WriteFile(filepath.Join(propDir, "buildpacks"), []byte("https://github.com/heroku/heroku-buildpack-nodejs.git\nhttps://github.com/heroku/heroku-buildpack-python.git\n"), 0644)).To(Succeed())
err := setBuildpackList("test-app", []string{"heroku/ruby"}, 0)
Expect(err).NotTo(HaveOccurred())
buildpacks, err := getBuildpacks("test-app")
Expect(err).NotTo(HaveOccurred())
Expect(buildpacks).To(HaveLen(1))
Expect(buildpacks[0]).To(Equal("https://github.com/heroku/heroku-buildpack-ruby.git"))
}
func TestSetBuildpackListInvalidLeavesListUnchanged(t *testing.T) {
RegisterTestingT(t)
tmpDir := setupTestEnvironment(t)
defer teardownTestEnvironment(tmpDir)
propDir := filepath.Join(tmpDir, "config", "buildpacks", "test-app")
Expect(os.MkdirAll(propDir, 0755)).To(Succeed())
Expect(os.WriteFile(filepath.Join(propDir, "buildpacks"), []byte("https://github.com/heroku/heroku-buildpack-nodejs.git\n"), 0644)).To(Succeed())
err := setBuildpackList("test-app", []string{"heroku/ruby", "nodejs"}, 0)
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("Invalid buildpack specified"))
buildpacks, err := getBuildpacks("test-app")
Expect(err).NotTo(HaveOccurred())
Expect(buildpacks).To(HaveLen(1))
Expect(buildpacks[0]).To(Equal("https://github.com/heroku/heroku-buildpack-nodejs.git"))
}
func TestSetBuildpackListRejectsIndex(t *testing.T) {
RegisterTestingT(t)
tmpDir := setupTestEnvironment(t)
defer teardownTestEnvironment(tmpDir)
err := setBuildpackList("test-app", []string{"heroku/ruby"}, 1)
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("--index"))
}
func TestSetBuildpackListRejectsEmpty(t *testing.T) {
RegisterTestingT(t)
tmpDir := setupTestEnvironment(t)
defer teardownTestEnvironment(tmpDir)
err := setBuildpackList("test-app", []string{}, 0)
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("buildpacks:clear"))
}

View File

@@ -25,6 +25,7 @@ Additional commands:`
buildpacks:remove <app> <buildpack>, Remove a buildpack set on the app
buildpacks:report [<app>] [<flag>], Displays a buildpack report for one or more apps
buildpacks:set [--index 1] <app> <buildpack>, Set new app buildpack at a given position defaulting to the first buildpack if no index is specified
buildpacks:set --replace <app> <buildpack> [<buildpack> ...], Replace the entire ordered buildpack list with the specified buildpacks
buildpacks:set-property [--global|<app>] <key> <value>, Set or clear a buildpacks property for an app`
)

View File

@@ -66,10 +66,11 @@ func main() {
case "set":
args := flag.NewFlagSet("buildpacks:set", flag.ExitOnError)
index := args.Int("index", 0, "--index: the 1-based index of the URL in the list of URLs")
replace := args.Bool("replace", false, "--replace: replace the entire ordered buildpack list")
args.Parse(os.Args[2:])
appName := args.Arg(0)
buildpack := args.Arg(1)
err = buildpacks.CommandSet(appName, buildpack, *index)
_, buildpackList := common.ShiftString(args.Args())
err = buildpacks.CommandSet(appName, buildpackList, *index, *replace)
case "set-property":
args := flag.NewFlagSet("buildpacks:set-property", flag.ExitOnError)
global := args.Bool("global", false, "--global: set a global property")

View File

@@ -162,12 +162,24 @@ func CommandReport(appName string, format string, infoFlag string) error {
}
// CommandSet implements buildpacks:set
func CommandSet(appName string, buildpack string, index int) error {
func CommandSet(appName string, buildpacks []string, index int, replace bool) error {
if err := common.VerifyAppName(appName); err != nil {
return err
}
buildpack, err := validBuildpackURL(buildpack)
if replace {
return setBuildpackList(appName, buildpacks, index)
}
if len(buildpacks) > 1 {
return errors.New("Setting multiple buildpacks requires the --replace flag")
}
if len(buildpacks) == 0 {
return errors.New("Must specify a buildpack")
}
buildpack, err := validBuildpackURL(buildpacks[0])
if err != nil {
return err
}
@@ -179,6 +191,28 @@ func CommandSet(appName string, buildpack string, index int) error {
return common.PropertyListSet("buildpacks", appName, "buildpacks", buildpack, index)
}
// setBuildpackList atomically replaces the entire ordered buildpack list
func setBuildpackList(appName string, buildpacks []string, index int) error {
if index != 0 {
return errors.New("The --index flag cannot be used with --replace")
}
if len(buildpacks) == 0 {
return errors.New("Must specify at least one buildpack, use buildpacks:clear to remove all buildpacks")
}
validated := make([]string, 0, len(buildpacks))
for _, buildpack := range buildpacks {
url, err := validBuildpackURL(buildpack)
if err != nil {
return err
}
validated = append(validated, url)
}
return common.PropertyListWrite("buildpacks", appName, "buildpacks", validated)
}
// CommandSetProperty implements buildpacks:set-property
func CommandSetProperty(appName string, property string, value string) error {
oldStack := ""