utils/compression.go

package utils

import "fmt"

var (
	pipeThroughProgram PipeThroughProgram
)

const DefaultPipeThroughProgram = "cat -"

type PipeThroughProgram struct {
	Name          string
	OutputCommand string
	InputCommand  string
	Extension     string
}

func InitializePipeThroughParameters(compress bool, compressionType string, compressionLevel int) {
	if !compress {
		pipeThroughProgram = PipeThroughProgram{Name: "cat", OutputCommand: DefaultPipeThroughProgram, InputCommand: DefaultPipeThroughProgram, Extension: ""}
		return
	}

	// backward compatibility for inputs without compressionType
	if compressionType == "" {
		compressionType = "gzip"
	}

	if compressionType == "gzip" {
		pipeThroughProgram = PipeThroughProgram{Name: "gzip", OutputCommand: fmt.Sprintf("gzip -c -%d", compressionLevel), InputCommand: "gzip -d -c", Extension: ".gz"}
		return
	}

	if compressionType == "zstd" {
		pipeThroughProgram = PipeThroughProgram{Name: "zstd", OutputCommand: fmt.Sprintf("zstd --compress -%d -c", compressionLevel), InputCommand: "zstd --decompress -c", Extension: ".zst"}
		return
	}
}

func GetPipeThroughProgram() PipeThroughProgram {
	return pipeThroughProgram
}

func SetPipeThroughProgram(compression PipeThroughProgram) {
	pipeThroughProgram = compression
}