+postgresql_user "prometheus" do
+ cluster database_cluster
+ superuser true
+end
+
+postgresql_database "promscale" do
+ cluster database_cluster
+ owner "prometheus"
+end
+
+directory "/opt/promscale" do
+ owner "root"
+ group "root"
+ mode "755"
+end
+
+directory "/opt/promscale/bin" do
+ owner "root"
+ group "root"
+ mode "755"
+end
+
+remote_file "/opt/promscale/bin/promscale" do
+ action :create
+ source "https://github.com/timescale/promscale/releases/download/#{promscale_version}/promscale_#{promscale_version}_Linux_x86_64"
+ owner "root"
+ group "root"
+ mode "755"
+end
+
+systemd_service "promscale" do
+ description "Promscale Connector"
+ type "simple"
+ user "prometheus"
+ exec_start "/opt/promscale/bin/promscale --db.uri postgresql:///promscale?host=/run/postgresql&port=5432 --db.connections-max 400"
+ limit_nofile 16384
+ private_tmp true
+ protect_system "strict"
+ protect_home true
+ no_new_privileges true
+end
+
+if node[:prometheus][:promscale]
+ service "promscale" do
+ action [:enable, :start]
+ subscribes :restart, "remote_file[/opt/promscale/bin/promscale]"
+ subscribes :restart, "systemd_service[promscale]"
+ end
+else
+ service "promscale" do
+ action [:disable, :stop]
+ end
+end
+
+search(:node, "roles:gateway") do |gateway|
+ allowed_ips = gateway.interfaces(:role => :internal).map do |interface|
+ "#{interface[:network]}/#{interface[:prefix]}"
+ end
+
+ node.default[:networking][:wireguard][:peers] << {
+ :public_key => gateway[:networking][:wireguard][:public_key],
+ :allowed_ips => allowed_ips,
+ :endpoint => "#{gateway.name}:51820"
+ }
+end
+
+jobs = {}
+snmp_targets = []
+
+search(:node, "recipes:prometheus\\:\\:default").sort_by(&:name).each do |client|
+ if client[:prometheus][:mode] == "wireguard"
+ node.default[:networking][:wireguard][:peers] << {
+ :public_key => client[:networking][:wireguard][:public_key],
+ :allowed_ips => client[:networking][:wireguard][:address],
+ :endpoint => "#{client.name}:51820"
+ }
+ end
+
+ client[:prometheus][:exporters].each do |key, exporter|
+ if exporter.is_a?(Hash)
+ name = exporter[:name]
+ address = exporter[:address]
+ sni = exporter[:sni]
+ metric_relabel = exporter[:metric_relabel] || []
+ else
+ name = key
+ address = exporter
+ sni = nil
+ metric_relabel = []
+ end