Search in sources :

Example 11 with GapicProductConfig

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

the class CodeGeneratorApi method process.

@Override
protected void process() throws Exception {
    // Read the YAML config and convert it to proto.
    List<String> configFileNames = options.get(GENERATOR_CONFIG_FILES);
    if (configFileNames.size() == 0) {
        error(String.format("--%s must be provided", GENERATOR_CONFIG_FILES.name()));
        return;
    }
    ConfigSource configSource = loadConfigFromFiles(configFileNames);
    if (configSource == null) {
        return;
    }
    ConfigProto configProto = (ConfigProto) configSource.getConfig();
    if (configProto == null) {
        return;
    }
    model.establishStage(Merged.KEY);
    List<String> adviceSuppressors = options.get(ADVICE_SUPPRESSORS);
    Adviser adviser = new Adviser(adviceSuppressors);
    adviser.advise(model, configProto);
    if (model.getDiagCollector().getErrorCount() > 0) {
        for (Diag diag : model.getDiagCollector().getDiags()) {
            System.err.println(diag.toString());
        }
        return;
    }
    PackageMetadataConfig packageConfig = null;
    if (!Strings.isNullOrEmpty(options.get(PACKAGE_CONFIG_FILE))) {
        String contents = new String(Files.readAllBytes(Paths.get(options.get(PACKAGE_CONFIG_FILE))), StandardCharsets.UTF_8);
        packageConfig = PackageMetadataConfig.createFromString(contents);
    }
    GeneratorProto generator = configProto.getGenerator();
    GapicProductConfig productConfig = GapicProductConfig.create(model, configProto);
    if (productConfig == null) {
        return;
    }
    if (generator != null) {
        String factory = generator.getFactory();
        String id = generator.getId();
        GapicProviderFactory<GapicProvider> providerFactory = createProviderFactory(model, factory);
        GapicGeneratorConfig generatorConfig = GapicGeneratorConfig.newBuilder().id(id).enabledArtifacts(options.get(ENABLED_ARTIFACTS)).build();
        String outputPath = options.get(OUTPUT_FILE);
        List<GapicProvider> providers = providerFactory.create(model, productConfig, generatorConfig, packageConfig);
        Map<String, Doc> outputFiles = Maps.newHashMap();
        for (GapicProvider provider : providers) {
            outputFiles.putAll(provider.generate());
        }
        writeCodeGenOutput(outputFiles, outputPath);
    }
}
Also used : GapicProductConfig(com.google.api.codegen.config.GapicProductConfig) Adviser(com.google.api.codegen.advising.Adviser) GapicProvider(com.google.api.codegen.gapic.GapicProvider) Diag(com.google.api.tools.framework.model.Diag) ConfigSource(com.google.api.tools.framework.model.ConfigSource) GapicGeneratorConfig(com.google.api.codegen.gapic.GapicGeneratorConfig) PackageMetadataConfig(com.google.api.codegen.config.PackageMetadataConfig) Doc(com.google.api.tools.framework.snippet.Doc)

Example 12 with GapicProductConfig

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

the class DiscoGapicGeneratorApi method getProviders.

/**
 * From config file paths, constructs the DiscoGapicProviders to run.
 */
@VisibleForTesting
static List<GapicProvider> getProviders(String discoveryDocPath, List<String> configFileNames, String packageConfigFile, List<String> enabledArtifacts) 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(packageConfigFile)) {
        String contents = new String(Files.readAllBytes(Paths.get(packageConfigFile)), StandardCharsets.UTF_8);
        packageConfig = PackageMetadataConfig.createFromString(contents);
    }
    GeneratorProto generator = configProto.getGenerator();
    String language = configProto.getLanguage();
    String defaultPackageName = configProto.getLanguageSettingsMap().get(language).getPackageName();
    SurfaceNamer surfaceNamer = null;
    if (language.equals(JAVA)) {
        surfaceNamer = new JavaSurfaceNamer(defaultPackageName, defaultPackageName, new JavaNameFormatter());
    }
    if (surfaceNamer == null) {
        throw new UnsupportedOperationException("DiscoGapicGeneratorApi: language \"" + language + "\" not yet supported");
    }
    DiscoApiModel model = new DiscoApiModel(Document.from(new DiscoveryNode(root)), defaultPackageName);
    GapicProductConfig productConfig = GapicProductConfig.create(model, configProto, surfaceNamer);
    String factory = generator.getFactory();
    String id = generator.getId();
    DiscoGapicProviderFactory providerFactory = createProviderFactory(factory);
    GapicGeneratorConfig generatorConfig = GapicGeneratorConfig.newBuilder().id(id).enabledArtifacts(enabledArtifacts).build();
    return providerFactory.create(model, productConfig, generatorConfig, packageConfig);
}
Also used : GapicProductConfig(com.google.api.codegen.config.GapicProductConfig) DiscoveryNode(com.google.api.codegen.discovery.DiscoveryNode) InputStreamReader(java.io.InputStreamReader) ConfigYamlReader(com.google.api.codegen.configgen.ConfigYamlReader) Reader(java.io.Reader) InputStreamReader(java.io.InputStreamReader) JsonNode(com.fasterxml.jackson.databind.JsonNode) IOException(java.io.IOException) JavaSurfaceNamer(com.google.api.codegen.transformer.java.JavaSurfaceNamer) FileInputStream(java.io.FileInputStream) JavaNameFormatter(com.google.api.codegen.util.java.JavaNameFormatter) DiscoGapicProviderFactory(com.google.api.codegen.discogapic.DiscoGapicProviderFactory) DiscoApiModel(com.google.api.codegen.config.DiscoApiModel) GapicGeneratorConfig(com.google.api.codegen.gapic.GapicGeneratorConfig) PackageMetadataConfig(com.google.api.codegen.config.PackageMetadataConfig) File(java.io.File) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) JavaSurfaceNamer(com.google.api.codegen.transformer.java.JavaSurfaceNamer) SurfaceNamer(com.google.api.codegen.transformer.SurfaceNamer) VisibleForTesting(com.google.common.annotations.VisibleForTesting)

Example 13 with GapicProductConfig

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

the class GapicTestBase2 method createTestConfig.

/**
 * Creates the constructor arguments to be passed onto this class (GapicTestBase2) to create test
 * methods. The idForFactory String is passed to GapicProviderFactory to get the GapicProviders
 * provided by that id, and then the snippet file names are scraped from those providers, and a
 * set of arguments is created for each combination of GapicProvider x snippet that
 * GapicProviderFactory returns.
 */
public static Object[] createTestConfig(String idForFactory, String[] gapicConfigFileNames, String packageConfigFileName, String apiName) {
    Model model = Model.create(Service.getDefaultInstance());
    GapicProductConfig productConfig = GapicProductConfig.createDummyInstance();
    PackageMetadataConfig packageConfig = PackageMetadataConfig.createDummyPackageMetadataConfig();
    GapicGeneratorConfig generatorConfig = GapicGeneratorConfig.newBuilder().id(idForFactory).enabledArtifacts(Arrays.asList("surface", "test")).build();
    List<GapicProvider> providers = MainGapicProviderFactory.defaultCreate(model, productConfig, generatorConfig, packageConfig);
    List<String> snippetNames = new ArrayList<>();
    for (GapicProvider provider : providers) {
        snippetNames.addAll(provider.getSnippetFileNames());
    }
    String baseline = idForFactory + "_" + apiName + ".baseline";
    return new Object[] { idForFactory, gapicConfigFileNames, packageConfigFileName, snippetNames, apiName, baseline };
}
Also used : GapicProductConfig(com.google.api.codegen.config.GapicProductConfig) GapicGeneratorConfig(com.google.api.codegen.gapic.GapicGeneratorConfig) Model(com.google.api.tools.framework.model.Model) PackageMetadataConfig(com.google.api.codegen.config.PackageMetadataConfig) GapicProvider(com.google.api.codegen.gapic.GapicProvider) ArrayList(java.util.ArrayList)

Example 14 with GapicProductConfig

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

the class GapicTestBase2 method run.

@Override
public Map<String, Doc> run() {
    model.establishStage(Merged.KEY);
    if (model.getDiagCollector().getErrorCount() > 0) {
        for (Diag diag : model.getDiagCollector().getDiags()) {
            System.err.println(diag.toString());
        }
        return null;
    }
    GapicProductConfig productConfig = GapicProductConfig.create(model, gapicConfig);
    if (productConfig == null) {
        for (Diag diag : model.getDiagCollector().getDiags()) {
            System.err.println(diag.toString());
        }
        return null;
    }
    List<String> enabledArtifacts = new ArrayList<>();
    if (hasSmokeTestConfig(productConfig)) {
        enabledArtifacts.addAll(Arrays.asList("surface", "test"));
    }
    GapicGeneratorConfig generatorConfig = GapicGeneratorConfig.newBuilder().id(idForFactory).enabledArtifacts(enabledArtifacts).build();
    List<GapicProvider> providers = MainGapicProviderFactory.defaultCreate(model, productConfig, generatorConfig, packageConfig);
    List<String> disabledGen = new ArrayList<>(snippetNames);
    for (GapicProvider provider : providers) {
        disabledGen.removeAll(provider.getSnippetFileNames());
    }
    for (String gen : disabledGen) {
        testOutput().printf("%s generation is not enabled for this test case.\n", gen);
    }
    // Don't run any providers we're not testing.
    ArrayList<GapicProvider> testedProviders = new ArrayList<>();
    for (GapicProvider provider : providers) {
        if (!Collections.disjoint(provider.getSnippetFileNames(), snippetNames)) {
            testedProviders.add(provider);
        }
    }
    boolean reportDiag = false;
    Map<String, Doc> output = new TreeMap<>();
    for (GapicProvider provider : testedProviders) {
        Map<String, Doc> out = provider.generate();
        if (output == null) {
            reportDiag = true;
        } else {
            if (!Collections.disjoint(out.keySet(), output.keySet())) {
                throw new IllegalStateException("file conflict");
            }
            output.putAll(out);
        }
    }
    if (reportDiag) {
        for (Diag diag : model.getDiagCollector().getDiags()) {
            testOutput().println(diag.toString());
        }
    }
    return output;
}
Also used : GapicProductConfig(com.google.api.codegen.config.GapicProductConfig) ArrayList(java.util.ArrayList) GapicProvider(com.google.api.codegen.gapic.GapicProvider) Diag(com.google.api.tools.framework.model.Diag) TreeMap(java.util.TreeMap) GapicGeneratorConfig(com.google.api.codegen.gapic.GapicGeneratorConfig) Doc(com.google.api.tools.framework.snippet.Doc)

Example 15 with GapicProductConfig

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

the class PhpGapicCodePathMapperTest method getOutputPathTest.

@Test
public void getOutputPathTest() {
    PhpGapicCodePathMapper pathMapper = PhpGapicCodePathMapper.newBuilder().setPrefix("prefix").setSuffix("suffix").build();
    GapicProductConfig configWithGoogleCloud = GapicProductConfig.createDummyInstance(ImmutableMap.<String, InterfaceConfig>of(), "Google\\Cloud\\Sample\\Package\\V1", "", null);
    Truth.assertThat(pathMapper.getOutputPath(null, configWithGoogleCloud)).isEqualTo("prefix/Sample/Package/V1/suffix");
    GapicProductConfig configWithGoogleNonCloud = GapicProductConfig.createDummyInstance(ImmutableMap.<String, InterfaceConfig>of(), "Google\\NonCloud\\Sample\\Package\\V1", "", null);
    Truth.assertThat(pathMapper.getOutputPath(null, configWithGoogleNonCloud)).isEqualTo("prefix/NonCloud/Sample/Package/V1/suffix");
    GapicProductConfig configWithAlphabet = GapicProductConfig.createDummyInstance(ImmutableMap.<String, InterfaceConfig>of(), "Alphabet\\Google\\Cloud\\Sample\\Package\\V1", "", null);
    Truth.assertThat(pathMapper.getOutputPath(null, configWithAlphabet)).isEqualTo("prefix/Alphabet/Google/Cloud/Sample/Package/V1/suffix");
}
Also used : GapicProductConfig(com.google.api.codegen.config.GapicProductConfig) PhpGapicCodePathMapper(com.google.api.codegen.php.PhpGapicCodePathMapper) Test(org.junit.Test)

Aggregations

GapicProductConfig (com.google.api.codegen.config.GapicProductConfig)24 PackageMetadataConfig (com.google.api.codegen.config.PackageMetadataConfig)11 ProtoApiModel (com.google.api.codegen.config.ProtoApiModel)10 ViewModel (com.google.api.codegen.viewmodel.ViewModel)9 List (java.util.List)8 ArrayList (java.util.ArrayList)7 ImmutableList (com.google.common.collect.ImmutableList)6 TargetLanguage (com.google.api.codegen.common.TargetLanguage)5 GapicInterfaceContext (com.google.api.codegen.config.GapicInterfaceContext)5 InterfaceContext (com.google.api.codegen.config.InterfaceContext)5 MethodModel (com.google.api.codegen.config.MethodModel)5 Model (com.google.api.tools.framework.model.Model)5 Map (java.util.Map)5 Function (java.util.function.Function)5 CodeGenerator (com.google.api.codegen.common.CodeGenerator)4 SampleContext (com.google.api.codegen.config.SampleContext)4 GapicCodePathMapper (com.google.api.codegen.gapic.GapicCodePathMapper)4 GapicGeneratorConfig (com.google.api.codegen.gapic.GapicGeneratorConfig)4 Streams (com.google.common.collect.Streams)4 Paths (java.nio.file.Paths)4