From e693af3c426dcfaa6ddbfd0e2ba58764c121b339 Mon Sep 17 00:00:00 2001
From: Coto Cisternas <cotocisternas@gmail.com>
Date: Fri, 15 Jan 2016 17:35:19 -0300
Subject: [PATCH] debian support, fix exec validations and repository

---
 manifests/config/file.pp |  7 +++++--
 manifests/init.pp        |  6 ++++++
 manifests/params.pp      | 20 +++++++++++---------
 manifests/repo/apt.pp    | 35 ++++++++++++++++-------------------
 4 files changed, 38 insertions(+), 30 deletions(-)

diff --git a/manifests/config/file.pp b/manifests/config/file.pp
index 3de5abb..d237cf7 100644
--- a/manifests/config/file.pp
+++ b/manifests/config/file.pp
@@ -17,9 +17,12 @@ define fluentd::config::file (
 
   # clean up to ensure priority changes take effect
   exec { "apply priority change for ${base_name}":
-    command => "rm ${::fluentd::conf_dir}/*-${base_name}",
-    onlyif  => "ls ${::fluentd::conf_dir}/*-${base_name} | grep -v ${config_name}",
+    path    => '/bin:/usr/bin:/usr/local/bin',
+    cwd     => "${::fluentd::conf_dir}",
+    command => "rm *-${base_name}",
+    onlyif  => "ls *-${base_name} | grep -v ${config_name}",
     before  => File[$config_path],
+    # require => Class['Fluentd::Config'],
     notify  => Class['Fluentd::Service'],
   }
 
diff --git a/manifests/init.pp b/manifests/init.pp
index ad36270..84a0395 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -54,6 +54,12 @@
 #
 class fluentd (
   $manage_repo             = $::fluentd::params::manage_repo,
+  $repo_gpgcheck           = $::fluentd::params::repo_gpgcheck,
+  $repo_gpgkey             = $::fluentd::params::repo_gpgkey,
+  $repo_gpgkeyid           = $::fluentd::params::repo_gpgkeyid,
+  $repo_url                = $::fluentd::params::repo_url,
+  $repo_name               = $::fluentd::params::repo_name,
+  $repo_desc               = $::fluentd::params::repo_desc,
   $package_ensure          = $::fluentd::params::package_ensure,
   $package_name            = $::fluentd::params::package_name,
   $package_install_options = $::fluentd::params::package_install_options,
diff --git a/manifests/params.pp b/manifests/params.pp
index 7a8a98f..e11b80a 100644
--- a/manifests/params.pp
+++ b/manifests/params.pp
@@ -17,17 +17,19 @@ class fluentd::params {
   $user_name   = 'td-agent'
   $user_group  = 'td-agent'
   $user_groups = ['adm']
+  # repo params
+  $repo_gpgcheck = true
+  $repo_gpgkey = 'http://packages.treasure-data.com/GPG-KEY-td-agent'
+  $repo_gpgkeyid = 'C901622B5EC4AF820C38AB861093DB45A12E206F'
+  $repo_name = 'treasuredata'
+  $repo_desc = 'TreasureData'
 
   case $::osfamily {
-    'Debian': {
-      case $::operatingsystem {
-        'Ubuntu': {
-          $manage_repo = true
-        }
-        default: {
-          fail("Unsupported osfamily: ${::operatingsystem}")
-        }
-      }
+    'debian': {
+      $manage_repo = true
+      $distro_id = downcase($::lsbdistid)
+      $distro_codename = $::lsbdistcodename
+      $repo_url = "http://packages.treasure-data.com/2/${distro_id}/${distro_codename}/"
     }
     default: {
       fail("Unsupported osfamily: ${::osfamily}")
diff --git a/manifests/repo/apt.pp b/manifests/repo/apt.pp
index 9b72ab6..db31836 100644
--- a/manifests/repo/apt.pp
+++ b/manifests/repo/apt.pp
@@ -1,23 +1,20 @@
 # Configure apt::source
 #
-class fluentd::repo::apt (
-  $ensure   = 'present',
-  $location = downcase("https://packages.treasuredata.com/2/${::operatingsystem}/${::lsbdistcodename}"),
-  $release  = $::lsbdistcodename,
-  $repos    = 'contrib',
-  $key      = {
-    'id'     => 'C901622B5EC4AF820C38AB861093DB45A12E206F',
-    'source' => 'https://packages.treasuredata.com/GPG-KEY-td-agent'
-  },
-) {
+class fluentd::repo::apt inherits fluentd {
 
-  include '::apt'
-
-  apt::source { 'treasure-data':
-    ensure   => $ensure,
-    location => $location,
-    release  => $release,
-    repos    => $repos,
-    key      => $key,
+  apt::source { $fluentd::repo_name:
+    location      => $fluentd::repo_url,
+    comment       => $fluentd::repo_desc,
+    repos         => 'contrib',
+    architecture  => 'amd64',
+    release       => $fluentd::distro_codename,
+    key           => {
+      id      => $fluentd::repo_gpgkeyid,
+      source  => $fluentd::repo_gpgkey,
+    },
+    include       => {
+      'src'   => false,
+      'deb'   => true,
+    },
   }
-}
+}
\ No newline at end of file