From b4c20fd7d6841e53ad8acb208e93379238759915 Mon Sep 17 00:00:00 2001
From: Ben Vincent <ben@unkin.net>
Date: Mon, 10 Jun 2024 21:24:48 +1000
Subject: [PATCH] feat: add sonarr module

---
 modules/sonarr/manifests/init.pp            | 21 ++++++++
 modules/sonarr/manifests/install.pp         | 58 +++++++++++++++++++++
 modules/sonarr/manifests/params.pp          | 27 ++++++++++
 modules/sonarr/manifests/service.pp         | 20 +++++++
 modules/sonarr/templates/sonarr.service.erb | 14 +++++
 5 files changed, 140 insertions(+)
 create mode 100644 modules/sonarr/manifests/init.pp
 create mode 100644 modules/sonarr/manifests/install.pp
 create mode 100644 modules/sonarr/manifests/params.pp
 create mode 100644 modules/sonarr/manifests/service.pp
 create mode 100644 modules/sonarr/templates/sonarr.service.erb

diff --git a/modules/sonarr/manifests/init.pp b/modules/sonarr/manifests/init.pp
new file mode 100644
index 0000000..7dc312b
--- /dev/null
+++ b/modules/sonarr/manifests/init.pp
@@ -0,0 +1,21 @@
+# manage sonarr
+class sonarr (
+  $packages       = $sonarr::params::packages,
+  $user           = $sonarr::params::user,
+  $group          = $sonarr::params::user,
+  $base_path      = $sonarr::params::base_path,
+  $install_path   = $sonarr::params::install_path,
+  $config_folder  = $sonarr::params::config_folder,
+  $app_folder     = $sonarr::params::app_folder,
+  $archive_name   = $sonarr::params::archive_name,
+  $archive_url    = $sonarr::params::archive_url,
+  $executable     = $sonarr::params::executable,
+  $service_enable = $sonarr::params::service_enable,
+  $service_name   = $sonarr::params::service_name,
+) inherits sonarr::params {
+
+  include sonarr::install
+  include sonarr::service
+
+  Class['sonarr::install'] -> Class['sonarr::service']
+}
diff --git a/modules/sonarr/manifests/install.pp b/modules/sonarr/manifests/install.pp
new file mode 100644
index 0000000..0aa2b01
--- /dev/null
+++ b/modules/sonarr/manifests/install.pp
@@ -0,0 +1,58 @@
+# instsall sonarr
+class sonarr::install (
+  $packages      = $sonarr::packages,
+  $user          = $sonarr::user,
+  $group         = $sonarr::user,
+  $base_path     = $sonarr::base_path,
+  $install_path  = $sonarr::install_path,
+  $config_folder = $sonarr::config_folder,
+  $app_folder    = $sonarr::app_folder,
+  $archive_name  = $sonarr::archive_name,
+  $archive_url   = $sonarr::archive_url,
+  $executable    = $sonarr::executable,
+) {
+
+  $_packages = $packages ? {
+    Array   => true,
+    default => false,
+  }
+
+  if $_packages {
+    ensure_packages($packages, {ensure => 'installed'})
+  }
+
+  group { $group:
+    ensure => present,
+  }
+
+  user { $user:
+    ensure     => present,
+    shell      => '/sbin/nologin',
+    groups     => $group,
+    managehome => true,
+  }
+
+  file { [ $base_path, $install_path, $config_folder, $app_folder ]:
+    ensure => directory,
+    owner  => $user,
+    group  => $group,
+  }
+
+  archive { $archive_name:
+    path         => "/tmp/${archive_name}",
+    source       => "${archive_url}${archive_name}",
+    extract      => true,
+    extract_path => $install_path,
+    creates      => "${install_path}/${executable}",
+    cleanup      => true,
+    require      => File[$install_path],
+    user         => $user,
+    group        => $group,
+    notify       => Exec['move_sonarr_files'],
+  }
+
+  exec { 'move_sonarr_files':
+    command => "/usr/bin/mv ${install_path}/NzbDrone/* ${install_path}",
+    creates => "${install_path}/${executable}",
+  }
+}
diff --git a/modules/sonarr/manifests/params.pp b/modules/sonarr/manifests/params.pp
new file mode 100644
index 0000000..9ece7a9
--- /dev/null
+++ b/modules/sonarr/manifests/params.pp
@@ -0,0 +1,27 @@
+# sonarr params
+class sonarr::params (
+  $packages        = [
+    'mediainfo',
+    'libzen',
+    'libmediainfo',
+    'gettext',
+    'sqlite.x86_64',
+    'par2cmdline',
+    'python3-feedparser',
+    'python3-configobj',
+    'python3-cheetah',
+    'python3-dbus',
+    'libxslt-devel'
+  ],
+  $user            = 'sonarr',
+  $base_path       = '/opt/sonarr',
+  $install_path    = '/opt/sonarr/bin',
+  $config_folder   = "/home/${user}/.config",
+  $app_folder      = "/home/${user}/.config/Sonarr",
+  $archive_version = '4.0.5',
+  $archive_name    = 'Sonarr.main.linux-x64.tar.gz',
+  $archive_url     = "https://git.query.consul/api/packages/unkinben/generic/sonarr/${archive_version}/",
+  $executable      = 'Sonarr',
+  $service_enable  = true,
+  $service_name    = 'sonarr',
+){}
diff --git a/modules/sonarr/manifests/service.pp b/modules/sonarr/manifests/service.pp
new file mode 100644
index 0000000..875440a
--- /dev/null
+++ b/modules/sonarr/manifests/service.pp
@@ -0,0 +1,20 @@
+# manage sonarr service
+class sonarr::service (
+  $service_enable = $sonarr::service_enable,
+  $service_name   = $sonarr::service_name,
+  $user           = $sonarr::user,
+  $group          = $sonarr::user,
+  $install_path   = $sonarr::install_path,
+  $executable     = $sonarr::executable,
+  $base_path      = $sonarr::base_path,
+) {
+  if $service_enable {
+    include ::systemd
+
+    systemd::unit_file { "${service_name}.service":
+      content => template('sonarr/sonarr.service.erb'),
+      enable  => true,
+      active  => true,
+    }
+  }
+}
diff --git a/modules/sonarr/templates/sonarr.service.erb b/modules/sonarr/templates/sonarr.service.erb
new file mode 100644
index 0000000..fa79541
--- /dev/null
+++ b/modules/sonarr/templates/sonarr.service.erb
@@ -0,0 +1,14 @@
+[Unit]
+Description=<%= @service_name %> Daemon
+After=syslog.target network.target
+
+[Service]
+User=<%= @user %>
+Group=<%= @group %>
+Type=simple
+ExecStart=<%= @install_path %>/<%= @executable %> -nobrowser -data=<%= @base_path %>
+KillMode=process
+Restart=on-failure
+TimeoutStopSec=20
+[Install]
+WantedBy=multi-user.target