fix: read release init response (#2075)

Fixes #2057

Author: JoeWang1127

internal/config/config.go

@@ -51,9 +51,11 @@ const (
 	LibrarianDir = ".librarian"
 	// ReleaseInitRequest is a JSON file that describes which library to release.
 	ReleaseInitRequest = "release-init-request.json"
-
-	pipelineStateFile = "state.yaml"
-	versionCmdName    = "version"
+	// ReleaseInitResponse is a JSON file that describes which library to change
+	// after release.
+	ReleaseInitResponse = "release-init-response.json"
+	pipelineStateFile   = "state.yaml"
+	versionCmdName      = "version"
 )
 
 // are variables so it can be replaced during testing.

internal/librarian/mocks_test.go

@@ -125,6 +125,7 @@ type mockContainerClient struct {
 	noBuildResponse     bool
 	noConfigureResponse bool
 	noGenerateResponse  bool
+	noReleaseResponse   bool
 	noInitVersion       bool
 	wantErrorMsg        bool
 	// Set this value if you want library files
@@ -264,6 +265,25 @@ func (m *mockContainerClient) Generate(ctx context.Context, request *docker.Gene
 
 func (m *mockContainerClient) ReleaseInit(ctx context.Context, request *docker.ReleaseInitRequest) error {
 	m.initCalls++
+	if m.noReleaseResponse {
+		return m.initErr
+	}
+	// Write a release-init-response.json unless we're configured not to.
+	if err := os.MkdirAll(filepath.Join(request.PartialRepoDir, ".librarian"), 0755); err != nil {
+		return err
+	}
+
+	library := &config.LibraryState{}
+	if m.wantErrorMsg {
+		library.ErrorMessage = "simulated error message"
+	}
+	b, err := json.MarshalIndent(library, "", " ")
+	if err != nil {
+		return err
+	}
+	if err := os.WriteFile(filepath.Join(request.PartialRepoDir, ".librarian", config.ReleaseInitResponse), b, 0755); err != nil {
+		return err
+	}
 	return m.initErr
 }
 

internal/librarian/release_init.go

@@ -204,6 +204,12 @@ func (r *initRunner) runInitCommand(ctx context.Context, outputDir string) error
 		return err
 	}
 
+	// Read the response file.
+	if _, err := readLibraryState(
+		filepath.Join(initRequest.PartialRepoDir, config.LibrarianDir, config.ReleaseInitResponse)); err != nil {
+		return err
+	}
+
 	for _, library := range r.state.Libraries {
 		if r.cfg.Library != "" {
 			if r.cfg.Library != library.ID {

internal/librarian/release_init_test.go

@@ -338,7 +338,6 @@ func TestInitRun(t *testing.T) {
 				},
 			},
 		},
-
 		{
 			name: "docker command returns error",
 			runner: &initRunner{
@@ -357,6 +356,24 @@ func TestInitRun(t *testing.T) {
 			wantErr:    true,
 			wantErrMsg: "simulated init error",
 		},
+		{
+			name: "release response contains error message",
+			runner: &initRunner{
+				workRoot: t.TempDir(),
+				containerClient: &mockContainerClient{
+					wantErrorMsg: true,
+				},
+				cfg:   &config.Config{},
+				state: &config.LibrarianState{},
+				repo: &MockRepository{
+					Dir: t.TempDir(),
+				},
+				partialRepo:     t.TempDir(),
+				librarianConfig: &config.LibrarianConfig{},
+			},
+			wantErr:    true,
+			wantErrMsg: "failed with error message: simulated error message",
+		},
 		{
 			name: "invalid work root",
 			runner: &initRunner{