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()
