com.google.api.codegen.packagegen.py.PythonGrpcPackageGenerator.java Source code

Introduction

Here is the source code for com.google.api.codegen.packagegen.py.PythonGrpcPackageGenerator.java

Source

/* Copyright 2017 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.google.api.codegen.packagegen.py;

import com.google.api.codegen.common.CodeGenerator;
import com.google.api.codegen.common.GeneratedResult;
import com.google.api.codegen.config.PackageMetadataConfig;
import com.google.api.codegen.config.ProtoApiModel;
import com.google.api.codegen.rendering.CommonSnippetSetRunner;
import com.google.api.codegen.viewmodel.metadata.PackageMetadataView;
import com.google.api.tools.framework.model.Model;
import com.google.api.tools.framework.snippet.Doc;
import com.google.api.tools.framework.tools.ToolOptions;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/** Performs gRPC package generation for Python */
public class PythonGrpcPackageGenerator implements CodeGenerator<Doc> {

    private final ToolOptions options;
    private final Model model;
    private final PackageMetadataConfig config;

    public PythonGrpcPackageGenerator(ToolOptions options, Model model, PackageMetadataConfig config) {
        this.options = options;
        this.model = model;
        this.config = config;
    }

    @Override
    public List<String> getInputFileNames() {
        return ImmutableList.of();
    }

    @Override
    public Map<String, GeneratedResult<Doc>> generate() throws IOException {
        ImmutableMap.Builder<String, GeneratedResult<Doc>> results = new ImmutableMap.Builder<>();
        ArrayList<PackageMetadataView> metadataViews = new ArrayList<>();

        PythonPackageCopier copier = new PythonPackageCopier();
        PythonPackageCopierResult copierResult = copier.run(options, config);

        results.putAll(copierResult.results());
        PythonGrpcPackageTransformer pythonTransformer = new PythonGrpcPackageTransformer(copierResult);
        ProtoApiModel apiModel = new ProtoApiModel(model);
        metadataViews.addAll(pythonTransformer.transform(apiModel, config));

        for (PackageMetadataView view : metadataViews) {
            CommonSnippetSetRunner runner = new CommonSnippetSetRunner(view, false);
            results.putAll(runner.generate(view));
        }
        return results.build();
    }
}