Method DownloadContainerFile

agent/app/service/container.go:1417–1474

(req dto.ContainerFileReq)

}

func (u *ContainerService) DownloadContainerFile(req dto.ContainerFileReq) (io.ReadCloser, string, string, error) {
	if len(req.Path) == 0 {
		req.Path = "/"
	}
	cli, err := docker.NewDockerClient()
	if err != nil {
		return nil, "", "", err
	}

	ctx := context.Background()
	stat, err := cli.ContainerStatPath(ctx, req.ContainerID, req.Path)
	if err != nil {
		_ = cli.Close()
		return nil, "", "", normalizeContainerFileError(err)
	}

	fileName := stat.Name
	if len(fileName) == 0 {
		fileName = "container-file"
	}
	if stat.Mode.IsDir() {
		if _, err := runContainerCommand(cli, req.ContainerID, []string{"tar", "--help"}); err != nil {
			_ = cli.Close()
			return nil, "", "", fmt.Errorf("tar command not found in container")
		}

		targetPath := path.Clean(req.Path)
		parentPath := path.Dir(targetPath)
		targetName := path.Base(targetPath)
		if parentPath == "." || parentPath == "" {
			parentPath = "/"
		}
		tarStream, err := runContainerCommandStream(cli, req.ContainerID, []string{
			"tar", "-czf", "-", "-C", parentPath, "--", targetName,
		})
		if err != nil {
			_ = cli.Close()
			return nil, "", "", err
		}
		if !strings.HasSuffix(fileName, ".tar.gz") {
			fileName += ".tar.gz"
		}
		return &closeHookReader{
			ReadCloser: tarStream,
			onClose:    cli.Close,
		}, fileName, "application/gzip", nil
	}

	fileStream, err := runContainerCommandStream(cli, req.ContainerID, []string{"cat", "--", req.Path})
	if err != nil {
		_ = cli.Close()
		return nil, "", "", err
	}
	return &closeHookReader{
		ReadCloser: fileStream,
		onClose:    cli.Close,
	}, fileName, "application/octet-stream", nil
}