--########################################################################--
--
-- Load billing_cost_estimate parameters (can be overriden in rates_file)
--
--########################################################################--

CONF_DIR       = '/etc/slurm/job_submit'
CONF_FILES     = {'job_submit.conf', 'cluster.conf', 'rates.conf'}

for index, file in ipairs(CONF_FILES) do
  filetoload = CONF_DIR.."/"..file
  file_fh    = io.open(filetoload, "r")
  if file_fh == nil then
    slurm.log_info("slurm_job_modify: No readable %s found!", filetoload)
  else
    io.close(file_fh)
    dofile(filetoload)
  end
end



--- require fonctions verbose_mode, track_gres, scitas_cost.lua, partition_setting.lua ---

if FX_VERBOSE then
  require('job_submit_fx/verbose_mode.lua')
end

if FX_TRACK_GRES then
  require('job_submit_fx/track_gres.lua')
end


if FX_SCITAS_COST then
  require('job_submit_fx/scitas_cost.lua')
end

if FX_PARTITION then
  require('job_submit_fx/partition_setting.lua')
end

if FX_FORCE_ACCOUNT then
  require('job_submit_fx/force_account.lua')
end

require('job_submit_fx/validate_job.lua')


function slurm_job_submit(job_desc, part_list, submit_uid)
  --- Verbose mode ---
  --- To enable verbose mode, you must set the verbose_mode variable to 1
  if FX_VERBOSE then
    verbose_mode(job_desc)
  end

  if FX_PARTITION then
    local partition = partition_setting (job_desc, submit_uid, INFINITE, SEVENTY, PARALLEL_PARTITION, SERIAL_PARTITION)
    if job_desc.partition ~= partition then
      job_desc.partition = partition
      slurm.log_info("slurm_job_modify: for user %u , setting partition: %s",
      submit_uid, partition)
    end
  end

  if FX_TRACK_GRES then
    status = track_gres(job_desc, submit_uid)
    if status ~= 0 then
      return status
    end
  end

  if FX_SCITAS_COST then
    scitas_cost(job_desc, CPU_COST, GPU_COST, CORES_PER_NODE, GPUS_PER_NODE, DEFAULT_WTIME, DEFAULT_PARTITION, PARALLEL_PARTITION, SERIAL_PARTITION, submit_uid)
  end

  if FX_FORCE_ACCOUNT then
    force_account(job_desc)
  end

  validate_job(job_desc)

  return slurm.SUCCESS
end


-- The other required function
function slurm_job_modify(job_desc, job_rec, part_list, modify_uid)
  return slurm.SUCCESS
end

slurm.log_info("job_submit_plugin initialized")
return slurm.SUCCESS