Support Windows

data/common.yaml

@@ -6,7 +6,6 @@ lookup_options:
   '^promtail::password_file_content$':
     convert_to: 'Sensitive'
 
-promtail::bin_dir: '/usr/local/bin'
 promtail::service_ensure: 'running'
 promtail::service_enable: true
 promtail::version: 'v2.0.0'

manifests/config.pp

@@ -6,20 +6,22 @@
 class promtail::config {
   case $facts['kernel'] {
     'Linux': {
-      $config_dir = '/etc/promtail'
+      $config_file = "${promtail::config_dir}/config.yaml"
+    }
+    'windows': {
+      $config_file = "${promtail::config_dir}\\config.yaml"
     }
     default: { fail("${facts['kernel']} is not supported") }
   }
 
-  file { $config_dir:
+  file { $promtail::config_dir:
     ensure => directory,
   }
 
-  $config_file = "${config_dir}/config.yaml"
-
   concat { $config_file:
-    ensure => present,
-    notify => Systemd::Unit_file['promtail.service'],
+    ensure  => present,
+    require => File[$promtail::config_dir],
+    notify  => $promtail::service::service_to_notify,
   }
 
   concat::fragment { 'config_header':

manifests/init.pp

@@ -29,6 +29,12 @@
 # @param [Stdlib::Absolutepath] bin_dir
 #   The directory in which to create a symlink to the promtail binary
 #
+# @param [Stdlib::Absolutepath] config_dir
+#   The directory in which the promtail config is stored
+#
+# @param [Stdlib::Absolutepath] data_dir
+#   The directory in which to versions of promtail are stored and updated
+#
 # @param [String[1]] checksum
 #   The checksum of the promtail binary.
 #   Note: each platform has its own checksum.
@@ -128,15 +134,17 @@
   Hash                           $clients_config_hash,
   Hash                           $positions_config_hash,
   Hash                           $scrape_configs_hash,
-  Stdlib::Absolutepath           $bin_dir,
+  Stdlib::Absolutepath           $bin_dir               = promtail::params::bin_dir,
+  Stdlib::Absolutepath           $data_dir              = promtail::params::data_dir,
+  Stdlib::Absolutepath           $config_dir            = promtail::params::config_dir,
   String[1]                      $checksum,
   String[1]                      $version,
   Optional[Hash]                 $server_config_hash    = undef,
   Optional[Hash]                 $target_config_hash    = undef,
   Optional[Stdlib::Absolutepath] $password_file_path    = undef,
   Optional[Sensitive[String[1]]] $password_file_content = undef,
   Stdlib::HTTPUrl                $source_url            = 'https://github.com/grafana/loki/releases/download',
-) {
+) inherits promtail::params {
   Class['promtail::install']
   -> Class['promtail::config']
   -> Class['promtail::service']

manifests/install.pp

@@ -15,13 +15,19 @@
 
   case $facts['kernel'] {
     'Linux': {
-      $data_dir = '/usr/local/promtail_data'
       if versioncmp($promtail::version, 'v0.3.0') > 0 {
         $release_file_name = "promtail-linux-${arch}"
       } else {
         $release_file_name = "promtail_linux_${arch}"
       }
     }
+    'windows': {
+      if versioncmp($promtail::version, 'v0.3.0') > 0 {
+        $release_file_name = "promtail-windows-${arch}.exe"
+      } else {
+        $release_file_name = "promtail_windows_${arch}.exe"
+      }
+    }
     default: { fail("${facts['kernel']} is not yet supported") }
   }
 
@@ -31,14 +37,20 @@
     $archive_type = 'gz'
   }
 
-  $version_dir = "${data_dir}/promtail-${promtail::version}"
-  $binary_path = "${version_dir}/${release_file_name}"
+  $version_dir = $facts['kernel'] ? {
+    'windows' => "${promtail::data_dir}\\${promtail::version}",
+    default => "${promtail::data_dir}/promtail-${promtail::version}"
+  }
+  $binary_path = $facts['kernel'] ? {
+    'windows' => "${version_dir}\\${release_file_name}",
+    default => "${version_dir}/${release_file_name}"
+  }
 
-  file { [$data_dir, $version_dir]:
+  file { [$promtail::bin_dir, $promtail::data_dir, $version_dir]:
     ensure => directory,
   }
 
-  archive { "${binary_path}.gz":
+  archive { "${binary_path}.${archive_type}":
     ensure        => present,
     source        => "${promtail::source_url}/${promtail::version}/${release_file_name}.${archive_type}",
     extract       => true,
@@ -49,17 +61,21 @@
     cleanup       => false,
   }
 
+  $binary_link_path = $facts['kernel'] ? {
+    'windows' => "${promtail::bin_dir}\\promtail.exe",
+    default => "${promtail::bin_dir}/promtail"
+  }
   file {
     $binary_path:
       ensure  => file,
       mode    => '0755',
-      require => Archive["${binary_path}.gz"],
+      require => Archive["${binary_path}.${archive_type}"],
       ;
-    "${promtail::bin_dir}/promtail":
+    $binary_link_path:
       ensure  => link,
       target  => $binary_path,
       require => File[$binary_path],
-      notify  => Systemd::Unit_file['promtail.service'],
+      notify  => $promtail::service::service_to_notify
       ;
   }
 }

manifests/params.pp

@@ -0,0 +1,20 @@
+# @summary Default promtail variables
+#
+# Sets default variables per OS
+#
+# @api private
+class promtail::params {
+  case $facts['kernel'] {
+    'Linux': {
+      $bin_dir = '/usr/local/bin'
+      $data_dir = '/usr/local/promtail_data'
+      $config_dir = '/etc/promtail'
+    }
+    'windows': {
+      $bin_dir = 'C:\\Program Files\\promtail'
+      $data_dir = "${bin_dir}\\versions"
+      $config_dir = "${bin_dir}\\config"
+    }
+    default: { fail("${facts['kernel']} is not yet supported") }
+  }
+}

manifests/service.pp

@@ -4,14 +4,30 @@
 #
 # @api private
 class promtail::service {
+  $service_to_notify = $facts['kernel'] ? {
+    'Linux'   => Systemd::Unit_file['promtail.service'],
+    'windows' => Service['promtail'],
+    default   => undef
+  }
+
   case $facts['kernel'] {
     'Linux': {
-      include systemd::systemctl::daemon_reload
       systemd::unit_file { 'promtail.service':
-        ensure => present,
         source => 'puppet:///modules/promtail/promtail.service',
+        notify => Service['promtail'],
+      }
+
+      service { 'promtail':
+        ensure  => $promtail::service_ensure,
+        enable  => $promtail::service_enable,
+        require => Systemd::Unit_file['promtail.service'],
+      }
+    }
+    'windows': {
+      service { 'promtail':
+        ensure => $promtail::service_ensure,
         enable => $promtail::service_enable,
-        active => $promtail::service_ensure == 'running'
+        start  => "${promtail::install::binary_link_path} --config.file ${promtail::config::config_file}"
       }
     }
     default: { fail("${facts['kernel']} is not supported") }