# Copyright 2018 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
#
#     http://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.

"""This script is used to synthesize generated parts of this library."""

import re

import synthtool as s
from synthtool import gcp

gapic = gcp.GAPICGenerator()

versions = ['v1beta2', 'v1']


for version in versions:
    library = gapic.py_library('dataproc', version)
    s.move(
        library,
        excludes=[
            'docs/index.rst', 'nox.py', 'README.rst', 'setup.py'])

    s.replace(
        f'google/cloud/dataproc_{version}/gapic/cluster_controller_client.py',
        'metadata_type=operations_pb2.ClusterOperationMetadata,',
        'metadata_type=proto_operations_pb2.ClusterOperationMetadata,')

    s.replace(
        f'google/cloud/dataproc_{version}/gapic/cluster_controller_client.py',
        '\s+<strong>Note:</strong>.*\n(.*\n)+?.*types.FieldMask.',
        f'''

                .. note::

                    Currently, only the following fields can be updated:

                    * ``labels``: Update labels
                    * ``config.worker_config.num_instances``: Resize primary
                      worker group
                    * ``config.secondary_worker_config.num_instances``: Resize
                      secondary worker group

                    If a dict is provided, it must be of the same form as the protobuf
                    message :class:`~google.cloud.dataproc_{version}.types.FieldMask`'''
    )

s.replace(
    'google/cloud/dataproc_v1beta2/proto/clusters_pb2.py',
    '# Generated by the protocol buffer compiler.  DO NOT EDIT!',
    '# -*- coding: utf-8 -*-\n'
    '# Generated by the protocol buffer compiler.  DO NOT EDIT!')

s.replace(
    'google/cloud/dataproc_v1beta2/proto/workflow_templates_pb2.py',
    ', and must\n\s+conform to the following PCRE regular expression:'
    '(.*\n)+?.*No more than 32',
    '. Label values must be between\n'
    '          1 and 63 characters long. No more than 32'
)
s.replace(
    'google/cloud/dataproc_v1beta2/proto/workflow_templates_pb2.py',
    ', and must conform to\n'
    '\s+the following regular expression:(.*\n)+?.* No more than',
    '. Label values must be between\n'
    '          1 and 63 characters long. No more than'
)

s.replace(
    f'google/cloud/dataproc_v1/gapic/cluster_controller_client.py',
    '}\n(\s+Similarly, to change the number of preemptible workers in a cluster to 5,)',
    '}\n\n\g<1>'
)
