added script for table creation

coder · Kira-Pilot · Jan 26, 2023 · Jan 25, 2023 · Jan 25, 2023 · Jan 25, 2023
commit 51c7aeeb64f0a62fe8cf769ade4659e6d94f1e36
diff --git a/enterprise/audit/generate.sh b/enterprise/audit/generate.sh
@@ -15,5 +15,5 @@ PROJECT_ROOT=$(cd "$SCRIPT_DIR" && git rev-parse --show-toplevel)
 
 (
 	cd "$PROJECT_ROOT"
-	go run ./scripts/auditgen ./coderd/database "$@"
+	go run ./scripts/audittypegen ./coderd/database "$@"
 )
diff --git a/scripts/auditdocgen/main.go b/scripts/auditdocgen/main.go
@@ -0,0 +1,82 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"log"
+	"strconv"
+	"strings"
+
+	"github.com/coder/coder/enterprise/audit"
+)
+
+func main() {
+	auditableResourcesMap, err := readAuditableResources()
+	if err != nil {
+		log.Fatal("can't read auditableResources: ", err)
+	}
+
+	doc, err := readAuditDoc()
+	if err != nil {
+		log.Fatal("can't read audit doc: ", err)
+	}
+
+	doc, err = updateAuditDoc(doc, auditableResourcesMap)
+	if err != nil {
+		log.Fatal("can't update audit doc: ", err)
+	}
+
+	err = writeAuditDoc(doc)
+	if err != nil {
+		log.Fatal("can't write updated audit doc: ", err)
+	}
+}
+
+type AuditableResourcesMap map[string]map[string]bool
+
+func readAuditableResources() (AuditableResourcesMap, error) {
+	auditableResourcesMap := make(AuditableResourcesMap)
+
+	for resourceName, resourceFields := range audit.AuditableResources {
+		friendlyResourceName := strings.Split(resourceName, ".")[2]
+		fieldNameMap := make(map[string]bool)
+		for fieldName, action := range resourceFields {
+			fieldNameMap[fieldName] = action != audit.ActionIgnore
+			auditableResourcesMap[friendlyResourceName] = fieldNameMap
+		}
+	}
+
+	return auditableResourcesMap, nil
+}
+
+func readAuditDoc() ([]byte, error) {
+	var doc []byte
+	return doc, nil
+}
+
+func updateAuditDoc(doc []byte, auditableResourcesMap AuditableResourcesMap) ([]byte, error) {
+	var updatedDoc []byte
+
+	var buffer bytes.Buffer
+	buffer.WriteByte('\n')
+	buffer.WriteString("|<b>Resource<b>||\n")
+	buffer.WriteString("|--|-----------------|\n")
+
+	for resourceName, resourceFields := range auditableResourcesMap {
+
+		buffer.Write([]byte("|" + resourceName + "|<table><thead><tr><th>Field</th><th>Tracked</th></tr></thead><tbody>"))
+
+		for fieldName, isTracked := range resourceFields {
+			buffer.Write([]byte("<tr><td>" + fieldName + "</td><td>" + strconv.FormatBool(isTracked) + "</td></tr>"))
+		}
+
+		buffer.WriteString("</tbody></table>\n")
+	}
+
+	fmt.Println("updated table", buffer.String())
+	return updatedDoc, nil
+}
+
+func writeAuditDoc(doc []byte) error {
+	return nil
+}
diff --git a/scripts/auditgen/main.go → scripts/audittypegen/main.go b/scripts/auditgen/main.go → scripts/audittypegen/main.go