#!/usr/bin/python3
"""
Inputs for individual files

Provides all the files, named via their content hash, specified
via `references` in a new directory.

The returned data in `refs` is a dictionary where the keys are
the content hash of a file and the values dictionaries with
metadata for it. The input itself currently does not set any
metadata itself, but will forward any metadata set via the
`metadata` property. Keys in that must start with a prefix,
like `rpm.` to avoid namespace clashes. This is enforced via
schema validation.
"""


import json
import sys
import subprocess

from osbuild.objectstore import StoreClient


SCHEMA = r"""
"additionalProperties": false,
"required": ["type", "origin", "references"],
"properties": {
  "type": {
    "enum": ["org.osbuild.files"]
  },
  "origin": {
    "description": "The origin of the input (must be 'org.osbuild.source')",
    "type": "string",
    "enum": ["org.osbuild.source"]
  },
  "references": {
    "description": "Checksums of files to use as files input",
    "oneOf": [{
      "type": "array",
      "items": {
        "type": "string"
      }
    }, {
      "type": "object",
      "additionalProperties": false,
      "minProperties": 1,
      "patternProperties": {
        ".*": {
          "type": "object",
          "additionalProperties": false,
          "properties": {
            "metadata": {
              "description": "Additinal metadata to forward to the stage",
              "type": "object",
              "additionalProperties": false,
              "patternProperties": {
                "^\\w+[.]{1}\\w+$": {
                  "additionalProperties": false
                }
              }
            }
          }
        }
      }
    }]
  }
}
"""


def main():
    args = json.load(sys.stdin)
    refs = args["refs"]

    store = StoreClient(connect_to=args["api"]["store"])
    source = store.source("org.osbuild.files")
    output = store.mkdtemp(prefix="files-input-")

    for checksum in refs:
        try:
            subprocess.run(
                [
                    "ln",
                    f"{source}/{checksum}",
                    f"{output}/{checksum}",
                ],
                check=True,
            )
        except subprocess.CalledProcessError as e:
            json.dump({"error": e.output}, sys.stdout)
            return 1


    reply = {
        "path": output,
        "data": {
          "refs": refs
        }
    }

    json.dump(reply, sys.stdout)
    return 0


if __name__ == '__main__':
    r = main()
    sys.exit(r)