Examples with ImportPackageInfo org.wso2.ballerinalang.programfile.ImportPackageInfo used on opensource projects

Example 1 with ImportPackageInfo

use of org.wso2.ballerinalang.programfile.ImportPackageInfo in project ballerina by ballerina-lang.

the class CodeGenerator method visit.

public void visit(BLangPackage pkgNode) {
    if (pkgNode.completedPhases.contains(CompilerPhase.CODE_GEN)) {
        if (!buildCompiledPackage) {
            programFile.packageInfoMap.put(pkgNode.symbol.pkgID.bvmAlias(), pkgNode.symbol.packageInfo);
        }
        return;
    }
    // TODO Improve this design without if/else
    PackageInfo packageInfo = new PackageInfo();
    pkgNode.symbol.packageInfo = packageInfo;
    if (buildCompiledPackage) {
        // Generating the BALO
        pkgNode.imports.forEach(impPkgNode -> {
            int impPkgNameCPIndex = addUTF8CPEntry(packageInfo, impPkgNode.symbol.name.value);
            // TODO Improve the import package version once it is available
            int impPkgVersionCPIndex = addUTF8CPEntry(packageInfo, PackageID.DEFAULT.version.value);
            ImportPackageInfo importPkgInfo = new ImportPackageInfo(impPkgNameCPIndex, impPkgVersionCPIndex);
            packageInfo.importPkgInfoSet.add(importPkgInfo);
            packageFile.packageInfo = packageInfo;
        });
    } else {
        // Generating a BALX
        // first visit all the imports
        pkgNode.imports.forEach(impPkgNode -> genNode(impPkgNode, this.env));
        // TODO We need to create identifier for both name and the version
        programFile.packageInfoMap.put(pkgNode.symbol.pkgID.bvmAlias(), packageInfo);
    }
    // Add the current package to the program file
    BPackageSymbol pkgSymbol = pkgNode.symbol;
    currentPkgID = pkgSymbol.pkgID;
    currentPkgInfo = packageInfo;
    currentPkgInfo.nameCPIndex = addUTF8CPEntry(currentPkgInfo, currentPkgID.bvmAlias());
    currentPkgInfo.versionCPIndex = addUTF8CPEntry(currentPkgInfo, currentPkgID.version.value);
    // Insert the package reference to the constant pool of the current package
    currentPackageRefCPIndex = addPackageRefCPEntry(currentPkgInfo, currentPkgID);
    // This attribute keep track of line numbers
    int lineNoAttrNameIndex = addUTF8CPEntry(currentPkgInfo, AttributeInfo.Kind.LINE_NUMBER_TABLE_ATTRIBUTE.value());
    lineNoAttrInfo = new LineNumberTableAttributeInfo(lineNoAttrNameIndex);
    // This attribute keep package-level variable information
    int pkgVarAttrNameIndex = addUTF8CPEntry(currentPkgInfo, AttributeInfo.Kind.LOCAL_VARIABLES_ATTRIBUTE.value());
    currentPkgInfo.addAttributeInfo(AttributeInfo.Kind.LOCAL_VARIABLES_ATTRIBUTE, new LocalVariableAttributeInfo(pkgVarAttrNameIndex));
    pkgNode.globalVars.forEach(this::createPackageVarInfo);
    pkgNode.structs.forEach(this::createStructInfoEntry);
    pkgNode.enums.forEach(this::createEnumInfoEntry);
    pkgNode.connectors.forEach(this::createConnectorInfoEntry);
    pkgNode.functions.forEach(this::createFunctionInfoEntry);
    pkgNode.services.forEach(this::createServiceInfoEntry);
    pkgNode.functions.forEach(this::createFunctionInfoEntry);
    pkgNode.transformers.forEach(this::createTransformerInfoEntry);
    // Visit package builtin function
    visitBuiltinFunctions(pkgNode.initFunction);
    visitBuiltinFunctions(pkgNode.startFunction);
    visitBuiltinFunctions(pkgNode.stopFunction);
    pkgNode.topLevelNodes.stream().filter(pkgLevelNode -> pkgLevelNode.getKind() != NodeKind.VARIABLE && pkgLevelNode.getKind() != NodeKind.XMLNS).forEach(pkgLevelNode -> genNode((BLangNode) pkgLevelNode, this.env));
    currentPkgInfo.addAttributeInfo(AttributeInfo.Kind.LINE_NUMBER_TABLE_ATTRIBUTE, lineNoAttrInfo);
    currentPackageRefCPIndex = -1;
    currentPkgID = null;
    pkgNode.completedPhases.add(CompilerPhase.CODE_GEN);
}