feat(amazonq): Enable checkbox to accept or block code fix with references (#5509)

* /review: passing referenceTrackerConfiguration to StartCodeFixJob

* Update changelog description

* Update .changes/next-release/feature-21a9e26b-edbf-4ad3-a5f4-2662420d4f0d.json

Co-authored-by: Richard Li <[email protected]>

---------

Co-authored-by: Richard Li <[email protected]>

Author: atonaamz

.changes/next-release/feature-21a9e26b-edbf-4ad3-a5f4-2662420d4f0d.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "/review: Respect user option to allow code suggestions with references"
+}

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/AmazonQCodeFixSession.kt

@@ -14,6 +14,7 @@ import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobR
 import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobResponse
 import software.amazon.awssdk.services.codewhispererruntime.model.Position
 import software.amazon.awssdk.services.codewhispererruntime.model.Range
+import software.amazon.awssdk.services.codewhispererruntime.model.RecommendationsWithReferencesPreference
 import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobRequest
 import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobResponse
 import software.amazon.awssdk.services.codewhispererruntime.model.UploadContext
@@ -28,6 +29,7 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWh
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererZipUploadManager
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.getTelemetryErrorMessage
+import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
 import software.aws.toolkits.resources.message
 import java.io.File
 import java.nio.file.Path
@@ -161,12 +163,19 @@ class AmazonQCodeFixSession(val project: Project) {
         codeFixName: String? = null,
         ruleId: String? = null,
     ): StartCodeFixJobResponse {
+        val includeCodeWithReference = if (CodeWhispererSettings.getInstance().isIncludeCodeWithReference()) {
+            RecommendationsWithReferencesPreference.ALLOW
+        } else {
+            RecommendationsWithReferencesPreference.BLOCK
+        }
+
         val request = StartCodeFixJobRequest.builder()
             .uploadId(uploadId)
             .snippetRange(snippetRange)
             .codeFixName(codeFixName)
             .ruleId(ruleId)
             .description(description)
+            .referenceTrackerConfiguration { it.recommendationsWithReferences(includeCodeWithReference) }
             .build()
 
         return try {