#!/usr/bin/python3
#
# Determine if this tool can run a test based on a test spec.
#

import datetime
import sys

import pscheduler

json = pscheduler.json_load(exit_on_error=True);

# TODO: Should be checking the package schema and the task spec schema.
# TODO: Should be validating.

try:
    if json['type'] != 'rtt':
        pscheduler.succeed_json({
            "can-run": False,
            "reasons": [ "Unsupported test type" ]
        })
except KeyError:
    pscheduler.succeed_json({
        "can-run": False,
        "reasons": [ "Missing test type" ]
    })


try:
    spec = json["spec"]
    pscheduler.json_check_schema(spec, 4)
except KeyError:
    pscheduler.succeed_json({
        "can-run": False,
        "reasons": ["Missing test specification"]
    })
except ValueError as ex:
    pscheduler.succeed_json({
        "can-run": False,
        "reasons": [str(ex)]
    })


errors = []

source = spec.get("source-node", spec.get("source", None))

try:
    ip_version = spec['ip-version']
except KeyError:
    ip_version = 4


if 'flow-label' in spec and ip_version != 6:
    errors.append("Cannot apply flow labels except with IPv6")

if 'protocol' in spec and spec['protocol'] != 'icmp':
    errors.append("Only icmp protocol is supported")

if 'port' in spec:
    errors.append("Ports are not supported")


result = {
    "can-run": len(errors) == 0
}

if len(errors) > 0:
    result["reasons"] = errors

pscheduler.succeed_json(result)
