install.rb

#!/usr/bin/env ruby
# frozen_string_literal: true

require 'fileutils'
require 'pathname'
require 'erb'
require 'ostruct'
require 'yaml'

def env
  e = ENV['WEBMO_ENVIRONMENT']
  raise StandardError, "No environment variable 'WEBMO_ENVIRONMENT'. Don't know where to install." if e.nil?

  unless ['dev', 'test', 'production'].include?(e)
    raise StandardError, "WEBMO_ENVIRONMENT must be set to 'dev', 'test' or 'production'. '#{e}' is invalid."
  end

  e
end

def template(name)
  fname = "#{name}.int.erb"
  File.read("#{templates_dir}/#{fname}")
end

def deployment_base_dir
  "#{__dir__}/deployments/#{env}"
end

def user_base
  "#{deployment_base_dir}/user_data"
end

def install_base_dir
  "#{deployment_base_dir}/install"
end

def cgi_base
  "#{install_base_dir}/cgi-bin"
end

def templates_dir
  "#{__dir__}/templates"
end

def globals_result_location
  "#{interfaces_dir}/globals.int"
end

def interfaces_dir
  "#{cgi_base}/interfaces"
end

def pbs_conf_src
  "#{__dir__}/pbs.conf"
end

def pbs_conf_dest
  "#{interfaces_dir}/pbs.conf"
end

def enabled_libraries
  [ 'gamess', 'gaussian', 'mopac', 'qchem', 'tinker' ]
end

def license
  lic = ENV['WEBMO_LICENSE']
  raise StandardError, "No environment variable 'WEBMO_LICENSE'. Did you source .webmo.info?" if lic.nil?

  lic
end

def version
  webmo_tar.to_s.match(/WebMO\.(?<version>.+)\.tar\.gz/)[:version]
end

def template_context(name)
  data = {
    'globals' => {
      'cgi_base' =>   cgi_base,
      'html_base' =>  "#{install_base_dir}/source",
      'license' => license,
      'user_base' => user_base,
      'version' => version
    }
  }.merge(YAML.load_file("#{__dir__}/lib_config.yml", aliases: true))

  OpenStruct.new(data[name.to_s])
end

def prep_and_untar
  FileUtils.mkdir_p(user_base)
  FileUtils.mkdir_p(install_base_dir)

  Dir.chdir(install_base_dir) do
    system("tar -xzf #{webmo_tar} --strip-components=1")
  end

  add_shebangs
end

def add_shebangs
  Dir.glob("#{cgi_base}/**/*.{pm,cgi}") do |file|
    original_content = File.read(file)
    new_content = <<HEREDOC
#!/usr/bin/perl

# This script is copyright (c) 2019 by WebMO, LLC, all rights reserved
# Its use is subject to the license agreement that can be found at the following
# URL:  http://www.webmo.net/license

#{original_content}
HEREDOC

    File.open(file, 'w') { |f| f.write(new_content) }
  end
end

def configure_queues
  int_files = Dir.glob("#{interfaces_dir}/*.int.disabled")

  # cleanup the old directories if they exist
  if Dir.exist?("#{cgi_base}/queues")
    old_dirs = Dir.children("#{cgi_base}/queues").reject { |d| d == 'cpu' }
    old_dirs.each do |dir|
      FileUtils.rm_rf("#{cgi_base}/queues/#{dir}")
    end
  end

  ['cpu'].each do |queue|
    dir = "#{cgi_base}/queues/#{queue}"
    FileUtils.mkdir_p(dir)
    FileUtils.cp_r(int_files, dir)

    enabled_libraries.each do |lib|
      dest_file = "#{dir}/#{lib}.int"
      FileUtils.mv("#{dir}/#{lib}.int.disabled", dest_file)
      context = template_context(lib)
      context = fix_parallel_context(context) if queue == 'parallel'

      content = ERB.new(template(lib)).result(context.instance_eval { binding })
      File.open(dest_file, 'w') { |f| f.write(content) }
    end
  end
end

def fix_parallel_context(context)
  context.nodes_min = 2
  context.nodes_max = 10

  context
end

def add_webmo_profile
  FileUtils.copy("#{__dir__}/webmo_profile", "#{user_base}/.webmo_profile")
end

def configure
  content = ERB.new(template('globals')).result(template_context('globals').instance_eval { binding })
  File.open(globals_result_location, 'w') { |f| f.write(content) }

  FileUtils.copy(pbs_conf_src, pbs_conf_dest)

  configure_queues
  add_webmo_profile
end

def install
  prep_and_untar
  configure
end

def webmo_tar
  tar = Pathname.glob("#{__dir__}/WebMO.*.tar.gz").max
  raise StandardError, "Cannot find any WebMo.*.tar.gz in #{__dir__}" if tar.nil?

  tar
end

install