From 238c5b8547bc403bfffd74bcc94abd54f2860e58 Mon Sep 17 00:00:00 2001
From: Saurav Panda <sgp65@cornell.edu>
Date: Thu, 1 Feb 2024 01:12:52 -0500
Subject: [PATCH] fix: Update the path check

---
 README.md                             | 2 +-
 services/awshandler/generatereport.go | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 9cbe1f2..52ce56c 100644
--- a/README.md
+++ b/README.md
@@ -31,7 +31,7 @@ go build cmd/main.go
 
 To Gather all resources
 ```
-./main gather -provider=aws -region=us-east-1 --resource=vm
+./main gather -provider=aws -region=us-east-1
 ```
 This will store all the meta data for AWS resource in `output/` directory
 
diff --git a/services/awshandler/generatereport.go b/services/awshandler/generatereport.go
index f49d356..6042c1a 100644
--- a/services/awshandler/generatereport.go
+++ b/services/awshandler/generatereport.go
@@ -34,8 +34,9 @@ func GenerateAWSReport(outFolder string) {
 			if len(pathParts) < 4 {
 				return fmt.Errorf("unexpected path format: %s", path)
 			}
-			serviceName := pathParts[3]
-			regionName := pathParts[2]
+			path_len := len(pathParts)
+			serviceName := pathParts[path_len-2]
+			regionName := pathParts[path_len-3]
 
 			// Print the entire data for debugging
 			fmt.Printf("Service name %v, data is %v, \n", serviceName, data["stats"])
@@ -63,7 +64,7 @@ func GenerateAWSReport(outFolder string) {
 	// Print and write the report
 	utils.PrintNested(regionStats, "", 0)
 
-	err = utils.WriteJSONToFile("output/aws_report.json", regionStats)
+	err = utils.WriteJSONToFile(outFolder+"/aws_report.json", regionStats)
 	if err != nil {
 		fmt.Println("Failed to Write the report file to json")
 		fmt.Println(err)