Refactoring security patch for chmod folder TFO-Go-SDK generator

Author: zeroc0d3

cmd/generator-restfulapi/main.go

@@ -347,7 +347,7 @@ func runNew(cmd *cobra.Command, args []string) {
 	}
 
 	for _, dir := range dirs {
-		if err := os.MkdirAll(dir, 0755); err != nil {
+		if err := os.MkdirAll(dir, 0750); err != nil {
 			fmt.Fprintf(os.Stderr, "Failed to create directory %s: %v\n", dir, err)
 			os.Exit(1)
 		}
@@ -404,7 +404,7 @@ func runNew(cmd *cobra.Command, args []string) {
 	generateFromTemplate("pkg/response.go.tpl", data, filepath.Join(projectRoot, "pkg", "response", "response.go"))
 
 	// Create safefile package directory
-	if err := os.MkdirAll(filepath.Join(projectRoot, "pkg", "safefile"), 0755); err != nil {
+	if err := os.MkdirAll(filepath.Join(projectRoot, "pkg", "safefile"), 0750); err != nil {
 		fmt.Fprintf(os.Stderr, "Failed to create safefile directory: %v\n", err)
 	}
 	generateFromTemplate("pkg/safefile.go.tpl", data, filepath.Join(projectRoot, "pkg", "safefile", "safefile.go"))
@@ -537,7 +537,7 @@ func generateFromTemplate(templateName string, data interface{}, outputPath stri
 	// Sanitize the output path to prevent path traversal (G304)
 	cleanPath := filepath.Clean(outputPath)
 
-	if err := os.MkdirAll(filepath.Dir(cleanPath), 0755); err != nil {
+	if err := os.MkdirAll(filepath.Dir(cleanPath), 0750); err != nil {
 		fmt.Fprintf(os.Stderr, "Failed to create directory for %s: %v\n", cleanPath, err)
 		return
 	}

cmd/generator/main.go

@@ -135,7 +135,7 @@ func runInit(cmd *cobra.Command, args []string) {
 	}
 
 	for _, dir := range dirs {
-		if err := os.MkdirAll(dir, 0755); err != nil {
+		if err := os.MkdirAll(dir, 0750); err != nil {
 			fmt.Fprintf(os.Stderr, "Failed to create directory %s: %v\n", dir, err)
 			os.Exit(1)
 		}

tests/unit/generator/basic/generator_test.go

@@ -136,7 +136,7 @@ func TestDirectoryCreation(t *testing.T) {
 
 		// Create directories
 		for _, dir := range dirs {
-			err := os.MkdirAll(dir, 0755)
+			err := os.MkdirAll(dir, 0750)
 			require.NoError(t, err)
 		}
 
@@ -160,11 +160,11 @@ func TestDirectoryCreation(t *testing.T) {
 		dir := filepath.Join(tmpDir, "telemetry")
 
 		// Create directory first time
-		err = os.MkdirAll(dir, 0755)
+		err = os.MkdirAll(dir, 0750)
 		require.NoError(t, err)
 
 		// Should not error on second creation
-		err = os.MkdirAll(dir, 0755)
+		err = os.MkdirAll(dir, 0750)
 		require.NoError(t, err)
 	})
 }

tests/unit/generator/restapi/restapi_test.go

@@ -651,7 +651,7 @@ func TestProjectFileGeneration(t *testing.T) {
 		}
 
 		for _, dir := range dirs {
-			err := os.MkdirAll(dir, 0755)
+			err := os.MkdirAll(dir, 0750)
 			require.NoError(t, err)
 		}
 

tests/unit/infrastructure/template/template_test.go

@@ -242,7 +242,7 @@ func TestTemplateFileOperations(t *testing.T) {
 		outputPath := filepath.Join(tmpDir, "test", "output.go")
 
 		// Create directory structure
-		err := os.MkdirAll(filepath.Dir(outputPath), 0755)
+		err := os.MkdirAll(filepath.Dir(outputPath), 0750)
 		require.NoError(t, err)
 
 		// Create file
@@ -285,7 +285,7 @@ func TestTemplateFileOperations(t *testing.T) {
 		tmpDir := t.TempDir()
 		nestedPath := filepath.Join(tmpDir, "a", "b", "c", "file.go")
 
-		err := os.MkdirAll(filepath.Dir(nestedPath), 0755)
+		err := os.MkdirAll(filepath.Dir(nestedPath), 0750)
 		require.NoError(t, err)
 
 		err = os.WriteFile(nestedPath, []byte("content"), 0644)