stages: add org.osbuild.gzip to compress files

Add a new stage that will take a file from the input and compress
it via gzip.
This commit is contained in:
Diaa Sami 2021-08-18 16:50:20 +02:00 committed by Christian Kellner
parent a40cf616f5
commit b93dedf353
6 changed files with 1495 additions and 0 deletions

71
stages/org.osbuild.gzip Executable file
View file

@ -0,0 +1,71 @@
#!/usr/bin/python3
"""
Compress a file using gzip
Buildhost commands used: `gzip`.
"""
import os
import subprocess
import sys
import osbuild.api
SCHEMA_2 = r"""
"inputs": {
"type": "object",
"additionalProperties": false,
"required": ["file"],
"properties": {
"file": {
"type": "object",
"additionalProperties": true
}
}
},
"options": {
"additionalProperties": false,
"required": ["filename"],
"properties": {
"filename": {
"description": "Filename to use for the compressed file",
"type": "string"
}
}
}
"""
def parse_input(inputs):
image = inputs["file"]
files = image["data"]["files"]
assert len(files) == 1
filename, _ = files.popitem()
filepath = os.path.join(image["path"], filename)
return filepath
def main(inputs, output, options):
filename = options["filename"].lstrip("/")
source = parse_input(inputs)
target = os.path.join(output, filename)
with open(target, "w") as f:
cmd = [
"gzip", "--stdout", "-1", source
]
subprocess.run(
cmd, stdout=f, check=True
)
return 0
if __name__ == '__main__':
args = osbuild.api.arguments()
r = main(args["inputs"], args["tree"], args["options"])
sys.exit(r)