From 92c2d4840be1042236eeb3b6caf7ace43e1abfc9 Mon Sep 17 00:00:00 2001
From: n0ts <me@n0ts.org>
Date: Wed, 21 Dec 2016 15:16:16 +0900
Subject: [PATCH] Add DD_VERSION

---
 .../datadog-agent/source/install_agent.sh     | 27 ++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/packaging/datadog-agent/source/install_agent.sh b/packaging/datadog-agent/source/install_agent.sh
index c9b6647a90..fdf689772f 100755
--- a/packaging/datadog-agent/source/install_agent.sh
+++ b/packaging/datadog-agent/source/install_agent.sh
@@ -55,6 +55,12 @@ else
     no_start=false
 fi
 
+if [ -n "$DD_VERSION" ]; then
+    dd_version=$DD_VERSION
+else
+    dd_version=
+fi
+
 if [ ! $apikey ]; then
     printf "\033[31mAPI key not available in DD_API_KEY environment variable.\033[0m\n"
     exit 1;
@@ -133,7 +139,12 @@ if [ $OS = "RedHat" ]; then
             $sudo_cmd yum -y remove datadog-agent-base
         fi
     fi
-    $sudo_cmd yum -y --disablerepo='*' --enablerepo='datadog' install datadog-agent || $sudo_cmd yum -y install datadog-agent
+    if [ -n "$dd_version" ]; then
+        dd_agent=datadog-agent-$dd_version
+    else
+        dd_agent=datadog-agent
+    fi
+    $sudo_cmd yum -y --disablerepo='*' --enablerepo='datadog' install $dd_agent || $sudo_cmd yum -y install $dd_agent
 elif [ $OS = "Debian" ]; then
     printf "\033[34m\n* Installing apt-transport-https\n\033[0m\n"
     $sudo_cmd apt-get update || printf "\033[31m'apt-get update' failed, the script will not install the latest version of apt-transport-https.\033[0m\n"
@@ -159,7 +170,12 @@ determine the cause.
 If the cause is unclear, please contact Datadog support.
 *****
 "
-    $sudo_cmd apt-get install -y --force-yes datadog-agent
+    if [ -n "$dd_version" ]; then
+        dd_agent=datadog-agent=$dd_version
+    else
+        dd_agent=datadog-agent
+    fi
+    $sudo_cmd apt-get install -y --force-yes $dd_agent
     ERROR_MESSAGE=""
 elif [ $OS = "SUSE" ]; then
   UNAME_M=$(uname -m)
@@ -174,8 +190,13 @@ elif [ $OS = "SUSE" ]; then
   echo -e "\033[34m\n* Refreshing repositories\n\033[0m"
   $sudo_cmd zypper --non-interactive refresh
 
+    if [ -n "$dd_version" ]; then
+        dd_agent=datadog-agent-$dd_version
+    else
+        dd_agent=datadog-agent
+    fi
   echo -e "\033[34m\n* Installing Datadog Agent\n\033[0m"
-  $sudo_cmd zypper --non-interactive install datadog-agent
+  $sudo_cmd zypper --non-interactive install $dd_agent
 
 else
     printf "\033[31mYour OS or distribution are not supported by this install script.