Resolve declared file targets in Run
This commit is contained in:
@@ -704,6 +704,47 @@ func TestScopedFileExecutorBoundsReads(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestDeclaredFileTargetKeyReadsAndWritesRelativeToDeploymentRoot(t *testing.T) {
|
||||
workspaceRoot := t.TempDir()
|
||||
serverRoot := t.TempDir()
|
||||
filePath := filepath.Join(serverRoot, "SCUM", "Saved", "Config", "WindowsServer", "ServerSettings.ini")
|
||||
if err := os.MkdirAll(filepath.Dir(filePath), 0o700); err != nil {
|
||||
t.Fatalf("mkdir declared file fixture: %v", err)
|
||||
}
|
||||
if err := os.WriteFile(filePath, []byte("[General]\nscum.MaxPlayers=63\n"), 0o600); err != nil {
|
||||
t.Fatalf("write declared file fixture: %v", err)
|
||||
}
|
||||
executor, err := NewFileExecutor(workspaceRoot)
|
||||
if err != nil {
|
||||
t.Fatalf("new file executor: %v", err)
|
||||
}
|
||||
assignment := executionAssignment(protocol.RunCapabilityFilesRead)
|
||||
assignment.TargetKey = "scum-server-settings"
|
||||
assignment.ExecutionInput.FileTargetKey = "SCUM/Saved/Config/WindowsServer/ServerSettings.ini"
|
||||
assignment.ExecutionInput.Deployment = &protocol.ServerDeploymentExecution{SchemaVersion: "1", Mode: "guided-install", ServerRoot: serverRoot, Revision: 1}
|
||||
read := executor.Execute(context.Background(), assignment)
|
||||
if read.State != lifecycleResultStateSucceeded || read.ExecutionResult.Kind != "file.read" || read.ExecutionResult.Content != "[General]\nscum.MaxPlayers=63\n" {
|
||||
t.Fatalf("expected declared target read, got %+v", read)
|
||||
}
|
||||
assignment.Capability = protocol.RunCapabilityFilesWrite
|
||||
assignment.InputRef = "input://server-execution/settings-write"
|
||||
assignment.ExecutionInput.Content = "[General]\nscum.MaxPlayers=80\n"
|
||||
assignment.ExecutionInput.ExpectedVersion = read.ExecutionResult.Version
|
||||
assignment.ExecutionInput.ExpectedChecksum = read.ExecutionResult.Checksum
|
||||
write := executor.Execute(context.Background(), assignment)
|
||||
if write.State != lifecycleResultStateSucceeded || write.ExecutionResult.Kind != "file.write" {
|
||||
t.Fatalf("expected declared target write, got %+v", write)
|
||||
}
|
||||
body, err := os.ReadFile(filePath)
|
||||
if err != nil || string(body) != assignment.ExecutionInput.Content {
|
||||
t.Fatalf("declared target write used the wrong path, body=%q err=%v", body, err)
|
||||
}
|
||||
assignment.ExecutionInput.FileTargetKey = "../outside.ini"
|
||||
if unsafe := executor.Execute(context.Background(), assignment); unsafe.State != lifecycleResultStateFailed {
|
||||
t.Fatalf("expected unsafe declared target rejection, got %+v", unsafe)
|
||||
}
|
||||
}
|
||||
|
||||
func TestScopedFileExecutorListsDirectories(t *testing.T) {
|
||||
root := t.TempDir()
|
||||
executor, err := NewFileExecutor(root)
|
||||
|
||||
@@ -106,8 +106,8 @@ func (executor *FileExecutor) existingReadTargetForAssignment(assignment protoco
|
||||
if err != nil {
|
||||
return "", "", false, err
|
||||
}
|
||||
targetKey := assignment.TargetKey
|
||||
if deploymentRoot {
|
||||
targetKey := fileTargetKeyForAssignment(assignment)
|
||||
if deploymentRoot && assignment.ExecutionInput.FileTargetKey == "" {
|
||||
targetKey = deploymentTargetKey(targetKey)
|
||||
filePath, err := existingDeploymentTarget(scope, targetKey)
|
||||
return scope, filePath, deploymentRoot, err
|
||||
@@ -245,8 +245,8 @@ func (executor *FileExecutor) read(ctx context.Context, scope string, deployment
|
||||
if err := ctx.Err(); err != nil {
|
||||
return lifecycleExecutionFailure("file_cancelled", "file read cancelled", false)
|
||||
}
|
||||
targetKey := assignment.TargetKey
|
||||
if deploymentRoot {
|
||||
targetKey := fileTargetKeyForAssignment(assignment)
|
||||
if deploymentRoot && assignment.ExecutionInput.FileTargetKey == "" {
|
||||
targetKey = deploymentTargetKey(targetKey)
|
||||
}
|
||||
var filePath string
|
||||
@@ -292,8 +292,8 @@ func (executor *FileExecutor) write(ctx context.Context, scope string, deploymen
|
||||
if err := ctx.Err(); err != nil {
|
||||
return lifecycleExecutionFailure("file_cancelled", "file write cancelled", false)
|
||||
}
|
||||
targetKey := assignment.TargetKey
|
||||
if deploymentRoot {
|
||||
targetKey := fileTargetKeyForAssignment(assignment)
|
||||
if deploymentRoot && assignment.ExecutionInput.FileTargetKey == "" {
|
||||
targetKey = deploymentTargetKey(targetKey)
|
||||
}
|
||||
var filePath string
|
||||
@@ -365,6 +365,13 @@ func (executor *FileExecutor) write(ctx context.Context, scope string, deploymen
|
||||
return LifecycleExecutionResult{State: lifecycleResultStateSucceeded, Progress: protocol.RunJobProgressReport{Percent: 100, Message: "file write completed"}, Message: "file write completed", ExecutionResult: protocol.RunJobExecutionResult{Kind: "file.write", Version: next.Version, Checksum: checksum, SizeBytes: int64(len(content)), Summary: "atomic compare-and-swap file write"}}
|
||||
}
|
||||
|
||||
func fileTargetKeyForAssignment(assignment protocol.RunJobAssignment) string {
|
||||
if targetKey := strings.TrimSpace(assignment.ExecutionInput.FileTargetKey); targetKey != "" {
|
||||
return targetKey
|
||||
}
|
||||
return assignment.TargetKey
|
||||
}
|
||||
|
||||
func deploymentTargetKey(value string) string {
|
||||
cleaned := path.Clean(strings.TrimPrefix(strings.ReplaceAll(strings.TrimSpace(value), "\\", "/"), "/"))
|
||||
if cleaned == "." || cleaned == "" {
|
||||
|
||||
Reference in New Issue
Block a user