101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
// copy.go
|
|
//
|
|
// This source file is part of the FoundationDB open source project
|
|
//
|
|
// Copyright 2021 Apple Inc. and the FoundationDB project authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/go-logr/logr"
|
|
)
|
|
|
|
// copyFile copies a file into the output directory.
|
|
func copyFile(logger logr.Logger, inputPath string, outputPath string, required bool) error {
|
|
logger.Info("Copying file", "inputPath", inputPath, "outputPath", outputPath)
|
|
inputFile, err := os.Open(inputPath)
|
|
if err != nil {
|
|
logger.Error(err, "Error opening file", "path", inputPath)
|
|
return err
|
|
}
|
|
defer inputFile.Close()
|
|
|
|
inputInfo, err := inputFile.Stat()
|
|
if err != nil {
|
|
logger.Error(err, "Error getting stats for file", "path", inputPath)
|
|
return err
|
|
}
|
|
|
|
if required && inputInfo.Size() == 0 {
|
|
return fmt.Errorf("File %s is empty", inputPath)
|
|
}
|
|
|
|
outputDir := path.Dir(outputPath)
|
|
|
|
tempFile, err := os.CreateTemp(outputDir, "")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer tempFile.Close()
|
|
|
|
_, err = tempFile.ReadFrom(inputFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = tempFile.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = os.Chmod(tempFile.Name(), inputInfo.Mode())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = os.Rename(tempFile.Name(), outputPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// CopyFiles copies a list of files into the output directory.
|
|
func CopyFiles(logger logr.Logger, outputDir string, copyDetails map[string]string, requiredCopies map[string]bool) error {
|
|
for inputPath, outputSubpath := range copyDetails {
|
|
if outputSubpath == "" {
|
|
outputSubpath = path.Base(inputPath)
|
|
}
|
|
outputPath := path.Join(outputDir, outputSubpath)
|
|
err := os.MkdirAll(path.Dir(outputPath), os.ModeDir|os.ModePerm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
required := requiredCopies[inputPath]
|
|
err = copyFile(logger, inputPath, outputPath, required)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|