#!/bin/bash

PATHS='/etc/ood/config/apps/dashboard/initializers/fix_slurm_18.rb /var/www/ood/apps/sys/activejobs/config/initializers/fix_slurm_18.rb /etc/ood/config/apps/myjobs/initializers/fix_slurm_18.rb'

function monkey_patch_patch {
  read -r -d '' SCRIPT_CONTENT <<'EOF'
require 'ood_core/job/adapters/slurm'

class OodCore::Job::Adapters::Slurm::Batch
  alias_method :orig_fields, :fields

  def fields
    orig_fields.tap { |f| f.delete(:gres) }
  end
end
EOF

  for file in $PATHS; do
    DIR=$(echo "$file" | sed 's/fix_slurm_18.rb//')
    echo "Creating directory: $DIR"
    mkdir -p "$DIR"
    echo "Writing file: $file"
    echo "$SCRIPT_CONTENT" > $file
  done  
}

function monkey_patch_unpatch {
  for file in $PATHS; do
    echo "Removing file: $file"
    rm -f "$file"
  done
}

function usage {
  cat <<EOF
Usage:
  bash patch_slurm18.sh patch|unpatch

  patch: creates initializer files that remove the field 'gres' from the Slurm Adapter's native options
  unpatch: removes the initializer files, but leaves the directories intact
EOF
}

function main {
  case "$1" in
    "patch" )
      monkey_patch_patch;;
    "unpatch" )
      monkey_patch_unpatch;;
    * )
      usage;;
  esac  
}

main "$@"