updated the flow to correspond to the sequence diagram

internal/file/file_manager_service.go

@@ -10,7 +10,9 @@ import (
 	"errors"
 	"fmt"
 	"log/slog"
+	"maps"
 	"os"
+	"slices"
 	"sync"
 	"sync/atomic"
 
@@ -133,14 +135,6 @@ func (fms *FileManagerService) UpdateOverview(
 			return nil, validatedError
 		}
 
-		response.Overview = &mpi.FileOverview{
-			Files: filesToUpdate,
-			ConfigVersion: &mpi.ConfigVersion{
-				InstanceId: instanceID,
-				Version:    files.GenerateConfigVersion(filesToUpdate),
-			},
-		}
-
 		return response, nil
 	}
 
@@ -154,6 +148,19 @@ func (fms *FileManagerService) UpdateOverview(
 
 	slog.DebugContext(newCtx, "UpdateOverview response", "response", response)
 
+	// calculate the response.Overview delta vs what is on disk
+	delta, _, compareErr := fms.DetermineFileActions(fms.currentFilesOnDisk,
+		files.ConvertToMapOfFiles(response.GetOverview().GetFiles()))
+
+	if compareErr != nil {
+		return compareErr
+	}
+
+	if len(delta) != 0 {
+		diffFiles := slices.Collect(maps.Values(delta))
+		return fms.UpdateOverview(ctx, instanceID, diffFiles)
+	}
+
 	return err
 }
 

internal/file/file_manager_service_test.go

@@ -37,10 +37,12 @@ func TestFileManagerService_UpdateOverview(t *testing.T) {
 	overview := protos.FileOverview(filePath, fileHash)
 
 	fakeFileServiceClient := &v1fakes.FakeFileServiceClient{}
-	fakeFileServiceClient.UpdateOverviewReturns(&mpi.UpdateOverviewResponse{
+	fakeFileServiceClient.UpdateOverviewReturnsOnCall(0, &mpi.UpdateOverviewResponse{
 		Overview: overview,
 	}, nil)
 
+	fakeFileServiceClient.UpdateOverviewReturnsOnCall(1, &mpi.UpdateOverviewResponse{}, nil)
+
 	fileManagerService := NewFileManagerService(fakeFileServiceClient, types.AgentConfig())
 	fileManagerService.SetIsConnected(true)
 
@@ -51,7 +53,7 @@ func TestFileManagerService_UpdateOverview(t *testing.T) {
 	})
 
 	require.NoError(t, err)
-	assert.Equal(t, 1, fakeFileServiceClient.UpdateOverviewCallCount())
+	assert.Equal(t, 2, fakeFileServiceClient.UpdateOverviewCallCount())
 }
 
 func TestFileManagerService_UpdateFile(t *testing.T) {