1ed85dc790
Create a `InputService` class with an abstract method called `map`, meant to be implemented by all inputs. An `unmap` method may be optionally overridden by inputs to cleanup resources. Instantiate a `host.ServiceManager` in the `Stage.run` section and pass the to the host side input code so it can be used to spawn the input services. Convert all existing inputs to the new service framework.
78 lines
1.7 KiB
Python
Executable file
78 lines
1.7 KiB
Python
Executable file
#!/usr/bin/python3
|
|
"""
|
|
Tree inputs
|
|
|
|
Open the tree produced by the pipeline supplied via the
|
|
first and only entry in `references`. The tree is opened
|
|
in read only mode. If the id is `null` or the empty
|
|
string it returns an empty tree.
|
|
"""
|
|
|
|
import sys
|
|
|
|
from osbuild import inputs
|
|
|
|
|
|
SCHEMA = """
|
|
"additionalProperties": false,
|
|
"required": ["type", "origin", "references"],
|
|
"properties": {
|
|
"type": {
|
|
"enum": ["org.osbuild.tree"]
|
|
},
|
|
"origin": {
|
|
"description": "The origin of the input (must be 'org.osbuild.pipeline')",
|
|
"type": "string",
|
|
"enum": ["org.osbuild.pipeline"]
|
|
},
|
|
"references": {
|
|
"description": "Exactly one pipeline identifier to ues as tree input",
|
|
"oneOf": [{
|
|
"type": "array",
|
|
"additionalItems": false,
|
|
"items": [{
|
|
"type": "string"
|
|
}]
|
|
}, {
|
|
"type": "object",
|
|
"additionalProperties": false,
|
|
"patternProperties": {
|
|
".*": {
|
|
"type": "object",
|
|
"additionalProperties": false
|
|
}
|
|
},
|
|
"minProperties": 1,
|
|
"maxProperties": 1
|
|
}]
|
|
}
|
|
}
|
|
"""
|
|
|
|
|
|
class TreeInput(inputs.InputService):
|
|
|
|
def map(self, store, _origin, refs, target, _options):
|
|
|
|
# input verification *must* have been done via schema
|
|
# verification. It is expected that origin is a pipeline
|
|
# and we have exactly one reference, i.e. a pipeline id
|
|
pid, _ = refs.popitem()
|
|
|
|
if pid:
|
|
path = store.read_tree_at(pid, target)
|
|
|
|
if not path:
|
|
raise ValueError(f"Unknown pipeline '{pid}'")
|
|
|
|
reply = {"path": target}
|
|
return reply
|
|
|
|
|
|
def main():
|
|
service = TreeInput.from_args(sys.argv[1:])
|
|
service.main()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|