[Security Policies] Support DataYaml updates for the bundle

elastic/csp-security-policies#89


---NOTE---
This is an imported commit, it was initially committed to the
csp-security-policies repo which was then merged into cloudbeat. See:
#1405

Author: yashtewari

bundle/server.go

@@ -7,26 +7,14 @@ import (
 	"path/filepath"
 )
 
-const BundlesFolder = "tmpBundles"
-const BundlePathPrefix = "/bundles/"
-
-type Server struct {
-	mux *http.ServeMux
-}
-
-func NewServer() *Server {
-	mux := http.NewServeMux()
-	staticFileServer := http.FileServer(http.Dir(BundlesFolder))
-	mux.Handle(BundlePathPrefix, http.StripPrefix(BundlePathPrefix, staticFileServer))
-
-	return &Server{
-		mux: mux,
-	}
-}
+const (
+	BundlesFolder    = "tmpBundles"
+	BundlePathPrefix = "/bundles/"
+)
 
 // HostBundle writes the given bundle to the disk in order to serve it later
 // Consequent calls to HostBundle with the same name will override the file
-func (s *Server) HostBundle(name string, files map[string]string) error {
+func HostBundle(name string, files map[string]string) error {
 	if _, err := os.Stat(BundlesFolder); os.IsNotExist(err) {
 		err := os.Mkdir(BundlesFolder, os.ModePerm)
 		if err != nil {
@@ -51,6 +39,27 @@ func (s *Server) HostBundle(name string, files map[string]string) error {
 	return nil
 }
 
+func HostBundleWithDataYaml(name string, files map[string]string, dataYaml string) error {
+	files["data.yaml"] = dataYaml
+
+	err := HostBundle(name, files)
+	return err
+}
+
+type Server struct {
+	mux *http.ServeMux
+}
+
+func NewServer() *Server {
+	mux := http.NewServeMux()
+	staticFileServer := http.FileServer(http.Dir(BundlesFolder))
+	mux.Handle(BundlePathPrefix, http.StripPrefix(BundlePathPrefix, staticFileServer))
+
+	return &Server{
+		mux: mux,
+	}
+}
+
 func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	s.mux.ServeHTTP(w, r)
 }

bundle/server_test.go

@@ -13,16 +13,16 @@ func TestServer(t *testing.T) {
 	assert := assert.New(t)
 
 	handler := NewServer()
-	err := handler.HostBundle("empty", map[string]string{})
+	err := HostBundle("empty", map[string]string{})
 	assert.NoError(err)
 
-	err = handler.HostBundle("otherBundle", map[string]string{"a.txt": "some text"})
+	err = HostBundle("otherBundle", map[string]string{"a.txt": "some text"})
 	assert.NoError(err)
 
-	err = handler.HostBundle("overridenBundle", map[string]string{})
+	err = HostBundle("overridenBundle", map[string]string{})
 	assert.NoError(err)
 
-	err = handler.HostBundle("overridenBundle", map[string]string{})
+	err = HostBundle("overridenBundle", map[string]string{})
 	assert.NoError(err)
 
 	server := httptest.NewServer(handler)

server/host.go

@@ -13,7 +13,7 @@ func HostCISKubernetes(path string) (http.Handler, error) {
 	}
 
 	server := bundle.NewServer()
-	err = server.HostBundle(path, policies)
+	err = bundle.HostBundle(path, policies)
 	if err != nil {
 		return nil, err
 	}