96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"archive/zip"
|
|
"context"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"gitlab.com/gitlab-org/labkit/mask"
|
|
|
|
"gitlab.com/gitlab-org/gitlab/workhorse/internal/zipartifacts"
|
|
)
|
|
|
|
const progName = "gitlab-zip-cat"
|
|
|
|
var Version = "unknown"
|
|
|
|
var printVersion = flag.Bool("version", false, "Print version and exit")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
version := fmt.Sprintf("%s %s", progName, Version)
|
|
if *printVersion {
|
|
fmt.Println(version)
|
|
os.Exit(0)
|
|
}
|
|
|
|
archivePath := os.Getenv("ARCHIVE_PATH")
|
|
encodedFileName := os.Getenv("ENCODED_FILE_NAME")
|
|
|
|
if len(os.Args) != 1 || archivePath == "" || encodedFileName == "" {
|
|
fmt.Fprintf(os.Stderr, "Usage: %s\n", progName)
|
|
fmt.Fprintf(os.Stderr, "Env: ARCHIVE_PATH=https://path.to/archive.zip or /path/to/archive.zip\n")
|
|
fmt.Fprintf(os.Stderr, "Env: ENCODED_FILE_NAME=base64-encoded-file-name\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
scrubbedArchivePath := mask.URL(archivePath)
|
|
|
|
fileName, err := zipartifacts.DecodeFileEntry(encodedFileName)
|
|
if err != nil {
|
|
fatalError(fmt.Errorf("decode entry %q", encodedFileName), err)
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
archive, err := zipartifacts.OpenArchive(ctx, archivePath)
|
|
if err != nil {
|
|
fatalError(errors.New("open archive"), err)
|
|
}
|
|
|
|
file := findFileInZip(fileName, archive)
|
|
if file == nil {
|
|
fatalError(fmt.Errorf("find %q in %q: not found", fileName, scrubbedArchivePath), zipartifacts.ErrorCode[zipartifacts.CodeEntryNotFound])
|
|
}
|
|
// Start decompressing the file
|
|
reader, err := file.Open()
|
|
if err != nil {
|
|
fatalError(fmt.Errorf("open %q in %q", fileName, scrubbedArchivePath), err)
|
|
}
|
|
defer reader.Close()
|
|
|
|
if _, err := fmt.Printf("%d\n", file.UncompressedSize64); err != nil {
|
|
fatalError(fmt.Errorf("write file size invalid"), err)
|
|
}
|
|
|
|
if _, err := io.Copy(os.Stdout, reader); err != nil {
|
|
fatalError(fmt.Errorf("write %q from %q to stdout", fileName, scrubbedArchivePath), err)
|
|
}
|
|
}
|
|
|
|
func findFileInZip(fileName string, archive *zip.Reader) *zip.File {
|
|
for _, file := range archive.File {
|
|
if file.Name == fileName {
|
|
return file
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func fatalError(contextErr error, statusErr error) {
|
|
code := zipartifacts.ExitCodeByError(statusErr)
|
|
|
|
fmt.Fprintf(os.Stderr, "%s error: %v - %v, code: %d\n", progName, statusErr, contextErr, code)
|
|
|
|
if code > 0 {
|
|
os.Exit(code)
|
|
} else {
|
|
os.Exit(1)
|
|
}
|
|
}
|