Examples with Doc com.google.api.tools.framework.snippet.Doc used on opensource projects

Example 11 with Doc

use of com.google.api.tools.framework.snippet.Doc in project toolkit by googleapis.

the class DiscoConfigGeneratorApi method generateConfig.

private Map<String, Doc> generateConfig(String outputPath) {
    Document document = setupDocument();
    ViewModel viewModel = new DiscoConfigTransformer().generateConfig(new DiscoApiModel(document, ""), outputPath);
    Doc generatedConfig = new CommonSnippetSetRunner(new CommonRenderingUtil()).generate(viewModel);
    return ImmutableMap.of(outputPath, generatedConfig);
}

Example 12 with Doc

use of com.google.api.tools.framework.snippet.Doc in project toolkit by googleapis.

the class DiscoGapicProvider method generate.

public Map<String, Doc> generate(String snippetFileName) {
    Map<String, Doc> docs = new LinkedTreeMap<>();
    for (DocumentToViewTransformer transformer : transformers) {
        List<ViewModel> surfaceDocs = transformer.transform(model, productConfig);
        for (ViewModel surfaceDoc : surfaceDocs) {
            if (snippetFileName != null && !surfaceDoc.templateFileName().equals(snippetFileName)) {
                continue;
            }
            Doc doc = snippetSetRunner.generate(surfaceDoc);
            if (doc == null) {
                // generation failed; failures are captured in the model.
                continue;
            }
            docs.put(surfaceDoc.outputPath(), doc);
        }
    }
    return docs;
}

Example 13 with Doc

use of com.google.api.tools.framework.snippet.Doc 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);
    }
}

Example 14 with Doc

use of com.google.api.tools.framework.snippet.Doc in project toolkit by googleapis.

the class DiscoGapicGeneratorApi method run.

public void run() throws Exception {
    String discoveryDocPath = options.get(DISCOVERY_DOC);
    List<String> configFileNames = options.get(GENERATOR_CONFIG_FILES);
    String packageConfigFile = options.get(PACKAGE_CONFIG_FILE);
    List<String> enabledArtifacts = options.get(ENABLED_ARTIFACTS);
    List<GapicProvider> providers = getProviders(discoveryDocPath, configFileNames, packageConfigFile, enabledArtifacts);
    Map<String, Doc> outputFiles = Maps.newHashMap();
    for (GapicProvider provider : providers) {
        outputFiles.putAll(provider.generate());
    }
    ToolUtil.writeFiles(outputFiles, options.get(OUTPUT_FILE));
}

Example 15 with Doc

use of com.google.api.tools.framework.snippet.Doc 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;
}