From afc135647e72c270a2b10803bba3469bd98c51de Mon Sep 17 00:00:00 2001
From: Rikard Danielsson <richir@sunet.se>
Date: Thu, 23 Jan 2025 15:58:40 +0100
Subject: [PATCH] invent.sh script from puppet

---
 client/invent.sh | 79 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 client/invent.sh

diff --git a/client/invent.sh b/client/invent.sh
new file mode 100644
index 0000000..f87d616
--- /dev/null
+++ b/client/invent.sh
@@ -0,0 +1,79 @@
+#!/bin/bash
+
+# Source env vars
+source /etc/default/invent-client
+
+datadir="${INVENT_DIR}/data"
+export_endpoint="${INVENT_EXPORT_ENDPOINT}"
+fact_dir="/var/lib/puppet/facts.d"
+filename="${datadir}/data-$(date +%Y%m%dT%H%M%S).json"
+host_os="${INVENT_HOST_OS}"
+latest="${datadir}/latest.json"
+retention_days="${INVENT_RETENTION_DAYS}"
+
+# Gather packages
+parse_command="awk -v q='\"' '{print \"{\"q\"name\"q\": \"q\$1q\",\"q\"version\"q\": \"q\$2q\"}\"}' | jq -s ."
+
+case "${host_os}" in
+  alpine)
+    query_command="apk list -q"
+    parse_command="awk '{print \$1}' | sed 's/\-\\([0-9]\\)/ \1/' | ${parse_command}"
+    ;;
+  centos | fedora | redhat)
+    query_command="rpm -qa"
+    parse_command="sed 's/\-\\([0-9]\\)/ \1/' |${parse_command}"
+    ;;
+  debian | ubuntu)
+    query_command="dpkg-query -W"
+    ;;
+  *)
+    query_command='echo {\"unknown\": \"none\"}'
+    parse_command='cat -'
+esac
+
+# Gather structured data kernel fact
+kernel_fact="${fact_dir}/kernel.json"
+uname -rvmo | sed -e 's/ #/;#/' -e 's/ \([^ ]\+\) \([^ ]\+\)$/;\1;\2/'| \
+  awk -F ';' '{print "{ \"running-kernel\": { \"kernel-release\": \""$1"\",\"kernel-version\": \""$2"\", \"machine\": \""$3"\", \"operating-system\": \""$4"\" }}"}' | \
+  jq . > ${kernel_fact}
+# Gather structured data package fact
+package_fact="${fact_dir}/packages.json"
+echo "{
+        \"packages\": $(eval ${query_command} | eval ${parse_command} 2> /dev/null | jq -s .)
+      }" \
+    | jq . > ${package_fact}
+
+# Only run if we have docker
+if [ $(which docker) ]; then
+  # Gather structured data docker fact
+  docker_fact="${fact_dir}/docker_ps.json"
+  for container in $(docker ps -q); do
+    docker ps  --format '{{json . }}' --filter "id=${container}" | jq '. |= . + '{"ImageId":$(docker inspect --format '{{json .Image }}' ${container})'}';
+  done | jq -s |jq -s  '{docker_ps: add}' > ${docker_fact}
+
+fi
+
+# Export facts
+mkdir -p ${datadir}
+puppet facts --render-as json 2>/dev/null | jq . > ${filename}
+ln -f -s "${filename}" "${latest}"
+
+# Clean out old facts
+find ${datadir} -type f -mtime +${retention_days} -delete
+
+# Send data to inventory receiver
+if [[ -x /usr/bin/curl ]] && [[ -n ${export_endpoint} ]]; then
+  username=$(hostname -f)
+  pwfile="/opt/invent/passwd"
+  if [[ -f ${pwfile} ]]; then
+    password=$(cat ${pwfile})
+  else
+    mkdir -p $(basename ${pwfile})
+    password=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 256 | head -n 1)
+    echo ${password} > ${pwfile}
+  fi
+  curl -X POST -H 'accept: application/json' \
+    -F "file=@${filename}" \
+    --user ${username}:${password}  \
+    ${export_endpoint}/host/${username}
+fi