From 8d47395e53f9f9a474ea7a35a868d9b0300a2124 Mon Sep 17 00:00:00 2001
From: Jarrod Johnson <jjohnson2@lenovo.com>
Date: Wed, 9 Dec 2020 16:46:58 -0500
Subject: [PATCH] Add fetch of '<script>.d' scripts

This can be used by firstboot/post scripts to
get modularized scripts.
---
 confluent_server/confluent/selfservice.py | 29 +++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/confluent_server/confluent/selfservice.py b/confluent_server/confluent/selfservice.py
index 1f62fd66..5615b8d8 100644
--- a/confluent_server/confluent/selfservice.py
+++ b/confluent_server/confluent/selfservice.py
@@ -9,6 +9,7 @@ import confluent.discovery.handlers.xcc as xcc
 import confluent.discovery.handlers.tsm as tsm
 import crypt
 import json
+import os
 import time
 import yaml
 
@@ -261,6 +262,34 @@ def handle_request(env, start_response):
             nodename: {'deployment.sealedapikey': {'value': reqbody}}})
         start_response('200 OK', ())
         yield ''
+    elif env['PATH_INFO'].startswith('/self/scriptlist/'):
+        scriptcat = env['PATH_INFO'].replace('/self/scriptlist/', '')
+        if '..' in scriptcat:
+            start_response('400 Bad Requst', ())
+            yield ''
+            return
+        deployinfo = cfg.get_node_attributes(
+            nodename, ('deployment.*',))
+        deployinfo = deployinfo.get(nodename, {})
+        profile = deployinfo.get(
+            'deployment.pendingprofile', {}).get('value', '')
+        if not profile:
+            profile = deployinfo.get(
+            'deployment.stagedprofile', {}).get('value', '')
+        if not profile:
+            profile = deployinfo.get(
+            'deployment.profile', {}).get('value', '')
+        slist = None
+        try:
+            slist = os.listdir('/var/lib/confluent/public/os/{0}/scripts/{1}.d/'.format(profile, scriptcat))
+        except OSError:
+            pass
+        if slist:
+            start_response('200 OK', (('Content-Type', 'application/yaml'),))
+            yield yaml.safe_dump(util.natural_sort(slist), default_flow_style=False)
+        else:
+            start_response('200 OK', ())
+            yield ''
     else:
         start_response('404 Not Found', ())
         yield 'Not found'