Examples with PackagingConfig com.google.api.codegen.config.PackagingConfig used on opensource projects

Example 1 with PackagingConfig

use of com.google.api.codegen.config.PackagingConfig in project toolkit by googleapis.

the class DiscoGapicGeneratorApp method getGenerators.

/**
 * From config file paths, constructs the DiscoGapicGenerators to run.
 */
@VisibleForTesting
public static List<CodeGenerator<?>> getGenerators(String discoveryDocPath, List<String> configFileNames, String packageConfig2File, String dependencyConfigFile, String languageStr, List<String> enabledArtifacts, ArtifactType artifactType) throws IOException {
    if (!new File(discoveryDocPath).exists()) {
        throw new IOException("File not found: " + discoveryDocPath);
    }
    Reader reader = new InputStreamReader(new FileInputStream(new File(discoveryDocPath)));
    ObjectMapper mapper = new ObjectMapper();
    JsonNode root = mapper.readTree(reader);
    // Read the YAML config and convert it to proto.
    if (configFileNames.size() == 0) {
        throw new IOException(String.format("--%s must be provided", GENERATOR_CONFIG_FILES.name()));
    }
    ConfigProto configProto = loadConfigFromFiles(configFileNames);
    if (configProto == null) {
        throw new IOException("Failed to load config proto.");
    }
    PackageMetadataConfig packageConfig = null;
    if (!Strings.isNullOrEmpty(packageConfig2File)) {
        ApiDefaultsConfig apiDefaultsConfig = ApiDefaultsConfig.load();
        DependenciesConfig dependenciesConfig;
        if (dependencyConfigFile != null) {
            dependenciesConfig = DependenciesConfig.loadFromURL(new File(dependencyConfigFile).toURI().toURL());
        } else {
            dependenciesConfig = DependenciesConfig.load();
        }
        PackagingConfig packagingConfig = PackagingConfig.load(packageConfig2File);
        packageConfig = PackageMetadataConfig.createFromPackaging(apiDefaultsConfig, dependenciesConfig, packagingConfig);
    }
    TargetLanguage language;
    if (!Strings.isNullOrEmpty(languageStr)) {
        language = TargetLanguage.fromString(languageStr.toUpperCase());
    } else {
        throw new IllegalArgumentException("Language not set by --language option.");
    }
    String defaultPackageName = configProto.getLanguageSettingsMap().get(languageStr).getPackageName();
    DiscoApiModel model = new DiscoApiModel(Document.from(new DiscoveryNode(root)), defaultPackageName);
    GapicProductConfig productConfig = GapicProductConfig.create(model, configProto, language);
    ArtifactFlags artifactFlags = new ArtifactFlags(enabledArtifacts, artifactType, false);
    return DiscoGapicGeneratorFactory.create(language, model, productConfig, packageConfig, artifactFlags);
}

Example 2 with PackagingConfig

use of com.google.api.codegen.config.PackagingConfig in project toolkit by googleapis.

the class GapicGeneratorApp method process.

@Override
protected void process() throws Exception {
    String protoPackage = Strings.emptyToNull(options.get(PROTO_PACKAGE));
    // Read the GAPIC config, if it was given, and convert it to proto.
    List<String> configFileNames = options.get(GENERATOR_CONFIG_FILES);
    ConfigProto configProto = null;
    if (configFileNames.size() > 0) {
        // Read the YAML config and convert it to proto.
        ConfigSource configSource = loadConfigFromFiles(configFileNames, ConfigProto.getDescriptor().getFullName(), ConfigProto.getDefaultInstance());
        if (configSource == null) {
            return;
        }
        configProto = (ConfigProto) configSource.getConfig();
        if (configProto == null) {
            return;
        }
    }
    // Consume gRPC Service Config if it is given with gapic_v2.
    String gRPCServiceConfigPath = options.get(GRPC_SERVICE_CONFIG);
    ServiceConfig gRPCServiceConfig = null;
    if (!Strings.isNullOrEmpty(gRPCServiceConfigPath) && configProto.getConfigSchemaVersion().equals("2.0.0")) {
        ServiceConfig.Builder builder = ServiceConfig.newBuilder();
        FileReader file = new FileReader(gRPCServiceConfigPath);
        JsonFormat.parser().merge(file, builder);
        gRPCServiceConfig = builder.build();
    }
    // Read the sample configs, if they are given, and convert them to protos.
    SampleConfigProto sampleConfigProto = null;
    List<String> sampleConfigFileNames = options.get(SAMPLE_CONFIG_FILES);
    if (sampleConfigFileNames.size() > 0) {
        ConfigSource configSource = loadConfigFromFiles(SampleConfigSanitizer.sanitize(sampleConfigFileNames), SampleConfigProto.getDescriptor().getFullName(), SampleConfigProto.getDefaultInstance());
        // TODO(hzyi): Verify this works for repeated fields as well
        // TODO(hzyi): Allow users to put arbitrary top-level directives not
        // used by gapic-generator
        sampleConfigProto = (SampleConfigProto) configSource.getConfig();
    }
    model.establishStage(Merged.KEY);
    if (model.getDiagReporter().getDiagCollector().getErrorCount() > 0) {
        for (Diag diag : model.getDiagReporter().getDiagCollector().getDiags()) {
            System.err.println(diag.toString());
        }
        return;
    }
    ApiDefaultsConfig apiDefaultsConfig = ApiDefaultsConfig.load();
    DependenciesConfig dependenciesConfig = DependenciesConfig.load();
    TargetLanguage language;
    if (!Strings.isNullOrEmpty(options.get(LANGUAGE))) {
        language = TargetLanguage.fromString(options.get(LANGUAGE).toUpperCase());
    } else {
        throw new IllegalArgumentException("Language not set by --language option.");
    }
    String clientPackage = Strings.emptyToNull(options.get(CLIENT_PACKAGE));
    String transport = options.get(TRANSPORT).toLowerCase();
    TransportProtocol tp;
    if (transport.equals("grpc")) {
        tp = TransportProtocol.GRPC;
    } else if (transport.equals("rest")) {
        tp = TransportProtocol.HTTP;
    } else {
        throw new IllegalArgumentException("Unknown transport protocol: " + transport);
    }
    GapicProductConfig productConfig = GapicProductConfig.create(model, configProto, sampleConfigProto, protoPackage, clientPackage, language, gRPCServiceConfig, tp);
    if (productConfig == null) {
        ToolUtil.reportDiags(model.getDiagReporter().getDiagCollector(), true);
        return;
    }
    PackagingConfig packagingConfig;
    if (!Strings.isNullOrEmpty(options.get(PACKAGE_CONFIG2_FILE))) {
        packagingConfig = PackagingConfig.load(options.get(PACKAGE_CONFIG2_FILE));
    } else {
        packagingConfig = PackagingConfig.loadFromProductConfig(productConfig.getInterfaceConfigMap());
    }
    PackageMetadataConfig packageConfig = PackageMetadataConfig.createFromPackaging(apiDefaultsConfig, dependenciesConfig, packagingConfig);
    // TODO(hzyi-google): Once we switch to sample configs, require an
    // additional check to generate samples:
    // `sampleConfigProto != null`
    ArtifactFlags artifactFlags = new ArtifactFlags(options.get(ENABLED_ARTIFACTS), artifactType, options.get(DEV_SAMPLES));
    List<CodeGenerator<?>> generators = GapicGeneratorFactory.create(language, model, productConfig, packageConfig, artifactFlags);
    ImmutableMap.Builder<String, GeneratedResult<?>> generatedResults = ImmutableMap.builder();
    for (CodeGenerator<?> generator : generators) {
        Map<String, ? extends GeneratedResult<?>> generatorResult = generator.generate();
        for (Map.Entry<String, ? extends GeneratedResult<?>> entry : generatorResult.entrySet()) {
            generatedResults.put(entry.getKey(), entry.getValue());
        }
    }
    gapicWriter.writeCodeGenOutput(generatedResults.build(), model.getDiagReporter().getDiagCollector());
}

Example 3 with PackagingConfig

use of com.google.api.codegen.config.PackagingConfig in project toolkit by googleapis.

the class PackageGeneratorApp method generate.

protected Map<String, GeneratedResult<Doc>> generate(Model model) throws IOException {
    TargetLanguage language = TargetLanguage.fromString(options.get(LANGUAGE));
    PackageMetadataConfig config = null;
    ApiDefaultsConfig apiDefaultsConfig = ApiDefaultsConfig.load();
    DependenciesConfig dependenciesConfig;
    if (dependenciesYamlUrl != null) {
        dependenciesConfig = DependenciesConfig.loadFromURL(dependenciesYamlUrl);
    } else {
        dependenciesConfig = DependenciesConfig.load();
    }
    PackagingConfig packagingConfig = null;
    if (!Strings.isNullOrEmpty(options.get(PACKAGE_CONFIG2_FILE))) {
        packagingConfig = PackagingConfig.load(options.get(PACKAGE_CONFIG2_FILE));
    } else {
    // TODO(andrealin): Get PackageMetadataConfig from proto annotations.
    }
    config = PackageMetadataConfig.createFromPackaging(apiDefaultsConfig, dependenciesConfig, packagingConfig);
    Preconditions.checkNotNull(config);
    PackagingArtifactType artifactType = options.get(PackageGeneratorApp.ARTIFACT_TYPE);
    if (artifactType == null) {
        artifactType = config.artifactType();
    }
    CodeGenerator<Doc> generator = PackageGeneratorFactory.create(language, artifactType, options, model, config);
    return generator.generate();
}

Example 4 with PackagingConfig

use of com.google.api.codegen.config.PackagingConfig in project toolkit by googleapis.

the class GapicTestBase2 method run.

@Override
public Map<String, ?> run() throws IOException {
    model.establishStage(Merged.KEY);
    if (model.getDiagReporter().getDiagCollector().getErrorCount() > 0) {
        for (Diag diag : model.getDiagReporter().getDiagCollector().getDiags()) {
            System.err.println(diag.toString());
        }
        return null;
    }
    if (sampleConfig == null) {
        sampleConfig = SampleConfigProto.getDefaultInstance();
    }
    GapicProductConfig productConfig = GapicProductConfig.create(model, gapicConfig, sampleConfig, protoPackage, clientPackage, language, grpcServiceConfig, transportProtocol);
    if (productConfig == null) {
        for (Diag diag : model.getDiagReporter().getDiagCollector().getDiags()) {
            System.err.println(diag.toString());
        }
        return null;
    }
    List<String> enabledArtifacts = new ArrayList<>();
    if (hasSmokeTestConfig(productConfig)) {
        enabledArtifacts.addAll(Arrays.asList("surface", "test", "samples"));
    }
    ArtifactFlags artifactFlags = new ArtifactFlags(enabledArtifacts, ArtifactType.LEGACY_GAPIC_AND_PACKAGE, true);
    PackagingConfig actualPackagingConfig = packagingConfig;
    if (actualPackagingConfig == null) {
        actualPackagingConfig = PackagingConfig.loadFromProductConfig(productConfig.getInterfaceConfigMap());
    }
    PackageMetadataConfig packageConfig = PackageMetadataConfig.createFromPackaging(apiDefaultsConfig, dependenciesConfig, actualPackagingConfig);
    List<CodeGenerator<?>> generators = GapicGeneratorFactory.create(language, model, productConfig, packageConfig, artifactFlags);
    // Don't run any generators we're not testing.
    ArrayList<CodeGenerator<?>> testedGenerators = new ArrayList<>();
    for (CodeGenerator<?> generator : generators) {
        if (!Collections.disjoint(generator.getInputFileNames(), snippetNames)) {
            testedGenerators.add(generator);
        }
    }
    Map<String, Object> output = new TreeMap<>();
    for (CodeGenerator<?> generator : testedGenerators) {
        Map<String, ? extends GeneratedResult<?>> out = generator.generate();
        if (!Collections.disjoint(out.keySet(), output.keySet())) {
            throw new IllegalStateException("file conflict");
        }
        for (Map.Entry<String, ? extends GeneratedResult<?>> entry : out.entrySet()) {
            Object value = (entry.getValue().getBody() instanceof byte[]) ? "Static or binary file content is not shown." : entry.getValue().getBody();
            output.put(entry.getKey(), value);
        }
    }
    return output;
}