|
2 | 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. |
3 | 3 | * SPDX-License-Identifier: Apache-2.0 |
4 | 4 | */ |
| 5 | + |
5 | 6 | import { CodewhispererLanguage } from '../../shared/telemetry/telemetry.gen' |
| 7 | +import { createConstantMap, ConstantMap } from '../../shared/utilities/tsUtils' |
| 8 | +import * as codewhispererClient from '../client/codewhisperer' |
6 | 9 | import * as CodeWhispererConstants from '../models/constants' |
7 | 10 |
|
8 | | -interface RuntimeLanguageContextData { |
| 11 | +export class RuntimeLanguageContext { |
9 | 12 | /** |
10 | | - * collection of all language runtime versions |
| 13 | + * A map storing cwspr supporting programming language with key: vscLanguageId and value: cwsprLanguageId |
| 14 | + * Key: vscLanguageId |
| 15 | + * Value: CodeWhispererLanguageId |
11 | 16 | */ |
12 | | - languageContexts: { |
13 | | - [language in CodewhispererLanguage as string]: { |
14 | | - /** |
15 | | - * the language of the current file |
16 | | - */ |
17 | | - language: CodewhispererLanguage |
18 | | - } |
| 17 | + private supportedLanguageMap: ConstantMap<CodeWhispererConstants.SupportedLanguage, CodewhispererLanguage> |
| 18 | + |
| 19 | + // A set contains vscode languageId and CodeWhispererLanguage |
| 20 | + private supportedLanguageSet = new Set<string>() |
| 21 | + |
| 22 | + constructor() { |
| 23 | + this.supportedLanguageMap = createConstantMap<CodeWhispererConstants.SupportedLanguage, CodewhispererLanguage>({ |
| 24 | + java: 'java', |
| 25 | + python: 'python', |
| 26 | + javascriptreact: 'jsx', |
| 27 | + typescript: 'javascript', |
| 28 | + javascript: 'javascript', |
| 29 | + }) |
| 30 | + |
| 31 | + const values = Array.from(this.supportedLanguageMap.values()) |
| 32 | + const keys = Array.from(this.supportedLanguageMap.keys()) |
| 33 | + values.forEach(item => this.supportedLanguageSet.add(item)) |
| 34 | + keys.forEach(item => this.supportedLanguageSet.add(item)) |
19 | 35 | } |
20 | | -} |
21 | 36 |
|
22 | | -export class RuntimeLanguageContext { |
23 | | - private runtimeLanguageContext: RuntimeLanguageContextData = { |
24 | | - languageContexts: { |
25 | | - ['plaintext']: { |
26 | | - language: 'plaintext', |
27 | | - }, |
28 | | - ['java']: { |
29 | | - language: 'java', |
30 | | - }, |
31 | | - ['python']: { |
32 | | - language: 'python', |
33 | | - }, |
34 | | - ['javascript']: { |
35 | | - language: 'javascript', |
36 | | - }, |
37 | | - }, |
| 37 | + /** |
| 38 | + * |
| 39 | + * @param vscLanguageId : official vscode languageId |
| 40 | + * @returns corresponding CodewhispererLanguage ID if any, otherwise undefined |
| 41 | + */ |
| 42 | + public mapVscLanguageToCodeWhispererLanguage(vscLanguageId?: string): CodewhispererLanguage | undefined { |
| 43 | + return this.supportedLanguageMap.get(vscLanguageId) ?? undefined |
38 | 44 | } |
39 | 45 |
|
40 | | - public getLanguageContext(languageId?: string) { |
41 | | - const languageName = this.convertLanguage(languageId) |
42 | | - if (languageName in this.runtimeLanguageContext.languageContexts) { |
43 | | - return this.runtimeLanguageContext.languageContexts[languageName] |
| 46 | + /** |
| 47 | + * @param vscLanguageId : official vscode languageId |
| 48 | + * @returns An object with a field language: CodewhispererLanguage, if no corresponding CodewhispererLanguage ID, plaintext is returned |
| 49 | + */ |
| 50 | + public getLanguageContext(vscLanguageId?: string): { language: CodewhispererLanguage } { |
| 51 | + return { language: this.mapVscLanguageToCodeWhispererLanguage(vscLanguageId) ?? 'plaintext' } |
| 52 | + } |
| 53 | + |
| 54 | + /** |
| 55 | + * Mapping the field ProgrammingLanguage of codewhisperer generateRecommendationRequest | listRecommendationRequest to |
| 56 | + * its Codewhisperer runtime language e.g. jsx -> typescript, typescript -> typescript |
| 57 | + * @param request : cwspr generateRecommendationRequest | ListRecommendationRequest |
| 58 | + * @returns request with source language name mapped to cwspr runtime language |
| 59 | + */ |
| 60 | + public mapToRuntimeLanguage< |
| 61 | + T extends codewhispererClient.ListRecommendationsRequest | codewhispererClient.GenerateRecommendationsRequest |
| 62 | + >(request: T): T { |
| 63 | + const fileContext = request.fileContext |
| 64 | + const childLanguage = request.fileContext.programmingLanguage |
| 65 | + let parentLanguage: codewhispererClient.ProgrammingLanguage |
| 66 | + switch (childLanguage.languageName) { |
| 67 | + case 'typescript': |
| 68 | + parentLanguage = { languageName: CodeWhispererConstants.javascript } |
| 69 | + break |
| 70 | + case 'jsx': |
| 71 | + parentLanguage = { languageName: CodeWhispererConstants.javascript } |
| 72 | + break |
| 73 | + default: |
| 74 | + parentLanguage = childLanguage |
| 75 | + break |
44 | 76 | } |
| 77 | + |
45 | 78 | return { |
46 | | - language: languageName as CodewhispererLanguage, |
| 79 | + ...request, |
| 80 | + fileContext: { |
| 81 | + ...fileContext, |
| 82 | + programmingLanguage: parentLanguage, |
| 83 | + }, |
47 | 84 | } |
48 | 85 | } |
49 | 86 |
|
50 | | - public convertLanguage(languageId?: string) { |
51 | | - languageId = languageId === CodeWhispererConstants.typescript ? CodeWhispererConstants.javascript : languageId |
52 | | - if (!languageId) { |
53 | | - return 'plaintext' |
54 | | - } |
55 | | - |
56 | | - return languageId |
| 87 | + /** |
| 88 | + * |
| 89 | + * @param languageId: either vscodeLanguageId or CodewhispererLanguage |
| 90 | + * @returns ture if the language is supported by CodeWhisperer otherwise false |
| 91 | + */ |
| 92 | + public isLanguageSupported(languageId: string): boolean { |
| 93 | + return this.supportedLanguageSet.has(languageId) |
57 | 94 | } |
58 | 95 | } |
59 | 96 |
|
|
0 commit comments