@@ -17,18 +17,18 @@ import kotlinx.coroutines.ensureActive
1717import kotlinx.coroutines.isActive
1818import kotlinx.coroutines.time.withTimeout
1919import kotlinx.coroutines.withContext
20- import software.amazon.awssdk.services.codewhisperer.model.ArtifactType
21- import software.amazon.awssdk.services.codewhisperer.model.CodeScanFindingsSchema
22- import software.amazon.awssdk.services.codewhisperer.model.CodeScanStatus
23- import software.amazon.awssdk.services.codewhisperer.model.CodeWhispererException
24- import software.amazon.awssdk.services.codewhisperer.model.CreateCodeScanRequest
25- import software.amazon.awssdk.services.codewhisperer.model.CreateCodeScanResponse
26- import software.amazon.awssdk.services.codewhisperer.model.GetCodeScanRequest
27- import software.amazon.awssdk.services.codewhisperer.model.GetCodeScanResponse
28- import software.amazon.awssdk.services.codewhisperer.model.ListCodeScanFindingsRequest
29- import software.amazon.awssdk.services.codewhisperer.model.ListCodeScanFindingsResponse
20+ import software.amazon.awssdk.services.codewhispererruntime.model.ArtifactType
21+ import software.amazon.awssdk.services.codewhispererruntime.model.CodeAnalysisFindingsSchema
22+ import software.amazon.awssdk.services.codewhispererruntime.model.CodeAnalysisStatus
23+ import software.amazon.awssdk.services.codewhispererruntime.model.CodeWhispererRuntimeException
3024import software.amazon.awssdk.services.codewhispererruntime.model.CreateUploadUrlResponse
25+ import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeAnalysisRequest
26+ import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeAnalysisResponse
27+ import software.amazon.awssdk.services.codewhispererruntime.model.ListCodeAnalysisFindingsRequest
28+ import software.amazon.awssdk.services.codewhispererruntime.model.ListCodeAnalysisFindingsResponse
3129import software.amazon.awssdk.services.codewhispererruntime.model.Reference
30+ import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeAnalysisRequest
31+ import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeAnalysisResponse
3232import software.aws.toolkits.core.utils.Waiters.waitUntil
3333import software.aws.toolkits.core.utils.debug
3434import software.aws.toolkits.core.utils.getLogger
@@ -146,7 +146,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
146146 }
147147 }
148148 var codeScanStatus = createCodeScanResponse.status()
149- if (codeScanStatus == CodeScanStatus .FAILED ) {
149+ if (codeScanStatus == CodeAnalysisStatus .FAILED ) {
150150 if (isProjectScope()) {
151151 LOG .debug {
152152 " CodeWhisperer service error occurred. Something went wrong when creating a code review: $createCodeScanResponse " +
@@ -166,7 +166,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
166166
167167 // 5. Keep polling the API GetCodeScan to wait for results for a given timeout period.
168168 waitUntil(
169- succeedOn = { codeScanStatus == CodeScanStatus .COMPLETED },
169+ succeedOn = { codeScanStatus == CodeAnalysisStatus .COMPLETED },
170170 maxDuration = Duration .ofSeconds(sessionContext.sessionConfig.overallJobTimeoutInSeconds())
171171 ) {
172172 currentCoroutineContext.ensureActive()
@@ -183,7 +183,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
183183 }
184184 }
185185 delay(CODE_SCAN_POLLING_INTERVAL_IN_SECONDS * TOTAL_MILLIS_IN_SECOND )
186- if (codeScanStatus == CodeScanStatus .FAILED ) {
186+ if (codeScanStatus == CodeAnalysisStatus .FAILED ) {
187187 if (isProjectScope()) {
188188 LOG .debug {
189189 " CodeWhisperer service error occurred. Something went wrong fetching results for code review: $getCodeScanResponse " +
@@ -206,12 +206,12 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
206206 )
207207
208208 val documents = mutableListOf<String >()
209- documents.add(listCodeScanFindingsResponse.codeScanFindings ())
209+ documents.add(listCodeScanFindingsResponse.codeAnalysisFindings ())
210210 // coroutineContext helps to actively cancel the bigger projects quickly
211211 withContext(currentCoroutineContext) {
212212 while (listCodeScanFindingsResponse.nextToken() != null && currentCoroutineContext.isActive) {
213213 listCodeScanFindingsResponse = listCodeScanFindings(jobId, listCodeScanFindingsResponse.nextToken())
214- documents.add(listCodeScanFindingsResponse.codeScanFindings ())
214+ documents.add(listCodeScanFindingsResponse.codeAnalysisFindings ())
215215 }
216216 }
217217
@@ -230,7 +230,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
230230 codeScanResponseContext = codeScanResponseContext.copy(reason = " Succeeded" )
231231 return CodeScanResponse .Success (issues, codeScanResponseContext)
232232 } catch (e: Exception ) {
233- val exception = e as ? CodeWhispererException
233+ val exception = e as ? CodeWhispererRuntimeException
234234 val awsError = exception?.awsErrorDetails()
235235
236236 if (awsError != null ) {
@@ -254,7 +254,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
254254 }
255255 }
256256
257- fun createCodeScan (language : String , codeScanName : String ): CreateCodeScanResponse {
257+ fun createCodeScan (language : String , codeScanName : String ): StartCodeAnalysisResponse {
258258 val artifactsMap = mapOf (
259259 ArtifactType .SOURCE_CODE to urlResponse[ArtifactType .SOURCE_CODE ]?.uploadId(),
260260 ArtifactType .BUILT_JARS to urlResponse[ArtifactType .BUILT_JARS ]?.uploadId()
@@ -268,7 +268,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
268268
269269 try {
270270 return clientAdaptor.createCodeScan(
271- CreateCodeScanRequest .builder()
271+ StartCodeAnalysisRequest .builder()
272272 .clientToken(clientToken.toString())
273273 .programmingLanguage { it.languageName(language) }
274274 .artifacts(artifactsMap)
@@ -283,9 +283,9 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
283283 }
284284 }
285285
286- fun getCodeScan (jobId : String ): GetCodeScanResponse = try {
286+ fun getCodeScan (jobId : String ): GetCodeAnalysisResponse = try {
287287 clientAdaptor.getCodeScan(
288- GetCodeScanRequest .builder()
288+ GetCodeAnalysisRequest .builder()
289289 .jobId(jobId)
290290 .build()
291291 )
@@ -295,11 +295,11 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
295295 throw codeScanServerException(" GetCodeReviewException: $errorMessage " )
296296 }
297297
298- fun listCodeScanFindings (jobId : String , nextToken : String? ): ListCodeScanFindingsResponse = try {
298+ fun listCodeScanFindings (jobId : String , nextToken : String? ): ListCodeAnalysisFindingsResponse = try {
299299 clientAdaptor.listCodeScanFindings(
300- ListCodeScanFindingsRequest .builder()
300+ ListCodeAnalysisFindingsRequest .builder()
301301 .jobId(jobId)
302- .codeScanFindingsSchema( CodeScanFindingsSchema . CODESCAN_FINDINGS_1_0 )
302+ .codeAnalysisFindingsSchema( CodeAnalysisFindingsSchema . CODEANALYSIS_FINDINGS_1_0 )
303303 .nextToken(nextToken)
304304 .build()
305305 )
0 commit comments