library(optparse)
library(msPurity)
print(sessionInfo())

# Get the parameter
option_list <- list(
  make_option(c("-s","--sm_resultPth"),type="character"),
  make_option(c("-m","--metfrag_resultPth"),type="character"),
  make_option(c("-c","--sirius_csi_resultPth"),type="character"),
  make_option(c("-p","--probmetab_resultPth"),type="character"),
  make_option(c("-l","--ms1_lookup_resultPth"),type="character"),

  make_option("--ms1_lookup_checkAdducts", action="store_true"),
  make_option("--ms1_lookup_keepAdducts", type="character", default=NA),
  make_option("--ms1_lookup_dbSource", type="character", default="hmdb"),

  make_option("--sm_weight", type="numeric"),
  make_option("--metfrag_weight", type="numeric"),
  make_option("--sirius_csi_weight", type="numeric"),
  make_option("--probmetab_weight", type="numeric"),
  make_option("--ms1_lookup_weight", type="numeric"),
  make_option("--biosim_weight", type="numeric"),
  
  make_option("--summaryOutput", action="store_true"),
  
  make_option("--create_new_database", action="store_true"),
  make_option("--outdir", type="character", default="."),

  make_option("--compoundDbType", type="character", default="sqlite"),
  make_option("--compoundDbPth", type="character", default=NA),
  make_option("--compoundDbHost", type="character", default=NA)
)
opt <- parse_args(OptionParser(option_list=option_list))

print(opt)

if (!is.null(opt$create_new_database)){
  sm_resultPth <-  file.path(opt$outdir, 'combined_annotations.sqlite')
  file.copy(opt$sm_resultPth, sm_resultPth)
}else{
  sm_resultPth <- opt$sm_resultPth
}

if (is.null(opt$ms1_lookup_checkAdducts)){
  opt$ms1_lookup_checkAdducts <- FALSE
}
if (!is.null(opt$ms1_lookup_keepAdducts)){
    opt$ms1_lookup_keepAdducts <- gsub("__ob__", "[", opt$ms1_lookup_keepAdducts)
    opt$ms1_lookup_keepAdducts <- gsub("__cb__", "]", opt$ms1_lookup_keepAdducts)
    ms1_lookup_keepAdducts <- strsplit(opt$ms1_lookup_keepAdducts, ",")[[1]]
}

weights <-list('sm'=opt$sm_weight,
               'metfrag'=opt$metfrag_weight,
               'sirius_csifingerid'= opt$sirius_csi_weight,
               'probmetab'=opt$probmetab_weight,
               'ms1_lookup'=opt$ms1_lookup_weight,
               'biosim'=opt$biosim_weight
               )
print(weights)

if (is.null(opt$probmetab_resultPth)){
  opt$probmetab_resultPth = NA
}

if (round(!sum(unlist(weights),0)==1)){
  stop(paste0('The weights should sum to 1 not ', sum(unlist(weights))))
}

if (is.null(opt$summaryOutput)){
  summaryOutput = FALSE
}else{
 summaryOutput = TRUE
}

if (opt$compoundDbType=='local_config'){
  # load in compound config
  # Soure local function taken from workflow4metabolomics
  source_local <- function(fname){ argv <- commandArgs(trailingOnly=FALSE); base_dir <- dirname(substring(argv[grep("--file=", argv)], 8)); source(paste(base_dir, fname, sep="/")) }
  source_local("dbconfig.R")
}else{
  compoundDbPth = opt$compoundDbPth
  compoundDbType = opt$compoundDbType
  compoundDbName = NA
  compoundDbHost = NA
  compoundDbPort = NA
  compoundDbUser = NA
  compoundDbPass = NA
}

summary_output <- msPurity::combineAnnotations(
                            sm_resultPth = sm_resultPth,
                            compoundDbPth = compoundDbPth,
                            metfrag_resultPth = opt$metfrag_resultPth,
                            sirius_csi_resultPth = opt$sirius_csi_resultPth,
                            probmetab_resultPth = opt$probmetab_resultPth,
                            ms1_lookup_resultPth = opt$ms1_lookup_resultPth,
                            ms1_lookup_keepAdducts = ms1_lookup_keepAdducts,
                            ms1_lookup_checkAdducts = opt$ms1_lookup_checkAdducts,

                            compoundDbType = compoundDbType,
                            compoundDbName = compoundDbName,
                            compoundDbHost = compoundDbHost,
                            compoundDbPort = compoundDbPort,
                            compoundDbUser = compoundDbUser,
                            compoundDbPass = compoundDbPass,
                            weights = weights,
                            summaryOutput = summaryOutput)
if (summaryOutput){
  write.table(summary_output, file.path(opt$outdir, 'combined_annotations.tsv'), sep = '\t', row.names = FALSE)
}


write.table(summary_output, file.path(opt$outdir, 'combined_annotations.tsv'), sep = '\t', row.names = FALSE)


closeAllConnections()