#Snakefile

if config["workflow"] == "complete":
    include:
        "workflows/Combine_PathoFact_workflow"
    rule all:
        input:
            expand(
                [
                    "{OUTDIR}/{project}/AMR/{input_file}_AMR_MGE_prediction_detailed.tsv",
                    "{OUTDIR}/{project}/Toxin_gene_library_{input_file}_report.tsv",
                    "{OUTDIR}/{project}/PathoFact_{input_file}_predictions.csv"
                ],
                OUTDIR=config["OUTDIR"], project=config["project"], input_file=config["input_file"]
            )
elif config["workflow"] == "Tox":
    include:
        "workflows/Toxin_workflow"
    rule all:
        input:
            expand(
                [
                    "{OUTDIR}/{project}/Toxin_prediction_{input_file}_report.csv",
                    "{OUTDIR}/{project}/Toxin_gene_library_{input_file}_report.tsv"
                ],
                OUTDIR=config["OUTDIR"], project=config["project"], input_file=config["input_file"]
            )
elif config["workflow"] == "Vir":
    include:
        "workflows/Virulence_workflow"
    rule all:
        input:
            expand(
                "{OUTDIR}/{project}/Virulence_prediction_{input_file}_report.csv",
                OUTDIR=config["OUTDIR"], project=config["project"], input_file=config["input_file"]
            )                    
elif config["workflow"] == "AMR":
    include:
        "workflows/AMR_workflow"
    rule all:
        input:
            expand(
                [
                    "{OUTDIR}/{project}/AMR_MGE_prediction_{input_file}_report.tsv",
                    "{OUTDIR}/{project}/AMR/{input_file}_AMR_MGE_prediction_detailed.tsv"
                ],
                OUTDIR=config["OUTDIR"], project=config["project"], input_file=config["input_file"]
            )
else:
    raise Exception("Unknown workflow option: %s" % config["workflow"])