ztp provisioning script as a template

metal-stack · Dec 12, 2024 · eb95178 · eb95178
1 parent 3384bc2
commit eb95178
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 11 deletions.
diff --git a/partition/roles/ztp/README.md b/partition/roles/ztp/README.md
@@ -40,13 +40,14 @@ The result of this is unpredictable and, in the worst case, the switch might rea
 
 ## Variables
 
-| Name                 | Mandatory | Description                                                 |
-| -------------------- | --------- | ----------------------------------------------------------- |
-| ztp_nginx_image_name | yes       | the docker image to use to serve ztp scripts.               |
-| ztp_nginx_image_tag  | yes       | the tag of the docker image to use to serve ztp scripts.    |
-| ztp_host_dir_path    |           | the path to serve ztp scripts from.                         |
-| ztp_listen_address   |           | the address used to serve ztp requests                      |
-| ztp_port             |           | the port to serve ztp scripts on.                           |
-| ztp_authorized_keys  | yes       | the authorized keys that should be installed by ztp.        |
-| ztp_admin_user       |           | the user for which the authorized keys will be provisioned. |
-| ztp_additional_files |           | puts additional files into serve directory.                 |
+| Name                    | Mandatory | Description                                                                                    |
+| ----------------------- | --------- | ---------------------------------------------------------------------------------------------- |
+| ztp_nginx_image_name    | yes       | the docker image to use to serve ztp scripts.                                                  |
+| ztp_nginx_image_tag     | yes       | the tag of the docker image to use to serve ztp scripts.                                       |
+| ztp_host_dir_path       |           | the path to serve ztp scripts from.                                                            |
+| ztp_listen_address      |           | the address used to serve ztp requests                                                         |
+| ztp_port                |           | the port to serve ztp scripts on.                                                              |
+| ztp_authorized_keys     | yes       | the authorized keys that should be installed by ztp.                                           |
+| ztp_admin_user          |           | the user for which the authorized keys will be provisioned.                                    |
+| ztp_additional_files    |           | puts additional files into serve directory.                                                    |
+| ztp_provisioning_script |           | jinja template for optional ztp-provisioning-script.sh to be executed at the end of `ztp.json` |
diff --git a/partition/roles/ztp/tasks/main.yaml b/partition/roles/ztp/tasks/main.yaml
@@ -26,6 +26,13 @@
     - ztp.json.j2
     - ztp.sh.j2
 
+- name: render ztp-provisioning-script
+  template:
+    content: "{{ ztp_provisioning_script }}"
+    dest: "{{ ztp_host_dir_path }}/config/ztp-provisioning-script.sh"
+    mode: 0644
+  when: ztp_provisioning_script is defined
+
 - name: copy frr.conf
   copy:
     dest: "{{ ztp_host_dir_path }}/config/frr.conf"

diff --git a/partition/roles/ztp/templates/ztp.json.j2 b/partition/roles/ztp/templates/ztp.json.j2
@@ -25,7 +25,7 @@
       },
       "clear-config": true
     },
-{% if ztp_additional_files | selectattr('name', 'equalto', 'ztp-provisioning-script.sh') %}
+{% if ztp_provisioning_script %}
     "05-provisioning-script": {
       "plugin": {
         "url": "http://{{ ztp_listen_address }}:{{ ztp_port }}/ztp-provisioning-script.sh"