#AMR

import glob
import os

PROJECT = config["project"]
INPUT   = config["input_file"]

##########################
#     AMR Prediction     #
##########################

rule combine_AMR_plasmid:
    input:
        AMR="{OUTDIR}/{project}/AMR/deepARG_results/{input_file}.out.mapping.ARG",
        AMR_translation="{OUTDIR}/{project}/renamed/{input_file}_translation.tsv",
        Plasmid="{OUTDIR}/{project}/MGE/plasmid/{input_file}_plasflow_prediction_final.tsv",
        Contig_translation="{OUTDIR}/{project}/renamed/{input_file}_Contig_translation.tsv",
        Contig_gene_list="{OUTDIR}/{input_file}.contig",
        VirFinder="{OUTDIR}/{project}/MGE/phage/{input_file}_VirFinder_aggregated.csv",
        VirSorter="{OUTDIR}/{project}/MGE/phage/{input_file}_VIRSorter_aggregated.csv"
    output:    
        Report_1=temp("{OUTDIR}/{project}/AMR/{input_file}_MGE_AMR_prediction_detail_temp.csv"),
        Report_2=temp("{OUTDIR}/{project}/AMR/{input_file}_MGE_AMR_prediction_report_temp.csv")
    log:
        "{OUTDIR}/{project}/AMR/{input_file}_MGE_AMR_prediction_detail_temp.log"
    params:
        outdir="{OUTDIR}"
    conda:
        "../../envs/R.yaml"
    script:
        "../../scripts/AMR_MGE.R"

rule modify_details:
    input:
        "{OUTDIR}/{project}/AMR/{input_file}_MGE_AMR_prediction_detail_temp.csv"
    output:
        "{OUTDIR}/{project}/AMR/{input_file}_AMR_MGE_prediction_detailed.tsv"
    params:
        outdir="{OUTDIR}"
    shell:
        """
        sed 's/"//g' {input} | sed 's/,/\t/g' > {output}
        """

rule modify_report:
    input:
        "{OUTDIR}/{project}/AMR/{input_file}_MGE_AMR_prediction_report_temp.csv"
    output:
        "{OUTDIR}/{project}/AMR_MGE_prediction_{input_file}_report.tsv"
    params:
        outdir="{OUTDIR}"
    shell:
        """
        sed 's/"//g' {input} | sed 's/,/\t/g' > {output}
        """