JS: convert K2JSTranslator to Kotlin
This commit is contained in:
Anton Bannykh
@@ -1,210 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2017 JetBrains s.r.o.
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package org.jetbrains.kotlin.js.facade;
|
|
||||||
|
|
||||||
import com.intellij.openapi.vfs.VfsUtilCore;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.annotations.Nullable;
|
|
||||||
import org.jetbrains.kotlin.config.CommonConfigurationKeys;
|
|
||||||
import org.jetbrains.kotlin.config.CommonConfigurationKeysKt;
|
|
||||||
import org.jetbrains.kotlin.config.LanguageVersionSettings;
|
|
||||||
import org.jetbrains.kotlin.descriptors.DeclarationDescriptor;
|
|
||||||
import org.jetbrains.kotlin.descriptors.ModuleDescriptor;
|
|
||||||
import org.jetbrains.kotlin.incremental.js.IncrementalResultsConsumer;
|
|
||||||
import org.jetbrains.kotlin.js.analyze.TopDownAnalyzerFacadeForJS;
|
|
||||||
import org.jetbrains.kotlin.js.analyzer.JsAnalysisResult;
|
|
||||||
import org.jetbrains.kotlin.js.backend.ast.JsImportedModule;
|
|
||||||
import org.jetbrains.kotlin.js.backend.ast.JsProgramFragment;
|
|
||||||
import org.jetbrains.kotlin.js.config.JSConfigurationKeys;
|
|
||||||
import org.jetbrains.kotlin.js.config.JsConfig;
|
|
||||||
import org.jetbrains.kotlin.js.coroutine.CoroutineTransformer;
|
|
||||||
import org.jetbrains.kotlin.js.facade.exceptions.TranslationException;
|
|
||||||
import org.jetbrains.kotlin.js.inline.JsInliner;
|
|
||||||
import org.jetbrains.kotlin.js.inline.clean.LabeledBlockToDoWhileTransformation;
|
|
||||||
import org.jetbrains.kotlin.js.inline.clean.RemoveDuplicateImportsKt;
|
|
||||||
import org.jetbrains.kotlin.js.inline.clean.RemoveUnusedImportsKt;
|
|
||||||
import org.jetbrains.kotlin.js.inline.clean.ResolveTemporaryNamesKt;
|
|
||||||
import org.jetbrains.kotlin.js.sourceMap.SourceFilePathResolver;
|
|
||||||
import org.jetbrains.kotlin.js.translate.general.AstGenerationResult;
|
|
||||||
import org.jetbrains.kotlin.js.translate.general.Translation;
|
|
||||||
import org.jetbrains.kotlin.js.translate.utils.ExpandIsCallsKt;
|
|
||||||
import org.jetbrains.kotlin.metadata.ProtoBuf;
|
|
||||||
import org.jetbrains.kotlin.metadata.deserialization.BinaryVersion;
|
|
||||||
import org.jetbrains.kotlin.progress.ProgressIndicatorAndCompilationCanceledStatus;
|
|
||||||
import org.jetbrains.kotlin.psi.KtFile;
|
|
||||||
import org.jetbrains.kotlin.resolve.BindingTrace;
|
|
||||||
import org.jetbrains.kotlin.resolve.diagnostics.Diagnostics;
|
|
||||||
import org.jetbrains.kotlin.serialization.js.KotlinJavascriptSerializationUtil;
|
|
||||||
import org.jetbrains.kotlin.serialization.js.ast.JsAstSerializer;
|
|
||||||
import org.jetbrains.kotlin.utils.JsMetadataVersion;
|
|
||||||
|
|
||||||
import java.io.ByteArrayOutputStream;
|
|
||||||
import java.io.File;
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
import static org.jetbrains.kotlin.diagnostics.DiagnosticUtils.hasError;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* An entry point of translator.
|
|
||||||
*/
|
|
||||||
public final class K2JSTranslator {
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
private final JsConfig config;
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
private final IncrementalResultsConsumer incrementalResults;
|
|
||||||
|
|
||||||
public K2JSTranslator(@NotNull JsConfig config) {
|
|
||||||
this.config = config;
|
|
||||||
this.incrementalResults = config.getConfiguration().get(JSConfigurationKeys.INCREMENTAL_RESULTS_CONSUMER);
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
public TranslationResult translate(
|
|
||||||
@NotNull JsConfig.Reporter reporter,
|
|
||||||
@NotNull List<KtFile> files,
|
|
||||||
@NotNull MainCallParameters mainCallParameters
|
|
||||||
) throws TranslationException {
|
|
||||||
return translate(reporter, files, mainCallParameters, null);
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
public TranslationResult translate(
|
|
||||||
@NotNull JsConfig.Reporter reporter,
|
|
||||||
@NotNull List<KtFile> files,
|
|
||||||
@NotNull MainCallParameters mainCallParameters,
|
|
||||||
@Nullable JsAnalysisResult analysisResult
|
|
||||||
) throws TranslationException {
|
|
||||||
List<TranslationUnit> units = new ArrayList<>();
|
|
||||||
for (KtFile file : files) {
|
|
||||||
units.add(new TranslationUnit.SourceFile(file));
|
|
||||||
}
|
|
||||||
return translateUnits(reporter, units, mainCallParameters, analysisResult);
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
public TranslationResult translateUnits(
|
|
||||||
@NotNull JsConfig.Reporter reporter,
|
|
||||||
@NotNull List<TranslationUnit> units,
|
|
||||||
@NotNull MainCallParameters mainCallParameters
|
|
||||||
) throws TranslationException {
|
|
||||||
return translateUnits(reporter, units, mainCallParameters, null);
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
public TranslationResult translateUnits(
|
|
||||||
@NotNull JsConfig.Reporter reporter,
|
|
||||||
@NotNull List<TranslationUnit> units,
|
|
||||||
@NotNull MainCallParameters mainCallParameters,
|
|
||||||
@Nullable JsAnalysisResult analysisResult
|
|
||||||
) throws TranslationException {
|
|
||||||
List<KtFile> files = new ArrayList<>();
|
|
||||||
for (TranslationUnit unit : units) {
|
|
||||||
if (unit instanceof TranslationUnit.SourceFile) {
|
|
||||||
files.add(((TranslationUnit.SourceFile) unit).getFile());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (analysisResult == null) {
|
|
||||||
analysisResult = TopDownAnalyzerFacadeForJS.analyzeFiles(files, config);
|
|
||||||
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled();
|
|
||||||
}
|
|
||||||
|
|
||||||
BindingTrace bindingTrace = analysisResult.getBindingTrace();
|
|
||||||
TopDownAnalyzerFacadeForJS.checkForErrors(files, bindingTrace.getBindingContext());
|
|
||||||
ModuleDescriptor moduleDescriptor = analysisResult.getModuleDescriptor();
|
|
||||||
Diagnostics diagnostics = bindingTrace.getBindingContext().getDiagnostics();
|
|
||||||
|
|
||||||
SourceFilePathResolver pathResolver = SourceFilePathResolver.create(config);
|
|
||||||
|
|
||||||
AstGenerationResult translationResult = Translation.generateAst(
|
|
||||||
bindingTrace, units, mainCallParameters, moduleDescriptor, config, pathResolver);
|
|
||||||
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled();
|
|
||||||
if (hasError(diagnostics)) return new TranslationResult.Fail(diagnostics);
|
|
||||||
|
|
||||||
List<JsProgramFragment> newFragments = new ArrayList<>(translationResult.getNewFragments());
|
|
||||||
List<JsProgramFragment> allFragments = new ArrayList<>(translationResult.getFragments());
|
|
||||||
|
|
||||||
JsInliner.process(reporter, config, analysisResult.getBindingTrace(), translationResult.getInnerModuleName(),
|
|
||||||
allFragments, newFragments, translationResult.getImportStatements());
|
|
||||||
|
|
||||||
LabeledBlockToDoWhileTransformation.INSTANCE.apply(newFragments);
|
|
||||||
|
|
||||||
CoroutineTransformer coroutineTransformer = new CoroutineTransformer();
|
|
||||||
for (JsProgramFragment fragment : newFragments) {
|
|
||||||
coroutineTransformer.accept(fragment.getDeclarationBlock());
|
|
||||||
coroutineTransformer.accept(fragment.getInitializerBlock());
|
|
||||||
}
|
|
||||||
RemoveUnusedImportsKt.removeUnusedImports(translationResult.getProgram());
|
|
||||||
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled();
|
|
||||||
if (hasError(diagnostics)) return new TranslationResult.Fail(diagnostics);
|
|
||||||
|
|
||||||
ExpandIsCallsKt.expandIsCalls(newFragments);
|
|
||||||
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled();
|
|
||||||
JsAstSerializer serializer = new JsAstSerializer(file -> {
|
|
||||||
try {
|
|
||||||
return pathResolver.getPathRelativeToSourceRoots(file);
|
|
||||||
}
|
|
||||||
catch (IOException e) {
|
|
||||||
throw new RuntimeException("IO error occurred resolving path to source file", e);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
if (incrementalResults != null) {
|
|
||||||
KotlinJavascriptSerializationUtil serializationUtil = KotlinJavascriptSerializationUtil.INSTANCE;
|
|
||||||
|
|
||||||
for (KtFile file : files) {
|
|
||||||
JsProgramFragment fragment = translationResult.getFragmentMap().get(file);
|
|
||||||
assert fragment != null : "Could not find AST for file: " + file;
|
|
||||||
ByteArrayOutputStream output = new ByteArrayOutputStream();
|
|
||||||
serializer.serialize(fragment, output);
|
|
||||||
byte[] binaryAst = output.toByteArray();
|
|
||||||
|
|
||||||
List<DeclarationDescriptor> scope = translationResult.getFileMemberScopes().get(file);
|
|
||||||
assert scope != null : "Could not find descriptors for file: " + file;
|
|
||||||
BinaryVersion metadataVersion = config.getConfiguration().get(CommonConfigurationKeys.METADATA_VERSION);
|
|
||||||
ProtoBuf.PackageFragment packagePart = serializationUtil.serializeDescriptors(
|
|
||||||
bindingTrace.getBindingContext(), moduleDescriptor, scope, file.getPackageFqName(),
|
|
||||||
CommonConfigurationKeysKt.getLanguageVersionSettings(config.getConfiguration()),
|
|
||||||
metadataVersion != null ? metadataVersion : JsMetadataVersion.INSTANCE
|
|
||||||
);
|
|
||||||
|
|
||||||
File ioFile = VfsUtilCore.virtualToIoFile(file.getVirtualFile());
|
|
||||||
incrementalResults.processPackagePart(ioFile, packagePart.toByteArray(), binaryAst);
|
|
||||||
}
|
|
||||||
|
|
||||||
LanguageVersionSettings settings = CommonConfigurationKeysKt.getLanguageVersionSettings(config.getConfiguration());
|
|
||||||
incrementalResults.processHeader(serializationUtil.serializeHeader(moduleDescriptor, null, settings).toByteArray());
|
|
||||||
}
|
|
||||||
|
|
||||||
RemoveDuplicateImportsKt.removeDuplicateImports(translationResult.getProgram());
|
|
||||||
ResolveTemporaryNamesKt.resolveTemporaryNames(translationResult.getProgram());
|
|
||||||
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled();
|
|
||||||
if (hasError(diagnostics)) return new TranslationResult.Fail(diagnostics);
|
|
||||||
|
|
||||||
List<String> importedModules = new ArrayList<>();
|
|
||||||
for (JsImportedModule module : translationResult.getImportedModuleList()) {
|
|
||||||
importedModules.add(module.getExternalName());
|
|
||||||
}
|
|
||||||
|
|
||||||
return new TranslationResult.Success(config, files, translationResult.getProgram(), diagnostics, importedModules,
|
|
||||||
moduleDescriptor, bindingTrace.getBindingContext());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,174 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2017 JetBrains s.r.o.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.jetbrains.kotlin.js.facade
|
||||||
|
|
||||||
|
import com.intellij.openapi.vfs.VfsUtilCore
|
||||||
|
import org.jetbrains.kotlin.config.CommonConfigurationKeys
|
||||||
|
import org.jetbrains.kotlin.config.*
|
||||||
|
import org.jetbrains.kotlin.incremental.js.IncrementalResultsConsumer
|
||||||
|
import org.jetbrains.kotlin.js.analyze.TopDownAnalyzerFacadeForJS
|
||||||
|
import org.jetbrains.kotlin.js.analyzer.JsAnalysisResult
|
||||||
|
import org.jetbrains.kotlin.js.config.JSConfigurationKeys
|
||||||
|
import org.jetbrains.kotlin.js.config.JsConfig
|
||||||
|
import org.jetbrains.kotlin.js.coroutine.CoroutineTransformer
|
||||||
|
import org.jetbrains.kotlin.js.facade.exceptions.TranslationException
|
||||||
|
import org.jetbrains.kotlin.js.inline.JsInliner
|
||||||
|
import org.jetbrains.kotlin.js.inline.clean.LabeledBlockToDoWhileTransformation
|
||||||
|
import org.jetbrains.kotlin.js.inline.clean.*
|
||||||
|
import org.jetbrains.kotlin.js.sourceMap.SourceFilePathResolver
|
||||||
|
import org.jetbrains.kotlin.js.translate.general.Translation
|
||||||
|
import org.jetbrains.kotlin.js.translate.utils.*
|
||||||
|
import org.jetbrains.kotlin.progress.ProgressIndicatorAndCompilationCanceledStatus
|
||||||
|
import org.jetbrains.kotlin.psi.KtFile
|
||||||
|
import org.jetbrains.kotlin.serialization.js.KotlinJavascriptSerializationUtil
|
||||||
|
import org.jetbrains.kotlin.serialization.js.ast.JsAstSerializer
|
||||||
|
import org.jetbrains.kotlin.utils.JsMetadataVersion
|
||||||
|
|
||||||
|
import java.io.ByteArrayOutputStream
|
||||||
|
import java.io.IOException
|
||||||
|
import java.util.ArrayList
|
||||||
|
|
||||||
|
import org.jetbrains.kotlin.diagnostics.DiagnosticUtils.hasError
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An entry point of translator.
|
||||||
|
*/
|
||||||
|
class K2JSTranslator(private val config: JsConfig) {
|
||||||
|
|
||||||
|
private val incrementalResults = config.configuration.get(JSConfigurationKeys.INCREMENTAL_RESULTS_CONSUMER)
|
||||||
|
|
||||||
|
@Throws(TranslationException::class)
|
||||||
|
@JvmOverloads
|
||||||
|
fun translate(
|
||||||
|
reporter: JsConfig.Reporter,
|
||||||
|
files: List<KtFile>,
|
||||||
|
mainCallParameters: MainCallParameters,
|
||||||
|
analysisResult: JsAnalysisResult? = null
|
||||||
|
): TranslationResult {
|
||||||
|
val units = ArrayList<TranslationUnit>()
|
||||||
|
for (file in files) {
|
||||||
|
units.add(TranslationUnit.SourceFile(file))
|
||||||
|
}
|
||||||
|
return translateUnits(reporter, units, mainCallParameters, analysisResult)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Throws(TranslationException::class)
|
||||||
|
@JvmOverloads
|
||||||
|
fun translateUnits(
|
||||||
|
reporter: JsConfig.Reporter,
|
||||||
|
units: List<TranslationUnit>,
|
||||||
|
mainCallParameters: MainCallParameters,
|
||||||
|
analysisResult: JsAnalysisResult? = null
|
||||||
|
): TranslationResult {
|
||||||
|
var analysisResult = analysisResult
|
||||||
|
val files = ArrayList<KtFile>()
|
||||||
|
for (unit in units) {
|
||||||
|
if (unit is TranslationUnit.SourceFile) {
|
||||||
|
files.add(unit.file)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (analysisResult == null) {
|
||||||
|
analysisResult = TopDownAnalyzerFacadeForJS.analyzeFiles(files, config)
|
||||||
|
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled()
|
||||||
|
}
|
||||||
|
|
||||||
|
val bindingTrace = analysisResult.bindingTrace
|
||||||
|
TopDownAnalyzerFacadeForJS.checkForErrors(files, bindingTrace.bindingContext)
|
||||||
|
val moduleDescriptor = analysisResult.moduleDescriptor
|
||||||
|
val diagnostics = bindingTrace.bindingContext.diagnostics
|
||||||
|
|
||||||
|
val pathResolver = SourceFilePathResolver.create(config)
|
||||||
|
|
||||||
|
val translationResult = Translation.generateAst(
|
||||||
|
bindingTrace, units, mainCallParameters, moduleDescriptor, config, pathResolver
|
||||||
|
)
|
||||||
|
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled()
|
||||||
|
if (hasError(diagnostics)) return TranslationResult.Fail(diagnostics)
|
||||||
|
|
||||||
|
val newFragments = ArrayList(translationResult.newFragments)
|
||||||
|
val allFragments = ArrayList(translationResult.fragments)
|
||||||
|
|
||||||
|
JsInliner.process(
|
||||||
|
reporter, config, analysisResult.bindingTrace, translationResult.innerModuleName,
|
||||||
|
allFragments, newFragments, translationResult.importStatements
|
||||||
|
)
|
||||||
|
|
||||||
|
LabeledBlockToDoWhileTransformation.apply(newFragments)
|
||||||
|
|
||||||
|
val coroutineTransformer = CoroutineTransformer()
|
||||||
|
for (fragment in newFragments) {
|
||||||
|
coroutineTransformer.accept(fragment.declarationBlock)
|
||||||
|
coroutineTransformer.accept(fragment.initializerBlock)
|
||||||
|
}
|
||||||
|
removeUnusedImports(translationResult.program)
|
||||||
|
|
||||||
|
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled()
|
||||||
|
if (hasError(diagnostics)) return TranslationResult.Fail(diagnostics)
|
||||||
|
|
||||||
|
expandIsCalls(newFragments)
|
||||||
|
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled()
|
||||||
|
val serializer = JsAstSerializer { file ->
|
||||||
|
try {
|
||||||
|
pathResolver.getPathRelativeToSourceRoots(file)
|
||||||
|
} catch (e: IOException) {
|
||||||
|
throw RuntimeException("IO error occurred resolving path to source file", e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (incrementalResults != null) {
|
||||||
|
val serializationUtil = KotlinJavascriptSerializationUtil
|
||||||
|
|
||||||
|
for (file in files) {
|
||||||
|
val fragment = translationResult.fragmentMap[file] ?: error("Could not find AST for file: $file")
|
||||||
|
val output = ByteArrayOutputStream()
|
||||||
|
serializer.serialize(fragment, output)
|
||||||
|
val binaryAst = output.toByteArray()
|
||||||
|
|
||||||
|
val scope = translationResult.fileMemberScopes[file] ?: error("Could not find descriptors for file: $file")
|
||||||
|
val metadataVersion = config.configuration.get(CommonConfigurationKeys.METADATA_VERSION)
|
||||||
|
val packagePart = serializationUtil.serializeDescriptors(
|
||||||
|
bindingTrace.bindingContext, moduleDescriptor, scope, file.packageFqName,
|
||||||
|
config.configuration.languageVersionSettings,
|
||||||
|
metadataVersion ?: JsMetadataVersion.INSTANCE
|
||||||
|
)
|
||||||
|
|
||||||
|
val ioFile = VfsUtilCore.virtualToIoFile(file.virtualFile)
|
||||||
|
incrementalResults.processPackagePart(ioFile, packagePart.toByteArray(), binaryAst)
|
||||||
|
}
|
||||||
|
|
||||||
|
val settings = config.configuration.languageVersionSettings
|
||||||
|
incrementalResults.processHeader(serializationUtil.serializeHeader(moduleDescriptor, null, settings).toByteArray())
|
||||||
|
}
|
||||||
|
|
||||||
|
removeDuplicateImports(translationResult.program)
|
||||||
|
translationResult.program.resolveTemporaryNames()
|
||||||
|
|
||||||
|
ProgressIndicatorAndCompilationCanceledStatus.checkCanceled()
|
||||||
|
if (hasError(diagnostics)) return TranslationResult.Fail(diagnostics)
|
||||||
|
|
||||||
|
val importedModules = ArrayList<String>()
|
||||||
|
for (module in translationResult.importedModuleList) {
|
||||||
|
importedModules.add(module.externalName)
|
||||||
|
}
|
||||||
|
|
||||||
|
return TranslationResult.Success(
|
||||||
|
config, files, translationResult.program, diagnostics, importedModules,
|
||||||
|
moduleDescriptor, bindingTrace.bindingContext
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user