GetResourceType function added

opengovern · May 20, 2023 · ebccc51 · ebccc51
1 parent 50a35d0
commit ebccc51
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/external/states/statefile/resources.go b/external/states/statefile/resources.go
@@ -59,3 +59,21 @@ func GetResourcesArn(f io.Reader) []string {
 	}
 	return arns
 }
+
+// Returns the list of resource types
+func GetResourcesTypes(f io.Reader) []string {
+	result, err := Read(f)
+	if err != nil {
+		panic(err)
+	}
+
+	types := make([]string, 0)
+
+	state := result.State
+	for _, ms := range state.Modules {
+		for _, re := range ms.Resources {
+			types = append(types, re.Addr.Resource.Type)
+		}
+	}
+	return types
+}