fix: properly use multi name fields (#102)

Signed-off-by: Sarah Funkhouser <[email protected]>
theopenlane · Dec 13, 2024 · a003a04 · a003a04
1 parent 0f0cca2
commit a003a04
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/entfga/template.go b/entfga/template.go
@@ -42,9 +42,9 @@ func extractDefaultObjectType(val any) string {
 		return ""
 	}
 
-	schemaName = strings.ToLower(schemaName)
+	schemaName = strcase.SnakeCase(schemaName)
 
-	return strings.ReplaceAll(schemaName, "history", "")
+	return strings.Trim(strings.ReplaceAll(schemaName, "history", ""), "_")
 }
 
 // extractIDField gets the key that is used for the id field

diff --git a/entfga/template_test.go b/entfga/template_test.go
@@ -53,6 +53,11 @@ func TestExtractDefaultObjectType(t *testing.T) {
 			input:    "UserHistory",
 			expected: "user",
 		},
+		{
+			name:     "valid schema name",
+			input:    "ControlObjective",
+			expected: "control_objective",
+		},
 		{
 			name:     "valid schema name, lowercase",
 			input:    "userhistory",