From 81ba23094e8578ed11cb1aae48cf10b79dc2f3cd Mon Sep 17 00:00:00 2001
From: Arthur Petukhovsky <petuhovskiy@yandex.ru>
Date: Thu, 7 Apr 2022 20:38:26 +0300
Subject: [PATCH] Fix scripts to deploy sk4 on staging (#1476)

Adjust ansible scripts and inventory for sk4 on staging
---
 .circleci/ansible/deploy.yaml                | 24 ++++++++++++++++
 .circleci/ansible/scripts/init_safekeeper.sh | 30 ++++++++++++++++++++
 .circleci/ansible/staging.hosts              |  1 +
 3 files changed, 55 insertions(+)
 create mode 100644 .circleci/ansible/scripts/init_safekeeper.sh

diff --git a/.circleci/ansible/deploy.yaml b/.circleci/ansible/deploy.yaml
index b7ffd075a0..2112102aa7 100644
--- a/.circleci/ansible/deploy.yaml
+++ b/.circleci/ansible/deploy.yaml
@@ -116,6 +116,30 @@
 
   tasks:
 
+    - name: upload init script
+      when: console_mgmt_base_url is defined
+      ansible.builtin.template:
+        src: scripts/init_safekeeper.sh
+        dest: /tmp/init_safekeeper.sh
+        owner: root
+        group: root
+        mode: '0755'
+      become: true
+      tags:
+      - safekeeper
+
+    - name: init safekeeper
+      shell:
+        cmd: /tmp/init_safekeeper.sh
+      args:
+        creates: "/storage/safekeeper/data/safekeeper.id"
+      environment:
+        ZENITH_REPO_DIR: "/storage/safekeeper/data"
+        LD_LIBRARY_PATH: "/usr/local/lib"
+      become: true
+      tags:
+      - safekeeper
+
     # in the future safekeepers should discover pageservers byself
     # but currently use first pageserver that was discovered
     - name: set first pageserver var for safekeepers
diff --git a/.circleci/ansible/scripts/init_safekeeper.sh b/.circleci/ansible/scripts/init_safekeeper.sh
new file mode 100644
index 0000000000..2297788f59
--- /dev/null
+++ b/.circleci/ansible/scripts/init_safekeeper.sh
@@ -0,0 +1,30 @@
+#!/bin/sh
+
+# get instance id from meta-data service
+INSTANCE_ID=$(curl -s http://169.254.169.254/latest/meta-data/instance-id)
+
+# store fqdn hostname in var
+HOST=$(hostname -f)
+
+
+cat <<EOF | tee /tmp/payload
+{
+  "version": 1,
+  "host": "${HOST}",
+  "port": 6500,
+  "region_id": {{ console_region_id }},
+  "instance_id": "${INSTANCE_ID}",
+  "http_host": "${HOST}",
+  "http_port": 7676
+}
+EOF
+
+# check if safekeeper already registered or not
+if ! curl -sf -X PATCH -d '{}' {{ console_mgmt_base_url }}/api/v1/safekeepers/${INSTANCE_ID} -o /dev/null; then
+
+    # not registered, so register it now
+    ID=$(curl -sf -X POST {{ console_mgmt_base_url }}/api/v1/safekeepers -d@/tmp/payload | jq -r '.ID')
+
+    # init safekeeper
+    sudo -u safekeeper /usr/local/bin/safekeeper --id ${ID} --init -D /storage/safekeeper/data
+fi
diff --git a/.circleci/ansible/staging.hosts b/.circleci/ansible/staging.hosts
index e7414a6ba3..f6b7bf009f 100644
--- a/.circleci/ansible/staging.hosts
+++ b/.circleci/ansible/staging.hosts
@@ -6,6 +6,7 @@ zenith-us-stage-ps-2 console_region_id=27
 zenith-us-stage-sk-1 console_region_id=27
 zenith-us-stage-sk-2 console_region_id=27
 zenith-us-stage-sk-3 console_region_id=27
+zenith-us-stage-sk-4 console_region_id=27
 
 [storage:children]
 pageservers