Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: Add option to modify whether symlink if followed in ZIP archive #1803

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 24 additions & 3 deletions packages/artifact/__tests__/upload-zip-specification.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ const extraFileInFolderCPath = path.join(
'extra-file-in-folder-c.txt'
)
const amazingFileInFolderHPath = path.join(root, 'folder-h', 'amazing-item.txt')
const symlinkToFolderDPath = path.join(root, 'symlink-to-folder-d')

const artifactFilesToUpload = [
goodItem1Path,
Expand All @@ -46,7 +47,8 @@ const artifactFilesToUpload = [
goodItem4Path,
goodItem5Path,
extraFileInFolderCPath,
amazingFileInFolderHPath
amazingFileInFolderHPath,
symlinkToFolderDPath
]

describe('Search', () => {
Expand Down Expand Up @@ -89,6 +91,8 @@ describe('Search', () => {
await fs.writeFile(extraFileInFolderCPath, 'extra file')

await fs.writeFile(amazingFileInFolderHPath, 'amazing file')

await fs.symlink(path.join(root, 'folder-d'), symlinkToFolderDPath)
/*
Directory structure of files that get created:
root/
Expand All @@ -112,6 +116,7 @@ describe('Search', () => {
folder-i/
bad-item4.txt
bad-item5.txt
symlink-to-folder-d -> folder-d/
good-item5.txt
*/
})
Expand Down Expand Up @@ -168,7 +173,7 @@ describe('Search', () => {
artifactFilesToUpload,
root
)
expect(specifications.length).toEqual(7)
expect(specifications.length).toEqual(8)

const absolutePaths = specifications.map(item => item.sourcePath)
expect(absolutePaths).toContain(goodItem1Path)
Expand All @@ -178,6 +183,7 @@ describe('Search', () => {
expect(absolutePaths).toContain(goodItem5Path)
expect(absolutePaths).toContain(extraFileInFolderCPath)
expect(absolutePaths).toContain(amazingFileInFolderHPath)
expect(absolutePaths).toContain(symlinkToFolderDPath)

for (const specification of specifications) {
if (specification.sourcePath === goodItem1Path) {
Expand Down Expand Up @@ -213,6 +219,13 @@ describe('Search', () => {
expect(specification.destinationPath).toEqual(
path.join('/folder-h', 'amazing-item.txt')
)
} else if (specification.sourcePath === symlinkToFolderDPath) {
expect(specification.destinationPath).toEqual(
path.join('/symlink-to-folder-d')
)
expect(specification.symlinkTargetPath).toEqual(
path.join(root, '/folder-d')
)
} else {
throw new Error(
'Invalid specification found. This should never be reached'
Expand All @@ -227,7 +240,7 @@ describe('Search', () => {
artifactFilesToUpload,
rootWithSlash
)
expect(specifications.length).toEqual(7)
expect(specifications.length).toEqual(8)

const absolutePaths = specifications.map(item => item.sourcePath)
expect(absolutePaths).toContain(goodItem1Path)
Expand All @@ -237,6 +250,7 @@ describe('Search', () => {
expect(absolutePaths).toContain(goodItem5Path)
expect(absolutePaths).toContain(extraFileInFolderCPath)
expect(absolutePaths).toContain(amazingFileInFolderHPath)
expect(absolutePaths).toContain(symlinkToFolderDPath)

for (const specification of specifications) {
if (specification.sourcePath === goodItem1Path) {
Expand Down Expand Up @@ -272,6 +286,13 @@ describe('Search', () => {
expect(specification.destinationPath).toEqual(
path.join('/folder-h', 'amazing-item.txt')
)
} else if (specification.sourcePath === symlinkToFolderDPath) {
expect(specification.destinationPath).toEqual(
path.join('/symlink-to-folder-d')
)
expect(specification.symlinkTargetPath).toEqual(
path.join(root, '/folder-d')
)
} else {
throw new Error(
'Invalid specification found. This should never be reached'
Expand Down
7 changes: 7 additions & 0 deletions packages/artifact/src/internal/shared/interfaces.ts
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,13 @@ export interface UploadArtifactOptions {
* For large files that are not easily compressed, a value of 0 is recommended for significantly faster uploads.
*/
compressionLevel?: number

/**
* Whether or not symlinks in artifact ZIP should be followed.
*
* By default, artifact ZIP will follow symlinks.
*/
followSymlinks?: boolean
}

/**
Expand Down
3 changes: 2 additions & 1 deletion packages/artifact/src/internal/upload/upload-artifact.ts
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@ export async function uploadArtifact(

const zipUploadStream = await createZipUploadStream(
zipSpecification,
options?.compressionLevel
options?.compressionLevel,
options?.followSymlinks
)

// Upload zip to blob storage
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,11 @@ export interface UploadZipSpecification {
* The destination path in a zip for a file
*/
destinationPath: string

/**
* The relative path to a symlink target (file or directory) in a zip
*/
symlinkTargetPath?: string
}

/**
Expand Down Expand Up @@ -78,7 +83,8 @@ export function getUploadZipSpecification(
if (!fs.existsSync(file)) {
throw new Error(`File ${file} does not exist`)
}
if (!fs.statSync(file).isDirectory()) {
const fileLstat = fs.lstatSync(file)
if (!fileLstat.isDirectory()) {
// Normalize and resolve, this allows for either absolute or relative paths to be used
file = normalize(file)
file = resolve(file)
Expand All @@ -94,7 +100,10 @@ export function getUploadZipSpecification(

specification.push({
sourcePath: file,
destinationPath: uploadPath
destinationPath: uploadPath,
symlinkTargetPath: fileLstat.isSymbolicLink()
? fs.readlinkSync(file)
: undefined
})
} else {
// Empty directory
Expand Down
16 changes: 11 additions & 5 deletions packages/artifact/src/internal/upload/zip.ts
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,8 @@ export class ZipUploadStream extends stream.Transform {

export async function createZipUploadStream(
uploadSpecification: UploadZipSpecification[],
compressionLevel: number = DEFAULT_COMPRESSION_LEVEL
compressionLevel: number = DEFAULT_COMPRESSION_LEVEL,
followSymlinks = true
): Promise<ZipUploadStream> {
core.debug(
`Creating Artifact archive with compressionLevel: ${compressionLevel}`
Expand All @@ -42,10 +43,15 @@ export async function createZipUploadStream(

for (const file of uploadSpecification) {
if (file.sourcePath !== null) {
// Add a normal file to the zip
zip.file(file.sourcePath, {
name: file.destinationPath
})
if (file.symlinkTargetPath !== undefined && !followSymlinks) {
// Add a symlink to the zip
zip.symlink(file.destinationPath, file.symlinkTargetPath)
} else {
// Add a normal file to the zip
zip.file(file.sourcePath, {
name: file.destinationPath
})
}
} else {
// Add a directory to the zip
zip.append('', {name: file.destinationPath})
Expand Down
2 changes: 1 addition & 1 deletion packages/glob/package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion packages/http-client/package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.