// Code generated by sidekick. DO NOT EDIT.
//
// Copyright 2026 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.

import Foundation
#if canImport(FoundationNetworking)
  import FoundationNetworking
#endif
import GoogleCloudWkt
import GoogleCloudGax

extension Clients {
  final class MigrationServiceRetry: MigrationServiceStub {
    let inner: any MigrationServiceStub
    let options: GoogleCloudGax.ClientOptions

    public init(_ inner: any MigrationServiceStub, options: GoogleCloudGax.ClientOptions) {
      self.inner = inner
      self.options = options
    }

    func _intercept<Input, Output>(
      request: Input,
      options: GoogleCloudGax.RequestOptions,
      idempotent: Swift.Bool,
      action: (Input, GoogleCloudGax.RequestOptions) async throws -> Output,
    ) async throws -> Output {
      let loop = GoogleCloudGax._RetryLoop(
        options: options, withDefault: self.options, idempotent: idempotent,
      )
      let attempt = { (attemptTimeout: Swift.Duration?) async throws -> Output in
        var attemptOptions = options
        attemptOptions.attemptTimeout = attemptTimeout
        return try await action(request, attemptOptions)
      }
      return try await loop.run(attempt: attempt)
    }

    public func createMigrationWorkflow(
      request: CreateMigrationWorkflowRequest, options: GoogleCloudGax.RequestOptions
    ) async throws -> GoogleBigQueryMigrationV2.MigrationWorkflow {
      try await self._intercept(
        request: request,
        options: options,
        idempotent: false,
        action: {
          (r: CreateMigrationWorkflowRequest, o: GoogleCloudGax.RequestOptions) async throws
            -> GoogleBigQueryMigrationV2.MigrationWorkflow
          in
          return try await self.inner.createMigrationWorkflow(request: r, options: o)
        })
    }

    public func getMigrationWorkflow(
      request: GetMigrationWorkflowRequest, options: GoogleCloudGax.RequestOptions
    ) async throws -> GoogleBigQueryMigrationV2.MigrationWorkflow {
      try await self._intercept(
        request: request,
        options: options,
        idempotent: true,
        action: {
          (r: GetMigrationWorkflowRequest, o: GoogleCloudGax.RequestOptions) async throws
            -> GoogleBigQueryMigrationV2.MigrationWorkflow
          in
          return try await self.inner.getMigrationWorkflow(request: r, options: o)
        })
    }

    public func listMigrationWorkflows(
      request: ListMigrationWorkflowsRequest, options: GoogleCloudGax.RequestOptions
    ) async throws -> GoogleBigQueryMigrationV2.ListMigrationWorkflowsResponse {
      try await self._intercept(
        request: request,
        options: options,
        idempotent: true,
        action: {
          (r: ListMigrationWorkflowsRequest, o: GoogleCloudGax.RequestOptions) async throws
            -> GoogleBigQueryMigrationV2.ListMigrationWorkflowsResponse
          in
          return try await self.inner.listMigrationWorkflows(request: r, options: o)
        })
    }

    public func deleteMigrationWorkflow(
      request: DeleteMigrationWorkflowRequest, options: GoogleCloudGax.RequestOptions
    ) async throws {
      try await self._intercept(
        request: request,
        options: options,
        idempotent: false,
        action: {
          (r: DeleteMigrationWorkflowRequest, o: GoogleCloudGax.RequestOptions) async throws -> Void
          in
          return try await self.inner.deleteMigrationWorkflow(request: r, options: o)
        })
    }

    public func startMigrationWorkflow(
      request: StartMigrationWorkflowRequest, options: GoogleCloudGax.RequestOptions
    ) async throws {
      try await self._intercept(
        request: request,
        options: options,
        idempotent: false,
        action: {
          (r: StartMigrationWorkflowRequest, o: GoogleCloudGax.RequestOptions) async throws -> Void
          in
          return try await self.inner.startMigrationWorkflow(request: r, options: o)
        })
    }

    public func getMigrationSubtask(
      request: GetMigrationSubtaskRequest, options: GoogleCloudGax.RequestOptions
    ) async throws -> GoogleBigQueryMigrationV2.MigrationSubtask {
      try await self._intercept(
        request: request,
        options: options,
        idempotent: true,
        action: {
          (r: GetMigrationSubtaskRequest, o: GoogleCloudGax.RequestOptions) async throws
            -> GoogleBigQueryMigrationV2.MigrationSubtask
          in
          return try await self.inner.getMigrationSubtask(request: r, options: o)
        })
    }

    public func listMigrationSubtasks(
      request: ListMigrationSubtasksRequest, options: GoogleCloudGax.RequestOptions
    ) async throws -> GoogleBigQueryMigrationV2.ListMigrationSubtasksResponse {
      try await self._intercept(
        request: request,
        options: options,
        idempotent: true,
        action: {
          (r: ListMigrationSubtasksRequest, o: GoogleCloudGax.RequestOptions) async throws
            -> GoogleBigQueryMigrationV2.ListMigrationSubtasksResponse
          in
          return try await self.inner.listMigrationSubtasks(request: r, options: o)
        })
    }
  }
}
