make-install-script.erb

<% if install_info[:distribution] == "redhat" %>
echo "=============================="
echo " <%= install_info[:package_name] %> Installation Script "
echo "=============================="
echo "This script requires superuser access to install rpm packages."
echo "You will be prompted for your password by sudo."

# clear any previous sudo permission
sudo -k

# run inside sudo
sudo sh <<'SCRIPT'
<% if install_info[:channel_version] >= 6 %>
  # add fluent-release to access repository
  distribution=$(cat /etc/system-release-cpe | awk '{print substr($1, index($1, "o"))}' | cut -d: -f2)
  version=$(cat /etc/system-release-cpe | awk '{print substr($1, index($1, "o"))}' | cut -d: -f4 | cut -d. -f1)
  arch=$(rpm --eval %{_arch})
<% if install_info[:lts] %>
  curl --silent --location --output fluent-release.rpm <%= install_info[:base_url] %>/lts/<%= install_info[:channel_version]%>/redhat/${version}/${arch}/fluent-lts-release-<%= install_info[:fluent_release_version] %>.el${version}.noarch.rpm
  if [ -e /etc/yum.repos.d/fluent-package-lts.repo ]; then
    if ! rpm -qf /etc/yum.repos.d/fluent-package-lts.repo; then
      echo "Backup unmanaged .repo to fluent-package-lts.repo.rpmsave"
      mv /etc/yum.repos.d/fluent-package-lts.repo /etc/yum.repos.d/fluent-package-lts.repo.rpmsave
    fi
  fi
<% else %>
  curl --silent --location --output fluent-release.rpm <%= install_info[:base_url] %>/<%= install_info[:channel_version]%>/redhat/${version}/${arch}/fluent-release-<%= install_info[:fluent_release_version] %>.el${version}.noarch.rpm
  if [ -e /etc/yum.repos.d/fluent-package.repo ]; then
    if ! rpm -qf /etc/yum.repos.d/fluent-package.repo; then
      echo "Backup unmanaged .repo to fluent-package.repo.rpmsave"
      mv /etc/yum.repos.d/fluent-package.repo /etc/yum.repos.d/fluent-package.repo.rpmsave
    fi
  fi
<% end %>
  yum install -y ./fluent-release.rpm
  rm -f ./fluent-release.rpm
<% else %>
  # add GPG key
  rpm --import <%= install_info[:base_url] %>/GPG-KEY-td-agent
<% if install_info[:channel_version] >= 5 %>
  rpm --import <%= install_info[:base_url] %>/GPG-KEY-fluent-package
<% end %>

<% if install_info[:channel_version] == 5 %>
  # add <%= install_info[:package_name] %> repository to yum
<% else %>
  # add treasure data repository to yum
<% end %>
  cat >/etc/yum.repos.d/<%= install_info[:repo_file] %> <<'EOF';
[<%= install_info[:repo_label] %>]
name=<%= install_info[:repo_name] %>
<% if install_info[:channel_version] >= 2 %>
baseurl=<%= install_info[:base_url] %>/<% if install_info[:lts] %>lts/<% end %><%= install_info[:channel_version] %>/redhat/$releasever/$basearch
<% else %>
baseurl=<%= install_info[:base_url] %>/redhat/$basearch
<% end %>
<% if install_info[:channel_version] >= 5 %>
enabled=1
<% end %>
gpgcheck=1
gpgkey=<%= install_info[:base_url] %>/GPG-KEY-td-agent<% if install_info[:old_sha1] %>-old-sha1<% end%>
<% if install_info[:channel_version] > 4 %>
       <%= install_info[:base_url] %>/GPG-KEY-fluent-package
<% end %>
EOF
<% end %>

  # update your sources
  yum check-update

  # install the toolbelt
  yes | yum install -y <%= install_info[:package_name] %>

SCRIPT

# message
echo ""
echo "Installation completed. Happy Logging!"
echo ""
<% elsif install_info[:distribution] == 'amazon' %>
echo "=============================="
echo " <%= install_info[:package_name] %> Installation Script "
echo "=============================="
echo "This script requires superuser access to install rpm packages."
echo "You will be prompted for your password by sudo."

# clear any previous sudo permission
sudo -k

# run inside sudo
sudo sh <<'SCRIPT'

<% if install_info[:channel_version] >= 6 %>
  # add fluent-release to access repository
  version=$(cat /etc/system-release-cpe | awk '{print substr($1, index($1, "o"))}' | cut -d: -f4 | cut -d. -f1)
  arch=$(rpm --eval %{_arch})
<% if install_info[:lts] %>
  curl --location --output fluent-release.rpm <%= install_info[:base_url] %>/lts/<%= install_info[:channel_version]%>/amazon/${version}/${arch}/fluent-lts-release-<%= install_info[:fluent_release_version] %>.amzn${version}.noarch.rpm
  if [ -e /etc/yum.repos.d/fluent-package-lts.repo ]; then
    if ! rpm -qf /etc/yum.repos.d/fluent-package-lts.repo; then
      echo "Backup unmanaged .repo to fluent-package-lts.repo.rpmsave"
      mv /etc/yum.repos.d/fluent-package-lts.repo /etc/yum.repos.d/fluent-package-lts.repo.rpmsave
    fi
  fi
<% else %>
  curl --location --output fluent-release.rpm <%= install_info[:base_url] %>/<%= install_info[:channel_version]%>/amazon/${version}/${arch}/fluent-release-<%= install_info[:fluent_release_version] %>.amzn${version}.noarch.rpm
  if [ -e /etc/yum.repos.d/fluent-package.repo ]; then
    if ! rpm -qf /etc/yum.repos.d/fluent-package.repo; then
      echo "Backup unmanaged .repo to fluent-package.repo.rpmsave"
      mv /etc/yum.repos.d/fluent-package.repo /etc/yum.repos.d/fluent-package.repo.rpmsave
    fi
  fi
<% end %>
  yum install -y ./fluent-release.rpm
  rm -f ./fluent-release.rpm
<% else %>
  # add GPG key
  rpm --import <%= install_info[:base_url] %>/GPG-KEY-td-agent
<% if install_info[:channel_version] >= 5 %>
  rpm --import <%= install_info[:base_url] %>/GPG-KEY-fluent-package
<% end %>

<% if install_info[:channel_version] == 5 %>
  # add fluent package repository to yum
<% else %>
  # add treasure data repository to yum
<% end %>
  cat >/etc/yum.repos.d/<%= install_info[:repo_file] %> <<'EOF';
[<%= install_info[:repo_label] %>]
name=<%= install_info[:repo_name] %>
<% if install_info[:channel_version] == 3 and install_info[:version] <= 2 %>
baseurl=<%= install_info[:base_url] %>/<% if install_info[:lts] %>lts/<% end %><%= install_info[:channel_version] %>/<%= install_info[:distribution] %>/<%= install_info[:version] %>/$releasever/$basearch
<% else %>
baseurl=<%= install_info[:base_url] %>/<% if install_info[:lts] %>lts/<% end %><%= install_info[:channel_version] %>/<%= install_info[:distribution] %>/<%= install_info[:version] %>/$basearch
<% end %>
<% if install_info[:channel_version] >= 5 %>
enabled=1
<% end %>
gpgcheck=1
gpgkey=<%= install_info[:base_url] %>/GPG-KEY-td-agent
<% if install_info[:channel_version] >= 5 %>
       <%= install_info[:base_url] %>/GPG-KEY-fluent-package
<% end %>
EOF
<% end %>

  # update your sources
  yum check-update

  # install the toolbelt
  if rpm -q <%= install_info[:package_name] %>; then
    yum update -y <%= install_info[:package_name] %>
  else
    yum install -y <%= install_info[:package_name] %>
  fi

SCRIPT

# message
echo ""
echo "Installation completed. Happy Logging!"
echo ""
<% elsif ['debian', 'ubuntu'].include?(install_info[:distribution]) %>
echo "=============================="
echo " <%= install_info[:package_name] %> Installation Script "
echo "=============================="
echo "This script requires superuser access to install apt packages."
echo "You will be prompted for your password by sudo."

# clear any previous sudo permission
sudo -k

# run inside sudo
executed_distribution=$(cat /etc/os-release | grep "^ID=" | cut -d'=' -f2)
executed_release=$(cat /etc/os-release | grep VERSION_CODENAME | cut -d'=' -f2)
if [ ! "${executed_release}" = "<%= install_info[:version] %>" ]; then
  echo
  echo "[ERROR] Executed wrong installation script for <%= install_info[:distribution] %> <%= install_info[:version] %> on ${executed_distribution} ${executed_release}"
  echo
  exit 1
fi
sudo sh <<SCRIPT
<% if install_info[:channel_version] >= 5 %>
  # use apt-source package which contains keyring
  curl --location --output fluent-apt-source.deb <%= install_info[:base_url] %>/<% if install_info[:lts] %>lts/<% end %><%= install_info[:channel_version] %>/<%= install_info[:distribution] %>/<%= install_info[:version] %>/pool/contrib/f/<%= install_info[:apt_source_deb] %>
  apt install -y ./fluent-apt-source.deb
  rm -f ./fluent-apt-source.deb
<% elsif install_info[:channel_version] == 4 %>
<% if install_info[:distribution] == 'ubuntu' and install_info[:version] == 'xenial' %>
  
    # Deprecated
    curl <%= install_info[:base_url] %>/GPG-KEY-td-agent | apt-key add -
    # add treasure data repository to apt
    echo "deb <%= install_info[:base_url] %>/4/ubuntu/xenial/ xenial contrib" > /etc/apt/sources.list.d/treasure-data.list
  
<% else %>
  
    # use apt-source package which contains keyring
    curl --location --output td-agent-apt-source.deb <%= install_info[:base_url] %>/<%= install_info[:channel_version] %>/<%= install_info[:distribution] %>/<%= install_info[:version] %>/pool/contrib/f/<%= install_info[:apt_source_deb] %>
    apt install -y ./td-agent-apt-source.deb
  
<% end %>
<% elsif install_info[:channel_version] >= 2 %>
  curl <%= install_info[:base_url] %>/GPG-KEY-td-agent | apt-key add -

  # add treasure data repository to apt
  echo "deb <%= install_info[:base_url] %>/<%= install_info[:channel_version] %>/<%= install_info[:distribution] %>/<%= install_info[:version] %>/ <%= install_info[:version] %> contrib" > /etc/apt/sources.list.d/treasure-data.list

<% elsif install_info[:channel_version] == 1 %>
  curl <%= install_info[:base_url] %>/GPG-KEY-td-agent-old-sha1 | apt-key add -

  # add treasure data repository to apt
<% if install_info[:version] == 'lucid' %>
  echo "deb <%= install_info[:base_url] %>/<%= install_info[:distribution] %>/ <%= install_info[:version] %> contrib" > /etc/apt/sources.list.d/treasure-data.list
<% else %>
  echo "deb <%= install_info[:base_url] %>/<%= install_info[:version] %>/ <%= install_info[:version] %> contrib" > /etc/apt/sources.list.d/treasure-data.list
<% end %>

<% end %>
  # update your sources
  <%= install_info[:apt] %> update

  # install the toolbelt
  <%= install_info[:apt] %> install -y <%= install_info[:package_name] %>

SCRIPT

# message
<% if install_info[:channel_version] >= 2.5 %>
if [ $? -eq 0 ]; then
  echo ""
  echo "Installation completed. Happy Logging!"
  echo ""
  exit 0
else
  echo ""
  echo "Installation incompleted. Check above messages."
  echo ""
  exit 1
fi
<% else %>
echo ""
echo "Installation completed. Happy Logging!"
echo ""
exit 0
<% end %>
<% end %>