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);
}
}
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));
}
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;
}
Aggregations